Function TestStartSSHServerFailure

internal/codespaces/rpc/invoker_test.go:283–313

Test that the RPC invoker returns an error when the SSH server fails to start

(t *testing.T)

// Test that the RPC invoker returns an error when the SSH server fails to start
func TestStartSSHServerFailure(t *testing.T) {
	resp := ssh.StartRemoteServerResponse{
		ServerPort: strconv.Itoa(1234),
		User:       "test",
		Message:    "error message",
		Result:     false,
	}

	server := newMockServer()
	server.StartRemoteServerAsyncFunc = func(context.Context, *ssh.StartRemoteServerRequest) (*ssh.StartRemoteServerResponse, error) {
		return &resp, nil
	}

	invoker, stop, err := createTestInvoker(t, server)
	if err != nil {
		t.Fatalf("error connecting to internal server: %v", err)
	}
	defer stop()

	errorMessage := fmt.Sprintf("failed to start SSH server: %s", resp.Message)
	port, user, err := invoker.StartSSHServer(context.Background())
	if err.Error() != errorMessage {
		t.Fatalf("expected %v, got %v", errorMessage, err)
	}
	if port != 0 {
		t.Fatalf("expected %d, got %d", 0, port)
	}
	if user != "" {
		t.Fatalf("expected %s, got %s", "", user)
	}
}