f91de2ecd6
I develop this diff while working on https://github.com/ooni/probe/issues/1803#issuecomment-957323297. While there, make sure we don't have duplicate bogon code and always use the code inside netxlite.
38 lines
928 B
Go
38 lines
928 B
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 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")
|
|
}
|
|
}
|