e527f14bd2
- Using custom DNS internally (without TLS) to download Unbound files - Using then Unbound with DNS over TLS internally and system wide - Works even if you host system DNS is broken - Waits a few milliseconds for Unbound to start up
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
package dns
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"testing"
|
|
|
|
filesmocks "github.com/qdm12/golibs/files/mocks"
|
|
loggingmocks "github.com/qdm12/golibs/logging/mocks"
|
|
"github.com/qdm12/private-internet-access-docker/internal/constants"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_UseDNSSystemWide(t *testing.T) {
|
|
t.Parallel()
|
|
tests := map[string]struct {
|
|
data []byte
|
|
writtenData []byte
|
|
readErr error
|
|
writeErr error
|
|
err error
|
|
}{
|
|
"no data": {
|
|
writtenData: []byte("nameserver 127.0.0.1"),
|
|
},
|
|
"read error": {
|
|
readErr: fmt.Errorf("error"),
|
|
err: fmt.Errorf("error"),
|
|
},
|
|
"write error": {
|
|
writtenData: []byte("nameserver 127.0.0.1"),
|
|
writeErr: fmt.Errorf("error"),
|
|
err: fmt.Errorf("error"),
|
|
},
|
|
"lines without nameserver": {
|
|
data: []byte("abc\ndef\n"),
|
|
writtenData: []byte("abc\ndef\nnameserver 127.0.0.1"),
|
|
},
|
|
"lines with nameserver": {
|
|
data: []byte("abc\nnameserver abc def\ndef\n"),
|
|
writtenData: []byte("abc\nnameserver 127.0.0.1\ndef"),
|
|
},
|
|
}
|
|
for name, tc := range tests {
|
|
tc := tc
|
|
t.Run(name, func(t *testing.T) {
|
|
t.Parallel()
|
|
fileManager := &filesmocks.FileManager{}
|
|
fileManager.On("ReadFile", string(constants.ResolvConf)).
|
|
Return(tc.data, tc.readErr).Once()
|
|
if tc.readErr == nil {
|
|
fileManager.On("WriteToFile", string(constants.ResolvConf), tc.writtenData).
|
|
Return(tc.writeErr).Once()
|
|
}
|
|
logger := &loggingmocks.Logger{}
|
|
logger.On("Info", "%s: using DNS address %s system wide", logPrefix, "127.0.0.1").Once()
|
|
c := &configurator{
|
|
fileManager: fileManager,
|
|
logger: logger,
|
|
}
|
|
err := c.UseDNSSystemWide(net.IP{127, 0, 0, 1})
|
|
if tc.err != nil {
|
|
require.Error(t, err)
|
|
assert.Equal(t, tc.err.Error(), err.Error())
|
|
} else {
|
|
assert.NoError(t, err)
|
|
}
|
|
fileManager.AssertExpectations(t)
|
|
logger.AssertExpectations(t)
|
|
})
|
|
}
|
|
}
|