import shutil
import unittest

from packaging.version import Version

from tests import Interceptor
import tests.data.kernel as data

import ekernel

class Tests (unittest.TestCase):

    def setUp (self):
        data.setup()

    def check_vars (self, k, s):
        v = ekernel.version(s.name)
        self.assertEqual(k.version, v)
        self.assertEqual(k.src, s)
        self.assertEqual(k.config, s / ".config")
        self.assertEqual(k.bzImage, s / "arch/x86_64/boot/bzImage")
        self.assertEqual(k.modules, data.modules / f"{v.base_version}-gentoo")
        self.assertEqual(
            k.bkp,
            data.boot.parent / f"gentoo-{v.base_version}.efi"
        )

    def test_paths (self):
        self.assertEqual(ekernel.Kernel.src, data.src)
        self.assertEqual(ekernel.Kernel.linux, data.linux)
        self.assertEqual(ekernel.Kernel.modules, data.modules)

    def test_version (self):
        self.assertEqual(ekernel.version("0.0.1-gentoo"), Version("0.0.1"))

    def test_constructor (self):
        self.check_vars(
            ekernel.Kernel(data.current),
            ekernel.Kernel.current().src
        )

    def test_list (self):
        self.assertEqual(ekernel.Kernel.list(), data.kernels)

    def test_list_ascending (self):
        self.assertEqual(
            ekernel.Kernel.list(descending=False),
            list(reversed(data.kernels))
        )

    def test_current (self):
        self.check_vars(ekernel.Kernel.current(), data.current)

    def test_latest (self):
        self.check_vars(ekernel.Kernel.latest(), data.latest)

    def test_eq (self):
        self.assertEqual(
            ekernel.Kernel.current(),
            ekernel.Kernel(data.current)
        )

    def test_neq (self):
        self.assertNotEqual(ekernel.Kernel.current(), ekernel.Kernel.latest())

    def test_current_missing (self):
        kernel = ekernel.Kernel.current()
        shutil.rmtree(kernel.src)
        with self.assertRaises(ValueError) as e:
            ekernel.Kernel.current()
        self.assertEqual(str(e.exception), f"missing source: {kernel.src}")