@@ -96,15 +96,20 @@ 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 (
109
+ substring in tasks ["results" ][0 ]["details" ]["originalFilters" ]
110
+ for substring in ["uids" , "1" , "2" ]
111
+ )
112
+
108
113
109
114
def test_cancel_every_task (client ):
110
115
"""Tests cancel every task."""
@@ -114,8 +119,9 @@ def test_cancel_every_task(client):
114
119
assert isinstance (task , TaskInfo )
115
120
assert task .task_uid is not None
116
121
assert task .index_uid is None
117
- assert task .status == "enqueued" or "processing" or "succeeded"
122
+ assert task .status in { "enqueued" , "processing" , "succeeded" }
118
123
assert task .type == "taskCancelation"
119
- assert "statuses" in tasks ["results" ][0 ]["details" ]["originalFilters" ]
120
-
121
-
124
+ assert any (
125
+ substring in tasks ["results" ][0 ]["details" ]["originalFilters" ]
126
+ for substring in ["statuses" , "enqueued" , "processing" ]
127
+ )
0 commit comments