Add telemetry manager tests

smartcontractkit · Sep 17, 2023 · 232197d · 232197d
1 parent 00cc2b9
commit 232197d
Show file tree

Hide file tree

Showing 5 changed files with 550 additions and 4 deletions.
diff --git a/core/config/mocks/telemetry_ingress.go b/core/config/mocks/telemetry_ingress.go
diff --git a/core/config/mocks/telemetry_ingress_endpoint.go b/core/config/mocks/telemetry_ingress_endpoint.go
diff --git a/core/config/telemetry_ingress_config.go b/core/config/telemetry_ingress_config.go
@@ -5,6 +5,8 @@ import (
 	"time"
 )
 
+//go:generate mockery --quiet --name TelemetryIngress --output ./mocks/ --case=underscore --filename telemetry_ingress.go
+
 type TelemetryIngress interface {
 	Logging() bool
 	UniConn() bool
@@ -16,6 +18,7 @@ type TelemetryIngress interface {
 	Endpoints() []TelemetryIngressEndpoint
 }
 
+//go:generate mockery --quiet --name TelemetryIngressEndpoint --output ./mocks/ --case=underscore --filename telemetry_ingress_endpoint.go
 type TelemetryIngressEndpoint interface {
 	Network() string
 	ChainID() string

diff --git a/core/services/telemetry/manager.go b/core/services/telemetry/manager.go
@@ -45,14 +45,14 @@ type telemetryEndpoint struct {
 	Network string
 	URL     *url.URL
 	client  TelemetryClient
-	pubKey  string
+	PubKey  string
 }
 
-func NewManager(cfg config.TelemetryIngress, ks keystore.CSA, lggr logger.Logger) *Manager {
+func NewManager(cfg config.TelemetryIngress, csaKeyStore keystore.CSA, lggr logger.Logger) *Manager {
 	m := &Manager{
 		bufferSize:   cfg.BufferSize(),
 		endpoints:    nil,
-		ks:           ks,
+		ks:           csaKeyStore,
 		lggr:         lggr.Named("TelemetryManager"),
 		logging:      cfg.Logging(),
 		maxBatchSize: cfg.MaxBatchSize(),
@@ -152,7 +152,7 @@ func (m *Manager) addEndpoint(e config.TelemetryIngressEndpoint) error {
 		Network: strings.ToUpper(e.Network()),
 		ChainID: strings.ToUpper(e.ChainID()),
 		URL:     e.URL(),
-		pubKey:  e.ServerPubKey(),
+		PubKey:  e.ServerPubKey(),
 		client:  tClient,
 	}