@@ -96,15 +96,17 @@ def test_get_task_inexistent(client):
96
96
97
97
def test_cancel_tasks (client ):
98
98
"""Tests cancel a task with uid 1."""
99
- task = client .cancel_tasks ({"uids" : ["1" ]})
99
+ task = client .cancel_tasks ({"uids" : ["1" , "2" ]})
100
100
tasks = client .get_tasks ({"types" : "taskCancelation" })
101
101
102
102
assert isinstance (task , TaskInfo )
103
103
assert task .task_uid is not None
104
104
assert task .index_uid is None
105
- assert task .status == "enqueued" or "processing" or "succeeded"
105
+ assert task .status in { "enqueued" , "processing" , "succeeded" }
106
106
assert task .type == "taskCancelation"
107
107
assert "uids" in tasks ["results" ][0 ]["details" ]["originalFilters" ]
108
+ assert any (substring in tasks ["results" ][0 ]["details" ]["originalFilters" ] for substring in ["uids" , "1" , "2" ])
109
+
108
110
109
111
def test_cancel_every_task (client ):
110
112
"""Tests cancel every task."""
@@ -114,8 +116,6 @@ def test_cancel_every_task(client):
114
116
assert isinstance (task , TaskInfo )
115
117
assert task .task_uid is not None
116
118
assert task .index_uid is None
117
- assert task .status == "enqueued" or "processing" or "succeeded"
119
+ assert task .status in { "enqueued" , "processing" , "succeeded" }
118
120
assert task .type == "taskCancelation"
119
- assert "statuses" in tasks ["results" ][0 ]["details" ]["originalFilters" ]
120
-
121
-
121
+ assert any (substring in tasks ["results" ][0 ]["details" ]["originalFilters" ] for substring in ["statuses" , "enqueued" , "processing" ])
0 commit comments