Aleksandr Meshchriakov
345b1d0cc8
Some checks failed
CI/CD Pipeline / Run Tests (push) Failing after 45s
CI/CD Pipeline / Code Quality Checks (push) Failing after 48s
CI/CD Pipeline / Build Docker Images (push) Has been skipped
CI/CD Pipeline / Push to Gitea Registry (push) Has been skipped
CI/CD Pipeline / Deploy to Server (push) Has been skipped
124 lines
4.6 KiB
Python
124 lines
4.6 KiB
Python
"""Tests for organization API."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from datetime import date
|
||
|
||
from apps.registers.models import Register, RegisterUpload, RegistryMembershipPeriod
|
||
from django.test import override_settings
|
||
from rest_framework import status
|
||
from rest_framework.test import APITestCase
|
||
|
||
from tests.apps.organization.factories import OrganizationFactory
|
||
from tests.apps.user.factories import UserFactory
|
||
|
||
|
||
@override_settings(ROOT_URLCONF="core.urls")
|
||
class OrganizationApiTest(APITestCase):
|
||
"""Tests for organization endpoints."""
|
||
|
||
def setUp(self):
|
||
self.user = UserFactory.create_user()
|
||
self.client.force_authenticate(self.user)
|
||
|
||
def test_list_includes_only_active_registry_names(self):
|
||
organization = OrganizationFactory.create(name="АО Альфа")
|
||
active_register = Register.objects.create(name="Реестр ОПК")
|
||
closed_register = Register.objects.create(name="Архивный реестр")
|
||
active_upload = RegisterUpload.objects.create(
|
||
registry=active_register,
|
||
actual_date=date(2026, 3, 1),
|
||
file_name="active.xlsx",
|
||
file_hash="active-api-hash",
|
||
rows_count=1,
|
||
)
|
||
closed_upload = RegisterUpload.objects.create(
|
||
registry=closed_register,
|
||
actual_date=date(2026, 2, 1),
|
||
file_name="closed.xlsx",
|
||
file_hash="closed-api-hash",
|
||
rows_count=1,
|
||
)
|
||
|
||
RegistryMembershipPeriod.objects.create(
|
||
registry=active_register,
|
||
organization=organization,
|
||
started_at=date(2026, 3, 1),
|
||
started_by_upload=active_upload,
|
||
)
|
||
RegistryMembershipPeriod.objects.create(
|
||
registry=closed_register,
|
||
organization=organization,
|
||
started_at=date(2026, 2, 1),
|
||
ended_at=date(2026, 3, 1),
|
||
started_by_upload=closed_upload,
|
||
ended_by_upload=closed_upload,
|
||
)
|
||
|
||
response = self.client.get("/api/v1/organizations/")
|
||
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data["data"][0]["active_registry_names"], ["Реестр ОПК"])
|
||
|
||
def test_detail_includes_active_registries(self):
|
||
organization = OrganizationFactory.create()
|
||
register = Register.objects.create(name="Реестр Роскосмос")
|
||
upload = RegisterUpload.objects.create(
|
||
registry=register,
|
||
actual_date=date(2026, 3, 1),
|
||
file_name="detail.xlsx",
|
||
file_hash="detail-api-hash",
|
||
rows_count=1,
|
||
)
|
||
RegistryMembershipPeriod.objects.create(
|
||
registry=register,
|
||
organization=organization,
|
||
started_at=date(2026, 3, 1),
|
||
started_by_upload=upload,
|
||
)
|
||
|
||
response = self.client.get(f"/api/v1/organizations/{organization.id}/")
|
||
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(
|
||
response.data["data"]["active_registry_names"],
|
||
["Реестр Роскосмос"],
|
||
)
|
||
self.assertEqual(
|
||
response.data["data"]["active_registries"],
|
||
[{"id": str(register.id), "name": "Реестр Роскосмос"}],
|
||
)
|
||
|
||
def test_registry_filter_uses_only_active_memberships(self):
|
||
active_organization = OrganizationFactory.create(name="АО Актив")
|
||
closed_organization = OrganizationFactory.create(name="АО Закрыт")
|
||
register = Register.objects.create(name="Реестр Росатом")
|
||
upload = RegisterUpload.objects.create(
|
||
registry=register,
|
||
actual_date=date(2026, 3, 1),
|
||
file_name="filter.xlsx",
|
||
file_hash="filter-api-hash",
|
||
rows_count=2,
|
||
)
|
||
|
||
RegistryMembershipPeriod.objects.create(
|
||
registry=register,
|
||
organization=active_organization,
|
||
started_at=date(2026, 3, 1),
|
||
started_by_upload=upload,
|
||
)
|
||
RegistryMembershipPeriod.objects.create(
|
||
registry=register,
|
||
organization=closed_organization,
|
||
started_at=date(2026, 2, 1),
|
||
ended_at=date(2026, 3, 1),
|
||
started_by_upload=upload,
|
||
ended_by_upload=upload,
|
||
)
|
||
|
||
response = self.client.get(f"/api/v1/organizations/?registry={register.id}")
|
||
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(len(response.data["data"]), 1)
|
||
self.assertEqual(response.data["data"][0]["id"], str(active_organization.id))
|