From 0a6c082162f4f375e3b801a99e4771d3346e511d Mon Sep 17 00:00:00 2001
From: Oliver Herms <oliver.herms@exaring.de>
Date: Fri, 5 Oct 2018 20:10:47 +0200
Subject: [PATCH] Fixing tests

---
 protocols/bmp/packet/initiation_message.go       | 1 -
 protocols/bmp/packet/initiation_message_test.go  | 3 +++
 protocols/bmp/packet/termination_message_test.go | 3 +++
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/protocols/bmp/packet/initiation_message.go b/protocols/bmp/packet/initiation_message.go
index 6d4ee7ec..6fbc3f30 100644
--- a/protocols/bmp/packet/initiation_message.go
+++ b/protocols/bmp/packet/initiation_message.go
@@ -33,7 +33,6 @@ func decodeInitiationMessage(buf *bytes.Buffer, ch *CommonHeader) (Msg, error) {
 
 		im.TLVs = append(im.TLVs, tlv)
 		read += uint32(tlv.InformationLength) + MinInformationTLVLen
-		fmt.Printf("read: %d\n", read)
 	}
 
 	return im, nil
diff --git a/protocols/bmp/packet/initiation_message_test.go b/protocols/bmp/packet/initiation_message_test.go
index 4513a2be..d5c3a129 100644
--- a/protocols/bmp/packet/initiation_message_test.go
+++ b/protocols/bmp/packet/initiation_message_test.go
@@ -41,6 +41,9 @@ func TestDecodeInitiationMessage(t *testing.T) {
 			},
 			wantFail: false,
 			expected: &InitiationMessage{
+				CommonHeader: &CommonHeader{
+					MsgLength: CommonHeaderLen + 17,
+				},
 				TLVs: []*InformationTLV{
 					{
 						InformationType:   1,
diff --git a/protocols/bmp/packet/termination_message_test.go b/protocols/bmp/packet/termination_message_test.go
index 412704b8..9f33bbfe 100644
--- a/protocols/bmp/packet/termination_message_test.go
+++ b/protocols/bmp/packet/termination_message_test.go
@@ -41,6 +41,9 @@ func TestDecodeTerminationMessage(t *testing.T) {
 			},
 			wantFail: false,
 			expected: &TerminationMessage{
+				CommonHeader: &CommonHeader{
+					MsgLength: CommonHeaderLen + 17,
+				},
 				TLVs: []*InformationTLV{
 					{
 						InformationType:   1,
-- 
GitLab