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)