Resolved copilot review comments

Author: Prekshith-Microsoft

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

@@ -494,17 +494,13 @@ async def get_team_agent(self, team_id, agent_name): return None
 
         database = MockDatabase()
 
-        exception_raised = False
-        try:
+        with pytest.raises(ValueError, match="Test exception"):
             async with database:
                 assert database.initialized is True
                 # Raise an exception to test cleanup
                 raise ValueError("Test exception")
-        except ValueError:
-            exception_raised = True
 
         # Even with exception, close should have been called
-        assert exception_raised is True
         assert database.closed is True
 
 

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

@@ -347,7 +347,7 @@ async def cancel_task():
             _ = await task
 
         await cancel_task_handle
-        self.assertTrue(cancel_task_handle.done())
+        self.assertTrue(task.cancelled())
 
     async def test_wait_for_clarification_cancelled(self):
         """Test waiting for clarification when cancelled."""
@@ -368,7 +368,7 @@ async def cancel_task():
             _ = await task
 
         await cancel_task_handle
-        self.assertTrue(cancel_task_handle.done())
+        self.assertTrue(task.cancelled())
 
     def test_cleanup_approval(self):
         """Test cleanup approval."""