#!/usr/bin/python
from __future__ import division
import sys

import unittest

from check_kernel import clean_kernel_version, Version
import check_kernel


def patch_object(obj, function, return_value=None):
    """A simple replacement for unittest.mock.patch_object()

    It is not meant to used with the 'with' statement or as a decorator, unlike
    unittest.mock.patch_object(). To support Python <= 3.2."""
    def our_function():
        return return_value
    obj.__dict__[function] = our_function


# Patch up unittest.TestCase
if not hasattr(unittest.TestCase, 'assertGreater'):
    unittest.TestCase.assertGreater = lambda self, a, b: self.assertTrue(a > b)
if not hasattr(unittest.TestCase, 'assertLess'):
    unittest.TestCase.assertLess = lambda self, a, b: self.assertTrue(a < b)


class CleanKernelVersionTestCase(unittest.TestCase):
    def testDebianVanilla(self):
        versions = [('3.16.7-ckt9-3~deb8u2', '3.16.7-ckt9-3~deb8u2')]
        for dirty, clean in versions:
            self.assertEqual(clean_kernel_version(dirty), Version(clean))

    def testDebianProxmox(self):
        versions = [('5.13.19-14', '5.13.19-14')]
        for dirty, clean in versions:
            self.assertEqual(clean_kernel_version(dirty), Version(clean))

    def testFedora(self):
        versions = [
                ('4.0.4-301.fc22.x86_64', '4.0.4-301'),
                ('4.0.0-1.fc22', '4.0.0-1'),
        ]
        for dirty, clean in versions:
            self.assertEqual(clean_kernel_version(dirty), Version(clean))

    def testRHEL(self):
        versions = [
                ('4.18.0-348.12.2.el8_5', '4.18.0-348.12.2'),
                ('4.18.0-348.el8.0.2', '4.18.0-348')
        ]
        for dirty, clean in versions:
            self.assertEqual(clean_kernel_version(dirty), Version(clean))


class RunningKernelVersionTestCase(unittest.TestCase):
    def testFedora(self):
        patch_object(check_kernel, 'proc_version', return_value='Linux version 4.0.3-301.fc22.x86_64 (mockbuild@bkernel02.phx2.fedoraproject.org) (gcc version 5.1.1 20150422 (Red Hat 5.1.1-1) (GCC) ) #1 SMP Thu May 21 13:10:33 UTC 2015')
        self.assertEqual(check_kernel.running_kernel_version(),
                         Version('4.0.3-301'))

    def testDebianVanilla(self):
        patch_object(check_kernel, 'proc_version', return_value='Linux version 4.19.0-5-amd64 (debian-kernel@lists.debian.org) (gcc version 8.3.0 (Debian 8.3.0-7)) #1 SMP Debian 4.19.37-5 (2019-06-19)')
        self.assertEqual(check_kernel.running_kernel_version(),
                         Version('4.19.37-5'))

        patch_object(check_kernel, 'proc_version', return_value='Linux version 3.16.0-4-amd64 (debian-kernel@lists.debian.org) (gcc version 4.8.4 (Debian 4.8.4-1) ) #1 SMP Debian 3.16.7-ckt9-3~deb8u1 (2015-04-24)')
        self.assertEqual(check_kernel.running_kernel_version(),
                         Version('3.16.7-ckt9-3~deb8u1'))

        patch_object(check_kernel, 'proc_version', return_value='Linux version 2.6.32-5-amd64 (Debian 2.6.32-48squeeze11) (ben@decadent.org.uk) (gcc version 4.3.5 (Debian 4.3.5-4) ) #1 SMP Wed Feb 18 13:14:10 UTC 2015')
        self.assertEqual(check_kernel.running_kernel_version(),
                         Version('2.6.32-48squeeze11'))

        patch_object(check_kernel, 'proc_version', return_value='Linux version 5.10.0-8-amd64 (debian-kernel@lists.debian.org) (gcc-10 (Debian 10.2.1-6) 10.2.1 20210110, GNU ld (GNU Binutils for Debian) 2.35.2) #1 SMP Debian 5.10.46-4 (2021-08-03)')
        self.assertEqual(check_kernel.running_kernel_version(),
                         Version('5.10.46-4'))

    def testDebianProxmox(self):
        patch_object(check_kernel, 'proc_version', return_value='Linux version 5.4.162-1-pve (build@proxmox) (gcc version 8.3.0 (Debian 8.3.0-6)) #1 SMP PVE 5.4.162-2 (Thu, 20 Jan 2022 16:38:53 +0100)')
        self.assertEqual(check_kernel.running_kernel_version(),
                         Version('5.4.162-2'))

        patch_object(check_kernel, 'proc_version', return_value='Linux version 5.13.19-6-pve (build@proxmox) (gcc (Debian 10.2.1-6) 10.2.1 20210110, GNU ld (GNU Binutils for Debian) 2.35.2) #1 SMP PVE 5.13.19-14 (Thu, 10 Mar 2022 16:24:52 +0100)')
        self.assertEqual(check_kernel.running_kernel_version(),
                         Version('5.13.19-14'))


class VersionTestCase(unittest.TestCase):
    def testStr(self):
        self.assertEqual(str(Version('1.0')), '1.0')

    def testComparingTrivial(self):
        self.assertEqual(Version('1.0'), Version('1.0'))
        self.assertEqual(Version('2.6.32-5foo1'), Version('2.6.32-5foo1'))

        self.assertGreater(Version('2.0'), Version('1.0'))

    def testComparingNonTrivial(self):
        # lower < greater
        versions = [('2.2', '2.12'),
                    ('3.12-2', '3.12-12'),
                    ('2.6.32-5foo0', '2.6.32-5foo1'),
                    ('3.4.15-2', '3.16.7-ckt9-3~deb8u1'),
                    ('3.2.4', '3.2.4+2'),
                    ('4.18.0-348', '4.18.0-348.12.2')]

        for lower, greater in versions:
            self.assertGreater(Version(greater), Version(lower))
            self.assertLess(Version(lower), Version(greater))

    def testSorting(self):
        l = [Version('3.16.5-1'), Version('3.2.60-1+deb7u3'),
             Version('3.16.3-2'), Version('3.16.7-ckt9-3~deb8u1'),
             Version('3.14.15-2')]
        self.assertEqual(sorted(l)[0], Version('3.2.60-1+deb7u3'))
        self.assertEqual(sorted(l)[-1], Version('3.16.7-ckt9-3~deb8u1'))


if __name__ == '__main__':
    unittest.main()