def test_get_serialized_address(self):
score_sheet_addr = ScoreSheetAddressFactory(offer_year=self.offer_year,
entity_address_choice=None)
address = score_encoding_sheet._get_serialized_address(self.offer_year)
self.assertEqual(address.get('country'), score_sheet_addr.country.name)
for f in self.address_fields:
self.assertIn(f, address)
def test_case_customized_address(self):
address = ScoreSheetAddressFactory(offer_year=self.offer_year,
entity_address_choice=None)
self._assert_correct_address(address)
def test_case_address_from_entity_management(self):
ScoreSheetAddressFactory(
offer_year=self.offer_year,
entity_address_choice=score_sheet_address_choices.ENTITY_MANAGEMENT
)
self._assert_correct_address(self.entity_address_manag)
def test_case_address_from_entity_administration(self):
ScoreSheetAddressFactory(
offer_year=self.offer_year,
entity_address_choice=score_sheet_address_choices.
ENTITY_ADMINISTRATION)
self._assert_correct_address(self.entity_address_admin)
def test_get_address_as_dict(self):
address1 = ScoreSheetAddressFactory(offer_year=self.offer_year)
self._assert_address_fields_are_in_object(address1)
country = CountryFactory()
address2 = EntityFactory(country=country)
self._assert_address_fields_are_in_object(address2)
