Skip to content

Commit 4ff1586

Browse files
Merge pull request #1028 from Rafi-Mohammad-69/psl-codeqlfixes
chore: codeQL fixes
2 parents b313321 + 80ed897 commit 4ff1586

2 files changed

Lines changed: 7 additions & 4 deletions

File tree

src/tests/backend/common/database/test_database_base.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -493,13 +493,16 @@ async def delete_team_agent(self, team_id, agent_name): pass
493493
async def get_team_agent(self, team_id, agent_name): return None
494494

495495
database = MockDatabase()
496-
497-
with pytest.raises(ValueError, match="Test exception"):
496+
497+
async def run_database_context():
498498
async with database:
499499
assert database.initialized is True
500500
# Raise an exception to test cleanup
501501
raise ValueError("Test exception")
502502

503+
with pytest.raises(ValueError, match="Test exception"):
504+
await run_database_context()
505+
503506
# Even with exception, close should have been called
504507
assert database.closed is True
505508

src/tests/backend/v4/config/test_settings.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -344,7 +344,7 @@ async def cancel_task():
344344
cancel_task_handle = asyncio.create_task(cancel_task())
345345

346346
with self.assertRaises(asyncio.CancelledError):
347-
await task
347+
_ = await task
348348

349349
_ = await cancel_task_handle
350350
self.assertTrue(task.cancelled())
@@ -365,7 +365,7 @@ async def cancel_task():
365365
cancel_task_handle = asyncio.create_task(cancel_task())
366366

367367
with self.assertRaises(asyncio.CancelledError):
368-
await task
368+
_ = await task
369369

370370
_ = await cancel_task_handle
371371
self.assertTrue(task.cancelled())

0 commit comments

Comments
 (0)