WIP: fixing tests

Author: Filienko

migrations/migration.py

@@ -42,9 +42,7 @@ def dfs(node):
         for filename in migration_files:
             curr_migration = filename[:-3]
             prev_migration = self.get_previous_migration_id(filename)
-
-            if prev_migration is not None:
-                migration_sequence[curr_migration] = prev_migration
+            migration_sequence[curr_migration] = prev_migration
 
         # Perform DFS from each node
         for node in migration_sequence:

tests/versions/test_migration_0.py renamed to migrations/versions/test_migration_0.py

@@ -5,7 +5,7 @@
 
 @pytest.fixture
 def migration_instance():
-    return Migration(os.path.dirname(__file__))
+    return Migration()
 
 @pytest.fixture
 def mock_get_previous_migration_id():
@@ -26,8 +26,8 @@ def test_build_migration_sequence_empty(mock_get_previous_migration_id):
 
 def test_build_migration_sequence_with_dependencies(mock_get_previous_migration_id):
     # Mock the output of get_migration_files
-    fake_filenames = ['migration1.py', 'migration2.py', 'migration3.py']
-    with patch.object(Migration, 'get_migration_files', return_value=fake_filenames):
+    mock_filenames = ['migration1.py', 'migration2.py', 'migration3.py']
+    with patch.object(Migration, 'get_migration_files', return_value=mock_filenames):
         # Mock the output of get_previous_migration_id
         mock_get_previous_migration_id.side_effect = {
             'migration2.py': 'migration1',
@@ -47,8 +47,8 @@ def test_build_migration_sequence_with_dependencies(mock_get_previous_migration_
 
 def test_build_migration_sequence_with_circular_dependency(mock_get_previous_migration_id):
     # Mock the output of get_migration_files
-    fake_filenames = ['migration1.py', 'migration2.py']
-    with patch.object(Migration, 'get_migration_files', return_value=fake_filenames):
+    mock_filenames = ['migration1.py', 'migration2.py']
+    with patch.object(Migration, 'get_migration_files', return_value=mock_filenames):
         # Mock the output of get_previous_migration_id to create circular dependency
         mock_get_previous_migration_id.side_effect = {
             'migration2.py': 'migration1',