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
|
import datetime
import unittest
from mock import Mock, call
from trashcli.empty.clock import Clock
class TestClock(unittest.TestCase):
def setUp(self):
self.errors = Mock(spec=['print_error'])
def test_return_real_time(self):
clock = Clock(lambda: 'now', self.errors)
result = clock.get_now_value({})
assert (result, []) == ('now', self.errors.mock_calls)
def test_return_fake_time(self):
clock = Clock(lambda: 'now', self.errors)
result = clock.get_now_value({'TRASH_DATE': '2021-06-04T18:40:19'})
assert (result, []) == (datetime.datetime(2021, 6, 4, 18, 40, 19),
self.errors.mock_calls)
def test_return_true_now_whe_fake_time_is_invalid(self):
clock = Clock(lambda: 'now', self.errors)
result = clock.get_now_value({'TRASH_DATE': 'invalid'})
assert (result,
[call.print_error('invalid TRASH_DATE: invalid')]) == \
('now', self.errors.mock_calls)
|