def test__exec_ipmitool(self): pw_file = '/tmp/password_file' file_handle = open(pw_file, "w") self.mox.StubOutWithMock(ipmi, '_make_password_file') self.mox.StubOutWithMock(utils, 'execute') args = [ 'ipmitool', '-I', 'lanplus', '-H', self.info['address'], '-U', self.info['username'], '-f', file_handle, 'A', 'B', 'C', ] ipmi._make_password_file(self.info['password']).AndReturn(file_handle) utils.execute(*args, attempts=3).AndReturn((None, None)) self.mox.ReplayAll() ipmi._exec_ipmitool(self.info, 'A B C') self.mox.VerifyAll()
def test__make_password_file(self): with ipmi._make_password_file(self.info.get('password')) as pw_file: del_chk_pw_file = pw_file self.assertTrue(os.path.isfile(pw_file)) self.assertEqual(os.stat(pw_file)[stat.ST_MODE] & 0o777, 0o600) with open(pw_file, "r") as f: password = f.read() self.assertEqual(password, self.info.get('password')) self.assertFalse(os.path.isfile(del_chk_pw_file))