2399aa3b2d
Fixes https://github.com/redhat-performance/tuned/issues/241 Fedora is removing the unittest2 package: https://bugzilla.redhat.com/show_bug.cgi?id=1794222 tuned is not targeting Python 2.6: https://github.com/redhat-performance/tuned/issues/241#issuecomment-588320175 One thing not present in Python 2.7 is assertItemsEqual() -- renamed to assertCountEqual(). Related, mock was also listed as a requirement in the spec file, but all the imports are from unittest.mock, hence removed.
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
import unittest
|
|
from unittest.mock import Mock
|
|
import pyudev
|
|
|
|
from tuned.hardware.inventory import Inventory
|
|
|
|
subsystem_name = "test subsystem"
|
|
|
|
class InventoryTestCase(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls._context = pyudev.Context()
|
|
cls._inventory = Inventory(set_receive_buffer_size=False)
|
|
cls._dummy = DummyPlugin()
|
|
cls._dummier = DummyPlugin()
|
|
|
|
def test_get_device(self):
|
|
try:
|
|
device1 = pyudev.Devices.from_name(self._context, "tty", "tty0")
|
|
except AttributeError:
|
|
device1 = pyudev.Device.from_name(self._context, "tty", "tty0")
|
|
device2 = self._inventory.get_device("tty", "tty0")
|
|
self.assertEqual(device1,device2)
|
|
|
|
def test_get_devices(self):
|
|
device_list1 = self._context.list_devices(subsystem = "tty")
|
|
device_list2 = self._inventory.get_devices("tty")
|
|
try:
|
|
self.assertCountEqual(device_list1,device_list2)
|
|
except AttributeError: # Python 2
|
|
self.assertItemsEqual(device_list1,device_list2)
|
|
|
|
def test_subscribe(self):
|
|
self._inventory.subscribe(self._dummy,subsystem_name,
|
|
self._dummy.TestCallback)
|
|
self._inventory.subscribe(self._dummier,subsystem_name,
|
|
self._dummier.TestCallback)
|
|
device = Mock(subsystem = subsystem_name)
|
|
self._inventory._handle_udev_event("test event", device)
|
|
self.assertTrue(self._dummy.CallbackWasCalled)
|
|
self.assertTrue(self._dummier.CallbackWasCalled)
|
|
|
|
def test_unsubscribe(self):
|
|
self._dummy.CallbackWasCalled = False
|
|
self._dummier.CallbackWasCalled = False
|
|
self._inventory.unsubscribe(self._dummy)
|
|
device = Mock(subsystem = subsystem_name)
|
|
self._inventory._handle_udev_event("test event", device)
|
|
self.assertFalse(self._dummy.CallbackWasCalled)
|
|
self.assertTrue(self._dummier.CallbackWasCalled)
|
|
self._dummier.CallbackWasCalled = False
|
|
self._inventory.unsubscribe(self._dummier)
|
|
self._inventory._handle_udev_event("test event", device)
|
|
self.assertFalse(self._dummy.CallbackWasCalled)
|
|
self.assertFalse(self._dummier.CallbackWasCalled)
|
|
self.assertIsNone(self._inventory._monitor_observer)
|
|
|
|
class DummyPlugin():
|
|
def __init__(self):
|
|
self.CallbackWasCalled = False
|
|
|
|
def TestCallback(self, event, device):
|
|
self.CallbackWasCalled = True
|