test(bitswap/message) no duplicates

f758e76d · Brian Tiger Chow · 8cd17e8d · f758e76d
Commit f758e76d authored Oct 27, 2014 by Brian Tiger Chow
Show whitespace changes
Inline Side-by-side

Showing with 17 additions and 0 deletions

message/message_test.go message/message_test.go +17 -0

No files found.
--- a/message/message_test.go
+++ b/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")
+	}
+}