avm/mostovik-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix pre-commit
Some checks failed
CI/CD Pipeline / Telegram Notify Success (push) Has been cancelled
Details
CI/CD Pipeline / Run Tests (push) Has been cancelled
Details
CI/CD Pipeline / Code Quality Checks (push) Has been cancelled
Details
CI/CD Pipeline / Code Quality Checks (pull_request) Successful in 1m42s
Details
CI/CD Pipeline / Run Tests (pull_request) Successful in 2m25s
Details
CI/CD Pipeline / Telegram Notify Success (pull_request) Successful in 1m34s
Details

Browse Source
This commit is contained in:
Aleksandr Meshchriakov
2026-03-17 13:55:34 +01:00
parent 3d298ce352
commit 25176f31b4
31 changed files with 653 additions and 553 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
tests/apps/parsers/test_source_cards_service.py
View File

@@ -4,7 +4,6 @@ from types import SimpleNamespace
from unittest.mock import MagicMock, patch
from apps.parsers.source_cards import (
RefreshParamDefinition,
SourceCardDefinition,
SourceCardService,
SourceItemDefinition,
@@ -66,8 +65,14 @@ class SourceCardServiceUnitTest(SimpleTestCase):
@patch(
"apps.parsers.source_cards.SourceCardService._enqueue_task",
side_effect=[
{"task_id": "task-1", "task_name": "apps.parsers.tasks.parse_industrial_production"},
{"task_id": "task-2", "task_name": "apps.parsers.tasks.parse_industrial_products"},
{
"task_id": "task-1",
"task_name": "apps.parsers.tasks.parse_industrial_production",
},
{
"task_id": "task-2",
"task_name": "apps.parsers.tasks.parse_industrial_products",
},
{"task_id": "task-3", "task_name": "apps.parsers.tasks.parse_manufactures"},
],
)
@@ -78,14 +83,22 @@ class SourceCardServiceUnitTest(SimpleTestCase):
)
self.assertEqual(result["source_card"], "manufacturers-and-products")
self.assertEqual([item["task_id"] for item in result["tasks"]], ["task-1", "task-2", "task-3"])
self.assertEqual(
[item["task_id"] for item in result["tasks"]],
["task-1", "task-2", "task-3"],
)
self.assertEqual(enqueue_mock.call_count, 3)
@patch(
"apps.parsers.source_cards.SourceCardService._enqueue_task",
return_value={"task_id": "task-1", "task_name": "apps.parsers.tasks.sync_inspections"},
return_value={
"task_id": "task-1",
"task_name": "apps.parsers.tasks.sync_inspections",
},
)
def test_launch_refresh_for_inspections_passes_supported_kwargs_only(self, enqueue_mock):
def test_launch_refresh_for_inspections_passes_supported_kwargs_only(
self, enqueue_mock
):
definition = SourceCardService.get_definition("planned-inspections")
result = SourceCardService._launch_refresh(
@@ -99,7 +112,10 @@ class SourceCardServiceUnitTest(SimpleTestCase):
},
)
self.assertEqual(result, [{"task_id": "task-1", "task_name": "apps.parsers.tasks.sync_inspections"}])
self.assertEqual(
result,
[{"task_id": "task-1", "task_name": "apps.parsers.tasks.sync_inspections"}],
)
self.assertEqual(
enqueue_mock.call_args.kwargs["kwargs"],
{
@@ -112,7 +128,10 @@ class SourceCardServiceUnitTest(SimpleTestCase):
@patch(
"apps.parsers.source_cards.SourceCardService._enqueue_task",
return_value={"task_id": "task-9", "task_name": "apps.parsers.tasks.sync_procurements"},
return_value={
"task_id": "task-9",
"task_name": "apps.parsers.tasks.sync_procurements",
},
)
def test_refresh_card_for_procurements_uses_default_law_type(self, enqueue_mock):
result = SourceCardService.refresh_card(
@@ -156,34 +175,37 @@ class SourceCardServiceUnitTest(SimpleTestCase):
params={},
)
self.assertIn("Обновление для карточки не поддерживается", str(error.exception.detail))
self.assertIn(
"Обновление для карточки не поддерживается", str(error.exception.detail)
)
def test_enqueue_task_deletes_background_job_on_async_error(self):
task = MagicMock()
task.apply_async.side_effect = RuntimeError("broker down")
queryset = MagicMock()
with patch("apps.parsers.source_cards.uuid.uuid4", return_value="task-id-1"):
with patch("apps.parsers.source_cards.BackgroundJobService.create_job"):
with patch(
"apps.parsers.source_cards.BackgroundJobService.get_queryset",
return_value=queryset,
):
with self.assertRaisesMessage(RuntimeError, "broker down"):
SourceCardService._enqueue_task(
task=task,
task_name="apps.parsers.tasks.sync_procurements",
requested_by_id=5,
meta={"source_card": "public-procurements"},
kwargs={"region_code": "77"},
)
with patch(
"apps.parsers.source_cards.uuid.uuid4", return_value="task-id-1"
), patch("apps.parsers.source_cards.BackgroundJobService.create_job"), patch(
"apps.parsers.source_cards.BackgroundJobService.get_queryset",
return_value=queryset,
), self.assertRaisesMessage(RuntimeError, "broker down"):
SourceCardService._enqueue_task(
task=task,
task_name="apps.parsers.tasks.sync_procurements",
requested_by_id=5,
meta={"source_card": "public-procurements"},
kwargs={"region_code": "77"},
)
queryset.filter.assert_called_once_with(task_id="task-id-1")
queryset.filter.return_value.delete.assert_called_once_with()
def test_helper_methods_cover_unknown_codes_and_status_variants(self):
self.assertEqual(SourceCardService._get_source_records_count("unknown"), 0)
self.assertEqual(SourceCardService._get_source_organizations_count("unknown"), 0)
self.assertEqual(
SourceCardService._get_source_organizations_count("unknown"), 0
)
self.assertIsNone(SourceCardService._get_source_data_timestamp("unknown"))
self.assertIsNone(SourceCardService._get_latest_load_by_source(None))
self.assertEqual(SourceCardService._get_status_label("custom"), "custom")