test_run_manifests.py

# 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 typing import Any
from unittest.mock import MagicMock, Mock, call, patch

import pytest

from run_manifests import main


class TestRunManifests(unittest.TestCase):
    @pytest.fixture(autouse=True)
    def _capfd(self, capfd: Any) -> None:
        self.capfd = capfd

    @patch("argparse._sys.argv", ["run_manifests.py", "--help"])
    def test_usage(self) -> None:
        with self.assertRaises(SystemExit):
            main()

        out, _ = self.capfd.readouterr()
        self.assertTrue(out.startswith("usage:"))

    @patch("argparse._sys.argv", ["run_manifests.py", "list"])
    @patch("run_manifests.logging", return_value=MagicMock())
    def test_main_list(self, mock_logging: Mock, *mocks: Any) -> None:
        main()

        mock_logging.info.assert_has_calls([
            call("OpenSearch 1.3.0"),
            call("OpenSearch 1.3.1"),
            call("OpenSearch 1.3.2"),
            call("OpenSearch 1.3.3"),
            call("OpenSearch 1.3.4")
        ])

        mock_logging.info.assert_has_calls([
            call("OpenSearch 2.0.0"),
            call("OpenSearch 2.0.1")
        ])

        mock_logging.info.assert_has_calls([
            call("OpenSearch 3.0.0")
        ])

        mock_logging.info.assert_has_calls([
            call("OpenSearch Dashboards 1.3.0")
        ])

        mock_logging.info.assert_has_calls([
            call("Done.")
        ])

    @patch("argparse._sys.argv", ["run_manifests.py", "update"])
    @patch("manifests_workflow.manifests_args.InputManifestsOpenSearch", return_value=MagicMock())
    @patch("manifests_workflow.manifests_args.InputManifestsOpenSearchDashboards", return_value=MagicMock())
    def test_main_update(self, mock_manifests_opensearch_dashboards: Mock, mock_manifests_opensearch: Mock, *mocks: Any) -> None:
        main()
        mock_manifests_opensearch_dashboards.return_value.update.assert_called()
        mock_manifests_opensearch.return_value.update.assert_called()