User Management backend - Base - DB migrations

.gitignore

@@ -17,3 +17,5 @@ _pycache_/
 # Python
 *.pyc
 *.orig
+*.pyc.*
+/.cache

tests/conftest.py

@@ -0,0 +1,33 @@
+import pytest
+import psycopg2
+
+
+@pytest.fixture
+def db_connection():
+    connection = psycopg2.connect(
+        host="localhost",
+        database="edison",
+        user="postgres",
+        password="edison")
+
+    yield connection
+    connection.close()
+
+@pytest.fixture
+def cursor(db_connection):
+    cursor = db_connection.cursor()
+    yield cursor
+    db_connection.rollback()
+
+@pytest.fixture
+def build_users_table(cursor):
+    cursor.execute("""
+    INSERT INTO users (username) VALUES (%s);
+    """,
+    ("ahinoam",))
+
+@pytest.fixture
+def reset_users_primary_key(cursor):
+    cursor.execute("""
+    ALTER SEQUENCE users_id_seq RESTART;
+    """)

tests/db_test.py

@@ -0,0 +1,17 @@
+import pytest
+import psycopg2
+
+
+def test_users_table_rows_count(cursor, build_users_table):
+    cursor.execute("""
+    SELECT * FROM users;
+    """)
+    result = cursor.fetchall()
+    assert len(result) == 1
+
+def test_username(cursor, build_users_table):
+    cursor.execute("""
+    SELECT username FROM users;
+    """)
+    result = cursor.fetchone()
+    assert result[0] == "ahinoam"