community
/
dns
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use compressionMapsEqual in TestPackDomainNameCompressionMap

This commit is contained in:
Tom Thorogood 2018-11-29 09:48:02 +10:30
parent 6aa28be819
commit c1ad186588
No known key found for this signature in database
GPG Key ID: 86C63CDA416C6D2F
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
msg_test.go
View File

@ -214,8 +214,15 @@ func TestUnpackDomainName(t *testing.T) {
}
func TestPackDomainNameCompressionMap(t *testing.T) {
msg := make([]byte, 256)
expected := map[string]struct{}{
`www.this.is.\131an.example.org.`: struct{}{},
`is.\131an.example.org.`: struct{}{},
"\x83an.example.org.": struct{}{},
`example.org.`: struct{}{},
`org.`: struct{}{},
}
msg := make([]byte, 256)
for _, compress := range []bool{true, false} {
compression := make(map[string]int)
@ -224,16 +231,8 @@ func TestPackDomainNameCompressionMap(t *testing.T) {
t.Fatalf("PackDomainName failed: %v", err)
}
for _, dname := range []string{
`www.this.is.\131an.example.org.`,
`is.\131an.example.org.`,
"\x83an.example.org.",
`example.org.`,
`org.`,
} {
if _, ok := compression[dname]; !ok {
t.Errorf("expected to find %q in compression map", dname)
}
if !compressionMapsEqual(expected, compression) {
t.Errorf("expected compression maps to be equal; expected %v, got %v", expected, compression)
}
}
}