sqs: Use unix_time in place of time.time() (#787)

unix_time() from moto.core.utils is used as the time source through moto,
and it is identical to time.time() in output. Hence, using unix_time()
since it makes mocking easier during testing (when time is mocked out).
This commit is contained in:
Raghavendra D Prabhu 2017-01-12 07:05:56 +05:30 committed by Steve Pulec
parent b5ff3345be
commit d07c646032

View File

@ -8,7 +8,7 @@ from xml.sax.saxutils import escape
import boto.sqs
from moto.core import BaseBackend
from moto.core.utils import camelcase_to_underscores, get_random_message_id, unix_time_millis
from moto.core.utils import camelcase_to_underscores, get_random_message_id, unix_time, unix_time_millis
from .utils import generate_receipt_handle
from .exceptions import (
ReceiptHandleIsInvalid,
@ -115,7 +115,7 @@ class Queue(object):
self.wait_time_seconds = wait_time_seconds or 0
self._messages = []
now = time.time()
now = unix_time()
self.created_timestamp = now
self.delay_seconds = 0
@ -281,7 +281,7 @@ class SQSBackend(BaseBackend):
queue = self.get_queue(queue_name)
result = []
polling_end = time.time() + wait_seconds_timeout
polling_end = unix_time() + wait_seconds_timeout
# queue.messages only contains visible messages
while True:
@ -295,7 +295,7 @@ class SQSBackend(BaseBackend):
if len(result) >= count:
break
if result or time.time() > polling_end:
if result or unix_time() > polling_end:
break
return result