More tests

protocols/bmp/packet/initiation_message.go

@@ -16,11 +16,6 @@ func (im *InitiationMessage) MsgType() uint8 {
 	return im.CommonHeader.MsgType
 }
 
-// SetCommonHeader sets the common header
-func (im *InitiationMessage) SetCommonHeader(ch *CommonHeader) {
-	im.CommonHeader = ch
-}
-
 func decodeInitiationMessage(buf *bytes.Buffer, ch *CommonHeader) (Msg, error) {
 	im := &InitiationMessage{
 		TLVs: make([]*InformationTLV, 0, 2),

protocols/bmp/packet/initiation_message_test.go

@@ -7,6 +7,17 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestInitiationMessageMsgType(t *testing.T) {
+	pd := &InitiationMessage{
+		CommonHeader: &CommonHeader{
+			MsgType: 100,
+		},
+	}
+
+	if pd.MsgType() != 100 {
+		t.Errorf("Unexpected result")
+	}
+}
 func TestDecodeInitiationMessage(t *testing.T) {
 	tests := []struct {
 		name     string

protocols/bmp/packet/peer_down_test.go

@@ -7,6 +7,18 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestPeerDownMsgType(t *testing.T) {
+	pd := &PeerDownNotification{
+		CommonHeader: &CommonHeader{
+			MsgType: 100,
+		},
+	}
+
+	if pd.MsgType() != 100 {
+		t.Errorf("Unexpected result")
+	}
+}
+
 func TestDecodePeerDownNotification(t *testing.T) {
 	tests := []struct {
 		name     string

protocols/bmp/packet/peer_up_test.go

@@ -7,6 +7,17 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestPeerUpMsgType(t *testing.T) {
+	pd := &PeerUpNotification{
+		CommonHeader: &CommonHeader{
+			MsgType: 100,
+		},
+	}
+
+	if pd.MsgType() != 100 {
+		t.Errorf("Unexpected result")
+	}
+}
 func TestDecodePeerUp(t *testing.T) {
 	tests := []struct {
 		name     string

protocols/bmp/packet/route_mirroring_test.go

@@ -7,6 +7,17 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestRouteMirrorType(t *testing.T) {
+	pd := &RouteMirroringMsg{
+		CommonHeader: &CommonHeader{
+			MsgType: 100,
+		},
+	}
+
+	if pd.MsgType() != 100 {
+		t.Errorf("Unexpected result")
+	}
+}
 func TestDecodeRouteMirroringMsg(t *testing.T) {
 	tests := []struct {
 		name     string

protocols/bmp/packet/route_monitoring_test.go

@@ -7,6 +7,17 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestRouteMonitoringMsgType(t *testing.T) {
+	pd := &RouteMonitoringMsg{
+		CommonHeader: &CommonHeader{
+			MsgType: 100,
+		},
+	}
+
+	if pd.MsgType() != 100 {
+		t.Errorf("Unexpected result")
+	}
+}
 func TestDecodeRouteMonitoringMsg(t *testing.T) {
 	tests := []struct {
 		name     string

protocols/bmp/packet/stats_report_test.go

@@ -7,6 +7,18 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestStatsReportMsgType(t *testing.T) {
+	pd := &StatsReport{
+		CommonHeader: &CommonHeader{
+			MsgType: 100,
+		},
+	}
+
+	if pd.MsgType() != 100 {
+		t.Errorf("Unexpected result")
+	}
+}
+
 func TestDecodeStatsReport(t *testing.T) {
 	tests := []struct {
 		name     string

protocols/bmp/packet/termination_message_test.go

@@ -7,6 +7,17 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestTerminationMessageType(t *testing.T) {
+	pd := &TerminationMessage{
+		CommonHeader: &CommonHeader{
+			MsgType: 100,
+		},
+	}
+
+	if pd.MsgType() != 100 {
+		t.Errorf("Unexpected result")
+	}
+}
 func TestDecodeTerminationMessage(t *testing.T) {
 	tests := []struct {
 		name     string