X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/bb175cc95898de420781867973b38d033c187e81..513175ce23674abffdd2ebe59fdb1a600c91cd01:/test/python/db/test_connection.py?ds=sidebyside diff --git a/test/python/db/test_connection.py b/test/python/db/test_connection.py index 00c29a43..dbba6109 100644 --- 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 @@ -20,6 +26,16 @@ def test_connection_table_exists(db, table_factory): 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')