def test_builder_joins_addresses_if_no_valid_address(): xstreet1 = 'Foo' xtreet2 = 'Bar' address_parser_mock = MagicMock(spec=address_parser) address_parser_mock.is_street_address = MagicMock(return_value=False) address_parser_mock.is_cross_street_address = MagicMock(return_value=False) builder = ListingBuilder(address_parser_mock, address=[xstreet1, xtreet2]) builder._build_address() address_parser_mock.join_cross_street.assert_called_with({'Foo', 'Bar'})
def test_builder_uses_firs_cross_street_address_to_populate(): address1 = '123 test st' address_parser_mock = MagicMock(spec=address_parser) address_parser_mock.is_street_address = MagicMock(return_value=False) address_parser_mock.is_cross_street_address = MagicMock(return_value=True) builder = ListingBuilder(address_parser_mock, address=address1) builder._build_address() address_attr = builder.listing_attrs_output[listing_builder.ADDRESS] assert address_attr is not None