@@ -189,25 +189,29 @@ def test_initial_schema_version_is_zero(self, migration_manager):
189189
190190 def test_set_and_get_schema_version (self , migration_manager ):
191191 """Test setting and retrieving schema versions."""
192- migration_manager .set_schema_version (5 )
192+ with pytest .deprecated_call ():
193+ migration_manager .set_schema_version (5 )
193194 version = migration_manager .get_schema_version ()
194195 assert version == 5
195196
196- migration_manager .set_schema_version (10 )
197+ with pytest .deprecated_call ():
198+ migration_manager .set_schema_version (10 )
197199 version = migration_manager .get_schema_version ()
198200 assert version == 10
199201
200202 def test_set_negative_version_fails (self , migration_manager ):
201203 """Test that setting negative version raises error."""
202204 with pytest .raises (ValueError , match = "version must be non-negative" ):
203- migration_manager .set_schema_version (- 1 )
205+ with pytest .deprecated_call ():
206+ migration_manager .set_schema_version (- 1 )
204207
205208 def test_version_persists_across_connections (self , temp_db_path ):
206209 """Test that schema version persists when reopening database."""
207210 # Set version in first instance
208211 manager1 = MigrationManager (db_path = temp_db_path )
209212 manager1 .initialize_schema () # Create tables first
210- manager1 .set_schema_version (7 )
213+ with pytest .deprecated_call ():
214+ manager1 .set_schema_version (7 )
211215 manager1 .close ()
212216
213217 # Verify version persists in second instance
@@ -240,22 +244,25 @@ def test_migrate_from_zero_to_one(self, migration_manager):
240244
241245 def test_migrate_same_version_is_noop (self , migration_manager ):
242246 """Test that migrating to current version is no-op."""
243- migration_manager .set_schema_version (3 )
247+ with pytest .deprecated_call ():
248+ migration_manager .set_schema_version (3 )
244249
245250 # Should not raise error and version should remain unchanged
246251 migration_manager .migrate_to_version (3 )
247252 assert migration_manager .get_schema_version () == 3
248253
249254 def test_migrate_to_lower_version_fails (self , migration_manager ):
250255 """Test that downgrading version is not allowed."""
251- migration_manager .set_schema_version (5 )
256+ with pytest .deprecated_call ():
257+ migration_manager .set_schema_version (5 )
252258
253259 with pytest .raises (ValueError , match = "Cannot downgrade from version 5 to 3" ):
254260 migration_manager .migrate_to_version (3 )
255261
256262 def test_unsupported_migration_path_fails (self , migration_manager ):
257263 """Test that unsupported migration paths raise errors."""
258- migration_manager .set_schema_version (2 )
264+ with pytest .deprecated_call ():
265+ migration_manager .set_schema_version (2 )
259266
260267 with pytest .raises (RuntimeError , match = "Migration from version 2 to 5 not supported" ):
261268 migration_manager .migrate_to_version (5 )
@@ -272,7 +279,8 @@ def test_ensure_current_schema_migrates_when_needed(self, migration_manager):
272279
273280 def test_ensure_current_schema_handles_newer_version (self , migration_manager ):
274281 """Test warning when database is newer than expected."""
275- migration_manager .set_schema_version (5 )
282+ with pytest .deprecated_call ():
283+ migration_manager .set_schema_version (5 )
276284
277285 # Should log warning but not fail
278286 with patch ('openspace.skill_engine.migration_manager.logger' ) as mock_logger :
@@ -340,17 +348,22 @@ def test_operations_fail_when_closed(self, migration_manager):
340348 """Test that operations fail after manager is closed."""
341349 migration_manager .close ()
342350
343- operations = [
344- lambda : migration_manager .initialize_schema (),
345- lambda : migration_manager .get_schema_version (),
346- lambda : migration_manager .set_schema_version (1 ),
347- lambda : migration_manager .migrate_to_version (1 ),
348- lambda : migration_manager .ensure_current_schema (1 ),
349- ]
350-
351- for operation in operations :
352- with pytest .raises (RuntimeError , match = "MigrationManager is closed" ):
353- operation ()
351+ # Test each operation individually
352+ with pytest .raises (RuntimeError , match = "MigrationManager is closed" ):
353+ migration_manager .initialize_schema ()
354+
355+ with pytest .raises (RuntimeError , match = "MigrationManager is closed" ):
356+ migration_manager .get_schema_version ()
357+
358+ with pytest .raises (RuntimeError , match = "MigrationManager is closed" ):
359+ with pytest .deprecated_call ():
360+ migration_manager .set_schema_version (1 )
361+
362+ with pytest .raises (RuntimeError , match = "MigrationManager is closed" ):
363+ migration_manager .migrate_to_version (1 )
364+
365+ with pytest .raises (RuntimeError , match = "MigrationManager is closed" ):
366+ migration_manager .ensure_current_schema (1 )
354367
355368
356369class TestSkillStoreFacadeIntegration :
0 commit comments