Skip to content

Commit

Permalink
WIP: fixing tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Filienko committed Apr 26, 2024
1 parent 84825bd commit 2ccefcd
Showing 1 changed file with 6 additions and 7 deletions.
13 changes: 6 additions & 7 deletions tests/migration_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,15 +91,14 @@ def test_build_migration_sequence_with_dependencies(mock_get_previous_migration_

def test_get_previous_migration_id_exists(migration_instance):
migration_name = "migration123.py"
migration_content = "down_revision = 'migration123'\n"
migration_content = "# down_revision = 'migration123'\n"
migration_path = os.path.join(migration_instance.migrations_dir, migration_name)
with open(migration_path, "w") as migration_file:
migration_file.write(migration_content)

with patch("builtins.open", create=True) as mock_open:
mock_file = mock_open.return_value.__enter__.return_value
mock_file.read.return_value = migration_content

down_revision = migration_instance.get_previous_migration_id(migration_name)
down_revision = migration_instance.get_previous_migration_id(migration_name)

assert down_revision == "migration123"
assert down_revision == "migration123"


def test_get_previous_migration_id_nonexistent_file(migration_instance):
Expand Down

0 comments on commit 2ccefcd

Please sign in to comment.