diff options
| author | Frederick Muriuki Muriithi | 2022-11-15 06:27:33 +0300 |
|---|---|---|
| committer | Frederick Muriuki Muriithi | 2022-11-15 06:27:33 +0300 |
| commit | d043cd59127a1031de20b12562712b6e3c50feef (patch) | |
| tree | befe3233dc4136158e300e8ffeef6db41b11e2ac /tests/unit/auth/test_migrations_drop_tables.py | |
| parent | 3dc9a8a4f413d142e84a81f9c1abafedb779d7dd (diff) | |
| download | genenetwork3-d043cd59127a1031de20b12562712b6e3c50feef.tar.gz | |
pylint: Fix a myriad of linting errors
Diffstat (limited to 'tests/unit/auth/test_migrations_drop_tables.py')
| -rw-r--r-- | tests/unit/auth/test_migrations_drop_tables.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/tests/unit/auth/test_migrations_drop_tables.py b/tests/unit/auth/test_migrations_drop_tables.py index dfab3f6..2362c77 100644 --- a/tests/unit/auth/test_migrations_drop_tables.py +++ b/tests/unit/auth/test_migrations_drop_tables.py @@ -1,9 +1,8 @@ """Test migrations that create tables""" -from contextlib import closing import pytest -import sqlite3 +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) @@ -15,7 +14,7 @@ test_params = ( @pytest.mark.unit_test @pytest.mark.parametrize("migration_file,the_table", test_params) def test_drop_table( - auth_testdb_path, auth_migrations_dir, backend, all_migrations, + auth_testdb_path, auth_migrations_dir, backend, migration_file, the_table): """ GIVEN: A database migration script to create table, `the_table` @@ -26,7 +25,7 @@ def test_drop_table( 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: + with db.connection(auth_testdb_path) as conn, db.cursor(conn) as cursor: cursor.execute("SELECT name FROM sqlite_schema WHERE type='table'") result_before_migration = cursor.fetchall() apply_single_migration(backend, the_migration) @@ -51,7 +50,7 @@ def test_rollback_drop_table( 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: + with db.connection(auth_testdb_path) as conn, db.cursor(conn) as cursor: apply_single_migration(backend, the_migration) cursor.execute("SELECT name FROM sqlite_schema WHERE type='table'") result_after_migration = cursor.fetchall() |