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

from build_workflow.build_target import BuildTarget
from build_workflow.builder_from_dist import BuilderFromDist
from manifests.build_manifest import BuildManifest
from manifests.input_manifest import InputComponentFromDist


class TestBuilderFromDist(unittest.TestCase):
    def __mock_builder(self, component_name: str) -> BuilderFromDist:
        return BuilderFromDist(
            InputComponentFromDist({"name": component_name, "dist": "url"}),
            BuildTarget(
                name="OpenSearch",
                version="1.1.0",
                platform="windows",
                architecture="x64",
                snapshot=False,
            ),
        )

    def test_builder(self) -> None:
        self.assertEqual(self.__mock_builder("common-utils").component.name, "common-utils")

    @patch("manifests.distribution.find_build_root")
    @patch("build_workflow.builder_from_dist.BuildManifest")
    def test_checkout(self, mock_manifest: Mock, find_build_root: Mock) -> None:
        builder = self.__mock_builder("common-utils")
        builder.checkout("dir")
        mock_manifest.from_url.assert_called_once()
        find_build_root.assert_called_once()
        self.assertIsNotNone(builder.distribution_url)

    def test_build(self) -> None:
        build_recorder = MagicMock()
        self.__mock_builder("common-utils").build(build_recorder)

    @patch("os.makedirs")
    @patch("urllib.request.urlretrieve")
    @patch("build_workflow.builder_from_dist.BuilderFromDist.ManifestGitRepository")
    def test_export_artifacts(self, mock_manifest_git_repository: Mock, mock_urllib: Mock, mock_makedirs: Mock) -> None:
        build_recorder = MagicMock()
        manifest_path = os.path.join(os.path.dirname(__file__), "data", "opensearch-build-windows-1.1.0.yml")
        mock_builder = self.__mock_builder("notifications")
        mock_builder.distribution_url = "dist_url"
        mock_builder.build_manifest = BuildManifest.from_path(manifest_path)
        mock_builder.export_artifacts(build_recorder)
        build_recorder.record_component.assert_called_with(
            "notifications", mock_manifest_git_repository.return_value)
        mock_makedirs.assert_called_with(
            os.path.realpath(os.path.join("builds", "plugins")),
            exist_ok=True
        )
        mock_urllib.assert_has_calls([
            call(
                'dist_url/plugins/opensearch-notifications-1.1.0.0.zip',
                os.path.realpath(os.path.join("builds", "plugins", "opensearch-notifications-1.1.0.0.zip")))
        ])

    @patch("os.makedirs")
    @patch("urllib.request.urlretrieve")
    @patch("build_workflow.builder_from_dist.BuilderFromDist.ManifestGitRepository")
    def test_export_artifacts_skips_maven_artifacts(self, mock_manifest_git_repository: Mock, mock_urllib: Mock, mock_makedirs: Mock) -> None:
        build_recorder = MagicMock()
        manifest_path = os.path.join(os.path.dirname(__file__), "data", "opensearch-build-windows-1.1.0.yml")
        mock_builder = self.__mock_builder("common-utils")
        mock_builder.build_manifest = BuildManifest.from_path(manifest_path)
        mock_builder.export_artifacts(build_recorder)
        build_recorder.record_component.assert_called_with("common-utils", mock_manifest_git_repository.return_value)
        mock_makedirs.assert_called_with("builds", exist_ok=True)
        mock_urllib.assert_not_called()