mirror of
https://github.com/complexcaresolutions/dak.c2s.git
synced 2026-03-17 20:43:41 +00:00
96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
"""Tests for the ICD service — normalize, validate, coding template."""
|
|
|
|
from io import BytesIO
|
|
from unittest.mock import MagicMock, PropertyMock
|
|
|
|
import pytest
|
|
from openpyxl import load_workbook
|
|
|
|
from app.services.icd_service import generate_coding_template, normalize_and_validate_icd
|
|
|
|
|
|
# ── normalize_and_validate_icd ─────────────────────────────────────
|
|
|
|
|
|
class TestNormalizeAndValidateICD:
|
|
def test_normalize_and_validate_single(self):
|
|
result = normalize_and_validate_icd("C50.1")
|
|
assert result == [("C50.1", "C50")]
|
|
|
|
def test_normalize_and_validate_multiple(self):
|
|
result = normalize_and_validate_icd("C50.1, C79.5")
|
|
assert result == [("C50.1", "C50"), ("C79.5", "C79")]
|
|
|
|
def test_normalize_and_validate_semicolon(self):
|
|
result = normalize_and_validate_icd("c50;D12.3")
|
|
assert result == [("C50", "C50"), ("D12.3", "D12")]
|
|
|
|
def test_normalize_and_validate_invalid(self):
|
|
with pytest.raises(ValueError, match="Invalid ICD code format"):
|
|
normalize_and_validate_icd("XYZ")
|
|
|
|
def test_normalize_and_validate_empty(self):
|
|
result = normalize_and_validate_icd("")
|
|
assert result == []
|
|
|
|
|
|
# ── generate_coding_template ──────────────────────────────────────
|
|
|
|
|
|
class TestGenerateCodingTemplate:
|
|
def test_generate_coding_template_returns_bytes(self):
|
|
"""Mock the DB session and verify the template is valid xlsx bytes."""
|
|
from datetime import date
|
|
|
|
# Create mock cases
|
|
mock_case = MagicMock()
|
|
mock_case.id = 1
|
|
mock_case.fall_id = "FALL-001"
|
|
mock_case.nachname = "Mustermann"
|
|
mock_case.vorname = "Max"
|
|
mock_case.fallgruppe = "onko"
|
|
mock_case.datum = date(2026, 2, 24)
|
|
|
|
# Build a mock query chain
|
|
mock_db = MagicMock(spec=["query"])
|
|
mock_query = MagicMock()
|
|
mock_db.query.return_value = mock_query
|
|
mock_query.filter.return_value = mock_query
|
|
mock_query.count.return_value = 1
|
|
mock_query.order_by.return_value = mock_query
|
|
mock_query.offset.return_value = mock_query
|
|
mock_query.limit.return_value = mock_query
|
|
mock_query.all.return_value = [mock_case]
|
|
|
|
result = generate_coding_template(mock_db, jahr=2026, fallgruppe="onko")
|
|
|
|
# Verify it is non-empty bytes
|
|
assert isinstance(result, bytes)
|
|
assert len(result) > 0
|
|
|
|
# Verify it is valid xlsx
|
|
wb = load_workbook(BytesIO(result))
|
|
ws = wb.active
|
|
assert ws.title == "ICD Coding"
|
|
|
|
# Verify header row
|
|
headers = [cell.value for cell in ws[1]]
|
|
assert headers == [
|
|
"Case_ID",
|
|
"Fall_ID",
|
|
"Nachname",
|
|
"Vorname",
|
|
"Fallgruppe",
|
|
"Datum",
|
|
"ICD",
|
|
]
|
|
|
|
# Verify data row
|
|
row2 = [cell.value for cell in ws[2]]
|
|
assert row2[0] == 1
|
|
assert row2[1] == "FALL-001"
|
|
assert row2[2] == "Mustermann"
|
|
assert row2[3] == "Max"
|
|
assert row2[4] == "onko"
|
|
assert row2[5] == "2026-02-24"
|
|
assert row2[6] is None # ICD column should be empty
|