diff options
author | Frederick Muriuki Muriithi | 2022-11-08 16:40:55 +0300 |
---|---|---|
committer | Frederick Muriuki Muriithi | 2022-11-08 16:40:55 +0300 |
commit | 3a44dd8750a212b5ed26cb0882dd1ad3dc3085d6 (patch) | |
tree | fb1b9be7e703347cb237c25f18ec65ff5e653545 /tests/unit/auth | |
parent | d5980f1daef38fe831a2113c4eeccf5bcca4388d (diff) | |
download | genenetwork3-3a44dd8750a212b5ed26cb0882dd1ad3dc3085d6.tar.gz |
Migrations: Init `resource_categories` with initial categories
* migrations/auth/20221108_04_CKcSL-init-data-in-resource-categories-table.py:
new migration.
* tests/unit/auth/test_migration_init_data_in_resource_categories_table.py:
test new migration.
Diffstat (limited to 'tests/unit/auth')
-rw-r--r-- | tests/unit/auth/test_migration_init_data_in_resource_categories_table.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/tests/unit/auth/test_migration_init_data_in_resource_categories_table.py b/tests/unit/auth/test_migration_init_data_in_resource_categories_table.py new file mode 100644 index 0000000..4b5a039 --- /dev/null +++ b/tests/unit/auth/test_migration_init_data_in_resource_categories_table.py @@ -0,0 +1,28 @@ +""" +Test that the `resource_categories` table is initialised with the startup data. +""" +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/20221108_04_CKcSL-init-data-in-resource-categories-table.py" + +@pytest.mark.unit_test +def test_apply_init_data(auth_testdb_path, auth_migrations_dir, backend): + older_migrations = migrations_up_to(migration_path, auth_migrations_dir) + the_migration = get_migration(migration_path) + apply_migrations(backend, older_migrations) + with closing(sqlite3.connect(auth_testdb_path)) as conn, closing(conn.cursor()) as cursor: + cursor.execute("SELECT * FROM resource_categories") + assert len(cursor.fetchall()) == 0, "Expected empty table." + apply_single_migration(auth_testdb_path, the_migration) + cursor.execute("SELECT * FROM resource_categories") + results = cursor.fetchall() + assert len(results) == 3, "Expected 3 rows of data." + + rollback_migrations(backend, older_migrations + [the_migration]) |