Function TestComponentCalls

pkg/components/state/pluggable_test.go:164–784

(t *testing.T)

}

func TestComponentCalls(t *testing.T) {
	getStateStore := func(srv *server) (statestore *grpcStateStore, cleanupf func(), err error) {
		withSvc := testingGrpc.TestServerWithDialer(testLogger, func(s *grpc.Server, svc *server) {
			proto.RegisterStateStoreServer(s, svc)
			proto.RegisterTransactionalStateStoreServer(s, svc)
			proto.RegisterQueriableStateStoreServer(s, svc)
		})
		dialer, cleanup, err := withSvc(srv)
		require.NoError(t, err)
		clientFactory := newGRPCStateStore(func(ctx context.Context, name string, opts ...grpc.DialOption) (*grpc.ClientConn, error) {
			return dialer(ctx, opts...)
		})
		client := clientFactory(testLogger).(*grpcStateStore)
		require.NoError(t, client.Init(t.Context(), state.Metadata{}))
		return client, cleanup, err
	}

	if runtime.GOOS != "windows" {
		t.Run("test init should populate features and call grpc init", func(t *testing.T) {
			const (
				fakeName          = "name"
				fakeType          = "type"
				fakeVersion       = "v1"
				fakeComponentName = "component"
				fakeSocketFolder  = "/tmp"
			)

			uniqueID := guuid.New().String()
			socket := fmt.Sprintf("%s/%s.sock", fakeSocketFolder, uniqueID)
			defer os.Remove(socket)

			connector := pluggable.NewGRPCConnector(socket, newStateStoreClient)
			defer connector.Close()

			listener, err := net.Listen("unix", socket)
			require.NoError(t, err)
			defer listener.Close()
			s := grpc.NewServer()
			srv := &server{}
			proto.RegisterStateStoreServer(s, srv)
			go func() {
				if serveErr := s.Serve(listener); serveErr != nil {
					testLogger.Debugf("Server exited with error: %v", serveErr)
				}
			}()

			ps := fromConnector(testLogger, connector)
			err = ps.Init(t.Context(), state.Metadata{
				Base: contribMetadata.Base{},
			})

			require.NoError(t, err)
			assert.Equal(t, int64(1), srv.featuresCalled.Load())
			assert.Equal(t, int64(1), srv.initCalled.Load())
		})
	} else {
		t.Logf("skipping pubsub pluggable component init test due to the lack of OS (%s) support", runtime.GOOS)
	}