Fixed tests

Acly · Sep 30, 2023 · 4cadd46 · 4cadd46
1 parent 94dc23e
commit 4cadd46
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/ai_diffusion/__init__.py b/ai_diffusion/__init__.py
@@ -1,6 +1,6 @@
 """Generative AI plugin for Krita using Stable Diffusion"""
 
-__version__ = "1.2.0"
+__version__ = "1.2.1"
 
 from . import util
 from .settings import Settings, Setting, settings, PerformancePreset, ServerBackend, ServerMode

diff --git a/tests/test_server.py b/tests/test_server.py
@@ -100,7 +100,10 @@ async def main():
 
 
 def test_python_version(qtapp):
-    py = server.get_python_version("python")
-    assert py.startswith("Python 3.")
-    pip = server.get_python_version("pip")
-    assert pip.startswith("pip ")
+    async def main():
+        py = await server.get_python_version("python")
+        assert py.startswith("Python 3.")
+        pip = await server.get_python_version("pip")
+        assert pip.startswith("pip ")
+
+    qtapp.run(main())
diff --git a/tests/test_workflow.py b/tests/test_workflow.py
@@ -53,8 +53,10 @@ def default_style(comfy, sd_ver=SDVersion.sd1_5):
 
 
 async def receive_images(comfy: Client, workflow: ComfyWorkflow):
-    job_id = await comfy.enqueue(workflow)
+    job_id = None
     async for msg in comfy.listen():
+        if not job_id:
+            job_id = await comfy.enqueue(workflow)
         if msg.event is ClientEvent.finished and msg.job_id == job_id:
             return msg.images
         if msg.event is ClientEvent.error and msg.job_id == job_id: