1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
"""
Test that the `resource_categories` table is initialised with the startup data.
"""
import pytest
from gn_auth.auth import db
from gn_auth.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):
"""
GIVEN: A migration script
WHEN: The migration is applied
THEN: Verify that the expected data exists in the table
"""
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, None) as conn, db.cursor(conn) as cursor:
cursor.execute("SELECT * FROM resource_categories")
assert len(cursor.fetchall()) == 0, "Expected empty table."
apply_single_migration(backend, the_migration)
cursor.execute("SELECT * FROM resource_categories")
results = cursor.fetchall()
assert len(results) == 3, "Expected 3 rows of data."
assert sorted(results) == sorted((
('fad071a3-2fc8-40b8-992b-cdefe7dcac79', 'mrna', 'mRNA Dataset'),
('548d684b-d4d1-46fb-a6d3-51a56b7da1b3', 'phenotype',
'Phenotype (Publish) Dataset'),
('48056f84-a2a6-41ac-8319-0e1e212cba2a', 'genotype',
'Genotype Dataset')))
rollback_migrations(backend, older_migrations + [the_migration])
@pytest.mark.unit_test
def test_rollback_init_data(auth_testdb_path, auth_migrations_dir, backend):
"""
GIVEN: A migration script
WHEN: The migration is rolled back
THEN: Verify that the table is empty
"""
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, None) as conn, db.cursor(conn) as cursor:
cursor.execute("SELECT * FROM resource_categories")
assert len(cursor.fetchall()) == 0, "Expected empty table."
apply_single_migration(backend, the_migration)
cursor.execute("SELECT * FROM resource_categories")
results = cursor.fetchall()
assert len(results) == 3, "Expected 3 rows of data."
rollback_single_migration(backend, the_migration)
cursor.execute("SELECT * FROM resource_categories")
assert len(cursor.fetchall()) == 0, "Expected empty table."
rollback_migrations(backend, older_migrations)
|
