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)