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

from build_workflow.build_target import BuildTarget


class TestBuildTarget(unittest.TestCase):
    def test_output_dir(self) -> None:
        self.assertEqual(BuildTarget(version="1.1.0", architecture="x86").output_dir, "artifacts")

    def test_build_id_hex(self) -> None:
        self.assertEqual(len(BuildTarget(version="1.1.0", architecture="x86").build_id), 32)

    @patch.dict(os.environ, {"BUILD_NUMBER": "id"})
    def test_build_id_from_env(self) -> None:
        self.assertEqual(BuildTarget(version="1.1.0", architecture="x86").build_id, "id")

    def test_build_id_from_arg(self) -> None:
        self.assertEqual(BuildTarget(version="1.1.0", architecture="x86", build_id="id").build_id, "id")

    def test_distribution_from_arg(self) -> None:
        self.assertEqual(BuildTarget(version="1.3.0", architecture="x86", distribution="rpm").distribution, "rpm")

    def test_opensearch_version(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.0", architecture="x86", snapshot=False).opensearch_version,
            "1.1.0",
        )

    def test_opensearch_version_snapshot(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.0", architecture="x86", snapshot=True).opensearch_version,
            "1.1.0-SNAPSHOT",
        )

    def test_opensearch_version_qualifier(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.0", architecture="x86", snapshot=False, qualifier="alpha1").opensearch_version,
            "1.1.0-alpha1",
        )

    def test_opensearch_version_snapshot_qualifier(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.0", architecture="x86", snapshot=True, qualifier="alpha1").opensearch_version,
            "1.1.0-alpha1-SNAPSHOT",
        )

    def test_compatible_min_versions(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=False).compatible_min_versions,
            ['1.1.2', '1.1.0', '1.1.1', '1.1.0-SNAPSHOT', '1.1.1-SNAPSHOT'],
        )

    def test_compatible_min_versions_qualifier(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=False, qualifier="alpha1").compatible_min_versions,
            ['1.1.2-alpha1', '1.1.0', '1.1.1', '1.1.0-alpha1-SNAPSHOT', '1.1.1-alpha1-SNAPSHOT'],
        )

    def test_compatible_min_versions_snapshot(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=True).compatible_min_versions,
            ['1.1.2-SNAPSHOT', '1.1.0', '1.1.1', '1.1.0-SNAPSHOT', '1.1.1-SNAPSHOT'],
        )

    def test_compatible_min_versions_snapshot_qualifier(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=True, qualifier="alpha1").compatible_min_versions,
            ['1.1.2-alpha1-SNAPSHOT', '1.1.0', '1.1.1', '1.1.0-alpha1-SNAPSHOT', '1.1.1-alpha1-SNAPSHOT'],
        )

    def test_component_version(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.0", architecture="x86", snapshot=False).component_version,
            "1.1.0.0",
        )

    def test_component_version_qualifier(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.0", architecture="x86", qualifier="alpha1", snapshot=False).component_version,
            "1.1.0.0-alpha1",
        )

    def test_compatible_component_versions(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=False).compatible_component_versions,
            ['1.1.2.0', '1.1.0.0', '1.1.1.0', '1.1.0.0-SNAPSHOT', '1.1.1.0-SNAPSHOT'],
        )

    def test_compatible_component_versions_qualifier(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=False, qualifier="alpha1").compatible_component_versions,
            ['1.1.2.0-alpha1', '1.1.0.0-alpha1', '1.1.1.0-alpha1', '1.1.0.0-alpha1-SNAPSHOT', '1.1.1.0-alpha1-SNAPSHOT'],
        )

    def test_compatible_component_versions_snapshot(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=True).compatible_component_versions,
            ['1.1.2.0-SNAPSHOT', '1.1.0.0', '1.1.1.0', '1.1.0.0-SNAPSHOT', '1.1.1.0-SNAPSHOT'],
        )

    def test_compatible_component_versions_snapshot_qualifier(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.2", architecture="x86", patches=["1.1.0", "1.1.1"], snapshot=True, qualifier="alpha1").compatible_component_versions,
            ['1.1.2.0-alpha1-SNAPSHOT', '1.1.0.0-alpha1', '1.1.1.0-alpha1', '1.1.0.0-alpha1-SNAPSHOT', '1.1.1.0-alpha1-SNAPSHOT'],
        )

    def test_component_version_snapshot(self) -> None:
        self.assertEqual(
            BuildTarget(version="1.1.0", architecture="x86", snapshot=True).component_version,
            "1.1.0.0-SNAPSHOT",
        )

    @patch("build_workflow.build_target.current_platform", return_value="value")
    def test_platform(self, value_platform: Mock) -> None:
        self.assertEqual(BuildTarget(version="1.1.0", snapshot=False).platform, "value")
        value_platform.assert_called_once()

    def test_platform_value(self) -> None:
        self.assertEqual(BuildTarget(version="1.1.0", platform="value", snapshot=False).platform, "value")

    @patch("build_workflow.build_target.current_architecture", return_value="value")
    def test_arch(self, value_architecture: Mock) -> None:
        self.assertEqual(BuildTarget(version="1.1.0", snapshot=False).architecture, "value")
        value_architecture.assert_called_once()

    def test_arch_value(self) -> None:
        self.assertEqual(BuildTarget(version="1.1.0", architecture="value", snapshot=False).architecture, "value")