def test_getReservedMemSize_invalid_suffix(self):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
gap = makeVM.getReservedMemSize(rh, '1024M', '128T')
self.assertEqual(gap, '0M')
self.assertEqual(rh.results['overallRC'], 4)
self.assertEqual(rh.results['rc'], 4)
self.assertEqual(rh.results['rs'], 205)
def test_getCPUCount_with_STANDBY(self, fake_smcli):
fake_smcli.return_value = {
'overallRC':
0,
'rc':
0,
'rs':
0,
'errno':
0,
'response':
'Partition mode: Z/VM\n\n'
'ADDRESS STATUS TYPE CORE_ID\n'
'0000 MASTER-PROCESSOR CP 0000\n'
'0002 ALTERNATE IFL 0001\n'
'0003 PARKED IFL 0001\n'
'0004 STANDBY IFL 0002\n',
'strError':
''
}
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
ret_total, ret_used = getHost.getCPUCount(rh)
print("return value2:", ret_total, ret_used)
self.assertEqual(4, ret_total)
self.assertEqual(3, ret_used)
def test_getReservedMemSize_max_less_than_initial(self):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
gap = makeVM.getReservedMemSize(rh, '64G', '32G')
self.assertEqual(gap, '0M')
self.assertEqual(rh.results['overallRC'], 4)
self.assertEqual(rh.results['rc'], 4)
self.assertEqual(rh.results['rs'], 206)
def test_getVM_directory_py3(self):
rh = ReqHandle.ReqHandle(captureLogs=False)
with mock.patch('subprocess.check_output') as exec_cmd:
# subprocess.check_output returns bytes in py3
exec_cmd.return_value = (
b"0 0 0 (details) None\n"
b"USER T9572493 LBYONLY 2048m 64G G\nINCLUDE ZCCDFLT\n"
b"COMMAND DEF STOR RESERVED 63488M\n"
b"CPU 00 BASE\nIPL 0100\nLOGONBY MAINT\nMACHINE ESA 32\n"
b"MDISK 0100 3390 48697 5500 OMB1B6 MR\n"
b"*DVHOPT LNK0 LOG1 RCM1 SMS0 NPW1 LNGAMENG PWC20180808 "
b"CRC\xf3:\n")
expected_resp = (
u"USER T9572493 LBYONLY 2048m 64G G\nINCLUDE ZCCDFLT\n"
u"COMMAND DEF STOR RESERVED 63488M\nCPU 00 BASE\nIPL 0100\n"
u"LOGONBY MAINT\nMACHINE ESA 32\n"
u"MDISK 0100 3390 48697 5500 OMB1B6 MR\n"
u"*DVHOPT LNK0 LOG1 RCM1 SMS0 NPW1 LNGAMENG PWC20180808 "
u"CRC\ufffd:\n")
res = vmUtils.invokeSMCLI(rh, "Image_Query_DM", ['-T', 'fakeuid'])
self.assertEqual(res['response'], expected_resp)
exec_cmd.assert_called_once_with([
'sudo', '/opt/zthin/bin/smcli', 'Image_Query_DM',
'--addRCheader', '-T', 'fakeuid'
],
close_fds=True)
def test_create_VM_swap_4096M(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1G',
'maxMemSize': '1G',
'privClasses': 'G',
'vdisk': '0102:4096M'
}
rh.parms = parms
rs = makeVM.createVM(rh)
self.assertEqual(4, rs)
def test_getCPUCount_with_overallRC_error(self, fake_smcli):
fake_smcli.return_value = {
'overallRC': 24,
'rc': 0,
'rs': 0,
'errno': 0,
'response': 'SMAPI API failed\n',
'strError': 'Input error'
}
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
ret_total, ret_used = getHost.getCPUCount(rh)
print("return value5:", ret_total, ret_used)
self.assertEqual(0, ret_total)
self.assertEqual(0, ret_used)
def test_create_VM_swap_256M(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1G',
'maxMemSize': '1G',
'privClasses': 'G',
'vdisk': '0102:256M'
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1G 1G G\nCPU 00 BASE\n'
b'MDISK 0102 FB-512 V-DISK 524288 MWV\n')
def test_create_VM_STOR_RESERVED_0M(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1G',
'maxMemSize': '1G',
'privClasses': 'G',
'setReservedMem': ''
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1G 1G G\nCPU 00 BASE\n'
b'COMMAND DEF STOR RESERVED 0M\n')
def test_getCPUCount_no_CPU(self, fake_smcli):
fake_smcli.return_value = {
'overallRC': 0,
'rc': 0,
'rs': 0,
'errno': 0,
'response': 'Partition mode: Z/VM\n\n'
'ADDRESS STATUS TYPE CORE_ID\n',
'strError': ''
}
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
ret_total, ret_used = getHost.getCPUCount(rh)
print("return value3:", ret_total, ret_used)
self.assertEqual(0, ret_total)
self.assertEqual(0, ret_used)
def test_create_with_share(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1024M',
'maxMemSize': '1G',
'privClasses': 'G',
'commandSetShare': 'RELATIVE 125'
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1024M 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'COMMAND SET SHARE &USERID RELATIVE 125\n')
def test_create_VM_swap_2048M(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1G',
'maxMemSize': '1G',
'privClasses': 'G',
'vdisk': '0102:2048M'
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1G 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'MDISK 0102 FB-512 V-DISK 4194296 MWV\n')
def test_create_VM_account(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1G',
'maxMemSize': '1G',
'privClasses': 'G',
'account': "acc acc1 acc2 dummy"
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1G 1G G\n'
b'ACCOUNT ACC ACC1 ACC2 DUMMY\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n')
def test_create_VM_profile(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1G',
'maxMemSize': '1G',
'privClasses': 'G',
'profName': "Profile1"
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1G 1G G\n'
b'INCLUDE PROFILE1\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n')
def test_create_with_logonby(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1024M',
'maxMemSize': '1G',
'privClasses': 'G',
'byUsers': ['USER1', 'USER2']
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1024M 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'LOGONBY USER1 USER2\n')
def test_create_with_pcif(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1024M',
'maxMemSize': '1G',
'privClasses': 'G',
'commandPcif': '100:200'
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1024M 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'COMMAND ATTACH PCIF 100 * AS 200\n')
def test_create_with_rdomain(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1024M',
'maxMemSize': '1G',
'privClasses': 'G',
'commandRDomain': 'Z15ONLY'
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1024M 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'COMMAND SET VMRELOCATE * DOMAIN Z15ONLY\n')
def test_create_with_cpupool(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1024M',
'maxMemSize': '1G',
'privClasses': 'G',
'commandSchedule': 'POOL1'
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1024M 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'COMMAND SCHEDULE * WITHIN POOL POOL1\n')
def test_create_VM_STOR_RESERVED_0M_diff_unit(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1024M',
'maxMemSize': '1G',
'privClasses': 'G',
'setReservedMem': ''
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1024M 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'COMMAND DEF STOR RESERVED 0M\n')
def test_create_with_profile(self, write):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
parms = {
'pw': 'pwd',
'priMemSize': '1024M',
'maxMemSize': '1G',
'privClasses': 'G',
'comment': 'comment1$@$@$this is comment2$@$@$'
}
rh.parms = parms
makeVM.createVM(rh)
write.assert_called_with(
mock.ANY, b'USER pwd 1024M 1G G\n'
b'COMMAND SET VCONFIG MODE LINUX\n'
b'COMMAND DEFINE CPU 00 TYPE IFL\n'
b'* COMMENT1\n'
b'* THIS IS COMMENT2\n')
def test_getReservedMemSize_gap_G(self):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
gap = makeVM.getReservedMemSize(rh, '512m', '9999G')
self.assertEqual(gap, '9998G')
self.assertEqual(rh.results['overallRC'], 0)
def test_getReservedMemSize_equal_size(self):
rh = ReqHandle.ReqHandle(captureLogs=False, smt=mock.Mock())
gap = makeVM.getReservedMemSize(rh, '1024M', '1G')
self.assertEqual(gap, '0M')
self.assertEqual(rh.results['overallRC'], 0)
