github.com/prysmaticlabs/prysm@v1.4.4/beacon-chain/p2p/testing/mock_broadcaster.go (about)

     1  package testing
     2  
     3  import (
     4  	"context"
     5  
     6  	ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
     7  	"google.golang.org/protobuf/proto"
     8  )
     9  
    10  // MockBroadcaster implements p2p.Broadcaster for testing.
    11  type MockBroadcaster struct {
    12  	BroadcastCalled   bool
    13  	BroadcastMessages []proto.Message
    14  }
    15  
    16  // Broadcast records a broadcast occurred.
    17  func (m *MockBroadcaster) Broadcast(_ context.Context, msg proto.Message) error {
    18  	m.BroadcastCalled = true
    19  	m.BroadcastMessages = append(m.BroadcastMessages, msg)
    20  	return nil
    21  }
    22  
    23  // BroadcastAttestation records a broadcast occurred.
    24  func (m *MockBroadcaster) BroadcastAttestation(_ context.Context, _ uint64, _ *ethpb.Attestation) error {
    25  	m.BroadcastCalled = true
    26  	return nil
    27  }