test: ✅ fixed test_refresh_status failure with timezone

Author: newgene

src/tests/test_controller.py

@@ -380,8 +380,8 @@ def test_refresh_status():
     assert mygene.webdoc.timestamp > datetime(2020, 1, 1).astimezone()
     _ts0 = mygene.webdoc.timestamp
 
-    original_last_updated = mygene.last_updated.replace(microsecond=0, tzinfo=None)
-    one_hour_before = (datetime.now() - timedelta(hours=1)).replace(microsecond=0)
+    original_last_updated = mygene.last_updated.replace(microsecond=0, tzinfo=timezone.utc)
+    one_hour_before = (datetime.now(timezone.utc) - timedelta(hours=1)).replace(microsecond=0)
     assert original_last_updated > one_hour_before
 
     mygene.save()
@@ -397,14 +397,14 @@ def test_refresh_status():
     assert "components" in mygene
     assert mygene.webdoc.timestamp > _ts0
 
-    current_last_updated = mygene.last_updated.replace(microsecond=0, tzinfo=None)
+    current_last_updated = mygene.last_updated.replace(microsecond=0, tzinfo=timezone.utc)
     assert current_last_updated == original_last_updated
 
     mygene.save()
     refresh(index=ES_INDEX_NAME)
 
     # confirm last_updated is not changed after refresh
-    assert mygene.last_updated.replace(microsecond=0, tzinfo=None) == current_last_updated
+    assert mygene.last_updated.replace(microsecond=0, tzinfo=timezone.utc) == current_last_updated
 
     mygene_doc = SmartAPIDoc.get(MYGENE_ID, index=ES_INDEX_NAME)
     assert mygene_doc._status.refresh_status == 200
@@ -419,7 +419,7 @@ def test_refresh_status():
     refresh(index=ES_INDEX_NAME)
 
     # confirm last_updated is not changed after refresh
-    assert mygene.last_updated.replace(microsecond=0, tzinfo=None) == current_last_updated
+    assert mygene.last_updated.replace(microsecond=0, tzinfo=timezone.utc) == current_last_updated
 
     mygene_doc = SmartAPIDoc.get(MYGENE_ID, index=ES_INDEX_NAME)
     assert mygene_doc._status.refresh_status == 404
@@ -446,7 +446,7 @@ def test_refresh_status():
     refresh(index=ES_INDEX_NAME)
 
     # confirm last_updated is not changed after refresh
-    assert mygene.last_updated.replace(microsecond=0, tzinfo=None) == current_last_updated
+    assert mygene.last_updated.replace(microsecond=0, tzinfo=timezone.utc) == current_last_updated
 
     mygene_doc = SmartAPIDoc.get(MYGENE_ID, index=ES_INDEX_NAME)
     assert mygene_doc._status.refresh_status == 499