github.com/TBD54566975/ftl@v0.219.0/internal/rpc/context_test.go (about) 1 package rpc 2 3 import ( 4 "context" 5 "net/http" 6 "testing" 7 8 "github.com/TBD54566975/ftl/backend/protos/xyz/block/ftl/v1/ftlv1connect" 9 "github.com/alecthomas/assert/v2" 10 ) 11 12 func TestRPCContext(t *testing.T) { 13 ctx := context.Background() 14 verbClient := ftlv1connect.NewVerbServiceClient(http.DefaultClient, "http://localhost:8080") 15 ctx = ContextWithClient(ctx, verbClient) 16 controllerClient := ftlv1connect.NewControllerServiceClient(http.DefaultClient, "http://localhost:8080") 17 ctx = ContextWithClient(ctx, controllerClient) 18 19 assert.Equal(t, verbClient, ClientFromContext[ftlv1connect.VerbServiceClient](ctx)) 20 assert.Equal(t, controllerClient, ClientFromContext[ftlv1connect.ControllerServiceClient](ctx)) 21 }