update tests

Author: marshallweekley-eng

apps/bfd-pipeline-idr/src/idr_pipeline/batch_worker.py

@@ -234,7 +234,7 @@ async def _worker_main(self) -> None:
                 timeout=600,  # See loader.py for explanation on timeout length
             ) as pool,
         ):
-            await pool.wait()
+            await pool.wait(timeout=600)
 
             stop = anyio.Event()
             tg.start_soon(self._run_queue_bridge, task_send, stop)

apps/bfd-pipeline-idr/src/idr_pipeline/loader.py

@@ -80,7 +80,7 @@ async def _async_load(
             # TODO: Investigate pool timeout further so that this can be removed
             timeout=600,
         ) as pool:
-            await pool.wait()
+            await pool.wait(timeout=600)
             return await BatchLoader(
                 fetch_results,
                 model,