Fix tests: list_versions() now has a time-zoned date

man-group · Aug 19, 2015 · e091661 · e091661
1 parent 1959f49
commit e091661
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/integration/store/test_version_store.py b/tests/integration/store/test_version_store.py
@@ -14,6 +14,7 @@
 from arctic.exceptions import NoDataFoundException, DuplicateSnapshotException
 
 from ...util import read_str_as_pandas
+from arctic.date._mktz import mktz
 
 
 ts1 = read_str_as_pandas("""         times | near
@@ -285,7 +286,7 @@ def test_query_version_as_of_int(library):
 def test_list_version(library):
     assert len(list(library.list_versions(symbol))) == 0
     dates = [None, None, None]
-    now = dt.utcnow()
+    now = dt.utcnow().replace(tzinfo=mktz('UTC'))
     for x in xrange(len(dates)):
         dates[x] = now - dtd(minutes=130 - x)
         with patch("bson.ObjectId", return_value=bson.ObjectId.from_datetime(dates[x])):
@@ -305,7 +306,7 @@ def test_list_version(library):
 def test_list_version_latest_only(library):
     assert len(list(library.list_versions(symbol))) == 0
     dates = [None, None, None]
-    now = dt.utcnow()
+    now = dt.utcnow().replace(tzinfo=mktz('UTC'))
     for x in xrange(len(dates)):
         dates[x] = now - dtd(minutes=20 - x)
         with patch("bson.ObjectId", return_value=bson.ObjectId.from_datetime(dates[x])):

diff --git a/tests/unit/store/test_version_store.py b/tests/unit/store/test_version_store.py
@@ -37,7 +37,7 @@ def test_list_versions_localTime():
                                        'symbol': 's', 'version': 10}]
 
     version = list(VersionStore.list_versions(vs, "symbol"))[0]
-    local_date = date.replace(tzinfo=mktz("UTC")).astimezone(mktz()).replace(tzinfo=None)
+    local_date = date.replace(tzinfo=mktz("UTC"))
     assert version == {'symbol': version['symbol'], 'version': version['version'],
                        # We return naive datetimes in 'default' time, which is London for us
                        'date': local_date,