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  }