Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
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
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
moto/sqs/models.py
View File

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