39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
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]
|