X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/7a874d5b97f65dc1f5a267fd53cc73c2cd5246b6..fb3353b854f40b8fd97ea942d3d9814e80e14779:/test/python/test_tools_database_import.py?ds=sidebyside

diff --git a/test/python/test_tools_database_import.py b/test/python/test_tools_database_import.py
index 2e76f5c5..e2852acb 100644
--- a/test/python/test_tools_database_import.py
+++ b/test/python/test_tools_database_import.py
@@ -209,17 +209,23 @@ def test_create_country_names(temp_db_conn, temp_db_cursor, def_config,
     temp_db_cursor.execute("""CREATE FUNCTION make_standard_name (name TEXT)
                                   RETURNS TEXT AS $$ SELECT 'a'::TEXT $$ LANGUAGE SQL
                                """)
+    temp_db_cursor.execute('CREATE TABLE country_name (country_code varchar(2), name hstore)')
+    temp_db_cursor.execute('CREATE TABLE word (code varchar(2))')
+    temp_db_cursor.execute("""INSERT INTO country_name VALUES ('us',
+                              '"name"=>"us","name:af"=>"us"')""")
     temp_db_cursor.execute("""CREATE OR REPLACE FUNCTION getorcreate_country(lookup_word TEXT,
-                                               lookup_country_code varchar(2))
+                            lookup_country_code varchar(2))
                             RETURNS INTEGER
                             AS $$
                             BEGIN
-                                INSERT INTO country_name VALUES (5, lookup_word);
+                                INSERT INTO word VALUES (lookup_country_code);
                                 RETURN 5;
                             END;
                             $$
                             LANGUAGE plpgsql;
                                """)
-    temp_db_cursor.execute('CREATE TABLE country_name (id int, country_code varchar(2), name hstore)')
     database_import.create_country_names(temp_db_conn, def_config)
-    assert temp_db_cursor.table_rows('country_name') == 4
+    if languages:
+        assert temp_db_cursor.table_rows('word') == 4
+    else:
+        assert temp_db_cursor.table_rows('word') == 5