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
|
from .ut_utils import TestCase
from .ut_utils import Checkpoint
from forge import Forge
from forge.exceptions import ExpectedEventsNotFound
class ForgeTest(TestCase):
def setUp(self):
super(ForgeTest, self).setUp()
self.forge = Forge()
def assertRecording(self):
self.assertFalse(self.forge.is_replaying())
self.assertTrue(self.forge.is_recording())
def assertReplaying(self):
self.assertTrue(self.forge.is_replaying())
self.assertFalse(self.forge.is_recording())
def test__replaying(self):
self.assertRecording()
self.forge.replay()
self.assertReplaying()
self.forge.verify()
self.assertReplaying()
def test__reset(self):
self.test__replaying()
self.forge.queue._queue = [1, 2, 3, 4]
self.forge.reset()
self.assertRecording()
self.assertTrue(self.forge.queue.is_empty())
def test__verified_replay_context(self):
self.assertRecording()
check_verify = Checkpoint()
check_reset = Checkpoint()
self.forge.verify = check_verify.trigger
def _fake_reset():
self.assertTrue(check_verify.called)
check_reset.trigger()
self.forge.reset = _fake_reset
with self.forge.verified_replay_context():
self.assertReplaying()
self.assertFalse(check_verify.called)
self.assertFalse(check_reset.called)
self.assertTrue(check_reset.called)
self.assertTrue(check_verify.called)
# we don't assertRecording, since we stubbed out reset
def test__verified_replay_context_checks_no_more_calls(self):
expected_call = self.forge.create_wildcard_function_stub()(1, 2, 3)
with self.assertRaises(ExpectedEventsNotFound):
with self.forge.verified_replay_context():
pass
|
