Fixes test when run with suite

PrefectHQ · Feb 2, 2023 · c5cec8e · c5cec8e
1 parent 02a4fb0
commit c5cec8e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/orion/models/test_work_queues.py b/tests/orion/models/test_work_queues.py
@@ -49,8 +49,9 @@ async def test_create_work_queue_when_no_default_pool(self, session):
         pool = await models.workers.read_work_pool_by_name(
             session=session, work_pool_name=DEFAULT_AGENT_WORK_POOL_NAME
         )
-        await models.workers.delete_work_pool(session=session, work_pool_id=pool.id)
-        await session.commit()
+        if pool is not None:
+            await models.workers.delete_work_pool(session=session, work_pool_id=pool.id)
+            await session.commit()
 
         work_queue = await models.work_queues.create_work_queue(
             session=session,