Add initial implementations for forms and organization apps with serializers, factories, and admin configurations

tests/utils/fixtures.py

@@ -0,0 +1,241 @@
+"""Fixture builders for integration tests (Faker-based)."""
+
+from __future__ import annotations
+
+import io
+import zipfile
+from dataclasses import dataclass
+from typing import Iterable
+
+from faker import Faker
+from openpyxl import Workbook
+
+fake = Faker("ru_RU")
+
+
+@dataclass(frozen=True)
+class CertificateRow:
+    issue_date: str
+    certificate_number: str
+    expiry_date: str
+    certificate_file_url: str
+    organisation_name: str
+    inn: str
+    ogrn: str
+
+
+@dataclass(frozen=True)
+class ManufacturerRow:
+    full_legal_name: str
+    inn: str
+    ogrn: str
+    address: str
+
+
+@dataclass(frozen=True)
+class InspectionRow:
+    registration_number: str
+    inn: str
+    ogrn: str
+    organisation_name: str
+    control_authority: str
+    inspection_type: str
+    inspection_form: str
+    start_date: str
+    end_date: str
+    status: str
+    legal_basis: str
+    result: str
+
+
+@dataclass(frozen=True)
+class ProcurementRow:
+    purchase_number: str
+    purchase_name: str
+    customer_inn: str
+    customer_kpp: str
+    customer_ogrn: str
+    customer_name: str
+    max_price: str
+    publish_date: str
+    end_date: str
+    status: str
+    href: str
+
+
+def _digits(length: int) -> str:
+    return "".join(str(fake.random_int(0, 9)) for _ in range(length))
+
+
+def build_minpromtorg_certificates_excel(count: int = 5) -> tuple[bytes, list[CertificateRow]]:
+    wb = Workbook()
+    ws = wb.active
+    ws.append(
+        [
+            "issue_date",
+            "certificate_number",
+            "expiry_date",
+            "certificate_file_url",
+            "organisation_name",
+            "inn",
+            "ogrn",
+        ]
+    )
+
+    rows: list[CertificateRow] = []
+    for _ in range(count):
+        row = CertificateRow(
+            issue_date=str(fake.date()),
+            certificate_number=f"{fake.bothify(text='??-####-#####')}",
+            expiry_date=str(fake.date()),
+            certificate_file_url=fake.url(),
+            organisation_name=fake.company(),
+            inn=_digits(10),
+            ogrn=_digits(13),
+        )
+        rows.append(row)
+        ws.append(
+            [
+                row.issue_date,
+                row.certificate_number,
+                row.expiry_date,
+                row.certificate_file_url,
+                row.organisation_name,
+                row.inn,
+                row.ogrn,
+            ]
+        )
+
+    buf = io.BytesIO()
+    wb.save(buf)
+    wb.close()
+    return buf.getvalue(), rows
+
+
+def build_minpromtorg_manufacturers_excel(
+    count: int = 5,
+) -> tuple[bytes, list[ManufacturerRow]]:
+    wb = Workbook()
+    ws = wb.active
+    ws.append(["full_legal_name", "inn", "ogrn", "address"])
+
+    rows: list[ManufacturerRow] = []
+    for _ in range(count):
+        row = ManufacturerRow(
+            full_legal_name=fake.company(),
+            inn=_digits(10),
+            ogrn=_digits(13),
+            address=fake.address().replace("\n", ", "),
+        )
+        rows.append(row)
+        ws.append([row.full_legal_name, row.inn, row.ogrn, row.address])
+
+    buf = io.BytesIO()
+    wb.save(buf)
+    wb.close()
+    return buf.getvalue(), rows
+
+
+def build_proverki_xml(count: int = 3) -> tuple[bytes, list[InspectionRow]]:
+    rows: list[InspectionRow] = []
+    parts = ["<?xml version='1.0' encoding='utf-8'?>", "<INSPECTIONS>"]
+
+    for _ in range(count):
+        row = InspectionRow(
+            registration_number=_digits(12),
+            inn=_digits(10),
+            ogrn=_digits(13),
+            organisation_name=fake.company(),
+            control_authority=fake.company(),
+            inspection_type=fake.word(),
+            inspection_form=fake.word(),
+            start_date=str(fake.date()),
+            end_date=str(fake.date()),
+            status=fake.word(),
+            legal_basis=fake.sentence(nb_words=4),
+            result=fake.sentence(nb_words=3),
+        )
+        rows.append(row)
+        parts.append(
+            "<INSPECTION "
+            f"ERPID=\"{row.registration_number}\" "
+            f"INN=\"{row.inn}\" "
+            f"OGRN=\"{row.ogrn}\" "
+            f"ORG_NAME=\"{row.organisation_name}\" "
+            f"FRGU_ORG_NAME=\"{row.control_authority}\" "
+            f"ITYPE_NAME=\"{row.inspection_type}\" "
+            f"ICARRYOUT_TYPE_NAME=\"{row.inspection_form}\" "
+            f"START_DATE=\"{row.start_date}\" "
+            f"END_DATE=\"{row.end_date}\" "
+            f"STATUS=\"{row.status}\" "
+            f"FZ_NAME=\"{row.legal_basis}\" "
+            f"RESULT=\"{row.result}\" />"
+        )
+
+    parts.append("</INSPECTIONS>")
+    xml = "".join(parts).encode("utf-8")
+    return xml, rows
+
+
+def build_zakupki_xml(count: int = 3) -> tuple[bytes, list[ProcurementRow]]:
+    rows: list[ProcurementRow] = []
+    parts = ["<?xml version='1.0' encoding='utf-8'?>", "<root>"]
+
+    for _ in range(count):
+        row = ProcurementRow(
+            purchase_number=_digits(19),
+            purchase_name=fake.sentence(nb_words=6),
+            customer_inn=_digits(10),
+            customer_kpp=_digits(9),
+            customer_ogrn=_digits(13),
+            customer_name=fake.company(),
+            max_price=str(fake.pydecimal(left_digits=7, right_digits=2, positive=True)),
+            publish_date=str(fake.date()),
+            end_date=str(fake.date()),
+            status=fake.word(),
+            href=fake.url(),
+        )
+        rows.append(row)
+        parts.append(
+            "<notification>"
+            f"<purchaseNumber>{row.purchase_number}</purchaseNumber>"
+            f"<purchaseObjectInfo>{row.purchase_name}</purchaseObjectInfo>"
+            "<customer>"
+            f"<INN>{row.customer_inn}</INN>"
+            f"<KPP>{row.customer_kpp}</KPP>"
+            f"<OGRN>{row.customer_ogrn}</OGRN>"
+            f"<fullName>{row.customer_name}</fullName>"
+            "</customer>"
+            f"<maxPrice>{row.max_price}</maxPrice>"
+            f"<publishDate>{row.publish_date}</publishDate>"
+            f"<endDate>{row.end_date}</endDate>"
+            f"<state>{row.status}</state>"
+            f"<href>{row.href}</href>"
+            "</notification>"
+        )
+
+    parts.append("</root>")
+    xml = "".join(parts).encode("utf-8")
+    return xml, rows
+
+
+def build_zip(files: Iterable[tuple[str, bytes]]) -> bytes:
+    buf = io.BytesIO()
+    with zipfile.ZipFile(buf, "w", zipfile.ZIP_DEFLATED) as zf:
+        for name, content in files:
+            zf.writestr(name, content)
+    return buf.getvalue()
+
+
+__all__ = [
+    "fake",
+    "CertificateRow",
+    "ManufacturerRow",
+    "InspectionRow",
+    "ProcurementRow",
+    "build_minpromtorg_certificates_excel",
+    "build_minpromtorg_manufacturers_excel",
+    "build_proverki_xml",
+    "build_zakupki_xml",
+    "build_zip",
+]