mirror of https://github.com/v2ray/v2ray-core
34 lines
956 B
Go
34 lines
956 B
Go
|
package internal_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
. "github.com/v2ray/v2ray-core/app/dns/internal"
|
||
|
|
apptesting "github.com/v2ray/v2ray-core/app/testing"
|
||
|
|
netassert "github.com/v2ray/v2ray-core/common/net/testing/assert"
|
||
|
|
"github.com/v2ray/v2ray-core/common/serial"
|
||
|
|
v2testing "github.com/v2ray/v2ray-core/testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDnsAdd(t *testing.T) {
|
||
|
|
v2testing.Current(t)
|
||
|
|
|
||
|
|
domain := "v2ray.com"
|
||
|
|
cache := NewCache(&CacheConfig{
|
||
|
|
TrustedTags: map[serial.StringLiteral]bool{
|
||
|
|
serial.StringLiteral("testtag"): true,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
ip := cache.Get(&apptesting.Context{}, domain)
|
||
|
|
netassert.IP(ip).IsNil()
|
||
|
|
|
||
|
|
cache.Add(&apptesting.Context{CallerTagValue: "notvalidtag"}, domain, []byte{1, 2, 3, 4})
|
||
|
|
ip = cache.Get(&apptesting.Context{}, domain)
|
||
|
|
netassert.IP(ip).IsNil()
|
||
|
|
|
||
|
|
cache.Add(&apptesting.Context{CallerTagValue: "testtag"}, domain, []byte{1, 2, 3, 4})
|
||
|
|
ip = cache.Get(&apptesting.Context{}, domain)
|
||
|
|
netassert.IP(ip).Equals(net.IP([]byte{1, 2, 3, 4}))
|
||
|
|
}
|