Unit test for websocket comm service

platform/view/core/endpoint/endpoint.go

@@ -9,7 +9,6 @@ package endpoint
 import (
 	"bytes"
 	"net"
-	"reflect"
 	"strings"
 	"sync"
 
@@ -19,7 +18,6 @@ import (
 	"github.com/hyperledger-labs/fabric-smart-client/platform/view/driver"
 	"github.com/hyperledger-labs/fabric-smart-client/platform/view/view"
 	"github.com/pkg/errors"
-	"go.uber.org/zap/zapcore"
 	"golang.org/x/exp/slices"
 )
 
@@ -69,17 +67,14 @@ type Service struct {
 	resolversMutex sync.RWMutex
 	bindingKVS     driver2.BindingStore
 
-	pkiExtractorsLock      sync.RWMutex
-	publicKeyExtractors    []driver.PublicKeyExtractor
-	publicKeyIDSynthesizer driver.PublicKeyIDSynthesizer
+	pkiExtractor *PKIExtractor
 }
 
 // NewService returns a new instance of the view-sdk endpoint service
 func NewService(bindingKVS driver2.BindingStore) (*Service, error) {
 	er := &Service{
-		bindingKVS:             bindingKVS,
-		publicKeyExtractors:    []driver.PublicKeyExtractor{},
-		publicKeyIDSynthesizer: DefaultPublicKeyIDSynthesizer{},
+		bindingKVS:   bindingKVS,
+		pkiExtractor: NewPKIExtractor(),
 	}
 	return er, nil
 }
@@ -89,7 +84,7 @@ func (r *Service) Resolve(party view.Identity) (driver.Resolver, []byte, error)
 	if err != nil {
 		return nil, nil, err
 	}
-	return resolver, r.pkiResolve(resolver), nil
+	return resolver, r.pkiExtractor.PkiResolve(resolver), nil
 }
 
 func (r *Service) GetResolver(party view.Identity) (driver.Resolver, error) {
@@ -161,7 +156,7 @@ func (r *Service) matchesResolver(endpoint string, pkID []byte, resolver *Resolv
 	}
 
 	return len(pkID) > 0 && (bytes.Equal(pkID, resolver.Id) ||
-		bytes.Equal(pkID, r.pkiResolve(resolver)))
+		bytes.Equal(pkID, r.pkiExtractor.PkiResolve(resolver)))
 }
 
 func (r *Service) AddResolver(name string, domain string, addresses map[string]string, aliases []string, id []byte) (view.Identity, error) {
@@ -203,52 +198,15 @@ func (r *Service) AddResolver(name string, domain string, addresses map[string]s
 }
 
 func (r *Service) AddPublicKeyExtractor(publicKeyExtractor driver.PublicKeyExtractor) error {
-	r.pkiExtractorsLock.Lock()
-	defer r.pkiExtractorsLock.Unlock()
-
-	if publicKeyExtractor == nil {
-		return errors.New("pki resolver should not be nil")
-	}
-	r.publicKeyExtractors = append(r.publicKeyExtractors, publicKeyExtractor)
-	return nil
+	return r.pkiExtractor.AddPublicKeyExtractor(publicKeyExtractor)
 }
 
 func (r *Service) SetPublicKeyIDSynthesizer(publicKeyIDSynthesizer driver.PublicKeyIDSynthesizer) {
-	r.publicKeyIDSynthesizer = publicKeyIDSynthesizer
-}
-
-func (r *Service) pkiResolve(resolver *Resolver) []byte {
-	resolver.PKILock.RLock()
-	if len(resolver.PKI) != 0 {
-		resolver.PKILock.RUnlock()
-		return resolver.PKI
-	}
-	resolver.PKILock.RUnlock()
-
-	resolver.PKILock.Lock()
-	defer resolver.PKILock.Unlock()
-	if len(resolver.PKI) == 0 {
-		resolver.PKI = r.ExtractPKI(resolver.Id)
-	}
-	return resolver.PKI
+	r.pkiExtractor.SetPublicKeyIDSynthesizer(publicKeyIDSynthesizer)
 }
 
 func (r *Service) ExtractPKI(id []byte) []byte {
-	r.pkiExtractorsLock.RLock()
-	defer r.pkiExtractorsLock.RUnlock()
-
-	for _, extractor := range r.publicKeyExtractors {
-		if pk, err := extractor.ExtractPublicKey(id); pk != nil {
-			logger.Debugf("pki resolved for [%s]", id)
-			return r.publicKeyIDSynthesizer.PublicKeyID(pk)
-		} else {
-			if logger.IsEnabledFor(zapcore.DebugLevel) {
-				logger.Debugf("pki not resolved by [%s] for [%s]: [%s]", getIdentifier(extractor), id, err)
-			}
-		}
-	}
-	logger.Warnf("cannot resolve pki for [%s]", id)
-	return nil
+	return r.pkiExtractor.ExtractPKI(id)
 }
 
 func (r *Service) rootEndpoint(party view.Identity) (*Resolver, error) {
@@ -294,14 +252,3 @@ func LookupIPv4(endpoint string) string {
 	port := s[1]
 	return net.JoinHostPort(addrS, port)
 }
-
-func getIdentifier(f any) string {
-	if f == nil {
-		return "<nil view>"
-	}
-	t := reflect.TypeOf(f)
-	for t.Kind() == reflect.Ptr {
-		t = t.Elem()
-	}
-	return t.PkgPath() + "/" + t.Name()
-}

platform/view/core/endpoint/endpoint_test.go

@@ -41,7 +41,7 @@ func TestPKIResolveConcurrency(t *testing.T) {
 	for i := 0; i < 100; i++ {
 		go func() {
 			defer wg.Done()
-			svc.pkiResolve(resolver)
+			svc.pkiExtractor.PkiResolve(resolver)
 		}()
 	}
 	wg.Wait()

platform/view/core/endpoint/extractor.go

@@ -0,0 +1,91 @@
+package endpoint
+
+/*
+Copyright IBM Corp. All Rights Reserved.
+
+SPDX-License-Identifier: Apache-2.0
+*/
+
+import (
+	"reflect"
+	"sync"
+
+	"github.com/hyperledger-labs/fabric-smart-client/platform/view/driver"
+	"github.com/pkg/errors"
+	"go.uber.org/zap/zapcore"
+)
+
+func NewPKIExtractor() *PKIExtractor {
+	return &PKIExtractor{
+		publicKeyExtractors:    []driver.PublicKeyExtractor{},
+		publicKeyIDSynthesizer: DefaultPublicKeyIDSynthesizer{},
+	}
+}
+
+type PKIExtractor struct {
+	pkiExtractorsLock      sync.RWMutex
+	publicKeyExtractors    []driver.PublicKeyExtractor
+	publicKeyIDSynthesizer driver.PublicKeyIDSynthesizer
+}
+
+func (r *PKIExtractor) AddPublicKeyExtractor(publicKeyExtractor driver.PublicKeyExtractor) error {
+	r.pkiExtractorsLock.Lock()
+	defer r.pkiExtractorsLock.Unlock()
+
+	if publicKeyExtractor == nil {
+		return errors.New("pki resolver should not be nil")
+	}
+	r.publicKeyExtractors = append(r.publicKeyExtractors, publicKeyExtractor)
+	return nil
+}
+
+func (r *PKIExtractor) SetPublicKeyIDSynthesizer(publicKeyIDSynthesizer driver.PublicKeyIDSynthesizer) {
+	r.publicKeyIDSynthesizer = publicKeyIDSynthesizer
+}
+
+func (r *PKIExtractor) PkiResolve(resolver *Resolver) []byte {
+	resolver.PKILock.RLock()
+	if len(resolver.PKI) != 0 {
+		resolver.PKILock.RUnlock()
+		return resolver.PKI
+	}
+	resolver.PKILock.RUnlock()
+
+	resolver.PKILock.Lock()
+	defer resolver.PKILock.Unlock()
+	if len(resolver.PKI) == 0 {
+		resolver.PKI = r.ExtractPKI(resolver.Id)
+	}
+	return resolver.PKI
+}
+
+func (r *PKIExtractor) ExtractPKI(id []byte) []byte {
+	r.pkiExtractorsLock.RLock()
+	defer r.pkiExtractorsLock.RUnlock()
+
+	for _, extractor := range r.publicKeyExtractors {
+		if pk, err := extractor.ExtractPublicKey(id); pk != nil {
+			if logger.IsEnabledFor(zapcore.DebugLevel) {
+				logger.Debugf("pki resolved for [%s]", id)
+			}
+			return r.publicKeyIDSynthesizer.PublicKeyID(pk)
+		} else {
+			if logger.IsEnabledFor(zapcore.DebugLevel) {
+				logger.Debugf("pki not resolved by [%s] for [%s]: [%s]", getIdentifier(extractor), id, err)
+			}
+		}
+	}
+	logger.Warnf("cannot resolve pki for [%s]", id)
+	return nil
+}
+
+func getIdentifier(f any) string {
+	if f == nil {
+		return "<nil view>"
+	}
+	t := reflect.TypeOf(f)
+	for t.Kind() == reflect.Ptr {
+		t = t.Elem()
+	}
+	return t.PkgPath() + "/" + t.Name()
+}

platform/view/sdk/dig/sdk.go

@@ -26,7 +26,6 @@ import (
 	tracing2 "github.com/hyperledger-labs/fabric-smart-client/platform/view/sdk/tracing"
 	"github.com/hyperledger-labs/fabric-smart-client/platform/view/sdk/web"
 	"github.com/hyperledger-labs/fabric-smart-client/platform/view/services/comm"
-	"github.com/hyperledger-labs/fabric-smart-client/platform/view/services/comm/host"
 	"github.com/hyperledger-labs/fabric-smart-client/platform/view/services/comm/provider"
 	"github.com/hyperledger-labs/fabric-smart-client/platform/view/services/crypto"
 	dbdriver "github.com/hyperledger-labs/fabric-smart-client/platform/view/services/db/driver"
@@ -135,9 +134,7 @@ func (p *SDK) Install() error {
 		p.Container().Provide(manager.New, dig.As(new(ViewManager), new(node.ViewManager), new(driver.ViewManager), new(driver.Registry))),
 		p.Container().Provide(view.NewManager),
 
-		p.Container().Provide(func(hostProvider host.GeneratorProvider, configProvider driver.ConfigService, endpointService *view.EndpointService, tracerProvider trace.TracerProvider, metricsProvider metrics2.Provider) (*comm.Service, error) {
-			return comm.NewService(hostProvider, endpointService, configProvider, tracerProvider, metricsProvider)
-		}),
+		p.Container().Provide(comm.NewService),
 		p.Container().Provide(digutils.Identity[*comm.Service](), dig.As(new(manager.CommLayer))),
 		p.Container().Provide(provider.NewHostProvider),
 		p.Container().Provide(view.NewSigService),

platform/view/services/comm/comm.go

@@ -30,21 +30,17 @@ type ConfigService interface {
 }
 
 type Service struct {
-	HostProvider    host.GeneratorProvider
-	EndpointService EndpointService
-	ConfigService   ConfigService
+	HostProvider host.GeneratorProvider
 
 	Node            *P2PNode
 	NodeSync        sync.RWMutex
 	tracerProvider  trace.TracerProvider
 	metricsProvider metrics.Provider
 }
 
-func NewService(hostProvider host.GeneratorProvider, endpointService EndpointService, configService ConfigService, tracerProvider trace.TracerProvider, metricsProvider metrics.Provider) (*Service, error) {
+func NewService(hostProvider host.GeneratorProvider, tracerProvider trace.TracerProvider, metricsProvider metrics.Provider) (*Service, error) {
 	s := &Service{
 		HostProvider:    hostProvider,
-		EndpointService: endpointService,
-		ConfigService:   configService,
 		tracerProvider:  tracerProvider,
 		metricsProvider: metricsProvider,
 	}