Checking if database exists and some initial fixtures which clean up

1 files changed, 24 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 00000000..2f45e2a5
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,24 @@
+import pytest
+from utils import *
+from pgcli.pgexecute import PGExecute
+
+
+@pytest.yield_fixture(scope="function")
+def connection():
+    create_db('_test_db')
+    connection = db_connection('_test_db')
+    yield connection
+
+    drop_tables(connection)
+    connection.close()
+
+
+@pytest.fixture
+def cursor(connection):
+    with connection.cursor() as cur:
+        return cur
+
+
+@pytest.fixture
+def executor(connection):
+    return PGExecute(database='_test_db', user=POSTGRES_USER, host=POSTGRES_HOST)