Python ListingBuilder._sanitize_address Examples

Example #1

File: test_listing_builder.py Project: huokedu/dynamic-scrapy

def test_builder_address_only_sanitized_if_untrustworthy():
  geo_service_mock = MagicMock(spec=listing_geo_service)

  listing_source_mock = MagicMock(trusted_geo_data=True)

  builder = ListingBuilder(listing_geo_service=geo_service_mock)

  builder.listing_attrs_output = MagicMock()

  builder.listing_attrs_output.get.return_value = listing_source_mock

  builder._sanitize_address()

  assert 0 == len(geo_service_mock.get_sanitized_address.call_args_list)

Example #2

File: test_listing_builder.py Project: huokedu/dynamic-scrapy

def test_builder_delegates_address_sanitization():
  geo_service_mock = MagicMock(spec=listing_geo_service)

  builder = ListingBuilder(listing_geo_service=geo_service_mock)

  listing_source_mock = MagicMock(trusted_geo_data=False)

  builder.listing_attrs_output = MagicMock()

  builder.listing_attrs_output.get.return_value = listing_source_mock

  builder._sanitize_address()

  geo_service_mock.get_sanitized_address.assert_called_with(ANY, ANY, ANY)