def test_one_geography_with_gss(self): org = OrganisationFactory() geo = OrganisationGeographyFactory(organisation=org, gss='X01000001') self.assertEqual(geo, org.get_geography(date.today())) self.assertEqual(geo, org.get_geography("doesn't even need to be a date")) self.assertEqual('https://mapit.mysociety.org/area/X01000001', org.format_geography_link()) geo.gss = '' geo.save() self.assertEqual(None, org.format_geography_link())
def test_multiple_geographies(self): org = OrganisationFactory(start_date=date(2001, 1, 1), end_date=None) OrganisationGeographyFactory(organisation=org, gss='X01000001', start_date=None, end_date='2001-01-01') OrganisationGeographyFactory(organisation=org, gss='X01000002', start_date='2001-01-02', end_date='2002-01-01') OrganisationGeographyFactory(organisation=org, gss='X01000003', start_date='2002-01-02', end_date=None) self.assertEqual('X01000001', org.get_geography(date(2001, 1, 1)).gss) self.assertEqual('X01000002', org.get_geography(date(2001, 7, 20)).gss) self.assertEqual('X01000003', org.get_geography(date(2099, 1, 1)).gss) with self.assertRaises(ValueError): org.get_geography(date(1900, 1, 1)) # before the org start date
def test_one_geography_no_gss(self): org = OrganisationFactory() geo = OrganisationGeographyFactory(organisation=org, gss='') self.assertEqual(geo, org.get_geography(date.today())) self.assertEqual(None, org.format_geography_link())
def test_no_geographies(self): org = OrganisationFactory() self.assertEqual(None, org.get_geography(date.today())) self.assertEqual(None, org.format_geography_link())