6d3a4f1db8
When preparing a tutorial for netxlite, I figured it is easier to tell people "hey, this is the package you should use for all low-level networking stuff" rather than introducing people to a set of packages working together where some piece of functionality is here and some other piece is there. Part of https://github.com/ooni/probe/issues/1591
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package resolver_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/resolver"
|
|
"github.com/ooni/probe-cli/v3/internal/netxlite"
|
|
)
|
|
|
|
func TestResolverIsBogon(t *testing.T) {
|
|
if resolver.IsBogon("antani") != true {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
if resolver.IsBogon("127.0.0.1") != true {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
if resolver.IsBogon("1.1.1.1") != false {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
if resolver.IsBogon("10.0.1.1") != true {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
}
|
|
|
|
func TestBogonAwareResolverWithBogon(t *testing.T) {
|
|
r := resolver.BogonResolver{
|
|
Resolver: resolver.NewFakeResolverWithResult([]string{"127.0.0.1"}),
|
|
}
|
|
addrs, err := r.LookupHost(context.Background(), "dns.google.com")
|
|
if !errors.Is(err, netxlite.ErrDNSBogon) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if len(addrs) > 0 {
|
|
t.Fatal("expected to see nil here")
|
|
}
|
|
}
|
|
|
|
func TestBogonAwareResolverWithoutBogon(t *testing.T) {
|
|
orig := []string{"8.8.8.8"}
|
|
r := resolver.BogonResolver{
|
|
Resolver: resolver.NewFakeResolverWithResult(orig),
|
|
}
|
|
addrs, err := r.LookupHost(context.Background(), "dns.google.com")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(addrs) != len(orig) || addrs[0] != orig[0] {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
}
|