def test_builder_gets_correct_bedroom_from_list(): bedroom_count = '2' builder = ListingBuilder(bedroom_count=[bedroom_count]) builder._build_bedroom_count() bedroom_count_attr = builder.listing_attrs_output[listing_builder.BEDROOM_COUNT] expected_bedroom_count = 2 assert bedroom_count_attr == expected_bedroom_count
def test_builder_gets_correct_bedroom_from_text(): home_parser_mock = MagicMock(spec=home_parser) expected_bedroom_count = 2 home_parser_mock.get_bedroom_count = MagicMock(return_value=expected_bedroom_count) builder = ListingBuilder(home_parser=home_parser_mock, bedroom_count='something') builder.listing_attrs_output = MagicMock() builder._build_bedroom_count() assert builder.listing_attrs_output.__setitem__.call_args_list[0] == call(listing_builder.BEDROOM_COUNT, expected_bedroom_count)
def test_builder_gets_correct_bedroom_even_if_0(): home_parser_mock = MagicMock(spec=home_parser) expected_bedroom_count = 0 home_parser_mock.get_bedroom_count = MagicMock(return_value=expected_bedroom_count) builder = ListingBuilder(home_parser=home_parser_mock) builder.listing_attrs_output = MagicMock() builder.listing_attrs_output.get.return_value = True builder._build_bedroom_count() assert builder.listing_attrs_output.__setitem__.call_args_list[0] == call(listing_builder.BEDROOM_COUNT, expected_bedroom_count)