feat: address QA feedback and limit Checko collection

tests/apps/parsers/test_checko_collection.py

@@ -0,0 +1,92 @@
+from datetime import date
+
+from apps.parsers import tasks as parser_tasks
+from apps.parsers.checko_collection import claim_monthly_collection, finish_collection
+from apps.parsers.models import CheckoCollectionAttempt
+from django.test import TestCase
+
+from tests.apps.parsers.organization_helpers import create_directory_organization
+
+
+class CheckoCollectionClaimTest(TestCase):
+    def setUp(self):
+        self.organization = create_directory_organization(
+            pn_name='ООО "Месячный лимит"',
+            mn_ogrn=1027700000199,
+            mn_inn=7701000199,
+            in_kpp=770101001,
+            mn_okpo="12345678",
+        )
+
+    def test_claim_is_unique_per_organization_source_and_calendar_month(self):
+        first = claim_monthly_collection(
+            organization_id=str(self.organization.id),
+            source=CheckoCollectionAttempt.Source.ARBITRATION,
+            at=date(2026, 7, 1),
+        )
+
+        duplicate = claim_monthly_collection(
+            organization_id=str(self.organization.id),
+            source=CheckoCollectionAttempt.Source.ARBITRATION,
+            at=date(2026, 7, 31),
+        )
+        other_source = claim_monthly_collection(
+            organization_id=str(self.organization.id),
+            source=CheckoCollectionAttempt.Source.INSPECTIONS,
+            at=date(2026, 7, 31),
+        )
+        next_month = claim_monthly_collection(
+            organization_id=str(self.organization.id),
+            source=CheckoCollectionAttempt.Source.ARBITRATION,
+            at=date(2026, 8, 1),
+        )
+
+        self.assertIsNotNone(first)
+        self.assertIsNone(duplicate)
+        self.assertIsNotNone(other_source)
+        self.assertIsNotNone(next_month)
+
+    def test_failed_attempt_keeps_monthly_claim(self):
+        attempt = claim_monthly_collection(
+            organization_id=str(self.organization.id),
+            source=CheckoCollectionAttempt.Source.CONTRACTS,
+            at=date(2026, 7, 19),
+        )
+        assert attempt is not None
+        finish_collection(attempt, records_count=0, error="quota exhausted")
+
+        repeated = claim_monthly_collection(
+            organization_id=str(self.organization.id),
+            source=CheckoCollectionAttempt.Source.CONTRACTS,
+            at=date(2026, 7, 20),
+        )
+
+        attempt.refresh_from_db()
+        self.assertIsNone(repeated)
+        self.assertEqual(attempt.status, CheckoCollectionAttempt.Status.FAILED)
+        self.assertEqual(attempt.error_message, "quota exhausted")
+
+    def test_due_filter_is_applied_before_lookup_limit(self):
+        self.organization.opk_registry_membership = True
+        self.organization.save(update_fields=["opk_registry_membership"])
+        second = create_directory_organization(
+            pn_name='ООО "Следующая организация"',
+            mn_ogrn=1027700000299,
+            mn_inn=7701000299,
+            in_kpp=770101001,
+            mn_okpo="87654321",
+        )
+        second.opk_registry_membership = True
+        second.save(update_fields=["opk_registry_membership"])
+        claimed = claim_monthly_collection(
+            organization_id=str(self.organization.id),
+            source=CheckoCollectionAttempt.Source.INSPECTIONS,
+        )
+        self.assertIsNotNone(claimed)
+
+        targets = parser_tasks._active_registry_lookup_targets(
+            limit=1,
+            checko_source=CheckoCollectionAttempt.Source.INSPECTIONS,
+        )
+
+        self.assertEqual([target.organization_id for target in targets], [str(second.id)])