tests: remember to exit loop for fake worker

Haidra-Org · Jul 7, 2024 · b63dc1d · b63dc1d
1 parent 8097e56
commit b63dc1d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py b/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py
@@ -151,9 +151,10 @@ async def fake_worker(
             )
 
             assert isinstance(job_submit_response, JobSubmitResponse)
-
             assert job_submit_response.reward is not None and job_submit_response.reward > 0
 
+            break
+
     @pytest.mark.api_side_ci
     @pytest.mark.asyncio
     async def test_basic_image_roundtrip(self, simple_image_gen_request: ImageGenerateAsyncRequest) -> None: