27
27
from opentelemetry .sdk .trace import TracerProvider , export
28
28
from opentelemetry .trace .status import StatusCanonicalCode
29
29
30
+ # set a high timeout for async executions due to issues in CI
31
+ ASYNC_GET_TIMEOUT = 120
32
+
30
33
31
34
class MyException (Exception ):
32
35
pass
@@ -38,7 +41,7 @@ def fn_task():
38
41
return 42
39
42
40
43
result = fn_task .apply_async ()
41
- assert result .get () == 42
44
+ assert result .get (timeout = ASYNC_GET_TIMEOUT ) == 42
42
45
43
46
spans = memory_exporter .get_finished_spans ()
44
47
assert len (spans ) == 2
@@ -144,7 +147,7 @@ def fn_task_parameters(user, force_logout=False):
144
147
result = fn_task_parameters .apply_async (
145
148
args = ["user" ], kwargs = {"force_logout" : True }
146
149
)
147
- assert result .get (timeout = 10 ) == ["user" , True ]
150
+ assert result .get (timeout = ASYNC_GET_TIMEOUT ) == ["user" , True ]
148
151
149
152
spans = memory_exporter .get_finished_spans ()
150
153
assert len (spans ) == 2
@@ -181,7 +184,7 @@ def fn_task():
181
184
results = [fn_task .delay () for _ in range (100 )]
182
185
183
186
for result in results :
184
- assert result .get (timeout = 1 ) == 42
187
+ assert result .get (timeout = ASYNC_GET_TIMEOUT ) == 42
185
188
186
189
spans = memory_exporter .get_finished_spans ()
187
190
@@ -194,7 +197,7 @@ def fn_task_parameters(user, force_logout=False):
194
197
return (user , force_logout )
195
198
196
199
result = fn_task_parameters .delay ("user" , force_logout = True )
197
- assert result .get (timeout = 10 ) == ["user" , True ]
200
+ assert result .get (timeout = ASYNC_GET_TIMEOUT ) == ["user" , True ]
198
201
199
202
spans = memory_exporter .get_finished_spans ()
200
203
assert len (spans ) == 2
@@ -448,7 +451,7 @@ def run(self, *args, **kwargs):
448
451
# avoid call loop
449
452
return
450
453
CelerySubClass .apply_async (args = [], kwargs = {"stop" : True }).get (
451
- timeout = 10
454
+ timeout = ASYNC_GET_TIMEOUT
452
455
)
453
456
454
457
class CelerySubClass (CelerySuperClass ):
0 commit comments