code: func (m *MockBroadcastServer) Broadcast(server po.AtomicBroadcast_BroadcastServer) error {
  	res, err := server.Recv()
  	if err == io.EOF {
  		return nil
  	}
  
  	if err != nil {
  		return err
  	}
  
  	if m.BroadcastError != nil {
  		return m.BroadcastError
  	}
  
  	if m.BroadcastInternalServerError {
  		return server.Send(broadcastResponseError)
  	}
  
  	if m.BroadcastCustomResponse != nil {
  		return server.Send(m.BroadcastCustomResponse)
  	}
  
  	err = server.Send(broadcastResponseSuccess)
  	if err != nil {
  		return err
  	}
  
  	return m.mockBlockDelivery(res.Payload)
  }
docs: // Broadcast mock broadcast