def test_disk_type(self, mock_exists): # Seems like lvm detection # if its in /dev ?? for p in ['/dev/b', '/dev/blah/blah']: d_type = libvirt_utils.get_disk_type(p) self.assertEqual('lvm', d_type) # Try rbd detection d_type = libvirt_utils.get_disk_type('rbd:pool/instance') self.assertEqual('rbd', d_type) # Try the other types template_output = """image: %(path)s file format: %(format)s virtual size: 64M (67108864 bytes) cluster_size: 65536 disk size: 96K """ path = '/myhome/disk.config' for f in ['raw', 'qcow2']: output = template_output % ({ 'format': f, 'path': path, }) with mock.patch('patron.utils.execute', return_value=(output, '')) as mock_execute: d_type = libvirt_utils.get_disk_type(path) mock_execute.assert_called_once_with('env', 'LC_ALL=C', 'LANG=C', 'qemu-img', 'info', path) self.assertEqual(f, d_type)
def test_disk_type(self, mock_exists): # Seems like lvm detection # if its in /dev ?? for p in ['/dev/b', '/dev/blah/blah']: d_type = libvirt_utils.get_disk_type(p) self.assertEqual('lvm', d_type) # Try rbd detection d_type = libvirt_utils.get_disk_type('rbd:pool/instance') self.assertEqual('rbd', d_type) # Try the other types template_output = """image: %(path)s file format: %(format)s virtual size: 64M (67108864 bytes) cluster_size: 65536 disk size: 96K """ path = '/myhome/disk.config' for f in ['raw', 'qcow2']: output = template_output % ({ 'format': f, 'path': path, }) with mock.patch('patron.utils.execute', return_value=(output, '')) as mock_execute: d_type = libvirt_utils.get_disk_type(path) mock_execute.assert_called_once_with( 'env', 'LC_ALL=C', 'LANG=C', 'qemu-img', 'info', path) self.assertEqual(f, d_type)
def test_get_disk_type(self, mock_execute, mock_exists): path = "disk.config" example_output = """image: disk.config file format: raw virtual size: 64M (67108864 bytes) cluster_size: 65536 disk size: 96K blah BLAH: bb """ mock_execute.return_value = (example_output, '') disk_type = libvirt_utils.get_disk_type(path) mock_execute.assert_called_once_with('env', 'LC_ALL=C', 'LANG=C', 'qemu-img', 'info', path) mock_exists.assert_called_once_with(path) self.assertEqual('raw', disk_type)