Adjust tests to SeenCache changes

libp2p · Aug 16, 2023 · 777ccf0 · 777ccf0
1 parent b512f27
commit 777ccf0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/libp2p/src/test/kotlin/io/libp2p/pubsub/SeenCacheTest.kt b/libp2p/src/test/kotlin/io/libp2p/pubsub/SeenCacheTest.kt
@@ -24,6 +24,7 @@ fun createMessage(number: Int): Rpc.Message {
 fun createPubsubMessage(number: Int) = TestPubsubMessage(createMessage(number))
 fun createPubsubMessage(number: Int, fastId: Int) =
     TestPubsubMessage(createMessage(number)).also { it.fastID = fastId }
+operator fun SeenCache<*>.minusAssign(msg: PubsubMessage) = this.remove(msg.messageId)
 
 fun assertContainsEntry(cache: SeenCache<String>, fakeMsg: Int) {
     assertThat(cache.isSeen(createPubsubMessage(fakeMsg))).isTrue()