fixed bot comment in an unit test

multiversx · Dec 18, 2019 · 4628de3 · 4628de3
1 parent c284681
commit 4628de3
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/p2p/antiflood/p2pAntiflood_test.go b/p2p/antiflood/p2pAntiflood_test.go
@@ -93,18 +93,10 @@ func TestP2pAntiflood_CanNotIncrementMessageOriginatorShouldError(t *testing.T)
 	}
 	afm, _ := antiflood.NewP2pAntiflood(&mock.FloodPreventerStub{
 		IncrementAddingToSumCalled: func(identifier string, size uint64) bool {
-			if identifier == fromConnectedPeer.Pretty() {
-				return true
-			}
-
-			return false
+			return identifier == fromConnectedPeer.Pretty()
 		},
 		IncrementCalled: func(identifier string, size uint64) bool {
-			if identifier == message.PeerField.Pretty() {
-				return false
-			}
-
-			return true
+			return identifier != message.PeerField.Pretty()
 		},
 	})