feat: restore registry procurement collection
This commit is contained in:
@@ -240,10 +240,12 @@ class SourceCardServiceUnitTest(SimpleTestCase):
|
||||
"apps.parsers.source_cards.SourceCardService._enqueue_task",
|
||||
return_value={
|
||||
"task_id": "task-9",
|
||||
"task_name": "apps.parsers.tasks.sync_procurements",
|
||||
"task_name": "apps.parsers.tasks.parse_registry_enrichment_sources",
|
||||
},
|
||||
)
|
||||
def test_refresh_card_for_procurements_uses_default_law_type(self, enqueue_mock):
|
||||
def test_refresh_card_for_procurements_uses_addressed_registry_task(
|
||||
self, enqueue_mock
|
||||
):
|
||||
result = SourceCardService.refresh_card(
|
||||
slug="public-procurements",
|
||||
requested_by_id=10,
|
||||
@@ -254,12 +256,7 @@ class SourceCardServiceUnitTest(SimpleTestCase):
|
||||
self.assertEqual(result["tasks"][0]["task_id"], "task-9")
|
||||
self.assertEqual(
|
||||
enqueue_mock.call_args.kwargs["kwargs"],
|
||||
{
|
||||
"requested_by_id": 10,
|
||||
"region_code": "77",
|
||||
"law_type": "44",
|
||||
"current_year": 2026,
|
||||
},
|
||||
{"requested_by_id": 10},
|
||||
)
|
||||
|
||||
@patch(
|
||||
@@ -540,6 +537,7 @@ class SourceCardServiceDatabaseTest(TestCase):
|
||||
self.assertEqual(card["records_count"], 3)
|
||||
self.assertEqual(card["organizations_count"], 2)
|
||||
source_items = {item["code"]: item for item in card["source_items"]}
|
||||
self.assertEqual(source_items["procurements"]["records_count"], 2)
|
||||
self.assertEqual(source_items["procurements_44fz"]["organizations_count"], 1)
|
||||
self.assertEqual(source_items["procurements_223fz"]["organizations_count"], 1)
|
||||
self.assertEqual(source_items["contracts"]["organizations_count"], 1)
|
||||
|
||||
Reference in New Issue
Block a user