Update tests

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

Author: zelinh

src/test_workflow/smoke_test/distribution.py

@@ -1,3 +1,4 @@
+# yamllint disable-file
 openapi: 3.1.0
 info:
   title: OpenSearch API Specification
@@ -32157,7 +32158,7 @@ components:
         params:
           description: |-
             Specifies any named parameters that are passed into the script as variables. 
-            Use parameters instead of hard-coded values to decrease compilation time. 
+            Use parameters instead of hard-coded values to decrease compilation time.
           type: object
           additionalProperties: true
     _common___ScriptField:

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

@@ -1,5 +1,5 @@
+# yamllint disable
 ---
-schema-version: '1.0'
 info:
   title: OpenSearch Core smoke tests
   version: default

src/test_workflow/smoke_test/smoke_tests_spec/opensearch.yml

@@ -0,0 +1,48 @@
+# Copyright OpenSearch Contributors
+# SPDX-License-Identifier: Apache-2.0
+#
+# The OpenSearch Contributors require contributions made to
+# this file be licensed under the Apache-2.0 license or a
+# compatible open source license.
+
+
+import os
+import unittest
+from typing import Any
+from unittest.mock import Mock, patch
+
+import pytest
+from pytest import CaptureFixture
+
+from run_smoke_test import main
+
+
+class TestRunSmokeTest(unittest.TestCase):
+
+    TEST_MANIFEST_PATH = os.path.join(
+        os.path.dirname(__file__),
+        "tests_test_workflow",
+        "test_smoke_workflow",
+        "smoke_test",
+        "data",
+        "test_manifest.yml"
+    )
+
+    @pytest.fixture(autouse=True)
+    def getCapfd(self, capfd: CaptureFixture) -> None:
+        self.capfd = capfd
+
+    @patch("argparse._sys.argv", ["run_smoke_test.py", "--help"])
+    def test_usage(self, *mocks: Any) -> None:
+        with self.assertRaises(SystemExit):
+            main()
+
+        out, _ = self.capfd.readouterr()
+        self.assertTrue(out.startswith("usage:"))
+
+    @patch("argparse._sys.argv", ["run_smoke_test.py", TEST_MANIFEST_PATH, "--paths", "opensearch=foo"])
+    @patch('run_smoke_test.SmokeTestRunners')
+    def test_main(self, mock_runners: Mock, *mocks: Any) -> None:
+
+        main()
+        mock_runners.from_test_manifest.assert_called()

tests/test_run_smoke_test.py

@@ -0,0 +1,31 @@
+# Copyright OpenSearch Contributors
+# SPDX-License-Identifier: Apache-2.0
+#
+# The OpenSearch Contributors require contributions made to
+# this file be licensed under the Apache-2.0 license or a
+# compatible open source license.
+
+import unittest
+from unittest.mock import MagicMock, patch
+
+from test_workflow.smoke_test.smoke_test_runners import SmokeTestRunners
+
+
+class TestSmokeTestRunners(unittest.TestCase):
+
+    def test_opensearch(self) -> None:
+
+        mock_args = MagicMock()
+        mock_test_manifest = MagicMock()
+        mock_test_manifest.name = "OpenSearch"
+
+        mock_opensearch_runner_object = MagicMock()
+        mock_opensearch_runner = MagicMock()
+        mock_opensearch_runner.return_value = mock_opensearch_runner_object
+
+        with patch.dict("test_workflow.smoke_test.smoke_test_runners.SmokeTestRunners.RUNNERS", {
+            "OpenSearch": mock_opensearch_runner
+        }):
+            runner = SmokeTestRunners.from_test_manifest(mock_args, mock_test_manifest)
+            self.assertEqual(runner, mock_opensearch_runner_object)
+            mock_opensearch_runner.assert_called_once_with(mock_args, mock_test_manifest)