mirror of https://github.com/hashicorp/consul
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package testing
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
|
|
"github.com/hashicorp/consul/acl/resolver"
|
|
svc "github.com/hashicorp/consul/agent/grpc-external/services/resource"
|
|
internal "github.com/hashicorp/consul/agent/grpc-internal"
|
|
"github.com/hashicorp/consul/internal/resource"
|
|
"github.com/hashicorp/consul/internal/storage/inmem"
|
|
"github.com/hashicorp/consul/proto-public/pbresource"
|
|
"github.com/hashicorp/consul/sdk/testutil"
|
|
)
|
|
|
|
// RunResourceService runs a Resource Service for the duration of the test and
|
|
// returns a client to interact with it. ACLs will be disabled.
|
|
func RunResourceService(t *testing.T, registerFns ...func(resource.Registry)) pbresource.ResourceServiceClient {
|
|
t.Helper()
|
|
|
|
backend, err := inmem.NewBackend()
|
|
require.NoError(t, err)
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
t.Cleanup(cancel)
|
|
go backend.Run(ctx)
|
|
|
|
registry := resource.NewRegistry()
|
|
for _, fn := range registerFns {
|
|
fn(registry)
|
|
}
|
|
|
|
server := grpc.NewServer()
|
|
|
|
svc.NewServer(svc.Config{
|
|
Backend: backend,
|
|
Registry: registry,
|
|
Logger: testutil.Logger(t),
|
|
ACLResolver: resolver.DANGER_NO_AUTH{},
|
|
}).Register(server)
|
|
|
|
pipe := internal.NewPipeListener()
|
|
go server.Serve(pipe)
|
|
t.Cleanup(server.Stop)
|
|
|
|
conn, err := grpc.Dial("",
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
grpc.WithContextDialer(pipe.DialContext),
|
|
grpc.WithBlock(),
|
|
)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = conn.Close() })
|
|
|
|
return pbresource.NewResourceServiceClient(conn)
|
|
}
|