initial

tests/test_search.py

@@ -0,0 +1,38 @@
+from finn_eiendom.search import extract_ad_links, extract_search_cards
+from tests.fixtures import SAMPLE_FINN_SEARCH_HTML, SAMPLE_FINN_SEARCH_HTML_NEW
+
+
+def test_extract_search_cards():
+    cards = extract_search_cards(SAMPLE_FINN_SEARCH_HTML)
+    assert len(cards) == 2
+    assert cards[0].finnkode == "462400360"
+    assert cards[0].url.endswith("finnkode=462400360")
+    assert cards[0].area_m2 == 77
+    assert cards[0].total_price == 7200991
+    assert cards[0].common_costs == 3500
+    assert cards[1].bedrooms == 2
+
+
+def test_extract_search_cards_new_format():
+    cards = extract_search_cards(SAMPLE_FINN_SEARCH_HTML_NEW)
+    assert len(cards) == 1
+    assert cards[0].finnkode == "462880791"
+    assert cards[0].url.endswith("finnkode=462880791")
+    assert cards[0].address == "Lofotgata 4B, Oslo"
+    assert cards[0].area_m2 == 62
+    assert cards[0].total_price == 7253377
+    assert cards[0].common_costs == 7067
+    assert cards[0].bedrooms == 2
+
+
+def test_extract_ad_links():
+    links = extract_ad_links(SAMPLE_FINN_SEARCH_HTML)
+    assert len(links) == 2
+    assert "finnkode=462400360" in links[0]
+    assert "finnkode=460784945" in links[1]
+
+
+def test_extract_ad_links_new_format():
+    links = extract_ad_links(SAMPLE_FINN_SEARCH_HTML_NEW)
+    assert len(links) == 1
+    assert "finnkode=462880791" in links[0]