36 lines
900 B
Python
36 lines
900 B
Python
import datetime
|
|
|
|
from dateutil import tz
|
|
from docker_custodian import args
|
|
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
|
|
|
|
|
|
def test_datetime_seconds_ago(now):
|
|
expected = datetime.datetime(2014, 1, 15, 10, 10, tzinfo=tz.tzutc())
|
|
with mock.patch(
|
|
'docker_custodian.args.datetime.datetime',
|
|
autospec=True,
|
|
) as mock_datetime:
|
|
mock_datetime.now.return_value = now
|
|
assert args.datetime_seconds_ago(24 * 60 * 60 * 5) == expected
|
|
|
|
|
|
def test_timedelta_type_none():
|
|
assert args.timedelta_type(None) is None
|
|
|
|
|
|
def test_timedelta_type(now):
|
|
expected = datetime.datetime(2014, 1, 15, 10, 10, tzinfo=tz.tzutc())
|
|
with mock.patch(
|
|
'docker_custodian.args.datetime.datetime',
|
|
autospec=True,
|
|
) as mock_datetime:
|
|
mock_datetime.now.return_value = now
|
|
assert args.timedelta_type('5 days') == expected
|