From 8e1f297c5f788ed7b15cddd2eb62ba982c5346da Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Tue, 8 Nov 2022 09:47:04 +0300
Subject: Tests: Test migration to create user_credentials table
* tests/unit/auth/test_create_user_credentials_table.py: new tests
---
.../auth/test_create_user_credentials_table.py | 47 ++++++++++++++++++++++
1 file changed, 47 insertions(+)
create mode 100644 tests/unit/auth/test_create_user_credentials_table.py
diff --git a/tests/unit/auth/test_create_user_credentials_table.py b/tests/unit/auth/test_create_user_credentials_table.py
new file mode 100644
index 0000000..72b7dd7
--- /dev/null
+++ b/tests/unit/auth/test_create_user_credentials_table.py
@@ -0,0 +1,47 @@
+"""Test the auth database initialisation migration."""
+from contextlib import closing
+
+import pytest
+import sqlite3
+
+from gn3.migrations import get_migration, apply_migrations, rollback_migrations
+from tests.unit.auth.conftest import (
+ apply_single_migration, rollback_single_migration, migrations_up_to)
+
+migration_path = "migrations/auth/20221103_02_sGrIs-create-user-credentials-table.py"
+
+@pytest.mark.unit_test
+def test_create_user_credentials_table(auth_testdb_path, backend, all_migrations):
+ """
+ GIVEN: A database migration script to create the `user_credentials` table
+ WHEN: The migration is applied
+ THEN: Ensure that the table is created
+ """
+ older_migrations = migrations_up_to(migration_path, "migrations/auth/")
+ apply_migrations(backend, older_migrations)
+ with closing(sqlite3.connect(auth_testdb_path)) as conn, closing(conn.cursor()) as cursor:
+ cursor.execute("SELECT name FROM sqlite_schema WHERE type='table'")
+ result = cursor.fetchall()
+ assert "users_credentials" not in [row[0] for row in cursor.fetchall()]
+ apply_single_migration(auth_testdb_path, get_migration(migration_path))
+ cursor.execute("SELECT name FROM sqlite_schema WHERE type='table'")
+ assert "user_credentials" in [row[0] for row in cursor.fetchall()]
+
+ rollback_migrations(backend, older_migrations)
+
+@pytest.mark.unit_test
+def test_rollback_create_user_credentials_table(auth_testdb_path, backend):
+ """
+ GIVEN: A database migration script to create the `user_credentials` table
+ WHEN: The migration is rolled back
+ THEN: Ensure that the `user_credentials` table no longer exists
+ """
+ older_migrations = migrations_up_to(migration_path, "migrations/auth/")
+ apply_migrations(backend, older_migrations)
+ with closing(sqlite3.connect(auth_testdb_path)) as conn, closing(conn.cursor()) as cursor:
+ apply_single_migration(auth_testdb_path, get_migration(migration_path))
+ rollback_single_migration(auth_testdb_path, get_migration(migration_path))
+ cursor.execute("SELECT name FROM sqlite_schema WHERE type='table'")
+ assert "user_credentials" not in [row[0] for row in cursor.fetchall()]
+
+ rollback_migrations(backend, older_migrations)
--
cgit v1.2.3