def test_extract_location_success():
""" Testing extracting location data from recipient hash using the DUNS table """
recipient_hash = "00077a9a-5a70-8919-fd19-330762af6b84"
mommy.make("recipient.RecipientLookup", **TEST_RECIPIENT_LOOKUPS[recipient_hash])
country_code = TEST_RECIPIENT_LOCATIONS[recipient_hash]["country_code"]
mommy.make("references.RefCountryCode", **TEST_REF_COUNTRY_CODE[country_code])
additional_blank_fields = ["address_line3", "foreign_province", "county_name", "foreign_postal_code"]
expected_location = TEST_RECIPIENT_LOCATIONS[recipient_hash].copy()
expected_location["country_name"] = TEST_REF_COUNTRY_CODE[country_code]["country_name"]
for additional_blank_field in additional_blank_fields:
expected_location[additional_blank_field] = None
for k in MAP_DUNS_TO_CONTRACT:
expected_location[MAP_DUNS_TO_CONTRACT[k]] = expected_location[k]
del expected_location[k]
location = recipients.extract_location(recipient_hash)
assert location == expected_location
def test_extract_location_success():
""" Testing extracting location data from recipient hash using the DUNS table """
recipient_hash = '00077a9a-5a70-8919-fd19-330762af6b84'
mommy.make(RecipientLookup, **TEST_RECIPIENT_LOOKUPS[recipient_hash])
country_code = TEST_RECIPIENT_LOCATIONS[recipient_hash]['country_code']
mommy.make(RefCountryCode, **TEST_REF_COUNTRY_CODE[country_code])
additional_blank_fields = ['address_line3', 'foreign_province', 'county_name', 'foreign_postal_code']
expected_location = TEST_RECIPIENT_LOCATIONS[recipient_hash].copy()
expected_location['country_name'] = TEST_REF_COUNTRY_CODE[country_code]['country_name']
for additional_blank_field in additional_blank_fields:
expected_location[additional_blank_field] = None
for k in MAP_DUNS_TO_CONTRACT:
expected_location[MAP_DUNS_TO_CONTRACT[k]] = expected_location[k]
del expected_location[k]
location = recipients.extract_location(recipient_hash)
assert location == expected_location