aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/unit/auth/test_migrations_add_data_to_table.py79
1 files changed, 79 insertions, 0 deletions
diff --git a/tests/unit/auth/test_migrations_add_data_to_table.py b/tests/unit/auth/test_migrations_add_data_to_table.py
new file mode 100644
index 0000000..acd1f6f
--- /dev/null
+++ b/tests/unit/auth/test_migrations_add_data_to_table.py
@@ -0,0 +1,79 @@
+"""Test data insertion when migrations are run."""
+import pytest
+
+from gn3.auth import db
+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)
+
+test_params = (
+ ("20221116_01_nKUmX-add-privileges-to-group-leader-role.py",
+ ("SELECT role_id, privilege_id FROM role_privileges "
+ "WHERE role_id=? AND privilege_id IN (?, ?, ?, ?)"),
+ ("a0e67630-d502-4b9f-b23f-6805d0f30e30",
+ "221660b1-df05-4be1-b639-f010269dbda9",
+ "7bcca363-cba9-4169-9e31-26bdc6179b28",
+ "5103cc68-96f8-4ebb-83a4-a31692402c9b",
+ "1c59eff5-9336-4ed2-a166-8f70d4cb012e"),
+ (("a0e67630-d502-4b9f-b23f-6805d0f30e30",
+ "221660b1-df05-4be1-b639-f010269dbda9"),
+ ("a0e67630-d502-4b9f-b23f-6805d0f30e30",
+ "7bcca363-cba9-4169-9e31-26bdc6179b28"),
+ ("a0e67630-d502-4b9f-b23f-6805d0f30e30",
+ "5103cc68-96f8-4ebb-83a4-a31692402c9b"),
+ ("a0e67630-d502-4b9f-b23f-6805d0f30e30",
+ "1c59eff5-9336-4ed2-a166-8f70d4cb012e"))),)
+
+@pytest.mark.unit_test
+@pytest.mark.parametrize("migration_file,query,query_params,data", test_params)
+def test_apply_insert(# pylint: disable=[too-many-arguments]
+ auth_migrations_dir, backend, auth_testdb_path, migration_file, query,
+ query_params, data):
+ """
+ GIVEN: a database migration script
+ WHEN: the script is applied
+ THEN: ensure the given data exists in the table
+ """
+ migration_path=f"{auth_migrations_dir}/{migration_file}"
+ older_migrations = migrations_up_to(migration_path, auth_migrations_dir)
+ the_migration = get_migration(migration_path)
+ apply_migrations(backend, older_migrations)
+ with db.connection(auth_testdb_path) as conn, db.cursor(conn) as cursor:
+ cursor.execute(query, query_params)
+ result_before_migration = cursor.fetchall()
+ apply_single_migration(backend, the_migration)
+ cursor.execute(query, query_params)
+ result_after_migration = cursor.fetchall()
+
+ rollback_migrations(backend, older_migrations + [the_migration])
+ assert len(result_before_migration) == 0, "Expected no results before migration"
+ assert sorted(result_after_migration) == sorted(data)
+
+@pytest.mark.unit_test
+@pytest.mark.parametrize("migration_file,query,query_params,data", test_params)
+def test_rollback_insert(# pylint: disable=[too-many-arguments]
+ auth_migrations_dir, backend, auth_testdb_path, migration_file, query,
+ query_params, data):
+ """
+ GIVEN: a database migration script
+ WHEN: the script is rolled back
+ THEN: ensure the given data no longer exists in the database
+ """
+ migration_path=f"{auth_migrations_dir}/{migration_file}"
+ older_migrations = migrations_up_to(migration_path, auth_migrations_dir)
+ the_migration = get_migration(migration_path)
+ apply_migrations(backend, older_migrations)
+ with db.connection(auth_testdb_path) as conn, db.cursor(conn) as cursor:
+ cursor.execute(query, query_params)
+ result_before_migration = cursor.fetchall()
+ apply_single_migration(backend, the_migration)
+ cursor.execute(query, query_params)
+ result_after_migration = cursor.fetchall()
+ rollback_single_migration(backend, the_migration)
+ cursor.execute(query, query_params)
+ result_after_rollback = cursor.fetchall()
+
+ rollback_migrations(backend, older_migrations)
+ assert len(result_before_migration) == 0, "Expected no results before migration"
+ assert sorted(result_after_migration) == sorted(data)
+ assert len(result_after_rollback) == 0, "Expected no results after rollback"