diff --git a/p2p/mock/peer.go b/p2p/mock/peer.go
index 10254c343..e2c8e3ce5 100644
--- a/p2p/mock/peer.go
+++ b/p2p/mock/peer.go
@@ -45,6 +45,8 @@ func NewPeer(ip net.IP) *Peer {
 func (mp *Peer) FlushStop()                              { mp.Stop() } //nolint:errcheck //ignore error
 func (mp *Peer) TrySend(chID byte, msgBytes []byte) bool { return true }
 func (mp *Peer) Send(chID byte, msgBytes []byte) bool    { return true }
+func (mp *Peer) NewTrySend(e p2p.Envelope) bool          { return true }
+func (mp *Peer) NewSend(e p2p.Envelope) bool             { return true }
 func (mp *Peer) NodeInfo() p2p.NodeInfo {
 	return p2p.DefaultNodeInfo{
 		DefaultNodeID: mp.addr.ID,
diff --git a/p2p/mocks/peer.go b/p2p/mocks/peer.go
index a9151c7d8..d2c3747a8 100644
--- a/p2p/mocks/peer.go
+++ b/p2p/mocks/peer.go
@@ -123,6 +123,34 @@ func (_m *Peer) IsRunning() bool {
 	return r0
 }
 
+// NewSend provides a mock function with given fields: _a0
+func (_m *Peer) NewSend(_a0 p2p.Envelope) bool {
+	ret := _m.Called(_a0)
+
+	var r0 bool
+	if rf, ok := ret.Get(0).(func(p2p.Envelope) bool); ok {
+		r0 = rf(_a0)
+	} else {
+		r0 = ret.Get(0).(bool)
+	}
+
+	return r0
+}
+
+// NewTrySend provides a mock function with given fields: _a0
+func (_m *Peer) NewTrySend(_a0 p2p.Envelope) bool {
+	ret := _m.Called(_a0)
+
+	var r0 bool
+	if rf, ok := ret.Get(0).(func(p2p.Envelope) bool); ok {
+		r0 = rf(_a0)
+	} else {
+		r0 = ret.Get(0).(bool)
+	}
+
+	return r0
+}
+
 // NodeInfo provides a mock function with given fields:
 func (_m *Peer) NodeInfo() p2p.NodeInfo {
 	ret := _m.Called()