# 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, patch

from assemble_workflow.bundle import Bundle
from manifests.build_manifest import BuildComponent, BuildManifest


class TestBundle(unittest.TestCase):
    class DummyBundle(Bundle):
        def install_plugin(self, plugin: BuildComponent) -> None:
            pass

    @patch("assemble_workflow.dist.Dist.extract")
    def test_bundle(self, dist_extract: Mock) -> None:
        manifest_path = os.path.join(os.path.dirname(__file__), "data/opensearch-build-linux-1.1.0.yml")
        artifacts_path = os.path.join(os.path.dirname(__file__), "data", "artifacts")
        bundle = self.DummyBundle(BuildManifest.from_path(manifest_path), artifacts_path, MagicMock())
        self.assertEqual(bundle.min_dist.name, "OpenSearch")
        self.assertEqual(bundle.min_dist.__class__.__name__, "DistTar")
        self.assertEqual(bundle.build.distribution, None)
        self.assertEqual(bundle.build.platform, "linux")
        self.assertEqual(len(bundle.components), 15)
        self.assertEqual(bundle.artifacts_dir, artifacts_path)
        self.assertIsNotNone(bundle.bundle_recorder)
        self.assertEqual(bundle.installed_plugins, [])
        self.assertTrue(bundle.min_dist.path.endswith("opensearch-min-1.1.0-linux-x64.tar.gz"))
        dist_extract.assert_called_once()

    @patch("assemble_workflow.dist.Dist.extract")
    def test_bundle_distribution(self, dist_extract: Mock) -> None:
        manifest_path = os.path.join(os.path.dirname(__file__), "data/opensearch-build-windows-1.3.0.yml")
        artifacts_path = os.path.join(os.path.dirname(__file__), "data", "artifacts")
        bundle = self.DummyBundle(BuildManifest.from_path(manifest_path), artifacts_path, MagicMock())
        self.assertEqual(bundle.min_dist.name, "OpenSearch")
        self.assertEqual(bundle.min_dist.__class__.__name__, "DistZip")
        self.assertEqual(bundle.build.distribution, "zip")
        self.assertEqual(bundle.build.platform, "windows")
        self.assertEqual(bundle.artifacts_dir, artifacts_path)
        self.assertIsNotNone(bundle.bundle_recorder)
        self.assertEqual(bundle.installed_plugins, [])
        self.assertTrue(bundle.min_dist.path.endswith("opensearch-min-1.3.0-windows-x64.zip"))
        dist_extract.assert_called_once()

    def test_bundle_does_not_exist_raises_error(self) -> None:
        manifest_path = os.path.join(os.path.dirname(__file__), "data/opensearch-build-linux-1.1.0.yml")
        with self.assertRaises(FileNotFoundError) as ctx:
            self.DummyBundle(
                BuildManifest.from_path(manifest_path),
                os.path.join(os.path.dirname(__file__), "data", "does-not-exist"),
                MagicMock(),
            )
        self.assertTrue("opensearch-min-1.1.0-linux-x64.tar.gz" in str(ctx.exception))