65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package aghnet
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
|
|
"github.com/AdguardTeam/dnsproxy/upstream"
|
|
"github.com/miekg/dns"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNewMultiAddrExchanger(t *testing.T) {
|
|
var e Exchanger
|
|
var err error
|
|
|
|
t.Run("empty", func(t *testing.T) {
|
|
e, err = NewMultiAddrExchanger([]string{}, nil, 0)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, e)
|
|
})
|
|
|
|
t.Run("successful", func(t *testing.T) {
|
|
e, err = NewMultiAddrExchanger([]string{"www.example.com"}, nil, 0)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, e)
|
|
})
|
|
|
|
t.Run("unsuccessful", func(t *testing.T) {
|
|
e, err = NewMultiAddrExchanger([]string{"invalid-proto://www.example.com"}, nil, 0)
|
|
require.Error(t, err)
|
|
assert.Nil(t, e)
|
|
})
|
|
}
|
|
|
|
func TestMultiAddrExchanger_Exchange(t *testing.T) {
|
|
e := &multiAddrExchanger{}
|
|
|
|
t.Run("error", func(t *testing.T) {
|
|
e.ups = []upstream.Upstream{&aghtest.TestErrUpstream{}}
|
|
|
|
resp, err := e.Exchange(nil)
|
|
require.Error(t, err)
|
|
assert.Nil(t, resp)
|
|
})
|
|
|
|
t.Run("success", func(t *testing.T) {
|
|
e.ups = []upstream.Upstream{&aghtest.TestUpstream{
|
|
Reverse: map[string][]string{
|
|
"abc": {"cba"},
|
|
},
|
|
}}
|
|
|
|
resp, err := e.Exchange(&dns.Msg{
|
|
Question: []dns.Question{{
|
|
Name: "abc",
|
|
Qtype: dns.TypePTR,
|
|
}},
|
|
})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.Answer, 1)
|
|
assert.Equal(t, "cba", resp.Answer[0].Header().Name)
|
|
})
|
|
}
|