def test_get_memory_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('cat', '/proc/meminfo', run_as_root=True)\
.AndRaise(exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._get_memory)
def test_make_directory_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo', 'mkdir', '-p', '/tmp/foo').AndReturn(
('',''))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._make_directory('/tmp/foo')
def test_attach_volumes_success(self):
conn = openvz_conn.OpenVzConnection(False)
self.mox.StubOutWithMock(conn, '_container_script_modify')
conn._container_script_modify(test_instance, None, mox.IgnoreArg(),
mox.IgnoreArg(), 'add')
self.mox.ReplayAll()
conn._attach_volumes(test_instance)
def test_set_perms_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo', 'chmod', 755, '/tmp/foo').AndRaise(
exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._set_perms('/tmp/foo', 755))
def test_make_directory_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo', 'mkdir', '-p', '/tmp/foo').AndRaise(
exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._make_directory, '/tmp/foo')
def before_tests(self):
"""Sets up the reboot tests."""
self.init('VERIFY_REBOOT_VZ_')
print("instance info local_id %s" % instance_info.local_id)
self.local_id = instance_info.local_id
self.name = "instance-00000001"
self.conn = openvz_conn.OpenVzConnection(False)
def test_get_cpuunits_usage_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzcpucheck', '-v', run_as_root=True)\
.AndRaise(exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._get_cpuunits_usage)
def test_attach_volumes_success(self):
conn = openvz_conn.OpenVzConnection(False)
self.mox.StubOutWithMock(conn, 'attach_volume')
conn.attach_volume(INSTANCE['name'], None,
mox.IgnoreArg())
self.mox.ReplayAll()
conn._attach_volumes(INSTANCE)
def test_write_to_file_failure(self):
self.mox.StubOutWithMock(openvz_conn, 'open')
openvz_conn.open('/tmp/foo', 'w').AndRaise(Exception)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._write_to_file,
'/tmp/foo', file_contents)
def test_write_to_file_success(self):
filehandle = FakeFile(file_contents)
self.mox.StubOutWithMock(openvz_conn, 'open')
openvz_conn.open('/tmp/foo', 'w').AndReturn(filehandle)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._write_to_file('/tmp/foo', file_contents)
def test_get_cpuunits_usage_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzcpucheck', '-v', run_as_root=True)\
.AndReturn((CPUCHECKCONT, ''))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._get_cpuunits_usage()
def test_set_diskspace_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo', 'vzctl', 'set', test_instance['id'],
'--save', '--diskspace',
mox.IgnoreArg()).AndReturn(('',None))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._set_diskspace(test_instance)
def test_get_cpuunits_capability_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo',
'vzcpucheck').AndRaise(
exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error,conn._get_cpuunits_capability)
def test_get_cpuunits_capability_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo',
'vzcpucheck').AndReturn(
(vzcpucheck_return_nocontainers,''))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._get_cpuunits_capability()
def test_set_diskspace_soft_and_hard_manual_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'set', INSTANCE['id'],
'--save', '--diskspace', '40G:50G',
run_as_root=True).AndReturn(('', None))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._set_diskspace(INSTANCE, 40, 50)
def test_set_diskspace_soft_and_hard_manual_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo', 'vzctl', 'set', test_instance['id'],
'--save', '--diskspace',
'40G:50G').AndReturn(('',None))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._set_diskspace(test_instance, (40,50))
def test_set_cpus_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'set', INSTANCE['id'], '--save',
'--cpus', mox.IgnoreArg(), run_as_root=True)\
.AndRaise(exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._set_cpus, INSTANCE)
def test_stop_failure_on_exec(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'stop', INSTANCE['id'],
run_as_root=True)\
.AndRaise(exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._stop, INSTANCE)
def test_configure_vz_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'set', INSTANCE['id'], '--save',
'--applyconfig', 'basic', run_as_root=True)\
.AndRaise(exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._configure_vz, INSTANCE)
def test_configure_vz_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'set', INSTANCE['id'], '--save',
'--applyconfig', 'basic', run_as_root=True)\
.AndReturn(('', ''))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._configure_vz(INSTANCE)
def test_set_diskspace_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'set', INSTANCE['id'], '--save',
'--diskspace', mox.IgnoreArg(),
run_as_root=True).AndReturn(('', None))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._set_diskspace(INSTANCE)
def test_set_vz_os_hint_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'set', INSTANCE['id'], '--save',
'--ostemplate', 'ubuntu', run_as_root=True)\
.AndReturn(('', ''))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._set_vz_os_hint(INSTANCE)
def test_create_vz_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'create', INSTANCE['id'],
'--ostemplate', INSTANCE['image_ref'],
run_as_root=True).AndReturn(('', ''))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._create_vz(INSTANCE)
def test_set_perms_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo', 'chmod', 755, '/tmp/foo').AndReturn(
('','')
)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._set_perms('/tmp/foo', 755)
def test_read_file_success(self):
file_contents = FakeFile(file_contents)
self.mox.StubOutWithMock(openvz_conn, 'open')
openvz_conn.open('/tmp/foo', 'r').AndReturn(file_contents)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
resp = conn._read_file('/tmp/foo')
self.assertTrue(isinstance(resp, list))
def test_gratuitous_arp_all_addresses(self):
conn = openvz_conn.OpenVzConnection(False)
self.mox.StubOutWithMock(conn, '_send_garp')
conn._send_garp(INSTANCE['id'],
mox.IgnoreArg(),
mox.IgnoreArg(),
mox.IgnoreArg()).MultipleTimes()
self.mox.ReplayAll()
conn._gratuitous_arp_all_addresses(INSTANCE, NETWORKINFO)
def test_set_vz_os_hint_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'set', INSTANCE['id'],
'--save', '--ostemplate', 'ubuntu',
run_as_root=True)\
.AndRaise(exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._set_vz_os_hint, INSTANCE)
def test_get_memory_success(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('cat', '/proc/meminfo', run_as_root=True)\
.AndReturn((MEMINFO, ''))
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
conn._get_memory()
self.assertEquals(int, type(conn.utility['MEMORY_MB']))
self.assertTrue(conn.utility['MEMORY_MB'] > 0)
def test_create_vz_fail(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('vzctl', 'create', INSTANCE['id'],
'--ostemplate', INSTANCE['image_ref'],
run_as_root=True)\
.AndRaise(exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._create_vz, INSTANCE)
def test_set_diskspace_failure(self):
self.mox.StubOutWithMock(openvz_conn.utils, 'execute')
openvz_conn.utils.execute('sudo', 'vzctl', 'set', test_instance['id'],
'--save', '--diskspace',
mox.IgnoreArg()).AndRaise(
exception.ProcessExecutionError)
self.mox.ReplayAll()
conn = openvz_conn.OpenVzConnection(False)
self.assertRaises(exception.Error, conn._set_diskspace, test_instance)
