Skip to content
Snippets Groups Projects
Commit a876bfc2 authored by Oliver Herms's avatar Oliver Herms
Browse files

Implement GetAFlag

parent 68c3468b
No related branches found
No related tags found
No related merge requests found
......@@ -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
}
......@@ -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())
}
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment