import pytest
from app.modules.invoices.application.invoice_truth.classifier import TruthClassifierMixin

class DummyClassifier(TruthClassifierMixin):
    pass

def test_classify_item_transport_service():
    classifier = DummyClassifier()
    res = classifier.classify_item(
        item_name="Cước bốc xúc, vận chuyển cát từ bến Cống Thống",
        unit="m3",
        quantity=2001.1,
        unit_price=62000,
        total_amount=124068200
    )
    assert res["classification"] == "TRANSPORT_SERVICE"