Function TestClient_ForwardsCapiOptionsToSessionRequests

go/client_test.go:214–251

(t *testing.T)

}

func TestClient_ForwardsCapiOptionsToSessionRequests(t *testing.T) {
	rpcClient, server, _ := newRuntimeShutdownRpcPair(t)
	t.Cleanup(server.Stop)
	client := &Client{
		client:   rpcClient,
		RPC:      rpc.NewServerRPC(rpcClient),
		sessions: make(map[string]*Session),
	}

	createParams := make(chan json.RawMessage, 1)
	server.SetRequestHandler("session.create", func(params json.RawMessage) (json.RawMessage, *jsonrpc2.Error) {
		createParams <- append(json.RawMessage(nil), params...)
		sessionID := sessionIDFromParams(t, params)
		return []byte(`{"sessionId":"` + sessionID + `","workspacePath":"/workspace"}`), nil
	})

	_, err := client.CreateSession(t.Context(), &SessionConfig{
		Capi: &CapiSessionOptions{EnableWebSocketResponses: Bool(false)},
	})
	if err != nil {
		t.Fatalf("CreateSession failed: %v", err)
	}
	assertCapiEnableWebSocketResponses(t, <-createParams)

	resumeParams := make(chan json.RawMessage, 1)
	server.SetRequestHandler("session.resume", func(params json.RawMessage) (json.RawMessage, *jsonrpc2.Error) {
		resumeParams <- append(json.RawMessage(nil), params...)
		return []byte(`{"sessionId":"resumed-capi","workspacePath":"/workspace"}`), nil
	})

	_, err = client.ResumeSessionWithOptions(t.Context(), "resumed-capi", &ResumeSessionConfig{
		Capi: &CapiSessionOptions{EnableWebSocketResponses: Bool(false)},
	})
	if err != nil {
		t.Fatalf("ResumeSessionWithOptions failed: %v", err)
	}
	assertCapiEnableWebSocketResponses(t, <-resumeParams)
}

func TestClient_ForwardsNewSessionOptionsToSessionRequests(t *testing.T) {
	rpcClient, server, _ := newRuntimeShutdownRpcPair(t)