test/python/conftest.py

   1 import sys
   2 from pathlib import Path
   3
   4 import psycopg2
   5 import pytest
   6
   7 # always test against the source
   8 sys.path.insert(0, str((Path(__file__) / '..' / '..' / '..').resolve()))
   9
  10 @pytest.fixture
  11 def temp_db(monkeypatch):
  12     name = 'test_nominatim_python_unittest'
  13     with psycopg2.connect(database='postgres') as conn:
  14         conn.set_isolation_level(0)
  15         with conn.cursor() as cur:
  16             cur.execute('DROP DATABASE IF EXISTS {}'.format(name))
  17             cur.execute('CREATE DATABASE {}'.format(name))
  18
  19     monkeypatch.setenv('NOMINATIM_DATABASE_DSN' , 'dbname=' + name)
  20
  21     yield name
  22
  23     with psycopg2.connect(database='postgres') as conn:
  24         conn.set_isolation_level(0)
  25         with conn.cursor() as cur:
  26             cur.execute('DROP DATABASE IF EXISTS {}'.format(name))