feat(registries): add SME and budget imports and fix source API workflows

tests/apps/organizations/test_canonical_search.py

@@ -0,0 +1,140 @@
+"""Organization search must use the same canonical identity in every table."""
+
+from django.core.cache import cache
+from django.urls import reverse
+from organizations.models import (
+    Organization,
+    OrganizationSourceRecord,
+    PlannedInspectionExtension,
+)
+from rest_framework.test import APITestCase
+
+from tests.apps.user.factories import UserFactory
+
+
+class CanonicalOrganizationSearchTest(APITestCase):
+    def setUp(self):
+        cache.clear()
+        self.client.force_authenticate(UserFactory.create_user())
+        self.organization = Organization.objects.create(
+            name="Росатом",
+            full_name="Полное наименование Росатома",
+            short_name="Краткое",
+            inn="7707083881",
+            kpp="770701081",
+            ogrn="1027700132181",
+            okpo="00003881",
+            opk_registry_membership=True,
+        )
+        extension = PlannedInspectionExtension.objects.create(
+            organization=self.organization,
+            title="Проверки",
+        )
+        for index in range(2):
+            OrganizationSourceRecord.objects.create(
+                extension=extension,
+                source="inspections",
+                record_type="inspection",
+                external_id=f"target-{index}",
+                status="active",
+            )
+        other = Organization.objects.create(name="Другая", opk_registry_membership=True)
+        other_extension = PlannedInspectionExtension.objects.create(
+            organization=other,
+            title="Росатом",
+        )
+        OrganizationSourceRecord.objects.create(
+            extension=other_extension,
+            source="inspections",
+            record_type="inspection",
+            external_id="Росатом",
+            title="Росатом",
+            status="active",
+            url="https://example.test/Росатом",
+            payload={"raw_text": "Росатом"},
+        )
+        self.organizations_url = reverse("api_v2:organizations:organizations-list")
+        self.records_url = reverse(
+            "api_v2:organizations:organization-source-records-list"
+        )
+
+    def test_all_canonical_names_and_identifiers_are_searchable_in_both_endpoints(self):
+        for search in (
+            "Росатом",
+            "Полное",
+            "Краткое",
+            "7707083881",
+            "770701081",
+            "1027700132181",
+            "00003881",
+        ):
+            with self.subTest(search=search):
+                organizations = self.client.get(
+                    self.organizations_url, {"search": search}
+                )
+                records = self.client.get(self.records_url, {"search": search})
+                self.assertEqual(organizations.status_code, 200)
+                self.assertEqual(records.status_code, 200)
+                self.assertEqual(
+                    organizations.data["meta"]["pagination"]["total_count"], 1
+                )
+                self.assertEqual(records.data["meta"]["pagination"]["total_count"], 2)
+                self.assertEqual(
+                    organizations.data["data"][0]["uid"], str(self.organization.uid)
+                )
+
+    def test_entrepreneur_identifier_is_searchable_in_both_endpoints(self):
+        entrepreneur = Organization.objects.create(
+            name="ИП Иванов",
+            inn="500100732259",
+            ogrip="304500116000181",
+            opk_registry_membership=True,
+        )
+        extension = PlannedInspectionExtension.objects.create(
+            organization=entrepreneur,
+            title="Проверки",
+        )
+        OrganizationSourceRecord.objects.create(
+            extension=extension,
+            source="inspections",
+            record_type="inspection",
+            external_id="entrepreneur-record",
+        )
+        for url in (self.organizations_url, self.records_url):
+            response = self.client.get(url, {"search": "304500116000181"})
+            self.assertEqual(response.status_code, 200)
+            self.assertEqual(response.data["meta"]["pagination"]["total_count"], 1)
+
+    def test_search_terms_and_source_filters_are_applied_before_pagination(self):
+        extension = self.organization.source_extensions.get()
+        OrganizationSourceRecord.objects.create(
+            extension=extension,
+            source="other-source",
+            record_type="inspection",
+            external_id="filtered-out",
+            status="active",
+        )
+        params = {
+            "search": "Росатом 7707083881",
+            "source_group": "planned_inspections",
+            "source": "inspections",
+            "record_type": "inspection",
+            "status": "active",
+            "ordering": "external_id",
+            "page_size": 1,
+        }
+        first = self.client.get(self.records_url, {**params, "page": 1})
+        second = self.client.get(self.records_url, {**params, "page": 2})
+        self.assertEqual(first.status_code, 200)
+        self.assertEqual(second.status_code, 200)
+        self.assertEqual(first.data["meta"]["pagination"]["total_count"], 2)
+        self.assertEqual(first.data["data"][0]["external_id"], "target-0")
+        self.assertEqual(second.data["data"][0]["external_id"], "target-1")
+
+    def test_technical_identity_is_not_a_search_field(self):
+        Organization.objects.filter(pk=self.organization.pk).update(
+            primary_identity="technical-only"
+        )
+        response = self.client.get(self.organizations_url, {"search": "technical-only"})
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.data["meta"]["pagination"]["total_count"], 0)