File: test_time.py

package info (click to toggle)
python-testfixtures 9.2.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,036 kB
  • sloc: python: 10,373; makefile: 76; sh: 9
file content (229 lines) | stat: -rw-r--r-- 8,328 bytes parent folder | download
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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
from datetime import timedelta
from unittest import TestCase

from testfixtures import mock_time, replace, compare, ShouldRaise
from testfixtures.datetime import MockTime
from .test_datetime import SampleTZInfo


class TestTime(TestCase):

    @replace('time.time', mock_time())
    def test_time_call(self) -> None:
        from time import time
        compare(time(), 978307200.0)
        compare(time(), 978307201.0)
        compare(time(), 978307203.0)

    @replace('time.time', mock_time(2002, 1, 1, 1, 2, 3))
    def test_time_supplied(self) -> None:
        from time import time
        compare(time(), 1009846923.0)

    @replace('time.time', mock_time(None))
    def test_time_sequence(self, t: MockTime) -> None:
        t.add(2002, 1, 1, 1, 0, 0)
        t.add(2002, 1, 1, 2, 0, 0)
        t.add(2002, 1, 1, 3, 0, 0)
        from time import time
        compare(time(), 1009846800.0)
        compare(time(), 1009850400.0)
        compare(time(), 1009854000.0)

    @replace('time.time', mock_time(None))
    def test_add_datetime_supplied(self, t: MockTime) -> None:
        from datetime import datetime
        from time import time
        t.add(datetime(2002, 1, 1, 2))
        compare(time(), 1009850400.0)
        tzinfo = SampleTZInfo()
        tzrepr = repr(tzinfo)
        with ShouldRaise(ValueError(
            'Cannot add datetime with tzinfo of %s as configured to use None' %(
                tzrepr
            ))):
            t.add(datetime(2001, 1, 1, tzinfo=tzinfo))

    def test_instantiate_with_datetime(self) -> None:
        from datetime import datetime
        t = mock_time(datetime(2002, 1, 1, 2))
        compare(t(), 1009850400.0)

    @replace('time.time', mock_time(None))
    def test_now_requested_longer_than_supplied(self, t: MockTime) -> None:
        t.add(2002, 1, 1, 1, 0, 0)
        t.add(2002, 1, 1, 2, 0, 0)
        from time import time
        compare(time(), 1009846800.0)
        compare(time(), 1009850400.0)
        compare(time(), 1009850401.0)
        compare(time(), 1009850403.0)

    @replace('time.time', mock_time())
    def test_call(self, t: MockTime) -> None:
        compare(t(), 978307200.0)
        from time import time
        compare(time(), 978307201.0)

    @replace('time.time', mock_time())
    def test_repr_time(self) -> None:
        from time import time
        assert repr(time).startswith('<testfixtures.datetime.MockTime object at ')

    @replace('time.time', mock_time(delta=10))
    def test_delta(self) -> None:
        from time import time
        compare(time(), 978307200.0)
        compare(time(), 978307210.0)
        compare(time(), 978307220.0)

    @replace('time.time', mock_time(delta_type='minutes'))
    def test_delta_type(self) -> None:
        from time import time
        compare(time(), 978307200.0)
        compare(time(), 978307260.0)
        compare(time(), 978307380.0)

    @replace('time.time', mock_time(None))
    def test_set(self, time_mock: MockTime) -> None:
        from time import time
        time_mock.set(2001, 1, 1, 1, 0, 1)
        compare(time(), 978310801.0)
        time_mock.set(2002, 1, 1, 1, 0, 0)
        compare(time(), 1009846800.0)
        compare(time(), 1009846802.0)

    @replace('time.time', mock_time(None))
    def test_set_datetime_supplied(self, t: MockTime) -> None:
        from datetime import datetime
        from time import time
        t.set(datetime(2001, 1, 1, 1, 0, 1))
        compare(time(), 978310801.0)
        tzinfo = SampleTZInfo()
        tzrepr = repr(tzinfo)
        with ShouldRaise(ValueError(
            'Cannot add datetime with tzinfo of %s as configured to use None' %(
                tzrepr
            ))):
            t.set(datetime(2001, 1, 1, tzinfo=tzinfo))

    @replace('time.time', mock_time(None))
    def test_set_kw(self, time_mock: MockTime) -> None:
        from time import time
        time_mock.set(year=2001, month=1, day=1, hour=1, second=1)
        compare(time(), 978310801.0)

    @replace('time.time', mock_time(None))
    def test_set_kw_tzinfo(self, time_mock: MockTime) -> None:
        with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
            time_mock.set(year=2001, tzinfo=SampleTZInfo())

    @replace('time.time', mock_time(None))
    def test_set_args_tzinfo(self, time_mock: MockTime) -> None:
        with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
            time_mock.set(2002, 1, 2, 3, 4, 5, 6, SampleTZInfo())  # type: ignore[arg-type]

    @replace('time.time', mock_time(None))
    def test_add_kw(self, time_mock: MockTime) -> None:
        from time import time
        time_mock.add(year=2001, month=1, day=1, hour=1, second=1)
        compare(time(), 978310801.0)

    @replace('time.time', mock_time(None))
    def test_add_tzinfo_kw(self, time_mock: MockTime) -> None:
        with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
            time_mock.add(year=2001, tzinfo=SampleTZInfo())

    @replace('time.time', mock_time(None))
    def test_add_tzinfo_args(self, time_mock: MockTime) -> None:
        with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
            time_mock.add(2001, 1, 2, 3, 4, 5, 6, SampleTZInfo())  # type: ignore[arg-type]

    @replace('time.time', mock_time(2001, 1, 2, 3, 4, 5, 600000))
    def test_max_number_args(self) -> None:
        from time import time
        compare(time(), 978404645.6)

    def test_max_number_tzinfo(self) -> None:
        with ShouldRaise(TypeError(
            "You don't want to use tzinfo with test_time"
                )):
            mock_time(2001, 1, 2, 3, 4, 5, 6, SampleTZInfo())  # type: ignore[arg-type]

    @replace('time.time', mock_time(2001, 1, 2))
    def test_min_number_args(self) -> None:
        from time import time
        compare(time(), 978393600.0)

    @replace('time.time', mock_time(
        year=2001,
        month=1,
        day=2,
        hour=3,
        minute=4,
        second=5,
        microsecond=6,
        ))
    def test_all_kw(self) -> None:
        from time import time
        compare(time(), 978404645.000006)

    def test_kw_tzinfo(self) -> None:
        with ShouldRaise(TypeError(
            "You don't want to use tzinfo with test_time"
                )):
            mock_time(year=2001, tzinfo=SampleTZInfo())  # type: ignore[arg-type]

    def test_instance_tzinfo(self) -> None:
        from datetime import datetime
        with ShouldRaise(TypeError(
            "You don't want to use tzinfo with test_time"
                )):
            mock_time(datetime(2001, 1, 1, tzinfo=SampleTZInfo()))

    def test_subsecond_deltas(self) -> None:
        time = mock_time(delta=0.5)
        compare(time(), 978307200.0)
        compare(time(), 978307200.5)
        compare(time(), 978307201.0)

    def test_ms_deltas(self) -> None:
        time = mock_time(delta=1000, delta_type='microseconds')
        compare(time(), 978307200.0)
        compare(time(), 978307200.001)
        compare(time(), 978307200.002)

    def test_tick_when_static(self) -> None:
        time = mock_time(delta=0)
        compare(time(), expected=978307200.0)
        time.tick(seconds=1)
        compare(time(), expected=978307201.0)

    def test_tick_when_dynamic(self) -> None:
        # hopefully not that common?
        time = mock_time()
        compare(time(), expected=978307200.0)
        time.tick(seconds=1)
        compare(time(), expected=978307202.0)

    def test_tick_with_timedelta_instance(self) -> None:
        time = mock_time(delta=0)
        compare(time(), expected=978307200.0)
        time.tick(timedelta(seconds=1))
        compare(time(), expected=978307201.0)

    def test_old_import(self) -> None:
        from testfixtures import test_time
        assert test_time is mock_time

    def test_addition_to_no_params_call_indirect(self) -> None:
        mock = mock_time()
        # Calling the mock without any parameters definitely gives you a float:
        value = mock()
        assert isinstance(value, float)
        compare(value + 3600, expected=978307200.0 + 3600.0)

    def test_addition_to_no_params_call_direct(self) -> None:
        mock = mock_time()
        # ...but if if you do it in one step, mypy gets confused:
        compare(mock() + 3600, expected=978307200.0 + 3600.0)