Merge pull request #91 from prodigyfinance/feature/fix-moto-datetime-…

…update Fix test HTTPHeaders returns to include date return
prodigyfinance · Jul 12, 2023 · 5ab5286 · 5ab5286
2 parents 2059c53 + d87a858
commit 5ab5286
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/tests/cli_commands/test_model.py b/tests/cli_commands/test_model.py
@@ -28,7 +28,7 @@ def test_help(self, cli_helper):
     def test_list_empty(self, cli_helper):
         cli_helper.invoke(["model", "list"], output_jsonl=[])
 
-    def test_create_and_list(self, cli_helper):
+    def test_create_and_list(self, cli_helper, fake_utcnow):
         cfg = self.cfg()
         model_output = json.loads(
             cli_helper.invoke(["model", "create", "mdl-0-1-12"], cfg=cfg)
@@ -39,7 +39,10 @@ def test_create_and_list(self, cli_helper):
             ),
             "ResponseMetadata": {
                 "HTTPStatusCode": 200,
-                "HTTPHeaders": {"server": "amazon.com"},
+                "HTTPHeaders": {
+                    "server": "amazon.com",
+                    "date": "Thu, 31 Jan 2019 12:00:02 GMT",
+                },
                 "RetryAttempts": 0,
             },
         }
@@ -63,7 +66,7 @@ def test_create_and_describe(self, cli_helper):
             describe_output["ExecutionRoleArn"] == "arn:aws:iam::12345:role/role-name"
         )
 
-    def test_create_and_delete(self, cli_helper):
+    def test_create_and_delete(self, cli_helper, fake_utcnow):
         cfg = self.cfg()
         cli_helper.invoke(["model", "create", "mdl-0-1-12"], cfg=cfg)
         delete_output = json.loads(
@@ -72,7 +75,10 @@ def test_create_and_delete(self, cli_helper):
         assert delete_output == {
             "ResponseMetadata": {
                 "HTTPStatusCode": 200,
-                "HTTPHeaders": {"server": "amazon.com"},
+                "HTTPHeaders": {
+                    "server": "amazon.com",
+                    "date": "Thu, 31 Jan 2019 12:00:02 GMT",
+                },
                 "RetryAttempts": 0,
             }
         }
diff --git a/tests/fixtures.py b/tests/fixtures.py
@@ -27,6 +27,10 @@ class fake_datetime(datetime.datetime):
         def utcnow(cls):
             return utcnow
 
+        @classmethod
+        def now(cls, **kwargs):
+            return datetime.datetime(2019, 1, 31, 12, 0, 2)
+
     monkeypatch.setattr(datetime, "datetime", fake_datetime)
     return utcnow