Fix unit test suite current timestamp

Signed-off-by: Brandon Myers <bmyers@mozilla.com>
This commit is contained in:
Brandon Myers 2016-10-19 20:05:11 -05:00
Родитель b4ffb4d42e
Коммит 44b5e8aa4a
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 8AA79AD83045BBC7
1 изменённых файлов: 5 добавлений и 3 удалений

Просмотреть файл

@ -5,6 +5,8 @@ from config import ES
sys.path.append(os.path.join(os.path.dirname(__file__), "../lib"))
from elasticsearch_client import ElasticsearchClient
from utilities.toUTC import toUTC
from datetime import datetime
from datetime import timedelta
from dateutil.parser import parse
@ -55,7 +57,7 @@ class UnitTestSuite(object):
@staticmethod
def current_timestamp():
return pytz.UTC.normalize(pytz.timezone("UTC").localize(datetime.now())).isoformat()
return toUTC(datetime.now()).isoformat()
@staticmethod
def subtract_from_timestamp(date_timedelta, timestamp=None):
@ -67,8 +69,8 @@ class UnitTestSuite(object):
@staticmethod
def current_timestamp_lambda():
return lambda: (pytz.UTC.normalize(pytz.timezone("UTC").localize(datetime.now())).isoformat())
return lambda: UnitTestSuite.current_timestamp()
@staticmethod
def subtract_from_timestamp_lambda(date_timedelta, timestamp=None):
return lambda: UnitTestSuite.subtract_from_timestamp(date_timedelta, timestamp)
return lambda: UnitTestSuite.subtract_from_timestamp(date_timedelta, timestamp)