Implement GetAFlag

protocols/bmp/packet/per_peer_header.go

@@ -65,3 +65,8 @@ func (p *PerPeerHeader) GetIPVersion() uint8 {
 	}
 	return 4
 }
+
+// GetAFlag checks if the A flag is set
+func (p *PerPeerHeader) GetAFlag() bool {
+	return p.PeerFlags&0x20 == 0x20
+}

protocols/bmp/packet/per_peer_header_test.go

@@ -157,3 +157,30 @@ func TestGetIPVersion(t *testing.T) {
 		assert.Equal(t, test.expected, v)
 	}
 }
+
+func TestGetAFlag(t *testing.T) {
+	tests := []struct {
+		name     string
+		input    *PerPeerHeader
+		expected bool
+	}{
+		{
+			name: "Test #1",
+			input: &PerPeerHeader{
+				PeerFlags: 0xdf, // 11011111
+			},
+			expected: false,
+		},
+		{
+			name: "Test #2",
+			input: &PerPeerHeader{
+				PeerFlags: 0x20,
+			},
+			expected: true,
+		},
+	}
+
+	for _, test := range tests {
+		assert.Equal(t, test.expected, test.input.GetAFlag())
+	}
+}