nodepeers testclient

attestantio · Oct 21, 2023 · c234237 · c234237
1 parent 65f6d3b
commit c234237
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/testclients/erroring.go b/testclients/erroring.go
@@ -560,6 +560,19 @@ func (s *Erroring) NodeSyncing(ctx context.Context) (*api.Response[*apiv1.SyncSt
 	return next.NodeSyncing(ctx)
 }
 
+// NodePeers provides the peers of the node
+func (s *Erroring) NodePeers(ctx context.Context, opts *api.PeerOpts) (*api.Response[*apiv1.Peers], error) {
+	if err := s.maybeError(ctx); err != nil {
+		return nil, err
+	}
+	next, isNext := s.next.(consensusclient.NodePeersProvider)
+	if !isNext {
+		return nil, fmt.Errorf("%s@%s does not support this call", s.next.Name(), s.next.Address())
+	}
+
+	return next.NodePeers(ctx, opts)
+}
+
 // ProposerDuties obtains proposer duties for the given epoch.
 func (s *Erroring) ProposerDuties(ctx context.Context,
 	opts *api.ProposerDutiesOpts,

diff --git a/testclients/sleepy.go b/testclients/sleepy.go
@@ -426,6 +426,17 @@ func (s *Sleepy) NodeSyncing(ctx context.Context) (*api.Response[*apiv1.SyncStat
 	return next.NodeSyncing(ctx)
 }
 
+// NodePeers provides the peers of the node
+func (s *Sleepy) NodePeers(ctx context.Context, opts *api.PeerOpts) (*api.Response[*apiv1.Peers], error) {
+	s.sleep(ctx)
+	next, isNext := s.next.(consensusclient.NodePeersProvider)
+	if !isNext {
+		return nil, errors.New("next does not support this call")
+	}
+
+	return next.NodePeers(ctx, opts)
+}
+
 // ProposerDuties obtains proposer duties for the given epoch.
 func (s *Sleepy) ProposerDuties(ctx context.Context,
 	opts *api.ProposerDutiesOpts,