Add a test when you try to add an existing message in the buffer (#36)

rstefan1 · May 7, 2019 · a5a376d · a5a376d
1 parent d50f396
commit a5a376d
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/pkg/internal/buffer/message_test.go b/pkg/internal/buffer/message_test.go
@@ -99,6 +99,21 @@ var _ = Describe("MessageBuffer interface", func() {
 			Expect(msgBuffer.Messages).To(HaveLen(1))
 			expectProperMessage(msgBuffer.Messages[0], msgID, msgMsg, msgGossipCount)
 		})
+
+		It("doesn't add the message if it already exists in buffer", func() {
+			msg := Message{
+				ID:          msgID,
+				Msg:         msgMsg,
+				GossipCount: msgGossipCount,
+			}
+			msgBuffer := NewMessageBuffer()
+			msgBuffer.Messages = append(msgBuffer.Messages, msg)
+
+			// try to add same message in buffer
+			msgBuffer.AddMessage(msg)
+			Expect(msgBuffer.Messages).To(HaveLen(1))
+			expectProperMessage(msgBuffer.Messages[0], msgID, msgMsg, msgGossipCount)
+		})
 	})
 
 	Describe("at DigestBuffer fuction call", func() {