wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-22 12:30:50 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix failing test

Browse Source
This commit is contained in:
Richard Yu 2025-12-04 14:07:49 -08:00
parent 5ed90a1583
commit b14ae80c75
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/execution/test_execution.py
View File

@ -99,7 +99,7 @@ class ComfyClient:
with urllib.request.urlopen(url) as response: with urllib.request.urlopen(url) as response:
return json.loads(response.read()) return json.loads(response.read())
def get_jobs(self, status=None, limit=None, offset=None, sort=None, order=None): def get_jobs(self, status=None, limit=None, offset=None, sort_by=None, sort_order=None):
url = "http://{}/api/jobs".format(self.server_address) url = "http://{}/api/jobs".format(self.server_address)
params = {} params = {}
if status is not None: if status is not None:
@ -108,10 +108,10 @@ class ComfyClient:
params["limit"] = limit params["limit"] = limit
if offset is not None: if offset is not None:
params["offset"] = offset params["offset"] = offset
if sort is not None: if sort_by is not None:
params["sort"] = sort params["sort_by"] = sort_by
if order is not None: if sort_order is not None:
params["order"] = order params["sort_order"] = sort_order
if params: if params:
url_values = urllib.parse.urlencode(params) url_values = urllib.parse.urlencode(params)
@ -983,8 +983,8 @@ class TestExecution:
for _ in range(3): for _ in range(3):
self._create_history_item(client, builder) self._create_history_item(client, builder)
desc_jobs = client.get_jobs(order="desc") desc_jobs = client.get_jobs(sort_order="desc")
asc_jobs = client.get_jobs(order="asc") asc_jobs = client.get_jobs(sort_order="asc")
if len(desc_jobs["jobs"]) >= 2: if len(desc_jobs["jobs"]) >= 2:
desc_times = [j["create_time"] for j in desc_jobs["jobs"] if j["create_time"]] desc_times = [j["create_time"] for j in desc_jobs["jobs"] if j["create_time"]]