Add 5 tests

Author: juggler31

backend/gn_module_zh/tests/__init__.py

@@ -0,0 +1,3 @@
+from geonature.tests.fixtures import *
+from geonature.tests.fixtures import _session, app, users, _app
+from geonature.tests.test_permissions import g_permissions

backend/gn_module_zh/tests/conftest.py

@@ -0,0 +1,89 @@
+import datetime
+import pytest
+from geonature.utils.env import db
+
+from shapely import Polygon
+import geoalchemy2
+
+import uuid
+
+
+def create_zh(
+    main_name,
+    code,
+    id_org,
+    id_role,
+    zh_date,
+    uuid_id_lim_list,
+    id_sdage,
+    polygon,
+    zh_area,
+    **kwargs,
+):
+    # Import here because TZH class need to be imported after "app instanced"
+    from gn_module_zh.model.zh_schema import TZH
+
+    zh = TZH(
+        main_name=main_name,
+        code=code,
+        id_org=id_org,
+        create_author=id_role,
+        update_author=id_role,
+        create_date=zh_date,
+        update_date=zh_date,
+        id_lim_list=uuid_id_lim_list,
+        id_sdage=id_sdage,
+        geom=polygon,
+        area=zh_area,
+    )
+    return zh
+
+
+@pytest.fixture(scope="function")
+def zh_data(users):
+    coords = ((0.0, 0.0), (0.0, 1.0), (1.0, 1.0), (1.0, 0.0), (0.0, 0.0))
+    polygon = Polygon(coords)
+    date = datetime.datetime(2024, 10, 2, 11, 22, 33)
+    id_sdage = 967
+    user = users["self_user"]
+    data = {}
+    with db.session.begin_nested():
+        for (
+            main_name,
+            code,
+            id_org,
+            id_role,
+            zh_date,
+            uuid_id_lim_list,
+            id_sdage,
+            geom,
+            zh_area,
+        ) in [
+            (
+                "zh1",
+                "05CEN0189",
+                2,  # id_org = 2 (not dynamic) because not same table bib_organismes used with the users
+                user.id_role,
+                date,
+                uuid.uuid4(),
+                id_sdage,
+                geoalchemy2.shape.from_shape(polygon),
+                polygon.area,
+            ),
+        ]:
+            kwargs = {}
+            s = create_zh(
+                main_name,
+                code,
+                id_org,
+                id_role,
+                zh_date,
+                uuid_id_lim_list,
+                id_sdage,
+                geom,
+                zh_area,
+                **kwargs,
+            )
+            db.session.add(s)
+            data[main_name] = s
+    return data

backend/gn_module_zh/tests/fixtures.py

@@ -0,0 +1,51 @@
+from pypnusershub.tests.utils import set_logged_user
+
+from flask import url_for
+
+from geonature.tests.fixtures import *
+from .fixtures import *
+
+
+@pytest.mark.usefixtures("client_class")
+class TestBdd:
+    def test_no_zh(self, users):
+        set_logged_user(self.client, users["self_user"])
+        response = self.client.get(url_for("pr_zh.get_zh"))
+        assert response.status_code == 200
+        data = response.get_json()
+        assert len(data["items"]["features"]) == 0
+
+    def test_get_zh(self, users, zh_data):
+        set_logged_user(self.client, users["self_user"])
+        response = self.client.get(url_for("pr_zh.get_zh"))
+        assert response.status_code == 200
+        data = response.get_json()
+        assert len(data["items"]["features"]) > 0
+
+    def test_get_zh_by_id(self, users, zh_data):
+        set_logged_user(self.client, users["self_user"])
+        response = self.client.get(url_for("pr_zh.get_zh_by_id", id_zh=zh_data.get("zh1").id_zh))
+        assert response.status_code == 200
+        data = response.get_json()
+        assert data["properties"]["id_zh"] == zh_data.get("zh1").id_zh
+        assert data["properties"]["main_name"] == zh_data.get("zh1").main_name
+
+    def test_get_zh_by_id_not_found(self, users):
+        set_logged_user(self.client, users["self_user"])
+        response = self.client.get(url_for("pr_zh.get_zh_by_id", id_zh=999999))
+        # TODO: renvoie une 500 si la zone de protection n'existe pas ce serait préférable une 404
+        assert response.status_code == 500
+        # data = response.get_json()
+        # print("data:", data)
+        # assert data["message"] == "Zone de protection non trouvée"
+
+    def test_get_zh_by_id_no_auth(self, users, zh_data):
+        set_logged_user(self.client, users["noright_user"])
+        response = self.client.get(url_for("pr_zh.get_zh_by_id", id_zh=zh_data.get("zh1").id_zh))
+        assert response.status_code == 403
+        data = response.get_json()
+        assert data["code"] == 403
+        assert (
+            data["description"]
+            == f"User {users['noright_user'].get_id()} has no permissions to R in ZONES_HUMIDES"
+        )