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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
from __future__ import print_function
import time
from unittest import TestCase, skipIf
from pyface.toolkit import toolkit_object
from ..i_timer import perf_counter
from ..do_later import DoLaterTimer, do_after, do_later
GuiTestAssistant = toolkit_object('util.gui_test_assistant:GuiTestAssistant')
no_gui_test_assistant = (GuiTestAssistant.__name__ == 'Unimplemented')
class ConditionHandler(object):
def __init__(self):
self.count = 0
self.times = []
def callback(self):
self.times.append(perf_counter())
self.count += 1
@skipIf(no_gui_test_assistant, 'No GuiTestAssistant')
class TestDoLaterTimer(TestCase, GuiTestAssistant):
""" Test the DoLaterTimer. """
def setUp(self):
GuiTestAssistant.setUp(self)
def tearDown(self):
GuiTestAssistant.tearDown(self)
def test_basic(self):
handler = ConditionHandler()
start_time = perf_counter()
length = 500
timer = DoLaterTimer(length, handler.callback, (), {})
try:
self.assertTrue(timer.active)
self.event_loop_helper.event_loop_until_condition(
lambda: not timer.active
)
self.assertFalse(timer.active)
finally:
timer.stop()
self.assertEqual(handler.count, 1)
# should take no less than 85% of time requested
expected_length = (length / 1000.0) * 0.85
expected_time = start_time + expected_length
self.assertLessEqual(
expected_time,
handler.times[0],
"Expected call after {} seconds, took {} seconds)".format(
expected_length,
handler.times[0] - start_time
)
)
@skipIf(no_gui_test_assistant, 'No GuiTestAssistant')
class TestDoLater(TestCase, GuiTestAssistant):
""" Test do_later. """
def setUp(self):
GuiTestAssistant.setUp(self)
def tearDown(self):
GuiTestAssistant.tearDown(self)
def test_basic(self):
handler = ConditionHandler()
timer = do_later(handler.callback)
try:
self.assertTrue(timer.active)
self.event_loop_helper.event_loop_until_condition(
lambda: not timer.active
)
self.assertFalse(timer.active)
finally:
timer.stop()
self.assertEqual(handler.count, 1)
@skipIf(no_gui_test_assistant, 'No GuiTestAssistant')
class TestDoAfter(TestCase, GuiTestAssistant):
""" Test do_after. """
def setUp(self):
GuiTestAssistant.setUp(self)
def tearDown(self):
GuiTestAssistant.tearDown(self)
def test_basic(self):
handler = ConditionHandler()
start_time = perf_counter()
length = 500
timer = do_after(length, handler.callback)
try:
self.assertTrue(timer.active)
self.event_loop_helper.event_loop_until_condition(
lambda: not timer.active
)
self.assertFalse(timer.active)
finally:
timer.stop()
self.assertEqual(handler.count, 1)
# should take no less than 85% of time requested
expected_length = (length / 1000.0) * 0.85
expected_time = start_time + expected_length
self.assertLessEqual(
expected_time,
handler.times[0],
"Expected call after {} seconds, took {} seconds)".format(
expected_length,
handler.times[0] - start_time
)
)
|
