class InteractivePartitioningInterfaceTestCase(unittest.TestCase):
"""Test DBus interface of the interactive partitioning module."""
def setUp(self):
"""Set up the module."""
self.module = InteractivePartitioningModule()
self.interface = InteractivePartitioningInterface(self.module)
@patch_dbus_publish_object
def get_device_tree_test(self, publisher):
"""Test GetDeviceTree."""
DeviceTreeInterface._tree_counter = 1
self.module.on_storage_reset(Mock())
tree_path = self.interface.GetDeviceTree()
publisher.assert_called_once()
object_path, obj = publisher.call_args[0]
self.assertEqual(tree_path, object_path)
self.assertIsInstance(obj, DeviceTreeInterface)
self.assertEqual(obj.implementation, self.module._device_tree_module)
self.assertEqual(obj.implementation.storage, self.module.storage)
self.assertTrue(tree_path.endswith("/DeviceTree/1"))
publisher.reset_mock()
self.assertEqual(tree_path, self.interface.GetDeviceTree())
self.assertEqual(tree_path, self.interface.GetDeviceTree())
self.assertEqual(tree_path, self.interface.GetDeviceTree())
publisher.assert_not_called()
@patch_dbus_publish_object
def configure_with_task_test(self, publisher):
"""Test ConfigureWithTask."""
self.module.on_storage_reset(Mock())
task_path = self.interface.ConfigureWithTask()
obj = check_task_creation(self, task_path, publisher,
InteractivePartitioningTask)
self.assertEqual(obj.implementation._storage, self.module.storage)
@patch_dbus_publish_object
def validate_with_task_test(self, publisher):
"""Test ValidateWithTask."""
self.module.on_storage_reset(Mock())
task_path = self.interface.ValidateWithTask()
obj = check_task_creation(self, task_path, publisher,
StorageValidateTask)
self.assertEqual(obj.implementation._storage, self.module.storage)
class InteractivePartitioningInterfaceTestCase(unittest.TestCase):
"""Test DBus interface of the interactive partitioning module."""
def setUp(self):
"""Set up the module."""
self.module = InteractivePartitioningModule()
self.interface = InteractivePartitioningInterface(self.module)
def publication_test(self):
"""Test the DBus representation."""
self.assertIsInstance(self.module.for_publication(),
InteractivePartitioningInterface)
def method_property_test(self):
"""Test Method property."""
self.assertEqual(self.interface.PartitioningMethod,
PARTITIONING_METHOD_INTERACTIVE)
@patch_dbus_publish_object
def get_device_tree_test(self, publisher):
"""Test GetDeviceTree."""
DeviceTreeContainer._counter = 0
self.module.on_storage_reset(Mock())
tree_path = self.interface.GetDeviceTree()
publisher.assert_called_once()
object_path, obj = publisher.call_args[0]
self.assertEqual(tree_path, object_path)
self.assertIsInstance(obj, DeviceTreeInterface)
self.assertEqual(obj.implementation, self.module._device_tree_module)
self.assertEqual(obj.implementation.storage, self.module.storage)
self.assertTrue(tree_path.endswith("/DeviceTree/1"))
publisher.reset_mock()
self.assertEqual(tree_path, self.interface.GetDeviceTree())
self.assertEqual(tree_path, self.interface.GetDeviceTree())
self.assertEqual(tree_path, self.interface.GetDeviceTree())
publisher.assert_not_called()
@patch_dbus_publish_object
def configure_with_task_test(self, publisher):
"""Test ConfigureWithTask."""
self.module.on_storage_reset(Mock())
task_path = self.interface.ConfigureWithTask()
obj = check_task_creation(self, task_path, publisher,
InteractivePartitioningTask)
self.assertEqual(obj.implementation._storage, self.module.storage)
def publish(self):
"""Publish the module."""
DBus.publish_object(INTERACTIVE_PARTITIONING.object_path,
InteractivePartitioningInterface(self))
def setUp(self):
"""Set up the module."""
self.module = InteractivePartitioningModule()
self.interface = InteractivePartitioningInterface(self.module)
def for_publication(self):
"""Return a DBus representation."""
return InteractivePartitioningInterface(self)