cli: import python modules for commands on demand

[nominatim.git] / test / python / conftest.py

diff --git a/test/python/conftest.py b/test/python/conftest.py
index 26bbb397614ef4b6c408002ba0d03408890733fe..1cc9ef9c718eb2884077270f6ce76c42dc267078 100644 (file)
--- a/test/python/conftest.py
+++ b/test/python/conftest.py
@@ -1,6 +1,26 @@
 import sys
 from pathlib import Path
 
 import sys
 from pathlib import Path
 

+import psycopg2
+import pytest
+

 # always test against the source
 sys.path.insert(0, str((Path(__file__) / '..' / '..' / '..').resolve()))
 
 # always test against the source
 sys.path.insert(0, str((Path(__file__) / '..' / '..' / '..').resolve()))
 

+@pytest.fixture
+def temp_db(monkeypatch):
+    name = 'test_nominatim_python_unittest'
+    with psycopg2.connect(database='postgres') as conn:
+        conn.set_isolation_level(0)
+        with conn.cursor() as cur:
+            cur.execute('DROP DATABASE IF EXISTS {}'.format(name))
+            cur.execute('CREATE DATABASE {}'.format(name))
+
+    monkeypatch.setenv('NOMINATIM_DATABASE_DSN' , 'dbname=' + name)
+
+    yield name
+
+    with psycopg2.connect(database='postgres') as conn:
+        conn.set_isolation_level(0)
+        with conn.cursor() as cur:
+            cur.execute('DROP DATABASE IF EXISTS {}'.format(name))