Add hierarchical geography and cloud-hosted profile images

2026-10-04 21:30:30 +08:00
parent a77fcc6671
commit 427fd51e14
27 changed files with 618 additions and 50 deletions
@@ -0,0 +1,58 @@
+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_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_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'],'火星')