UPDATE

2026-05-09 16:55:06 +03:30
parent 1679825ae2
commit cead7dafe2
51 changed files with 7514 additions and 1221 deletions
@@ -0,0 +1,114 @@
+from django.test import TestCase, override_settings
+
+from location_data.grid_analysis import create_or_get_analysis_grid_cells
+from location_data.models import AnalysisGridCell, BlockSubdivision, SoilLocation
+
+
+@override_settings(SUBDIVISION_CHUNK_SQM=900)
+class AnalysisGridServiceTests(TestCase):
+    def setUp(self):
+        self.boundary = {
+            "type": "Polygon",
+            "coordinates": [
+                [
+                    [51.389000, 35.689000],
+                    [51.389760, 35.689000],
+                    [51.389760, 35.689620],
+                    [51.389000, 35.689620],
+                    [51.389000, 35.689000],
+                ]
+            ],
+        }
+        self.location = SoilLocation.objects.create(
+            latitude="35.689310",
+            longitude="51.389380",
+            farm_boundary=self.boundary,
+        )
+        self.location.set_input_block_count(1)
+        self.location.save(update_fields=["input_block_count", "block_layout", "updated_at"])
+        self.subdivision = BlockSubdivision.objects.create(
+            soil_location=self.location,
+            block_code="block-1",
+            source_boundary=self.boundary,
+            chunk_size_sqm=900,
+            status="created",
+        )
+
+    def test_create_analysis_grid_cells_persists_30x30_cells(self):
+        result = create_or_get_analysis_grid_cells(
+            self.location,
+            block_code="block-1",
+            block_subdivision=self.subdivision,
+        )
+
+        self.assertTrue(result["created"])
+        self.assertEqual(result["chunk_size_sqm"], 900)
+        self.assertGreater(result["created_count"], 0)
+        self.assertEqual(result["created_count"], result["total_count"])
+
+        cells = list(
+            AnalysisGridCell.objects.filter(
+                soil_location=self.location,
+                block_code="block-1",
+                chunk_size_sqm=900,
+            ).order_by("cell_code")
+        )
+        self.assertEqual(len(cells), result["total_count"])
+        self.assertTrue(all(cell.block_subdivision_id == self.subdivision.id for cell in cells))
+        self.assertTrue(all(cell.geometry.get("type") == "Polygon" for cell in cells))
+        self.assertTrue(all(len(cell.geometry.get("coordinates", [[]])[0]) == 5 for cell in cells))
+
+        self.subdivision.refresh_from_db()
+        self.location.refresh_from_db()
+        self.assertEqual(
+            self.subdivision.metadata["analysis_grid"]["chunk_size_sqm"],
+            900,
+        )
+        self.assertEqual(
+            self.subdivision.metadata["analysis_grid"]["cell_count"],
+            result["total_count"],
+        )
+        self.assertEqual(
+            self.location.block_layout["blocks"][0]["analysis_grid_summary"]["chunk_size_sqm"],
+            900,
+        )
+
+    def test_create_analysis_grid_cells_is_idempotent(self):
+        first = create_or_get_analysis_grid_cells(
+            self.location,
+            block_code="block-1",
+            block_subdivision=self.subdivision,
+        )
+        second = create_or_get_analysis_grid_cells(
+            self.location,
+            block_code="block-1",
+            block_subdivision=self.subdivision,
+        )
+
+        self.assertTrue(first["created"])
+        self.assertFalse(second["created"])
+        self.assertEqual(second["created_count"], 0)
+        self.assertEqual(second["existing_count"], first["total_count"])
+        self.assertEqual(
+            AnalysisGridCell.objects.filter(
+                soil_location=self.location,
+                block_code="block-1",
+                chunk_size_sqm=900,
+            ).count(),
+            first["total_count"],
+        )
+
+    def test_create_analysis_grid_cells_uses_location_boundary_without_subdivision(self):
+        result = create_or_get_analysis_grid_cells(
+            self.location,
+            block_code="",
+        )
+
+        self.assertGreater(result["total_count"], 0)
+        self.assertTrue(
+            AnalysisGridCell.objects.filter(
+                soil_location=self.location,
+                block_code="",
+                chunk_size_sqm=900,
+            ).exists()
+        )