Update based on code review

tests/unit_tests/scheduler/conftest.py

@@ -34,30 +34,6 @@ def mock_driver():
     return MockDriver
 
 
-class MockSemaphore(asyncio.Semaphore):
-    def __init__(self, value: int):
-        super().__init__(value)
-        self._mock_locked = asyncio.Future()
-        self._mock_unlocked = asyncio.Future()
-
-    async def acquire(self) -> Coroutine[Any, Any, Literal[True]]:
-        if self._mock_locked.done():
-            self._mock_locked = asyncio.Future()
-        self._mock_locked.set_result(True)
-        return await super().acquire()
-
-    def release(self) -> None:
-        if self._mock_unlocked.done():
-            self._mock_unlocked = asyncio.Future()
-        self._mock_unlocked.set_result(True)
-        return super().release()
-
-
-@pytest.fixture
-def mock_semaphore():
-    return MockSemaphore
-
-
 class MockEvent(asyncio.Event):
     def __init__(self):
         self._mock_waited = asyncio.Future()

tests/unit_tests/scheduler/test_job.py

@@ -102,7 +102,7 @@ async def test_job_call(
     job.started.set()
     job.returncode.set_result(return_code)
 
-    await job._submit_and_run_once(asyncio.Semaphore(1))
+    await job._submit_and_run_once(asyncio.Semaphore())
 
     await assert_scheduler_events(
         scheduler,