Tweaks per review

This commit is contained in:
Peter Eckersley 2016-02-22 17:26:55 -08:00
parent 8d61c86c8c
commit 9a36439e1b
2 changed files with 6 additions and 2 deletions

View file

@ -4,6 +4,7 @@ import datetime
import heapq
import logging
import time
from email.utils import parsedate_tz
import six
from six.moves import http_client # pylint: disable=import-error
@ -17,7 +18,6 @@ from acme import jose
from acme import jws
from acme import messages
from email.utils import parsedate_tz
logger = logging.getLogger(__name__)

View file

@ -205,8 +205,12 @@ class ClientTest(unittest.TestCase):
datetime.datetime(2015, 3, 27, 0, 0, 10),
self.client.retry_after(response=self.response, default=10))
# wrong date -> ValueError
@mock.patch('acme.client.datetime')
def test_retry_after_overflow(self, dt_mock):
dt_mock.datetime.now.return_value = datetime.datetime(2015, 3, 27)
dt_mock.timedelta = datetime.timedelta
dt_mock.datetime.side_effect = datetime.datetime
self.response.headers['Retry-After'] = "Tue, 116 Feb 2016 11:50:00 MST"
self.assertEqual(
datetime.datetime(2015, 3, 27, 0, 0, 10),