Update tests

Signed-off-by: Zelin Hao <zelinhao@amazon.com>

src/test_workflow/smoke_test/smoke_test_runner_opensearch.py

@@ -37,9 +37,9 @@ def __init__(self, args: TestArgs, test_manifest: TestManifest) -> None:
             "Content-Type": "application/json"
         }
 
-    def download_spec(self, url, local_path):
+    def download_spec(self, url: str, local_path: str) -> None:
         try:
-            response = requests.get(url, timeout = 10)
+            response = requests.get(url, timeout=10)
             if response.status_code == 200:
                 with open(local_path, "wb") as file:
                     file.write(response.content)

src/test_workflow/smoke_test/smoke_tests_spec/opensearch-openapi.yaml

@@ -1,3 +1,4 @@
+# yamllint disable-file
 openapi: 3.1.0
 info:
   title: OpenSearch API Specification

tests/tests_test_workflow/test_smoke_workflow/smoke_test/test_smoke_test_runner_opensearch.py

@@ -7,9 +7,8 @@
 
 import unittest
 from pathlib import Path
-from unittest.mock import MagicMock, Mock, call, patch, mock_open
+from unittest.mock import MagicMock, Mock, call, mock_open, patch
 
-import os
 import requests
 from openapi_core.exceptions import OpenAPIError
 from requests.models import Response
@@ -96,9 +95,10 @@ def test_validate_response_swagger_with_invalid_response(self, mock_get: Mock, m
         with self.assertRaises(OpenAPIError):
             runner.validate_response_swagger(response)
 
+    @patch("test_workflow.smoke_test.smoke_test_runner.TestRecorder")
     @patch("requests.get")
     @patch("builtins.open", new_callable=mock_open)
-    def test_download_spec_success(self, mock_file, mock_get):
+    def test_download_spec_success(self, mock_file: Mock, mock_get: Mock, mock_test_recorder: Mock) -> None:
         mock_response = MagicMock()
         mock_response.status_code = 200
         mock_response.content = "Mock OpenSearch API Spec Yaml content"
@@ -114,27 +114,29 @@ def test_download_spec_success(self, mock_file, mock_get):
         mock_file.assert_any_call(runner.spec_path, "wb")
         mock_file().write.assert_called_once_with("Mock OpenSearch API Spec Yaml content")
 
+    @patch("test_workflow.smoke_test.smoke_test_runner.TestRecorder")
     @patch("requests.get")
     @patch("builtins.open", new_callable=mock_open)
-    def test_download_spec_fail_local(self, mock_file, mock_get):
+    def test_download_spec_fail_local(self, mock_file: Mock, mock_get: Mock, mock_test_recorder: Mock) -> None:
         # Mock request failure
         mock_get.side_effect = requests.RequestException
 
-        runner = SmokeTestRunnerOpenSearch(MagicMock(), MagicMock())
+        SmokeTestRunnerOpenSearch(MagicMock(), MagicMock())
 
         mock_get.assert_called_once()
 
         mock_file().write.assert_not_called()
 
+    @patch("test_workflow.smoke_test.smoke_test_runner.TestRecorder")
     @patch("requests.get")
     @patch("builtins.open", new_callable=mock_open)
-    def test_download_spec_https_fail(self, mock_file, mock_get):
+    def test_download_spec_https_fail(self, mock_file: Mock, mock_get: Mock, mock_test_recorder: Mock) -> None:
         mock_response = MagicMock()
         mock_response.status_code = 404
         mock_get.return_value = mock_response
 
-        runner = SmokeTestRunnerOpenSearch(MagicMock(), MagicMock())
+        SmokeTestRunnerOpenSearch(MagicMock(), MagicMock())
 
         mock_get.assert_called_once()
 
-        mock_file().write.assert_not_called()
+        mock_file().write.assert_not_called()