try:
    from unittest import mock
except ImportError:
    import mock
import os

import six

from tests import helpers

from pungi import metadata
from pungi.compose_metadata import discinfo


class DiscInfoTestCase(helpers.PungiTestCase):
    def setUp(self):
        super(DiscInfoTestCase, self).setUp()
        os.environ["SOURCE_DATE_EPOCH"] = "101010101"
        self.path = os.path.join(self.topdir, "compose/Server/x86_64/os/.discinfo")

    def test_write_discinfo_variant(self):
        compose = helpers.DummyCompose(
            self.topdir, {"release_name": "Test", "release_version": "1.0"}
        )

        metadata.write_discinfo(compose, "x86_64", compose.variants["Server"])

        with open(self.path) as f:
            self.assertEqual(
                f.read().strip().split("\n"), ["101010101", "Test 1.0", "x86_64", "ALL"]
            )

        self.assertEqual(
            discinfo.read_discinfo(self.path),
            {
                "timestamp": "101010101",
                "description": "Test 1.0",
                "disc_numbers": ["ALL"],
                "arch": "x86_64",
            },
        )

    def test_write_discinfo_custom_description(self):
        compose = helpers.DummyCompose(
            self.topdir,
            {
                "release_name": "Test",
                "release_version": "1.0",
                "release_discinfo_description": "Fuzzy %(variant_name)s.%(arch)s",
            },
        )
        compose.variants["Server"].name = "Server"

        metadata.write_discinfo(compose, "x86_64", compose.variants["Server"])

        with open(self.path) as f:
            self.assertEqual(
                f.read().strip().split("\n"),
                ["101010101", "Fuzzy Server.x86_64", "x86_64", "ALL"],
            )

    def test_write_discinfo_layered_product(self):
        compose = helpers.DummyCompose(
            self.topdir,
            {
                "release_name": "Test",
                "release_version": "1.0",
                "base_product_name": "Base",
                "base_product_version": 42,
            },
        )

        metadata.write_discinfo(compose, "x86_64", compose.variants["Server"])

        with open(self.path) as f:
            self.assertEqual(
                f.read().strip().split("\n"),
                ["101010101", "Test 1.0 for Base 42", "x86_64", "ALL"],
            )

    def test_write_discinfo_integrated_layered_product(self):
        compose = helpers.DummyCompose(
            self.topdir, {"release_name": "Test", "release_version": "1.0"}
        )
        compose.variants["ILP"] = mock.Mock(
            uid="Server",
            arches=["x86_64"],
            type="layered-product",
            is_empty=False,
            release_name="Integrated",
            release_version="2.1",
            parent=compose.variants["Server"],
        )

        metadata.write_discinfo(compose, "x86_64", compose.variants["ILP"])

        with open(self.path) as f:
            self.assertEqual(
                f.read().strip().split("\n"),
                ["101010101", "Integrated 2.1 for Test 1", "x86_64", "ALL"],
            )

    def test_addons_dont_have_discinfo(self):
        compose = helpers.DummyCompose(
            self.topdir, {"release_name": "Test", "release_version": "1.0"}
        )
        compose.variants["ILP"] = mock.Mock(
            uid="Server",
            arches=["x86_64"],
            type="addon",
            is_empty=False,
            parent=compose.variants["Server"],
        )

        metadata.write_discinfo(compose, "x86_64", compose.variants["ILP"])

        self.assertFalse(os.path.isfile(self.path))


class MediaRepoTestCase(helpers.PungiTestCase):
    def setUp(self):
        super(MediaRepoTestCase, self).setUp()
        self.path = os.path.join(self.topdir, "compose/Server/x86_64/os/media.repo")

    def test_write_media_repo(self):
        compose = helpers.DummyCompose(
            self.topdir, {"release_name": "Test", "release_version": "1.0"}
        )

        metadata.write_media_repo(
            compose, "x86_64", compose.variants["Server"], timestamp=123456
        )

        with open(self.path) as f:
            lines = f.read().strip().split("\n")
            self.assertEqual(lines[0], "[InstallMedia]")
            six.assertCountEqual(
                self,
                lines[1:],
                [
                    "name=Test 1.0",
                    "mediaid=123456",
                    "metadata_expire=-1",
                    "gpgcheck=0",
                    "cost=500",
                ],
            )

    def test_addons_dont_have_media_repo(self):
        compose = helpers.DummyCompose(
            self.topdir, {"release_name": "Test", "release_version": "1.0"}
        )
        compose.variants["ILP"] = mock.Mock(
            uid="Server",
            arches=["x86_64"],
            type="addon",
            is_empty=False,
            parent=compose.variants["Server"],
        )

        metadata.write_discinfo(compose, "x86_64", compose.variants["ILP"])

        self.assertFalse(os.path.isfile(self.path))


FOO_MD5 = {"md5": "acbd18db4cc2f85cedef654fccc4a4d8"}
BAR_MD5 = {"md5": "37b51d194a7513e45b56f6524f2d51f2"}


class TestPopulateExtraFiles(helpers.PungiTestCase):
    def setUp(self):
        super(TestPopulateExtraFiles, self).setUp()
        self.variant = mock.Mock(uid="Server")
        self.metadata = mock.Mock()

    def test_with_relative_root(self):
        helpers.touch(
            os.path.join(self.topdir, "compose/Server/x86_64/os/foo"), content="foo"
        )
        helpers.touch(
            os.path.join(self.topdir, "compose/Server/x86_64/os/bar"), content="bar"
        )

        metadata.populate_extra_files_metadata(
            self.metadata,
            self.variant,
            "x86_64",
            os.path.join(self.topdir, "compose/Server/x86_64/os"),
            ["foo", "bar"],
            ["md5"],
            relative_root=os.path.join(self.topdir, "compose"),
        )

        self.maxDiff = None

        six.assertCountEqual(
            self,
            self.metadata.mock_calls,
            [
                mock.call.add("Server", "x86_64", "Server/x86_64/os/foo", 3, FOO_MD5),
                mock.call.add("Server", "x86_64", "Server/x86_64/os/bar", 3, BAR_MD5),
                mock.call.dump_for_tree(
                    mock.ANY, "Server", "x86_64", "Server/x86_64/os/"
                ),
            ],
        )

    def test_without_relative_root(self):
        helpers.touch(os.path.join(self.topdir, "foo"), content="foo")
        helpers.touch(os.path.join(self.topdir, "bar"), content="bar")

        metadata.populate_extra_files_metadata(
            self.metadata, self.variant, "x86_64", self.topdir, ["foo", "bar"], ["md5"]
        )

        six.assertCountEqual(
            self,
            self.metadata.mock_calls,
            [
                mock.call.add("Server", "x86_64", "foo", 3, FOO_MD5),
                mock.call.add("Server", "x86_64", "bar", 3, BAR_MD5),
                mock.call.dump_for_tree(mock.ANY, "Server", "x86_64", ""),
            ],
        )