4165e0ef3a
Merge in DNS/adguard-home from 2574-external-tests-4 to master Close #2574. Squashed commit of the following: commit 0d06fce604750f76f4a319b2539105e936a248ce Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Apr 22 13:26:25 2021 +0300 home: imp tests, docs commit fc7b7f13f19bb8f183522a13d5726253eaae83d0 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Apr 22 12:20:15 2021 +0300 home: fix whois test
79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package home
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// fakeConn is a mock implementation of net.Conn to simplify testing.
|
|
//
|
|
// TODO(e.burkov): Search for other places in code where it may be used. Move
|
|
// into aghtest then.
|
|
type fakeConn struct {
|
|
// Conn is embedded here simply to make *fakeConn a net.Conn without
|
|
// actually implementing all methods.
|
|
net.Conn
|
|
data []byte
|
|
}
|
|
|
|
// Write implements net.Conn interface for *fakeConn. It always returns 0 and a
|
|
// nil error without mutating the slice.
|
|
func (c *fakeConn) Write(_ []byte) (n int, err error) {
|
|
return 0, nil
|
|
}
|
|
|
|
// Read implements net.Conn interface for *fakeConn. It puts the content of
|
|
// c.data field into b up to the b's capacity.
|
|
func (c *fakeConn) Read(b []byte) (n int, err error) {
|
|
return copy(b, c.data), io.EOF
|
|
}
|
|
|
|
// Close implements net.Conn interface for *fakeConn. It always returns nil.
|
|
func (c *fakeConn) Close() (err error) {
|
|
return nil
|
|
}
|
|
|
|
// SetReadDeadline implements net.Conn interface for *fakeConn. It always
|
|
// returns nil.
|
|
func (c *fakeConn) SetReadDeadline(_ time.Time) (err error) {
|
|
return nil
|
|
}
|
|
|
|
// fakeDial is a mock implementation of customDialContext to simplify testing.
|
|
func (c *fakeConn) fakeDial(ctx context.Context, network, addr string) (conn net.Conn, err error) {
|
|
return c, nil
|
|
}
|
|
|
|
func TestWhois(t *testing.T) {
|
|
const (
|
|
nl = "\n"
|
|
data = `OrgName: FakeOrg LLC` + nl +
|
|
`City: Nonreal` + nl +
|
|
`Country: Imagiland` + nl
|
|
)
|
|
|
|
fc := &fakeConn{
|
|
data: []byte(data),
|
|
}
|
|
|
|
w := Whois{
|
|
timeoutMsec: 5000,
|
|
dialContext: fc.fakeDial,
|
|
}
|
|
resp, err := w.queryAll(context.Background(), "1.2.3.4")
|
|
assert.NoError(t, err)
|
|
|
|
m := whoisParse(resp)
|
|
require.NotEmpty(t, m)
|
|
|
|
assert.Equal(t, "FakeOrg LLC", m["orgname"])
|
|
assert.Equal(t, "Imagiland", m["country"])
|
|
assert.Equal(t, "Nonreal", m["city"])
|
|
}
|