134 lines
6.0 KiB
Python
134 lines
6.0 KiB
Python
"""
|
||
Тесты builder.py — создание ZIP и mapping.csv.
|
||
|
||
Запуск: python3 tests/test_builder.py
|
||
"""
|
||
|
||
import io
|
||
import zipfile
|
||
import sys
|
||
import os
|
||
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from drhider.builder import build_zip, build_mapping_csv
|
||
|
||
passed = 0
|
||
failed = 0
|
||
|
||
def check(name, condition):
|
||
global passed, failed
|
||
if condition:
|
||
passed += 1
|
||
print(f" ✅ {name}")
|
||
else:
|
||
failed += 1
|
||
print(f" ❌ {name}")
|
||
|
||
|
||
# ═══════════════════════════════════════════════════════════════
|
||
# Test build_zip
|
||
# ═══════════════════════════════════════════════════════════════
|
||
print("\n=== build_zip ===")
|
||
|
||
# 1. Базовый ZIP с одним файлом
|
||
zip_data = build_zip([("test.md", b"content")])
|
||
with zipfile.ZipFile(io.BytesIO(zip_data)) as zf:
|
||
names = zf.namelist()
|
||
check("один файл в ZIP", len(names) == 1)
|
||
check("имя test.md", "test.md" in names)
|
||
check("контент верный", zf.read("test.md") == b"content")
|
||
|
||
# 2. ZIP с несколькими файлами
|
||
zip_data = build_zip([
|
||
("договор_obfuscated.md", "# Договор\nТекст".encode("utf-8")),
|
||
("спецификация_obfuscated.md", "| Услуга | Цена |".encode("utf-8")),
|
||
])
|
||
with zipfile.ZipFile(io.BytesIO(zip_data)) as zf:
|
||
names = zf.namelist()
|
||
check("два файла", len(names) == 2)
|
||
check("mapping.csv нет", "mapping.csv" not in names)
|
||
check("договор читается", "Договор" in zf.read("договор_obfuscated.md").decode("utf-8"))
|
||
|
||
# 3. ZIP с mapping.csv
|
||
zip_data = build_zip(
|
||
[("file.md", b"data")],
|
||
mapping_csv="тип,оригинал,замена\nperson,Иванов,Иванов_0001\n"
|
||
)
|
||
with zipfile.ZipFile(io.BytesIO(zip_data)) as zf:
|
||
names = zf.namelist()
|
||
check("mapping.csv есть", "mapping.csv" in names)
|
||
csv_content = zf.read("mapping.csv").decode("utf-8")
|
||
check("mapping с BOM", csv_content.startswith('\ufeff'))
|
||
check("данные в mapping", "Иванов_0001" in csv_content)
|
||
|
||
# 4. ZIP с кириллицей в именах
|
||
zip_data = build_zip([("договор_№123.md", "текст".encode("utf-8"))])
|
||
with zipfile.ZipFile(io.BytesIO(zip_data)) as zf:
|
||
check("кириллическое имя читается", "договор_№123.md" in zf.namelist())
|
||
|
||
# 5. ZIP с бинарными данными
|
||
binary = bytes(range(256))
|
||
zip_data = build_zip([("binary.bin", binary)])
|
||
with zipfile.ZipFile(io.BytesIO(zip_data)) as zf:
|
||
check("бинарный файл сохранён", zf.read("binary.bin") == binary)
|
||
|
||
# 6. Пустой mapping
|
||
zip_data = build_zip([("f.md", b"x")], mapping_csv="")
|
||
with zipfile.ZipFile(io.BytesIO(zip_data)) as zf:
|
||
check("пустой mapping не добавляется", "mapping.csv" not in zf.namelist())
|
||
|
||
|
||
# ═══════════════════════════════════════════════════════════════
|
||
# Test build_mapping_csv
|
||
# ═══════════════════════════════════════════════════════════════
|
||
print("\n=== build_mapping_csv ===")
|
||
|
||
# 1. Простой mapping
|
||
mapping = {"Иванов И.И.": "Иванов_0001"}
|
||
csv_str = build_mapping_csv(mapping)
|
||
check("заголовок есть", csv_str.startswith("тип_данных,оригинал,замена"))
|
||
check("данные есть", "Иванов_0001" in csv_str)
|
||
|
||
# 2. Пустой mapping
|
||
csv_str = build_mapping_csv({})
|
||
lines = csv_str.strip().split("\n")
|
||
check("пустой — только заголовок", len(lines) == 1)
|
||
|
||
# 3. Сортировка
|
||
mapping = {"Б": "Б_0001", "А": "А_0001"}
|
||
csv_str = build_mapping_csv(mapping)
|
||
check("сортировка А перед Б", csv_str.index("А_0001") < csv_str.index("Б_0001"))
|
||
|
||
# 4. Кавычки в значениях
|
||
mapping = {'ООО "Ромашка"': "ООО_Технология_0001"}
|
||
csv_str = build_mapping_csv(mapping)
|
||
check("кавычки экранированы", 'ООО "Ромашка"' in csv_str or 'ООО ""Ромашка""' in csv_str)
|
||
|
||
|
||
# ═══════════════════════════════════════════════════════════════
|
||
# Test: интеграция build_zip + build_mapping_csv
|
||
# ═══════════════════════════════════════════════════════════════
|
||
print("\n=== build_zip + build_mapping_csv ===")
|
||
|
||
mapping = {"Иванов И.И.": "Иванов_0001", "ООО Ромашка": "ООО_Технология_0001"}
|
||
csv = build_mapping_csv(mapping)
|
||
zip_data = build_zip([("doc.md", "# Договор с Иванов_0001".encode("utf-8"))], mapping_csv=csv)
|
||
|
||
with zipfile.ZipFile(io.BytesIO(zip_data)) as zf:
|
||
check("два файла в ZIP", len(zf.namelist()) == 2)
|
||
doc = zf.read("doc.md").decode("utf-8")
|
||
check("doc с заменой", "Иванов_0001" in doc)
|
||
csv_content = zf.read("mapping.csv").decode("utf-8")
|
||
check("CSV с обеими заменами", "Иванов_0001" in csv_content and "ООО_Технология_0001" in csv_content)
|
||
|
||
|
||
# ═══════════════════════════════════════════════════════════════
|
||
print(f"\n{'='*50}")
|
||
print(f"Результат: {passed} OK, {failed} FAIL из {passed + failed}")
|
||
if failed > 0:
|
||
print("❌ ТЕСТЫ ПРОВАЛЕНЫ")
|
||
sys.exit(1)
|
||
else:
|
||
print("✅ ВСЕ ТЕСТЫ ПРОЙДЕНЫ")
|