drgn/tests/test_util.py
Kevin Svetlitski d3c9e24115 tests: make all tests inherit from drgn's TestCase class
The majority of test cases already inherited from drgn's TestCase class.
The few outliers that inherited directly from unittest.TestCase have
been brought in line with the other tests.

Signed-off-by: Kevin Svetlitski <svetlitski@fb.com>
2022-01-11 17:28:17 -08:00

30 lines
1.1 KiB
Python

# Copyright (c) Meta Platforms, Inc. and affiliates.
# SPDX-License-Identifier: GPL-3.0-or-later
from functools import cmp_to_key
from tests import TestCase
from util import KernelVersion, verrevcmp
class TestUtil(TestCase):
def assertVersionSort(self, sorted_list):
self.assertEqual(sorted(sorted_list, key=cmp_to_key(verrevcmp)), sorted_list)
def test_verrevcmp(self):
self.assertVersionSort(
["0~", "0", "1", "1.0", "1.1~rc1", "1.1~rc2", "1.1", "1.2", "1.12"]
)
self.assertVersionSort(["a", "."])
self.assertVersionSort(["", "1"])
self.assertVersionSort(["~", "~1"])
self.assertVersionSort(["~~", "~~a", "~", "", "a"])
def test_kernel_version(self):
self.assertLess(KernelVersion("1.0"), KernelVersion("2.0"))
self.assertLess(KernelVersion("5.6.0-rc6"), KernelVersion("5.6.0-rc7"))
self.assertLess(KernelVersion("5.6.0-rc7"), KernelVersion("5.6.0"))
self.assertLess(
KernelVersion("5.6.0-rc7-vmtest2"), KernelVersion("5.6.0-vmtest1")
)
self.assertLess(KernelVersion("5.6.0-vmtest1"), KernelVersion("5.6.0-vmtest2"))