ucvl-home-vision/test_geography.py

import unittest
import geography
from test_app import app
import test_appearance as appearance_fixture


class GeographyUnitTests(unittest.TestCase):
    def norm(self, value): return geography.normalize(value, app.clean_text)

    def test_codes_resolve_canonical_names_and_reject_wrong_parent(self):
        v=dict(mode='china',provinceCode='37',cityCode='3706',districtCode='370682',city='假名字')
        result=self.norm(v)
        self.assertEqual((result['province'],result['city'],result['district']),('山东省','烟台市','莱阳市'))
        with self.assertRaises(ValueError):self.norm(dict(v,provinceCode='11'))
        with self.assertRaises(ValueError):self.norm(dict(v,cityCode=''))
        with self.assertRaises(ValueError):self.norm(dict(v,districtCode='999999'))

    def test_partial_unknown_and_manual_paths(self):
        self.assertEqual(self.norm(dict(mode='china'))['province'],'')
        self.assertEqual(self.norm(dict(mode='china',provinceCode='81',city='香港岛',district='中西区'))['district'],'中西区')
        result=self.norm(dict(mode='world',countryCode='US',province='California',city='Los Angeles'))
        self.assertEqual(result['country'],'美国')
        with self.assertRaises(ValueError):self.norm(dict(mode='world',countryCode='ZZ'))
        with self.assertRaises(ValueError):self.norm(dict(mode='world',countryCode='CN'))
        result=self.norm(dict(mode='custom',body='火星',path='基地 / 一区 / 一栋'))
        self.assertEqual(result['body'],'火星');self.assertEqual(result['countryCode'],'')

    def test_invalid_country_types_have_a_validation_error(self):
        for code in ([], {}, None, 42, True):
            with self.subTest(code=code),self.assertRaises(ValueError):
                self.norm(dict(mode='world',countryCode=code))

    def test_legacy_edits_drop_stale_codes_and_other_updates_preserve(self):
        old=dict(geography=self.norm(dict(mode='china',provinceCode='37',cityCode='3706')),province='山东省',city='烟台市')
        item={};geography.apply(item,{},old,app.clean_text,app.Problem)
        self.assertEqual(item['geography']['cityCode'],'3706')
        geography.apply(item,dict(city='登州府'),old,app.clean_text,app.Problem)
        self.assertEqual(item['geography']['mode'],'custom');self.assertEqual(item['geography']['city'],'登州府')
        self.assertEqual(item['geography']['cityCode'],'')


class GeographyIntegrationTests(unittest.TestCase):
    setUp=appearance_fixture.AppearanceTests.setUp
    stop_server=appearance_fixture.AppearanceTests.stop_server
    request=appearance_fixture.AppearanceTests.request

    def test_bad_country_input_returns_400_and_preserves_profile(self):
        old=app.get_object('households',self.aid)
        status,result=self.request('/api/households',self.root,dict(id=self.aid,name='不应写入',geography=dict(mode='world',countryCode=[])))
        self.assertEqual(status,400);self.assertIn('国家',result['error'])
        self.assertEqual(app.get_object('households',self.aid),old)

    def test_household_cover_and_path_roundtrip(self):
        v=dict(mode='china',provinceCode='37',cityCode='3706',districtCode='370682')
        status,saved=self.request('/api/households',self.root,dict(id=self.aid,name='封面与地域',geography=v,coverData=appearance_fixture.image_data()))
        self.assertEqual(status,200);self.assertEqual(saved['district'],'莱阳市')
        listed=self.request('/api/households',self.root)[1]['households']
        home=next(x for x in listed if x['id']==self.aid)
        self.assertEqual(home['geography']['districtCode'],'370682');self.assertTrue(home['coverUrl'])
        self.assertNotIn('coverData',home)
        code,_=self.request('/api/households',self.owner,dict(id=self.aid,name='错误地区',profileRevision=1,geography=dict(v,cityCode='1101')))
        self.assertEqual(code,400)
        self.assertEqual(app.get_object('households',self.aid)['name'],'封面与地域')

    def test_person_and_site_preserve_world_and_celestial_locations(self):
        person=app.FAMILY.person(dict(name='人物',birthGeography=dict(mode='world',countryCode='JP',province='东京都'),nativeGeography=dict(mode='custom',body='地球',path='登州府 / 某村')),None)
        self.assertEqual(person['birthCountry'],'日本');self.assertEqual(person['nativeGeography']['path'],'登州府 / 某村')
        site=app.save_site(dict(name='未来基地',geography=dict(mode='custom',body='火星',path='基地 / 居住区 / 一区')),self.root)
        self.assertEqual(site['geography']['body'],'火星')