BeaconNodes -> BeaconNodeSet

prysmaticlabs · Mar 25, 2021 · ec3fa52 · ec3fa52
1 parent 497fa52
commit ec3fa52
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 13 deletions.
diff --git a/endtoend/components/beacon_node.go b/endtoend/components/beacon_node.go
@@ -19,31 +19,31 @@ import (
 )
 
 var _ e2etypes.ComponentRunner = (*BeaconNode)(nil)
-var _ e2etypes.ComponentRunner = (*BeaconNodes)(nil)
+var _ e2etypes.ComponentRunner = (*BeaconNodeSet)(nil)
 
-// BeaconNodes represents set of beacon nodes.
-type BeaconNodes struct {
+// BeaconNodeSet represents set of beacon nodes.
+type BeaconNodeSet struct {
 	e2etypes.ComponentRunner
 	config  *e2etypes.E2EConfig
 	enr     string
 	started chan struct{}
 }
 
 // SetENR assigns ENR to the set of beacon nodes.
-func (s *BeaconNodes) SetENR(enr string) {
+func (s *BeaconNodeSet) SetENR(enr string) {
 	s.enr = enr
 }
 
 // NewBeaconNodes creates and returns a set of beacon nodes.
-func NewBeaconNodes(config *e2etypes.E2EConfig) *BeaconNodes {
-	return &BeaconNodes{
+func NewBeaconNodes(config *e2etypes.E2EConfig) *BeaconNodeSet {
+	return &BeaconNodeSet{
 		config:  config,
 		started: make(chan struct{}, 1),
 	}
 }
 
 // Start starts all the beacon nodes in set.
-func (s *BeaconNodes) Start(ctx context.Context) error {
+func (s *BeaconNodeSet) Start(ctx context.Context) error {
 	if s.enr == "" {
 		return errors.New("empty ENR")
 	}
@@ -82,7 +82,7 @@ func (s *BeaconNodes) Start(ctx context.Context) error {
 }
 
 // Started checks whether beacon node set is started and all nodes are ready to be queried.
-func (s *BeaconNodes) Started() <-chan struct{} {
+func (s *BeaconNodeSet) Started() <-chan struct{} {
 	return s.started
 }
 

diff --git a/endtoend/components/validator.go b/endtoend/components/validator.go
@@ -38,11 +38,6 @@ type ValidatorNodes struct {
 	started chan struct{}
 }
 
-// SetENR assigns ENR to the set of beacon nodes.
-func (s *BeaconNodes) SetParams(enr string) {
-	s.enr = enr
-}
-
 // NewValidatorNodes creates and returns a set of validator nodes.
 func NewValidatorNodes(config *e2etypes.E2EConfig) *ValidatorNodes {
 	return &ValidatorNodes{