def test_that_throws_error_when_bedrooms_not_found(self, get_anemities,
get_no_of_bedrooms,
get_property_type,
get_name):
get_name.return_value = 'test name'
get_property_type.return_value = 'test type'
get_no_of_bedrooms.return_value = 3
get_anemities.return_value = ['test1', 'test2']
with pytest.raises(ParsingException):
PropertyParser.parse('html')
def test_that_gets_correct_data_when_single_numbers(
self, page_html_single_values):
data = PropertyParser.parse(page_html_single_values)
assert data.name == 'Property Name'
assert data.property_type == 'Property Type'
assert data.no_of_bedrooms == 1
assert data.no_of_bathrooms == 2
def test_that_builds_property_model_correctly(self, get_anemities,
get_no_of_bathrooms,
get_no_of_bedrooms,
get_property_type, get_name):
get_name.return_value = 'test name'
get_property_type.return_value = 'test type'
get_no_of_bedrooms.return_value = 2
get_no_of_bathrooms.return_value = 3
get_anemities.return_value = ['test1', 'test2']
data = PropertyParser.parse("Some HTML")
assert data.name == "test name"
assert data.property_type == 'test type'
assert data.no_of_bedrooms == 2
assert data.no_of_bathrooms == 3
assert len(data.anemities) == 2
assert data.anemities_contains('test1', 'test2')