def test_get_mgmt_ip(self, mock_ssh):
with mock.patch("yardstick.ssh.SSH") as ssh:
ssh_mock = mock.Mock(autospec=ssh.SSH)
ssh_mock.execute = \
mock.Mock(return_value=(1, "1.2.3.4 00:00:00:00:00:01", ""))
ssh.return_value = ssh_mock
status = StandaloneContextHelper.get_mgmt_ip(ssh_mock,
"00:00:00:00:00:01",
"1.1.1.1/24", {})
self.assertIsNotNone(status)
def test_get_mgmt_ip_no(self, mock_ssh):
with mock.patch("yardstick.ssh.SSH") as ssh:
ssh_mock = mock.Mock(autospec=ssh.SSH)
ssh_mock.execute = \
mock.Mock(return_value=(1, "", ""))
ssh.return_value = ssh_mock
model.WAIT_FOR_BOOT = 0
status = StandaloneContextHelper.get_mgmt_ip(ssh_mock, "99",
"1.1.1.1/24", {})
self.assertIsNone(status)