from datetime import timedelta
from typing import Type, cast
from unittest import TestCase
from testfixtures import mock_time, replace, compare, ShouldRaise
from .test_datetime import SampleTZInfo
from ..datetime import MockTime
class TestTime(TestCase):
@replace('time.time', mock_time())
def test_time_call(self):
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):
from time import time
compare(time(), 1009846923.0)
@replace('time.time', mock_time(None))
def test_time_sequence(self, t: Type[MockTime]):
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: Type[MockTime]):
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):
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: Type[MockTime]):
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: Type[MockTime]):
compare(t(), 978307200.0)
from time import time
compare(time(), 978307201.0)
@replace('time.time', mock_time())
def test_repr_time(self):
from time import time
compare(repr(time), "<class 'testfixtures.datetime.MockTime'>")
@replace('time.time', mock_time(delta=10))
def test_delta(self):
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):
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):
from time import time
time = cast(Type[MockTime], time)
time.set(2001, 1, 1, 1, 0, 1)
compare(time(), 978310801.0)
time.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: Type[MockTime]):
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):
from time import time
time = cast(Type[MockTime], time)
time.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):
from time import time
time = cast(Type[MockTime], time)
with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
time.set(year=2001, tzinfo=SampleTZInfo())
@replace('time.time', mock_time(None))
def test_set_args_tzinfo(self):
from time import time
time = cast(Type[MockTime], time)
with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
time.set(2002, 1, 2, 3, 4, 5, 6, SampleTZInfo())
@replace('time.time', mock_time(None))
def test_add_kw(self):
from time import time
time = cast(Type[MockTime], time)
time.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):
from time import time
time = cast(Type[MockTime], time)
with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
time.add(year=2001, tzinfo=SampleTZInfo())
@replace('time.time', mock_time(None))
def test_add_tzinfo_args(self):
from time import time
time = cast(Type[MockTime], time)
with ShouldRaise(TypeError('Cannot add using tzinfo on MockTime')):
time.add(2001, 1, 2, 3, 4, 5, 6, SampleTZInfo())
@replace('time.time', mock_time(2001, 1, 2, 3, 4, 5, 600000))
def test_max_number_args(self):
from time import time
compare(time(), 978404645.6)
def test_max_number_tzinfo(self):
with ShouldRaise(TypeError(
"You don't want to use tzinfo with test_time"
)):
mock_time(2001, 1, 2, 3, 4, 5, 6, SampleTZInfo())
@replace('time.time', mock_time(2001, 1, 2))
def test_min_number_args(self):
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):
from time import time
compare(time(), 978404645.000006)
def test_kw_tzinfo(self):
with ShouldRaise(TypeError(
"You don't want to use tzinfo with test_time"
)):
mock_time(year=2001, tzinfo=SampleTZInfo())
def test_instance_tzinfo(self):
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):
time = mock_time(delta=0.5)
compare(time(), 978307200.0)
compare(time(), 978307200.5)
compare(time(), 978307201.0)
def test_ms_deltas(self):
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):
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):
# 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):
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):
from testfixtures import test_time
assert test_time is mock_time
|