Merge pull request #534 from adamtheturtle/fix-test-query

Fix some timezone errors

Author: adamtheturtle

tests/mock_vws/test_date_header.py

@@ -6,6 +6,7 @@
 from typing import Dict, Union
 
 import pytest
+import pytz
 import requests
 from freezegun import freeze_time
 from requests import codes
@@ -83,7 +84,8 @@ def test_incorrect_date_format(
         A `BAD_REQUEST` response is returned when the date given in the date
         header is not in the expected format (RFC 1123).
         """
-        with freeze_time(datetime.now()):
+        gmt = pytz.timezone('GMT')
+        with freeze_time(datetime.now(tz=gmt)):
             now = datetime.now()
             date_incorrect_format = now.strftime('%a %b %d %H:%M:%S %Y')
 
@@ -148,7 +150,8 @@ def test_date_out_of_range(
         """
         time_difference_from_now = timedelta(minutes=5, seconds=10)
         time_difference_from_now *= time_multiplier
-        with freeze_time(datetime.now() + time_difference_from_now):
+        gmt = pytz.timezone('GMT')
+        with freeze_time(datetime.now(tz=gmt) + time_difference_from_now):
             date = rfc_1123_date()
 
         endpoint_headers = dict(endpoint.prepared_request.headers)
@@ -204,7 +207,8 @@ def test_date_in_range(
         """
         time_difference_from_now = timedelta(minutes=4, seconds=50)
         time_difference_from_now *= time_multiplier
-        with freeze_time(datetime.now() + time_difference_from_now):
+        gmt = pytz.timezone('GMT')
+        with freeze_time(datetime.now(tz=gmt) + time_difference_from_now):
             date = rfc_1123_date()
 
         endpoint_headers = dict(endpoint.prepared_request.headers)

tests/mock_vws/test_invalid_json.py

@@ -5,6 +5,7 @@
 from datetime import datetime, timedelta
 
 import pytest
+import pytz
 import requests
 from freezegun import freeze_time
 from requests import codes
@@ -38,7 +39,9 @@ def test_invalid_json(
         # endpoints.
         is_summary_endpoint = endpoint.prepared_request.path_url == '/summary'
         content = b'a'
-        time_to_freeze = datetime.now() + timedelta(minutes=date_skew_minutes)
+        gmt = pytz.timezone('GMT')
+        now = datetime.now(tz=gmt)
+        time_to_freeze = now + timedelta(minutes=date_skew_minutes)
         with freeze_time(time_to_freeze):
             date = rfc_1123_date()