Use compressionMapsEqual in TestPackDomainNameCompressionMap
This commit is contained in:
parent
6aa28be819
commit
c1ad186588
21
msg_test.go
21
msg_test.go
|
@ -214,8 +214,15 @@ func TestUnpackDomainName(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestPackDomainNameCompressionMap(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} {
|
for _, compress := range []bool{true, false} {
|
||||||
compression := make(map[string]int)
|
compression := make(map[string]int)
|
||||||
|
|
||||||
|
@ -224,16 +231,8 @@ func TestPackDomainNameCompressionMap(t *testing.T) {
|
||||||
t.Fatalf("PackDomainName failed: %v", err)
|
t.Fatalf("PackDomainName failed: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, dname := range []string{
|
if !compressionMapsEqual(expected, compression) {
|
||||||
`www.this.is.\131an.example.org.`,
|
t.Errorf("expected compression maps to be equal; expected %v, got %v", expected, compression)
|
||||||
`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)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue