def testRADOSBlockDeviceCreate(self, attach_mock, run_cmd_mock):
"""Test for bdev.RADOSBlockDevice.Create() success"""
attach_mock.return_value = True
# This returns a successful RunCmd result
run_cmd_mock.return_value = _FakeRunCmd(True, "", "")
expect = bdev.RADOSBlockDevice(self.test_unique_id, [], 1024,
self.test_params, {})
got = bdev.RADOSBlockDevice.Create(self.test_unique_id, [], 1024, None,
self.test_params, False, {},
test_kwarg="test")
self.assertEqual(expect, got)
def testRADOSBlockDeviceExport(self, attach_mock):
"""Test for bdev.RADOSBlockDevice.Export()"""
# Set up the mock object return value
attach_mock.return_value = True
# Create a fake rbd volume
inst = bdev.RADOSBlockDevice(self.test_unique_id, [], 1024,
self.test_params, {})
# Desired output command
export_cmd = [constants.RBD_CMD, "export",
"-p", inst.rbd_pool,
inst.rbd_name, "-"]
self.assertEqual(inst.Export(), export_cmd)
def testRADOSBlockDeviceImport(self, attach_mock, unmap_mock, run_cmd_mock):
"""Test for bdev.RADOSBlockDevice.Import()"""
# Set up the mock objects return values
attach_mock.return_value = True
run_cmd_mock.return_value = _FakeRunCmd(True, "", "")
# Create a fake rbd volume
inst = bdev.RADOSBlockDevice(self.test_unique_id, [], 1024,
self.test_params, {})
# Desired output command
import_cmd = [constants.RBD_CMD, "import",
"-p", inst.rbd_pool,
"-", inst.rbd_name]
self.assertEqual(inst.Import(), import_cmd)