from finn_eiendom.models import EiendomUnit, FinnAd
from finn_eiendom.scoring import classify_ad, score_ad


def test_score_ad_and_classify():
    ad = FinnAd(
        finnkode="462400360",
        url="https://www.finn.no/realestate/homes/ad.html?finnkode=462400360",
        title="Flott 3-roms i Ferner",
    )
    unit = EiendomUnit(
        unit_code="c-gxw-xmyum-s2a",
        estimated_selling_price=7650000,
        listing_price=7200000,
        property_type="APARTMENT",
        usable_area=77,
        rooms=4,
    )
    scores = score_ad(ad, unit, [])
    assert scores["market_position"] >= 0
    categories = classify_ad(scores)
    assert isinstance(categories, list)