1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
import pickle
from datetime import datetime, timezone
from celery.exceptions import Reject, Retry
class test_Retry:
def test_when_datetime(self):
x = Retry('foo', KeyError(), when=datetime.now(timezone.utc))
assert x.humanize()
def test_pickleable(self):
x = Retry('foo', KeyError(), when=datetime.now(timezone.utc))
y = pickle.loads(pickle.dumps(x))
assert x.message == y.message
assert repr(x.exc) == repr(y.exc)
assert x.when == y.when
class test_Reject:
def test_attrs(self):
x = Reject('foo', requeue=True)
assert x.reason == 'foo'
assert x.requeue
def test_repr(self):
assert repr(Reject('foo', True))
def test_pickleable(self):
x = Retry('foo', True)
assert pickle.loads(pickle.dumps(x))
|
