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.
88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
import unittest
|
|
from unittest.mock import Mock
|
|
|
|
from tuned.exports.controller import ExportsController
|
|
import tuned.exports as exports
|
|
|
|
class ControllerTestCase(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls._controller = ExportsController()
|
|
|
|
def test_is_exportable_method(self):
|
|
self.assertFalse(self._controller._is_exportable_method( \
|
|
MockClass().NonExportableObject))
|
|
|
|
self.assertTrue(self._controller._is_exportable_method( \
|
|
MockClass().ExportableMethod))
|
|
|
|
def test_is_exportable_signal(self):
|
|
self.assertFalse(self._controller._is_exportable_signal( \
|
|
MockClass().NonExportableObject))
|
|
|
|
self.assertTrue(self._controller._is_exportable_signal( \
|
|
MockClass().ExportableSignal))
|
|
|
|
def test_initialize_exports(self):
|
|
local_controller = ExportsController()
|
|
exporter = MockExporter()
|
|
instance = MockClass()
|
|
local_controller.register_exporter(exporter)
|
|
local_controller.register_object(instance)
|
|
local_controller._initialize_exports()
|
|
self.assertEqual(exporter.exported_methods[0].method,\
|
|
instance.ExportableMethod)
|
|
self.assertEqual(exporter.exported_methods[0].args[0],\
|
|
"method_param1")
|
|
self.assertEqual(exporter.exported_methods[0].kwargs['kword'],\
|
|
"method_param2")
|
|
self.assertEqual(exporter.exported_signals[0].method,\
|
|
instance.ExportableSignal)
|
|
self.assertEqual(exporter.exported_signals[0].args[0],\
|
|
"signal_param1")
|
|
self.assertEqual(exporter.exported_signals[0].kwargs['kword'],\
|
|
"signal_param2")
|
|
|
|
def test_start_stop(self):
|
|
local_controller = ExportsController()
|
|
exporter = MockExporter()
|
|
local_controller.register_exporter(exporter)
|
|
local_controller.start()
|
|
self.assertTrue(exporter.is_running)
|
|
local_controller.stop()
|
|
self.assertFalse(exporter.is_running)
|
|
|
|
|
|
class MockExporter(object):
|
|
def __init__(self):
|
|
self.exported_methods = []
|
|
self.exported_signals = []
|
|
self.is_running = False
|
|
|
|
def export(self,method,*args,**kwargs):
|
|
object_to_export = Mock(\
|
|
method = method, args = args, kwargs = kwargs)
|
|
self.exported_methods.append(object_to_export)
|
|
|
|
def signal(self,method,*args,**kwargs):
|
|
object_to_export = Mock(\
|
|
method = method, args = args, kwargs = kwargs)
|
|
self.exported_signals.append(object_to_export)
|
|
|
|
def start(self):
|
|
self.is_running = True
|
|
|
|
def stop(self):
|
|
self.is_running = False
|
|
|
|
class MockClass(object):
|
|
@exports.export('method_param1', kword = 'method_param2')
|
|
def ExportableMethod(self):
|
|
return True
|
|
|
|
def NonExportableObject(self):
|
|
pass
|
|
|
|
@exports.signal('signal_param1', kword = 'signal_param2')
|
|
def ExportableSignal(self):
|
|
return True
|