github/weave-scope
1
0
Fork 0
mirror of https://github.com/weaveworks/scope.git synced 2026-07-28 01:31:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix test data.

Browse Source
This commit is contained in:
tiriplicamihai
2019-04-30 19:09:02 +03:00
parent 1fbe648e82
commit 92d3c1d5e9
1 changed files with 11 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
probe/endpoint/dns_snooper_test.go
View File

@@ -6,35 +6,32 @@ import (
"net"
"testing"
"github.com/bluele/gcache"
"github.com/google/gopacket/layers"
)
func TestprocessDNSMessageMultipleCNAME(t *testing.T) {
func TestProcessDNSMessageMultipleCNAME(t *testing.T) {
domain := "dummy.com"
question := layers.DNSQuestion{
Name: []byte(domain),
Type: layers.DNSTypeA,
Name: []byte(domain),
Type: layers.DNSTypeA,
Class: layers.DNSClassIN,
}
ipAddressCNAME := "127.0.0.1"
ipAddress := "127.0.1.1"
answers := []layers.DNSResourceRecord{
layers.DNSResourceRecord{
Name: []byte("api.dummy.com"),
CNAME: []byte("api.dummy.com"),
Type: layers.DNSTypeCNAME,
Class: layers.DNSClassIN,
},
layers.DNSResourceRecord{
Name: []byte("star.c10r.dummy.com"),
CNAME: []byte("star.c10r.dummy.com"),
Type: layers.DNSTypeCNAME,
Class: layers.DNSClassIN,
},
layers.DNSResourceRecord{
Name: []byte("dummy.com"),
Type: layers.DNSTypeA,
Class: layers.DNSClassIN,
IP: net.ParseIP(ipAddress),
},
layers.DNSResourceRecord{
Name: []byte("star.c10r.dummy.com"),
Type: layers.DNSTypeA,
@@ -44,12 +41,15 @@ func TestprocessDNSMessageMultipleCNAME(t *testing.T) {
}
dns := layers.DNS{
QR: true,
ResponseCode: layers.DNSResponseCodeNoErr,
Questions: []layers.DNSQuestion{question},
Answers: answers,
}
snooper := &DNSSnooper{}
snooper := &DNSSnooper{
reverseDNSCache: gcache.New(4).LRU().Build(),
}
snooper.processDNSMessage(&dns)
@@ -62,14 +62,4 @@ func TestprocessDNSMessageMultipleCNAME(t *testing.T) {
if _, ok := existingDomains.(map[string]struct{})[domain]; !ok {
t.Errorf("Domain %s should have been inserted", domain)
}
existingDomains, err = snooper.reverseDNSCache.Get(ipAddress)
if err != nil {
t.Errorf("A domain should have been inserted for the given IP:%v", err)
}
if _, ok := existingDomains.(map[string]struct{})[domain]; !ok {
t.Errorf("Domain %s should have been inserted", domain)
}
}