Add some debug_asserts for paritytech#8171 (paritytech#8181)

jam10o-new · Feb 28, 2021 · 4dca913 · 4dca913
1 parent a8cf203
commit 4dca913
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/client/network/src/transactions.rs b/client/network/src/transactions.rs
@@ -319,14 +319,16 @@ impl<B: BlockT + 'static, H: ExHashT> TransactionsHandler<B, H> {
 			},
 
 			Event::NotificationStreamOpened { remote, protocol, role } if protocol == self.protocol_name => {
-				self.peers.insert(remote, Peer {
+				let _was_in = self.peers.insert(remote, Peer {
 					known_transactions: LruHashSet::new(NonZeroUsize::new(MAX_KNOWN_TRANSACTIONS)
 						.expect("Constant is nonzero")),
 					role,
 				});
+				debug_assert!(_was_in.is_none());
 			}
 			Event::NotificationStreamClosed { remote, protocol } if protocol == self.protocol_name => {
-				self.peers.remove(&remote);
+				let _peer = self.peers.remove(&remote);
+				debug_assert!(_peer.is_some());
 			}
 
 			Event::NotificationsReceived { remote, messages } => {