# -*- coding: utf-8 -*-

from unittest import mock
import unittest

from pungi.arch import (
    get_compatible_arches,
    get_valid_arches,
    get_valid_multilib_arches,
    is_excluded,
    is_valid_arch,
    split_name_arch,
)


class TestArch(unittest.TestCase):
    def test_i386(self):
        arches = get_valid_arches("i386")
        self.assertEqual(arches, ["i686", "i586", "i486", "i386", "noarch"])

        arches = get_valid_arches("i386", multilib=False)
        self.assertEqual(arches, ["i686", "i586", "i486", "i386", "noarch"])

        arches = get_valid_arches("i386", add_src=True)
        self.assertEqual(arches, ["i686", "i586", "i486", "i386", "noarch", "src"])

    def test_x86_64(self):
        arches = get_valid_arches("x86_64")
        self.assertEqual(
            arches, ["x86_64", "athlon", "i686", "i586", "i486", "i386", "noarch"]
        )

        arches = get_valid_arches("x86_64", multilib=False)
        self.assertEqual(arches, ["x86_64", "noarch"])

        arches = get_valid_arches("x86_64", add_src=True)
        self.assertEqual(
            arches,
            ["x86_64", "athlon", "i686", "i586", "i486", "i386", "noarch", "src"],
        )

    def test_armhfp(self):
        arches = get_valid_arches("armhfp")
        self.assertEqual(arches, ["armv7hnl", "armv7hl", "armv6hl", "noarch"])

        arches = get_valid_arches("armhfp", multilib=False)
        self.assertEqual(arches, ["armv7hnl", "armv7hl", "armv6hl", "noarch"])

        arches = get_valid_arches("armhfp", add_src=True)
        self.assertEqual(arches, ["armv7hnl", "armv7hl", "armv6hl", "noarch", "src"])

    def test_get_compatible_arches(self):
        self.assertEqual(get_compatible_arches("noarch"), ["noarch"])
        self.assertEqual(get_compatible_arches("i386"), get_valid_arches("i386"))
        self.assertEqual(get_compatible_arches("i586"), get_valid_arches("i386"))
        self.assertEqual(
            get_compatible_arches("x86_64"), get_valid_arches("x86_64", multilib=False)
        )
        self.assertEqual(
            get_compatible_arches("ppc64p7"), get_valid_arches("ppc64", multilib=False)
        )
        self.assertEqual(
            get_compatible_arches("armhfp"),
            get_valid_arches("armv7hnl", multilib=False),
        )

    def test_is_valid_arch(self):
        self.assertEqual(is_valid_arch("i386"), True)
        self.assertEqual(is_valid_arch("x86_64"), True)
        self.assertEqual(is_valid_arch("noarch"), True)
        self.assertEqual(is_valid_arch("src"), True)
        self.assertEqual(is_valid_arch("nosrc"), True)
        self.assertEqual(is_valid_arch("foo"), False)
        self.assertEqual(is_valid_arch("armhfp"), False)

    def test_split_name_arch(self):
        self.assertEqual(split_name_arch("package"), ("package", None))
        self.assertEqual(split_name_arch("package.x86_64"), ("package", "x86_64"))
        self.assertEqual(split_name_arch("package.foo"), ("package.foo", None))
        self.assertEqual(
            split_name_arch("i386"), ("i386", None)
        )  # we suppose that $name is never empty

    def test_get_valid_multilib_arches(self):
        self.assertEqual(get_valid_multilib_arches("noarch"), [])
        self.assertEqual(get_valid_multilib_arches("athlon"), [])
        self.assertEqual(
            get_valid_multilib_arches("x86_64"),
            ["athlon", "i686", "i586", "i486", "i386"],
        )


class TestExclusiveExcludeArch(unittest.TestCase):
    def test_no_exclude(self):
        pkg = mock.Mock(excludearch=[], exclusivearch=[], file_name="pkg.rpm")
        self.assertFalse(is_excluded(pkg, ["x86_64"]))

    def test_exclude_arch(self):
        log = mock.Mock()
        pkg = mock.Mock(excludearch=["x86_64"], exclusivearch=[], file_name="pkg.rpm")
        self.assertTrue(is_excluded(pkg, ["x86_64"], logger=log))
        self.assertEqual(
            log.mock_calls,
            [mock.call.debug("Excluding (EXCLUDEARCH: ['x86_64']): pkg.rpm")],
        )

    def test_exclusive_arch(self):
        log = mock.Mock()
        pkg = mock.Mock(excludearch=[], exclusivearch=["aarch64"], file_name="pkg.rpm")
        self.assertTrue(is_excluded(pkg, ["x86_64"], logger=log))
        self.assertEqual(
            log.mock_calls,
            [mock.call.debug("Excluding (EXCLUSIVEARCH: ['aarch64']): pkg.rpm")],
        )