pymorizon/tests.py at master · limebrains/pymorizon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import sys

import morizon
import morizon.category
import morizon.offer
import morizon.utils
import pytest
from bs4 import BeautifulSoup

if sys.version_info < (3, 3):
    from mock import mock
else:
    from unittest import mock


@pytest.fixture
def offer_markup():
    with open('test_data/offer_source.htm', 'r') as page:
        page_markup = page.read()
    html_markup = BeautifulSoup(page_markup, "html.parser")
    return html_markup

@pytest.mark.parametrize('args, filter, expected_value', [
    (('mieszkania', 'Gdynia', 'Witomino Leśniczówka', 'do-wynajecia'), {'[number_of_rooms_from]': 2}, 'https://www.morizon.pl/do-wynajecia/mieszkania/gdynia/witomino-lesniczowka/?page=1&ps%5Bnumber_of_rooms_from%5D=2&'),
    (('domy', 'Warszawa', 'Śródmieście'), {'[number_of_rooms_from]': 4, }, 'https://www.morizon.pl/domy/warszawa/srodmiescie/?page=1&ps%5Bnumber_of_rooms_from%5D=4&'),
    (('działki', 'Opole'), {}, 'https://www.morizon.pl/dzialki/opole/?page=1&'),
    (('mieszkania', 'Szczecin', 'Pogodno', 'do-wynajecia'), {'[living_area_from]': 35,'[number_of_rooms_from]': 2}, 'https://www.morizon.pl/do-wynajecia/mieszkania/szczecin/pogodno/?page=1&ps%5Bliving_area_from%5D=35&ps%5Bnumber_of_rooms_from%5D=2&'),
])
def test_get_url(args, filter, expected_value):
    assert morizon.utils.URL(*args, filters=filter).get_url() == expected_value


@pytest.mark.parametrize('url, city', [
    ('https://www.morizon.pl/nieruchomosci/gdynia/witomino/?page=1&', 'gdynia'),
    ('https://www.morizon.pl/do-wynajecia/mieszkania/warszawa/stary-mokotow/?page=1&ps%5Bnumber_of_rooms_from%5D=2&ps%5Bnumber_of_rooms_to%5D=4&', 'warszawa'),
])
def test_url_parsing(url, city):
    url_obj = morizon.utils.URL.from_string(url)
    assert url_obj.city == city


@pytest.mark.parametrize('text, expected_value', [
    ('ĄĆĘŁŃÓŚŹŻ', 'acelnoszz'),
    ('Zielona Góra', 'zielona-gora'),
    ('Świętojańska', 'swietojanska'),
    ('PrZykłaDowY tEXt', 'przykladowy-text')
])
def test_encode_text(text, expected_value):
    assert morizon.utils.encode_text_to_url(text) == expected_value


def test_get_area(offer_markup):
    assert morizon.offer.get_surface_for_offer(offer_markup) == 56.34


def test_get_floor(offer_markup):
    assert morizon.offer.get_floor_for_offer(offer_markup) == 0


def test_get_price(offer_markup):
    assert morizon.offer.get_price_for_offer(offer_markup) == 1600.0


def test_get_phone(offer_markup):
    assert morizon.offer.get_phone_for_offer(offer_markup) == '508 861 700'


def test_get_voivodeship(offer_markup):
    assert morizon.offer.get_voivodeship_for_offer(offer_markup) == 'zachodniopomorskie'


def test_get_images(offer_markup):
    images = morizon.offer.get_images_for_offer(offer_markup)
    assert isinstance(images, list)
    for image in images:
        assert 'https://img' in image


@pytest.fixture
def offers_gdynia():
    with open('test_data/offers_witomino.htm', 'r') as page:
        page_markup = page.read()
    html_markup = BeautifulSoup(page_markup, 'html.parser')
    return html_markup


@pytest.fixture
def offers_szczecin():
    with open('test_data/offers_szczecin.html', 'r') as page:
        page_markup = page.read()
    return page_markup


def test_get_max_page(offers_szczecin):
    with mock.patch("morizon.utils.get_content_from_source") as get_content:
        get_content.return_value = offers_szczecin
        assert morizon.utils.get_max_page('https://www.morizon.pl/do-wynajecia/mieszkania/gdynia/witomino-lesniczowka/?page=1&ps%5Bnumber_of_rooms_from%5D=2&') == 3


def test_get_offers_from_page(offers_gdynia):
    with mock.patch("morizon.utils.get_content_from_source") as get_content:
        get_content.return_value = offers_gdynia
        assert type(morizon.category.get_offers_from_page('https://www.morizon.pl/do-wynajecia/mieszkania/gdynia/witomino-lesniczowka/?page=1')) == type([])


def test_get_offer_data(offer_markup):
    with mock.patch("morizon.utils.get_content_from_source") as get_content:
        get_content.return_value = offer_markup
        assert type(morizon.offer.get_offer_data('https://www.morizon.pl/oferta/wynajem-mieszkanie-szczecin-pogodno-somosierry-51m2-mzn2028886916')) == type({})