add mocks for GossipNeighbourSubset

weaveworks · Nov 5, 2019 · 640c5c3 · 640c5c3
1 parent 3780ce4
commit 640c5c3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/ipam/testutils_test.go b/ipam/testutils_test.go
@@ -65,6 +65,9 @@ func (m *mockGossipComms) GossipBroadcast(update mesh.GossipData) {
 	}
 }
 
+func (m *mockGossipComms) GossipNeighbourSubset(update mesh.GossipData) {
+	m.GossipBroadcast(update)
+}
 func equalByteBuffer(a, b []byte) bool {
 	if len(a) != len(b) {
 		return false

diff --git a/testing/gossip/mocks.go b/testing/gossip/mocks.go
@@ -199,3 +199,7 @@ func (client TestRouterClient) GossipUnicast(dstPeerName mesh.PeerName, buf []by
 func (client TestRouterClient) GossipBroadcast(update mesh.GossipData) {
 	client.router.gossipBroadcast(client.sender, update)
 }
+
+func (client TestRouterClient) GossipNeighbourSubset(update mesh.GossipData) {
+	client.router.gossip(client.sender, update)
+}