Added unit tests for loading country info from yaml file

diff --git a/test/python/tools/test_country_info.py b/test/python/tools/test_country_info.py
index 2abb97cd3d57e5b56927623d895109b25178b74b..ae3432e77eadecefa41ae5b0d4991615a7c6c4c2 100644 (file)
--- a/test/python/tools/test_country_info.py
+++ b/test/python/tools/test_country_info.py
@@ -59,3 +59,39 @@ def test_create_country_names(temp_db_with_extensions, temp_db_conn, temp_db_cur
     else:
         assert result_set == {'us' : set(('us', 'us1', 'us2', 'United States')),
                               'fr' : set(('fr', 'Fra', 'Fren'))}
+
+@pytest.mark.parametrize("yaml_file_content", (
+"""
+de:
+    partition: 3
+    names: 
+        name: 
+            default: Deutschland
+""",
+"""
+de:
+    partition: 3
+    languages: de
+    names:
+""",
+"""
+de:
+    partition: 3
+    languages: de
+"""
+))
+def test_load(project_env, def_config, yaml_file_content):
+    (project_env.project_dir / 'country_settings.yaml').write_text(yaml_file_content)
+    
+    country_info._COUNTRY_INFO._info = def_config.load_sub_configuration(
+        (project_env.project_dir / 'country_settings.yaml'))
+    
+    for prop in country_info._COUNTRY_INFO._info.values():
+        if 'languages' not in prop:
+            prop['languages'] = []
+            assert country_info._COUNTRY_INFO._info == {'de': {'partition': 3,
+            'languages': [], 'names': {'name': {'default': 'Deutschland'}}}}
+        if 'names' not in prop or prop['names'] is None:
+            prop['names'] = {'name': {}}
+            assert country_info._COUNTRY_INFO._info == {'de': {'partition': 3,
+            'languages': 'de', 'names': {'name': {}}}}
\ No newline at end of file