class TestVmdkTool(HelperUT):
"""Test cases for VmdkTool helper class."""
def setUp(self):
"""Test case setup function called automatically prior to each test."""
self.helper = VmdkTool()
super(TestVmdkTool, self).setUp()
def test_get_version(self):
"""Test .version getter logic."""
self.fake_output = "vmdktool version 1.4"
self.assertEqual(StrictVersion("1.4"), self.helper.version)
def test_install_helper_already_present(self):
"""Do nothing instead of re-installing."""
self.helper.install_helper()
self.assertEqual([], self.last_argv)
self.assertLogged(**self.ALREADY_INSTALLED)
@mock.patch('os.path.isdir')
@mock.patch('os.path.exists')
@mock.patch('os.makedirs')
def test_install_helper_apt_get(self,
mock_makedirs,
mock_exists,
mock_isdir):
"""Test installation via 'apt-get'."""
mock_isdir.return_value = False
mock_exists.return_value = False
mock_makedirs.side_effect = OSError
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['apt-get'] = True
Helper.PACKAGE_MANAGERS['port'] = False
Helper.PACKAGE_MANAGERS['yum'] = False
Helper._apt_updated = False
self.fake_output = 'is not installed and no information is available'
self.system = 'Linux'
os.environ['PREFIX'] = '/usr/local'
if 'DESTDIR' in os.environ:
del os.environ['DESTDIR']
self.helper.install_helper()
self.assertEqual([
['dpkg', '-s', 'make'],
['sudo', 'apt-get', '-q', 'update'],
['sudo', 'apt-get', '-q', 'install', 'make'],
['dpkg', '-s', 'zlib1g-dev'],
['sudo', 'apt-get', '-q', 'install', 'zlib1g-dev'],
['make', 'CFLAGS="-D_GNU_SOURCE -g -O -pipe"'],
['sudo', 'mkdir', '-p', '--mode=755', '/usr/local/man/man8'],
['sudo', 'mkdir', '-p', '--mode=755', '/usr/local/bin'],
['make', 'install', 'PREFIX=/usr/local'],
], self.last_argv)
self.assertTrue(Helper._apt_updated)
# Make sure we don't 'apt-get update/install' again unnecessarily
self.fake_output = 'install ok installed'
os.environ['PREFIX'] = '/opt/local'
os.environ['DESTDIR'] = '/home/cot'
self.last_argv = []
self.helper.install_helper()
self.assertEqual([
['dpkg', '-s', 'make'],
['dpkg', '-s', 'zlib1g-dev'],
['make', 'CFLAGS="-D_GNU_SOURCE -g -O -pipe"'],
['sudo', 'mkdir', '-p', '--mode=755',
'/home/cot/opt/local/man/man8'],
['sudo', 'mkdir', '-p', '--mode=755', '/home/cot/opt/local/bin'],
['make', 'install', 'PREFIX=/opt/local', 'DESTDIR=/home/cot'],
], self.last_argv)
def test_install_helper_port(self):
"""Test installation via 'port'."""
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['port'] = True
Helper._port_updated = False
self.helper.install_helper()
self.assertEqual([
['sudo', 'port', 'selfupdate'],
['sudo', 'port', 'install', 'vmdktool']
], self.last_argv)
self.assertTrue(Helper._port_updated)
# Make sure we don't 'port selfupdate' again unnecessarily
self.last_argv = []
self.helper.install_helper()
self.assertEqual([
['sudo', 'port', 'install', 'vmdktool']
], self.last_argv)
@mock.patch('os.path.isdir')
@mock.patch('os.path.exists')
@mock.patch('os.makedirs')
def test_install_helper_yum(self,
mock_makedirs,
mock_exists,
mock_isdir):
"""Test installation via 'yum'."""
mock_isdir.return_value = False
mock_exists.return_value = False
mock_makedirs.side_effect = OSError
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['apt-get'] = False
Helper.PACKAGE_MANAGERS['port'] = False
Helper.PACKAGE_MANAGERS['yum'] = True
self.system = 'Linux'
os.environ['PREFIX'] = '/usr/local'
if 'DESTDIR' in os.environ:
del os.environ['DESTDIR']
self.helper.install_helper()
self.assertEqual([
['sudo', 'yum', '--quiet', 'install', 'make'],
['sudo', 'yum', '--quiet', 'install', 'zlib-devel'],
['make', 'CFLAGS="-D_GNU_SOURCE -g -O -pipe"'],
['sudo', 'mkdir', '-p', '--mode=755', '/usr/local/man/man8'],
['sudo', 'mkdir', '-p', '--mode=755', '/usr/local/bin'],
['make', 'install', 'PREFIX=/usr/local'],
], self.last_argv)
def test_install_helper_unsupported(self):
"""Unable to install without a package manager."""
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['apt-get'] = False
Helper.PACKAGE_MANAGERS['port'] = False
Helper.PACKAGE_MANAGERS['yum'] = False
with self.assertRaises(NotImplementedError):
self.helper.install_helper()
def test_convert_unsupported(self):
"""Negative test - conversion to unsupported format/subformat."""
with self.assertRaises(NotImplementedError):
self.helper.convert_disk_image(self.blank_vmdk, self.temp_dir,
'qcow2')
with self.assertRaises(NotImplementedError):
self.helper.convert_disk_image(self.blank_vmdk, self.temp_dir,
'vmdk', 'monolithicSparse')
def setUp(self):
"""Test case setup function called automatically prior to each test."""
self.helper = VmdkTool()
super(TestVmdkTool, self).setUp()
class TestVmdkTool(HelperUT):
"""Test cases for VmdkTool helper class."""
def setUp(self):
"""Test case setup function called automatically prior to each test."""
self.helper = VmdkTool()
super(TestVmdkTool, self).setUp()
def test_get_version(self):
"""Test .version getter logic."""
self.fake_output = "vmdktool version 1.4"
self.assertEqual(StrictVersion("1.4"), self.helper.version)
def test_install_helper_already_present(self):
"""Do nothing instead of re-installing."""
self.helper.install_helper()
self.assertEqual([], self.last_argv)
self.assertLogged(**self.ALREADY_INSTALLED)
def test_install_helper_apt_get(self):
"""Test installation via 'apt-get'."""
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['apt-get'] = True
Helper.PACKAGE_MANAGERS['port'] = False
Helper.PACKAGE_MANAGERS['yum'] = False
Helper._apt_updated = False
self.system = 'Linux'
self.helper.install_helper()
self.assertEqual([
['sudo', 'apt-get', '-q', 'update'],
['sudo', 'apt-get', '-q', 'install', 'make'],
['sudo', 'apt-get', '-q', 'install', 'zlib1g-dev'],
['make', 'CFLAGS="-D_GNU_SOURCE -g -O -pipe"'],
['sudo', 'mkdir', '-p', '--mode=755', '/usr/local/man/man8'],
['sudo', 'make', 'install'],
], self.last_argv)
self.assertTrue(Helper._apt_updated)
# Make sure we don't 'apt-get update' again unnecessarily
self.last_argv = []
self.helper.install_helper()
self.assertEqual([
['sudo', 'apt-get', '-q', 'install', 'make'],
['sudo', 'apt-get', '-q', 'install', 'zlib1g-dev'],
['make', 'CFLAGS="-D_GNU_SOURCE -g -O -pipe"'],
['sudo', 'mkdir', '-p', '--mode=755', '/usr/local/man/man8'],
['sudo', 'make', 'install'],
], self.last_argv)
def test_install_helper_port(self):
"""Test installation via 'port'."""
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['port'] = True
Helper._port_updated = False
self.helper.install_helper()
self.assertEqual([
['sudo', 'port', 'selfupdate'],
['sudo', 'port', 'install', 'vmdktool']
], self.last_argv)
self.assertTrue(Helper._port_updated)
# Make sure we don't 'port selfupdate' again unnecessarily
self.last_argv = []
self.helper.install_helper()
self.assertEqual([
['sudo', 'port', 'install', 'vmdktool']
], self.last_argv)
def test_install_helper_yum(self):
"""Test installation via 'yum'."""
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['apt-get'] = False
Helper.PACKAGE_MANAGERS['port'] = False
Helper.PACKAGE_MANAGERS['yum'] = True
self.system = 'Linux'
self.helper.install_helper()
self.assertEqual([
['sudo', 'yum', '--quiet', 'install', 'make'],
['sudo', 'yum', '--quiet', 'install', 'zlib-devel'],
['make', 'CFLAGS="-D_GNU_SOURCE -g -O -pipe"'],
['sudo', 'mkdir', '-p', '--mode=755', '/usr/local/man/man8'],
['sudo', 'make', 'install'],
], self.last_argv)
def test_install_helper_unsupported(self):
"""Unable to install without a package manager."""
Helper.find_executable = self.stub_find_executable
Helper.PACKAGE_MANAGERS['apt-get'] = False
Helper.PACKAGE_MANAGERS['port'] = False
Helper.PACKAGE_MANAGERS['yum'] = False
with self.assertRaises(NotImplementedError):
self.helper.install_helper()
def test_convert_unsupported(self):
"""Negative test - conversion to unsupported format/subformat."""
with self.assertRaises(NotImplementedError):
self.helper.convert_disk_image(self.blank_vmdk, self.temp_dir,
'qcow2')
with self.assertRaises(NotImplementedError):
self.helper.convert_disk_image(self.blank_vmdk, self.temp_dir,
'vmdk', 'monolithicSparse')
