1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
from unittest.mock import Mock
class MockThread(Mock):
def __init__(self, *args, target=None, **kwargs):
super().__init__()
self._target = target
def start(self):
return self._target()
class MockTimer(MockThread):
def __init__(self, *args, **kwargs):
if 'function' in kwargs:
super().__init__(target=kwargs['function'])
elif len(args) >= 2:
super().__init__(target=args[1])
else:
raise AssertionError("%s couldn't find delayed function call"
% self.__class__.__name__)
|
