From 91a31a3e7dc7a5b6ca2d85df7bf7d114bc94d7bc Mon Sep 17 00:00:00 2001 From: Brian Tiger Chow Date: Mon, 27 Oct 2014 06:31:49 -0700 Subject: [PATCH] test(bitswap/message) no duplicates --- exchange/bitswap/message/message_test.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/exchange/bitswap/message/message_test.go b/exchange/bitswap/message/message_test.go index f98934b37fe..9c69136cd7c 100644 --- 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") + } +}