From 492260befa6de07972ed322a6a828d8fbd84118b Mon Sep 17 00:00:00 2001
From: gounux <contact@guilhemallaman.net>
Date: Fri, 20 Dec 2024 14:55:02 +0100
Subject: [PATCH] Add processprojectfile test for parsing invalid localized
 layers

---
 .../qfieldcloud/core/tests/test_jobs.py       | 93 +++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 docker-app/qfieldcloud/core/tests/test_jobs.py

diff --git a/docker-app/qfieldcloud/core/tests/test_jobs.py b/docker-app/qfieldcloud/core/tests/test_jobs.py
new file mode 100644
index 000000000..eaac048e6
--- /dev/null
+++ b/docker-app/qfieldcloud/core/tests/test_jobs.py
@@ -0,0 +1,93 @@
+import logging
+
+from qfieldcloud.authentication.models import AuthToken
+from qfieldcloud.core.models import (
+    Job,
+    Person,
+    Project,
+)
+from qfieldcloud.core.models import (
+    ProcessProjectfileJob,
+)
+from rest_framework import status
+from rest_framework.test import APITransactionTestCase
+
+from .utils import setup_subscription_plans, testdata_path, wait_for_project_ok_status
+
+logging.disable(logging.CRITICAL)
+
+VALID_LOCALIZED_LAYER_KEY = "apiary_853c29cb_ead8_4932_a60a_5cba33140b3d"
+INVALID_LAYER_KEY = "apiary_e7d1b542_1b1c_422d_ac75_c14d2b54e472"
+INVALID_LOCALIZED_LAYER_KEY = "area_a7b70aff_23f2_4876_9d4e_fa63036df0b2"
+
+
+class QfcTestCase(APITransactionTestCase):
+    def setUp(self):
+        setup_subscription_plans()
+
+        # Create a user
+        self.u1 = Person.objects.create_user(username="u1", password="abc123")
+        self.t1 = AuthToken.objects.get_or_create(user=self.u1)[0]
+
+        # Create a project
+        self.p1 = Project.objects.create(name="p1", is_public=False, owner=self.u1)
+
+    def test_create_job_bad_layer_handler_extracted_values(self):
+        # Test that BadLayerHandler is parsing data properly during package and process projectfile jobs
+        self.client.credentials(HTTP_AUTHORIZATION="Token " + self.t1.key)
+
+        # Push the QGIS project and datasource GPKG files
+        for local_filename, upload_filename in [
+            ("bumblebees.gpkg", "bumblebees.gpkg"),
+            ("simple_bumblebees_wrong_localized.qgs", "project.qgs"),
+        ]:
+            file_path = testdata_path(local_filename)
+            response = self.client.post(
+                f"/api/v1/files/{self.p1.id}/{upload_filename}/",
+                {
+                    "file": open(file_path, "rb"),
+                },
+                format="multipart",
+            )
+            self.assertTrue(status.is_success(response.status_code))
+
+        # Create ProcessProjectfileJob and wait for it to finish
+        processprojectfile_job = ProcessProjectfileJob.objects.create(
+            type=Job.Type.PROCESS_PROJECTFILE,
+            project=self.p1,
+            created_by=self.u1,
+        )
+        self.assertEqual(processprojectfile_job.status, Job.Status.PENDING)
+        wait_for_project_ok_status(self.p1)
+
+        processprojectfile_job.refresh_from_db()
+        self.assertEqual(processprojectfile_job.status, Job.Status.FINISHED)
+        self.assertIsNotNone(processprojectfile_job.feedback)
+
+        self.p1.refresh_from_db()
+        self.assertIsNotNone(self.p1.project_details)
+
+        # Project details extraction is usually step n°5
+        for step in processprojectfile_job.feedback["steps"]:
+            if step["id"] == "project_details":
+                details = step["returns"]["project_details"]
+        layers = details["layers_by_id"]
+
+        # "valid" localized layer -> QFC considers it as invalid
+        valid_localized_layer = layers[VALID_LOCALIZED_LAYER_KEY]
+        self.assertFalse(valid_localized_layer["is_valid"])
+        self.assertTrue(valid_localized_layer["is_localized"])
+
+        # invalid layer (datasource does not exist)
+        invalid_layer = layers[INVALID_LAYER_KEY]
+        self.assertFalse(invalid_layer["is_valid"])
+        self.assertFalse(invalid_layer["is_localized"])
+        self.assertEquals(invalid_layer["error_code"], "invalid_dataprovider")
+
+        # invalid localized layer
+        invalid_localized_layer = layers[INVALID_LOCALIZED_LAYER_KEY]
+        self.assertFalse(invalid_localized_layer["is_valid"])
+        self.assertTrue(invalid_localized_layer["is_localized"])
+        self.assertEquals(
+            invalid_localized_layer["error_code"], "localized_dataprovider"
+        )