Function TestStartSSHServerSuccess

internal/codespaces/rpc/invoker_test.go:249–280

Test that the RPC invoker returns the correct port and user when the SSH server starts successfully

(t *testing.T)

// Test that the RPC invoker returns the correct port and user when the SSH server starts successfully
func TestStartSSHServerSuccess(t *testing.T) {
	resp := ssh.StartRemoteServerResponse{
		ServerPort: strconv.Itoa(1234),
		User:       "test",
		Message:    "",
		Result:     true,
	}

	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()

	port, user, err := invoker.StartSSHServer(context.Background())
	if err != nil {
		t.Fatalf("expected %v, got %v", nil, err)
	}
	if strconv.Itoa(port) != resp.ServerPort {
		t.Fatalf("expected %s, got %d", resp.ServerPort, port)
	}
	if user != resp.User {
		t.Fatalf("expected %s, got %s", resp.User, user)
	}

	verifyNotifyCodespaceOfClientActivity(t, server)
}

// Test that the RPC invoker returns an error when the SSH server fails to start
func TestStartSSHServerFailure(t *testing.T) {