feat: add gosedo and media news sources
All checks were successful
All checks were successful
This commit is contained in:
@@ -191,14 +191,13 @@ class SourcesApiE2ETest(APITestCase):
|
||||
"task-procurements",
|
||||
],
|
||||
), patch(
|
||||
"apps.parsers.tasks.parse_industrial_production.apply_async",
|
||||
return_value=SimpleNamespace(id="task-industrial"),
|
||||
), patch(
|
||||
"apps.parsers.tasks.parse_industrial_products.apply_async",
|
||||
return_value=SimpleNamespace(id="task-products"),
|
||||
), patch(
|
||||
"apps.parsers.tasks.parse_manufactures.apply_async",
|
||||
return_value=SimpleNamespace(id="task-manufactures"),
|
||||
"celery.app.task.Task.apply_async",
|
||||
side_effect=[
|
||||
SimpleNamespace(id="task-industrial"),
|
||||
SimpleNamespace(id="task-products"),
|
||||
SimpleNamespace(id="task-manufactures"),
|
||||
SimpleNamespace(id="task-procurements"),
|
||||
],
|
||||
):
|
||||
minprom_response = self.client.post(
|
||||
reverse(
|
||||
@@ -209,18 +208,14 @@ class SourcesApiE2ETest(APITestCase):
|
||||
format="json",
|
||||
)
|
||||
|
||||
with patch(
|
||||
"apps.parsers.tasks.parse_registry_enrichment_sources.apply_async",
|
||||
return_value=SimpleNamespace(id="task-procurements"),
|
||||
):
|
||||
procurements_response = self.client.post(
|
||||
reverse(
|
||||
"api_v1:sources:source-cards-refresh",
|
||||
kwargs={"slug": "public-procurements"},
|
||||
),
|
||||
{"params": {"region_code": "77", "current_year": "2025"}},
|
||||
format="json",
|
||||
)
|
||||
procurements_response = self.client.post(
|
||||
reverse(
|
||||
"api_v1:sources:source-cards-refresh",
|
||||
kwargs={"slug": "public-procurements"},
|
||||
),
|
||||
{"params": {"region_code": "77", "current_year": "2025"}},
|
||||
format="json",
|
||||
)
|
||||
|
||||
self.assertEqual(minprom_response.status_code, status.HTTP_202_ACCEPTED)
|
||||
self.assertEqual(procurements_response.status_code, status.HTTP_202_ACCEPTED)
|
||||
@@ -228,13 +223,24 @@ class SourcesApiE2ETest(APITestCase):
|
||||
self.assertEqual(minprom_response.data["status"], "accepted")
|
||||
self.assertEqual(procurements_response.data["status"], "accepted")
|
||||
|
||||
self.assertEqual(set(minprom_response.data.keys()), {"task_id", "status"})
|
||||
self.assertEqual(
|
||||
set(minprom_response.data.keys()),
|
||||
{"task_id", "task_ids", "status"},
|
||||
)
|
||||
self.assertEqual(minprom_response.data["task_id"], "task-industrial")
|
||||
self.assertEqual(
|
||||
minprom_response.data["task_ids"],
|
||||
["task-industrial", "task-products", "task-manufactures"],
|
||||
)
|
||||
self.assertEqual(
|
||||
set(procurements_response.data.keys()),
|
||||
{"task_id", "status"},
|
||||
{"task_id", "task_ids", "status"},
|
||||
)
|
||||
self.assertEqual(procurements_response.data["task_id"], "task-procurements")
|
||||
self.assertEqual(
|
||||
procurements_response.data["task_ids"],
|
||||
["task-procurements"],
|
||||
)
|
||||
self.assertEqual(
|
||||
BackgroundJob.objects.filter(
|
||||
task_id__in=["task-industrial", "task-products", "task-manufactures"],
|
||||
|
||||
Reference in New Issue
Block a user