def test_ssh_vm(self, mock_ssh_utils, mock_do_op):
cmd = mock.Mock()
custom.ssh_vm(cmd, "rg", "vm", "ip", "public", "private", False)
mock_do_op.assert_called_once_with(cmd, "rg", "vm", "ip", "public",
"private", False,
mock_ssh_utils.start_ssh_connection)
def test_ssh_vm_debug(self, mock_info, mock_assert, mock_do_op):
cmd = mock.Mock()
cmd.cli_ctx.data = {'safe_params': ['--debug']}
custom.ssh_vm(cmd, "rg", "vm", "ip", "public", "private", False,
"username", "cert", "port", "ssh_folder", [])
mock_info.assert_called_once_with("rg", "vm", "ip", "public",
"private", False, "username", "cert",
"port", "ssh_folder", ['-v'])
mock_assert.assert_called_once_with("rg", "vm", "ip", "cert",
"username")
mock_do_op.assert_called_once_with(cmd, mock.ANY, None,
ssh_utils.start_ssh_connection)
def test_ssh_vm_debug(self, mock_type, mock_info, mock_assert, mock_do_op):
cmd = mock.Mock()
ssh_info = mock.Mock()
mock_info.return_value = ssh_info
cmd.cli_ctx.data = {'safe_params': ['--debug']}
custom.ssh_vm(cmd, "rg", "vm", "ip", "public", "private", False,
"username", "cert", "port", "ssh_folder", False, "type",
"proxy", False, [])
mock_info.assert_called_once_with("rg", "vm", "ip", "public",
"private", False, "username", "cert",
"port", "ssh_folder", ['-vvv'],
False, "type", "proxy", None, False)
mock_assert.assert_called_once_with("rg", "vm", "ip", "type", "cert",
"username")
mock_type.assert_called_once_with(cmd, ssh_info)
mock_do_op.assert_called_once_with(cmd, ssh_info,
ssh_utils.start_ssh_connection)
def test_ssh_vm_delete_credentials_cloudshell(self, mock_info, mock_assert,
mock_getenv, mock_type,
mock_op):
mock_getenv.return_value = "cloud-shell/1.0"
cmd = mock.Mock()
ssh_info = mock.Mock()
cmd.cli_ctx.data = {'safe_params': []}
mock_info.return_value = ssh_info
custom.ssh_vm(cmd, "rg", "vm", "ip", "public", "private", False,
"username", "cert", "port", "ssh_folder", True, "type",
"proxy", False, [])
mock_info.assert_called_once_with("rg", "vm", "ip", "public",
"private", False, "username", "cert",
"port", "ssh_folder", [], True,
"type", "proxy", None, False)
mock_assert.assert_called_once_with("rg", "vm", "ip", "type", "cert",
"username")
mock_type.assert_called_once_with(cmd, ssh_info)
mock_op.assert_called_once_with(cmd, ssh_info,
ssh_utils.start_ssh_connection)
