test(bitswap/message) no duplicates

ipfs · Oct 27, 2014 · cc7ec0f · cc7ec0f
1 parent 28db08d
commit cc7ec0f
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/exchange/bitswap/message/message_test.go b/exchange/bitswap/message/message_test.go
@@ -169,3 +169,20 @@ func contains(s []string, x string) bool {
 	}
 	return false
 }
+
+func TestDuplicates(t *testing.T) {
+	b := blocks.NewBlock([]byte("foo"))
+	msg := New()
+
+	msg.AddWanted(b.Key())
+	msg.AddWanted(b.Key())
+	if len(msg.Wantlist()) != 1 {
+		t.Fatal("Duplicate in BitSwapMessage")
+	}
+
+	msg.AddBlock(*b)
+	msg.AddBlock(*b)
+	if len(msg.Blocks()) != 1 {
+		t.Fatal("Duplicate in BitSwapMessage")
+	}
+}