Merge pull request #3384 from mtmail/geocodejson-admin-levels-only-boundaries

[nominatim.git] / test / python / db / test_connection.py

diff --git a/test/python/db/test_connection.py b/test/python/db/test_connection.py
index 00c29a43a9a8bdf32f9531199ac0b6335fb5a502..dbba61093816bf7bde2d07f9013eb46eee0972c5 100644 (file)
--- a/test/python/db/test_connection.py
+++ b/test/python/db/test_connection.py
@@ -1,5 +1,11 @@
+# SPDX-License-Identifier: GPL-2.0-only
+#
+# This file is part of Nominatim. (https://nominatim.org)
+#
+# Copyright (C) 2022 by the Nominatim developer community.
+# For a full list of authors see the git log.

 """
 """

-Tests for specialised conenction and cursor classes.
+Tests for specialised connection and cursor classes.

 """
 import pytest
 import psycopg2
 """
 import pytest
 import psycopg2


@@ -20,6 +26,16 @@ def test_connection_table_exists(db, table_factory):
     assert db.table_exists('foobar')
 
 
     assert db.table_exists('foobar')
 
 

+def test_has_column_no_table(db):
+    assert not db.table_has_column('sometable', 'somecolumn')
+
+
+@pytest.mark.parametrize('name,result', [('tram', True), ('car', False)])
+def test_has_column(db, table_factory, name, result):
+    table_factory('stuff', 'tram TEXT')
+
+    assert db.table_has_column('stuff', name) == result
+

 def test_connection_index_exists(db, table_factory, temp_db_cursor):
     assert not db.index_exists('some_index')
 
 def test_connection_index_exists(db, table_factory, temp_db_cursor):
     assert not db.index_exists('some_index')