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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
from six.moves.queue import Queue
from invoke.util import ExceptionWrapper, ExceptionHandlingThread as EHThread
# TODO: rename
class ExceptionHandlingThread_:
class via_target:
def setup(self):
def worker(q):
q.put(7)
self.worker = worker
def base_case(self):
queue = Queue()
t = EHThread(target=self.worker, args=[queue])
t.start()
t.join()
assert queue.get(block=False) == 7
assert queue.empty()
def catches_exceptions(self):
# Induce exception by submitting a bad queue obj
t = EHThread(target=self.worker, args=[None])
t.start()
t.join()
wrapper = t.exception()
assert isinstance(wrapper, ExceptionWrapper)
assert wrapper.kwargs == {"args": [None], "target": self.worker}
assert wrapper.type == AttributeError
assert isinstance(wrapper.value, AttributeError)
def exhibits_is_dead_flag(self):
# Spin up a thread that will except internally (can't put() on a
# None object)
t = EHThread(target=self.worker, args=[None])
t.start()
t.join()
# Excepted -> it's dead
assert t.is_dead
# Spin up a happy thread that can exit peacefully (it's not "dead",
# though...maybe we should change that terminology)
t = EHThread(target=self.worker, args=[Queue()])
t.start()
t.join()
# Not dead, just uh...sleeping?
assert not t.is_dead
class via_subclassing:
def setup(self):
class MyThread(EHThread):
def __init__(self, *args, **kwargs):
self.queue = kwargs.pop("queue")
super(MyThread, self).__init__(*args, **kwargs)
def _run(self):
self.queue.put(7)
self.klass = MyThread
def base_case(self):
queue = Queue()
t = self.klass(queue=queue)
t.start()
t.join()
assert queue.get(block=False) == 7
assert queue.empty()
def catches_exceptions(self):
# Induce exception by submitting a bad queue obj
t = self.klass(queue=None)
t.start()
t.join()
wrapper = t.exception()
assert isinstance(wrapper, ExceptionWrapper)
assert wrapper.kwargs == {}
assert wrapper.type == AttributeError
assert isinstance(wrapper.value, AttributeError)
def exhibits_is_dead_flag(self):
# Spin up a thread that will except internally (can't put() on a
# None object)
t = self.klass(queue=None)
t.start()
t.join()
# Excepted -> it's dead
assert t.is_dead
# Spin up a happy thread that can exit peacefully (it's not "dead",
# though...maybe we should change that terminology)
t = self.klass(queue=Queue())
t.start()
t.join()
# Not dead, just uh...sleeping?
assert not t.is_dead
|
