diff options
Diffstat (limited to 'tests/test_database.py')
| -rw-r--r-- | tests/test_database.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_database.py b/tests/test_database.py index 290b985..e387123 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -34,7 +34,7 @@ def test_migrate_records_version(tmp_path): with database.connect() as connection: rows = connection.execute("SELECT version FROM schema_migrations").fetchall() - assert rows == [(1,)] + assert rows == [(1,), (2,), (3,)] def test_migrate_is_idempotent(tmp_path): @@ -46,13 +46,13 @@ def test_migrate_is_idempotent(tmp_path): with database.connect() as connection: rows = connection.execute("SELECT version FROM schema_migrations").fetchall() - assert rows == [(1,)] + assert rows == [(1,), (2,), (3,)] def test_failed_migration_rolls_back(tmp_path): database = Database(tmp_path / "test.db") - migration = database.migrations_dir / "002_failing_migration.sql" + migration = database.migrations_dir / "999_failing_migration.sql" migration.write_text( """ THIS IS INVALID SQL; @@ -68,7 +68,7 @@ def test_failed_migration_rolls_back(tmp_path): "SELECT version FROM schema_migrations" ).fetchall() - assert migrations == [(1,)] + assert migrations == [(1,), (2,), (3,)] finally: migration.unlink() |