2019-06-10 09:33:19 +01:00
|
|
|
package home
|
2019-04-26 13:10:29 +01:00
|
|
|
|
2019-09-18 16:44:27 +01:00
|
|
|
import (
|
2019-12-23 13:59:02 +00:00
|
|
|
"net"
|
2022-10-14 13:29:44 +01:00
|
|
|
"net/netip"
|
2022-01-25 16:47:02 +00:00
|
|
|
"runtime"
|
2019-09-18 16:44:27 +01:00
|
|
|
"testing"
|
2019-12-23 13:59:02 +00:00
|
|
|
"time"
|
|
|
|
|
2020-10-30 10:32:02 +00:00
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/dhcpd"
|
2023-04-06 12:12:50 +01:00
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/filtering"
|
2023-06-21 10:53:53 +01:00
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/whois"
|
2019-09-18 16:44:27 +01:00
|
|
|
"github.com/stretchr/testify/assert"
|
2021-03-11 14:32:58 +00:00
|
|
|
"github.com/stretchr/testify/require"
|
2019-09-18 16:44:27 +01:00
|
|
|
)
|
2019-04-26 13:10:29 +01:00
|
|
|
|
2023-04-06 12:12:50 +01:00
|
|
|
// newClientsContainer is a helper that creates a new clients container for
|
|
|
|
// tests.
|
2023-06-27 16:03:07 +01:00
|
|
|
func newClientsContainer(t *testing.T) (c *clientsContainer) {
|
2023-04-06 12:12:50 +01:00
|
|
|
c = &clientsContainer{
|
|
|
|
testing: true,
|
|
|
|
}
|
|
|
|
|
2023-06-27 16:03:07 +01:00
|
|
|
err := c.Init(nil, nil, nil, nil, &filtering.Config{})
|
|
|
|
require.NoError(t, err)
|
2019-04-26 13:10:29 +01:00
|
|
|
|
2023-04-06 12:12:50 +01:00
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestClients(t *testing.T) {
|
2023-06-27 16:03:07 +01:00
|
|
|
clients := newClientsContainer(t)
|
2019-04-26 13:10:29 +01:00
|
|
|
|
2020-12-07 13:04:53 +00:00
|
|
|
t.Run("add_success", func(t *testing.T) {
|
2022-11-09 11:37:07 +00:00
|
|
|
var (
|
|
|
|
cliNone = "1.2.3.4"
|
|
|
|
cli1 = "1.1.1.1"
|
|
|
|
cli2 = "2.2.2.2"
|
|
|
|
|
|
|
|
cliNoneIP = netip.MustParseAddr(cliNone)
|
|
|
|
cli1IP = netip.MustParseAddr(cli1)
|
|
|
|
cli2IP = netip.MustParseAddr(cli2)
|
|
|
|
)
|
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
c := &Client{
|
2022-11-09 11:37:07 +00:00
|
|
|
IDs: []string{cli1, "1:2:3::4", "aa:aa:aa:aa:aa:aa"},
|
2020-12-07 13:04:53 +00:00
|
|
|
Name: "client1",
|
|
|
|
}
|
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
ok, err := clients.Add(c)
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.True(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
c = &Client{
|
2022-11-09 11:37:07 +00:00
|
|
|
IDs: []string{cli2},
|
2020-12-07 13:04:53 +00:00
|
|
|
Name: "client2",
|
|
|
|
}
|
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
ok, err = clients.Add(c)
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.True(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
c, ok = clients.Find(cli1)
|
2021-03-11 14:32:58 +00:00
|
|
|
require.True(t, ok)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.Equal(t, "client1", c.Name)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
c, ok = clients.Find("1:2:3::4")
|
2021-03-11 14:32:58 +00:00
|
|
|
require.True(t, ok)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.Equal(t, "client1", c.Name)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
c, ok = clients.Find(cli2)
|
2021-03-11 14:32:58 +00:00
|
|
|
require.True(t, ok)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.Equal(t, "client2", c.Name)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2023-02-10 13:40:36 +00:00
|
|
|
assert.Equal(t, clients.clientSource(cliNoneIP), ClientSourceNone)
|
|
|
|
assert.Equal(t, clients.clientSource(cli1IP), ClientSourcePersistent)
|
|
|
|
assert.Equal(t, clients.clientSource(cli2IP), ClientSourcePersistent)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("add_fail_name", func(t *testing.T) {
|
2021-03-11 14:32:58 +00:00
|
|
|
ok, err := clients.Add(&Client{
|
2020-12-07 13:04:53 +00:00
|
|
|
IDs: []string{"1.2.3.5"},
|
|
|
|
Name: "client1",
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.False(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("add_fail_ip", func(t *testing.T) {
|
2021-03-11 14:32:58 +00:00
|
|
|
ok, err := clients.Add(&Client{
|
2020-12-07 13:04:53 +00:00
|
|
|
IDs: []string{"2.2.2.2"},
|
|
|
|
Name: "client3",
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.Error(t, err)
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.False(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("update_fail_ip", func(t *testing.T) {
|
2023-05-25 15:45:55 +01:00
|
|
|
err := clients.Update(&Client{Name: "client1"}, &Client{
|
2020-12-07 13:04:53 +00:00
|
|
|
IDs: []string{"2.2.2.2"},
|
|
|
|
Name: "client1",
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
assert.Error(t, err)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("update_success", func(t *testing.T) {
|
2022-11-09 11:37:07 +00:00
|
|
|
var (
|
|
|
|
cliOld = "1.1.1.1"
|
|
|
|
cliNew = "1.1.1.2"
|
|
|
|
|
|
|
|
cliOldIP = netip.MustParseAddr(cliOld)
|
|
|
|
cliNewIP = netip.MustParseAddr(cliNew)
|
|
|
|
)
|
|
|
|
|
2023-05-25 15:45:55 +01:00
|
|
|
prev, ok := clients.list["client1"]
|
|
|
|
require.True(t, ok)
|
|
|
|
|
|
|
|
err := clients.Update(prev, &Client{
|
2022-11-09 11:37:07 +00:00
|
|
|
IDs: []string{cliNew},
|
2020-12-07 13:04:53 +00:00
|
|
|
Name: "client1",
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2023-02-10 13:40:36 +00:00
|
|
|
assert.Equal(t, clients.clientSource(cliOldIP), ClientSourceNone)
|
|
|
|
assert.Equal(t, clients.clientSource(cliNewIP), ClientSourcePersistent)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2023-05-25 15:45:55 +01:00
|
|
|
prev, ok = clients.list["client1"]
|
|
|
|
require.True(t, ok)
|
|
|
|
|
|
|
|
err = clients.Update(prev, &Client{
|
2022-11-09 11:37:07 +00:00
|
|
|
IDs: []string{cliNew},
|
2020-12-07 13:04:53 +00:00
|
|
|
Name: "client1-renamed",
|
|
|
|
UseOwnSettings: true,
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
c, ok := clients.Find(cliNew)
|
2021-03-11 14:32:58 +00:00
|
|
|
require.True(t, ok)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-01-13 13:56:05 +00:00
|
|
|
assert.Equal(t, "client1-renamed", c.Name)
|
2020-12-07 13:04:53 +00:00
|
|
|
assert.True(t, c.UseOwnSettings)
|
2021-03-11 14:32:58 +00:00
|
|
|
|
|
|
|
nilCli, ok := clients.list["client1"]
|
|
|
|
require.False(t, ok)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.Nil(t, nilCli)
|
|
|
|
|
|
|
|
require.Len(t, c.IDs, 1)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
assert.Equal(t, cliNew, c.IDs[0])
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("del_success", func(t *testing.T) {
|
2021-01-27 15:32:13 +00:00
|
|
|
ok := clients.Del("client1-renamed")
|
2021-03-11 14:32:58 +00:00
|
|
|
require.True(t, ok)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2023-02-10 13:40:36 +00:00
|
|
|
assert.Equal(t, clients.clientSource(netip.MustParseAddr("1.1.1.2")), ClientSourceNone)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("del_fail", func(t *testing.T) {
|
2021-01-27 15:32:13 +00:00
|
|
|
ok := clients.Del("client3")
|
|
|
|
assert.False(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("addhost_success", func(t *testing.T) {
|
2022-11-09 11:37:07 +00:00
|
|
|
ip := netip.MustParseAddr("1.1.1.1")
|
|
|
|
ok := clients.AddHost(ip, "host", ClientSourceARP)
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.True(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
ok = clients.AddHost(ip, "host2", ClientSourceARP)
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.True(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
ok = clients.AddHost(ip, "host3", ClientSourceHostsFile)
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.True(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
|
2023-02-10 13:40:36 +00:00
|
|
|
assert.Equal(t, clients.clientSource(ip), ClientSourceHostsFile)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
|
|
|
|
2021-06-10 12:54:47 +01:00
|
|
|
t.Run("dhcp_replaces_arp", func(t *testing.T) {
|
2022-11-09 11:37:07 +00:00
|
|
|
ip := netip.MustParseAddr("1.2.3.4")
|
|
|
|
ok := clients.AddHost(ip, "from_arp", ClientSourceARP)
|
2021-06-29 13:53:28 +01:00
|
|
|
assert.True(t, ok)
|
2023-02-10 13:40:36 +00:00
|
|
|
assert.Equal(t, clients.clientSource(ip), ClientSourceARP)
|
2021-06-10 12:54:47 +01:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
ok = clients.AddHost(ip, "from_dhcp", ClientSourceDHCP)
|
2021-06-29 13:53:28 +01:00
|
|
|
assert.True(t, ok)
|
2023-02-10 13:40:36 +00:00
|
|
|
assert.Equal(t, clients.clientSource(ip), ClientSourceDHCP)
|
2021-06-10 12:54:47 +01:00
|
|
|
})
|
|
|
|
|
2020-12-07 13:04:53 +00:00
|
|
|
t.Run("addhost_fail", func(t *testing.T) {
|
2022-11-09 11:37:07 +00:00
|
|
|
ip := netip.MustParseAddr("1.1.1.1")
|
|
|
|
ok := clients.AddHost(ip, "host1", ClientSourceRDNS)
|
2021-01-27 15:32:13 +00:00
|
|
|
assert.False(t, ok)
|
2020-12-07 13:04:53 +00:00
|
|
|
})
|
2019-04-26 13:10:29 +01:00
|
|
|
}
|
2019-10-11 14:58:10 +01:00
|
|
|
|
2021-06-18 16:13:36 +01:00
|
|
|
func TestClientsWHOIS(t *testing.T) {
|
2023-06-27 16:03:07 +01:00
|
|
|
clients := newClientsContainer(t)
|
2023-06-21 10:53:53 +01:00
|
|
|
whois := &whois.Info{
|
2021-04-02 15:30:39 +01:00
|
|
|
Country: "AU",
|
|
|
|
Orgname: "Example Org",
|
|
|
|
}
|
2021-03-11 14:32:58 +00:00
|
|
|
|
|
|
|
t.Run("new_client", func(t *testing.T) {
|
2022-10-24 15:49:52 +01:00
|
|
|
ip := netip.MustParseAddr("1.1.1.255")
|
2022-11-09 11:37:07 +00:00
|
|
|
clients.setWHOISInfo(ip, whois)
|
2022-10-24 15:49:52 +01:00
|
|
|
rc := clients.ipToRC[ip]
|
2021-06-29 13:53:28 +01:00
|
|
|
require.NotNil(t, rc)
|
2019-10-11 14:58:10 +01:00
|
|
|
|
2023-06-21 10:53:53 +01:00
|
|
|
assert.Equal(t, rc.WHOIS, whois)
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("existing_auto-client", func(t *testing.T) {
|
2022-10-24 15:49:52 +01:00
|
|
|
ip := netip.MustParseAddr("1.1.1.1")
|
2022-11-09 11:37:07 +00:00
|
|
|
ok := clients.AddHost(ip, "host", ClientSourceRDNS)
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.True(t, ok)
|
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
clients.setWHOISInfo(ip, whois)
|
2022-10-24 15:49:52 +01:00
|
|
|
rc := clients.ipToRC[ip]
|
2021-06-29 13:53:28 +01:00
|
|
|
require.NotNil(t, rc)
|
2019-10-11 14:58:10 +01:00
|
|
|
|
2023-06-21 10:53:53 +01:00
|
|
|
assert.Equal(t, rc.WHOIS, whois)
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
2019-12-23 13:59:02 +00:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
t.Run("can't_set_manually-added", func(t *testing.T) {
|
2022-10-24 15:49:52 +01:00
|
|
|
ip := netip.MustParseAddr("1.1.1.2")
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
ok, err := clients.Add(&Client{
|
|
|
|
IDs: []string{"1.1.1.2"},
|
|
|
|
Name: "client1",
|
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.True(t, ok)
|
2019-12-23 13:59:02 +00:00
|
|
|
|
2022-11-09 11:37:07 +00:00
|
|
|
clients.setWHOISInfo(ip, whois)
|
2022-10-24 15:49:52 +01:00
|
|
|
rc := clients.ipToRC[ip]
|
|
|
|
require.Nil(t, rc)
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.True(t, clients.Del("client1"))
|
|
|
|
})
|
|
|
|
}
|
2019-12-23 13:59:02 +00:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
func TestClientsAddExisting(t *testing.T) {
|
2023-06-27 16:03:07 +01:00
|
|
|
clients := newClientsContainer(t)
|
2019-12-23 13:59:02 +00:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
t.Run("simple", func(t *testing.T) {
|
2022-11-09 11:37:07 +00:00
|
|
|
ip := netip.MustParseAddr("1.1.1.1")
|
2021-06-29 13:53:28 +01:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
// Add a client.
|
|
|
|
ok, err := clients.Add(&Client{
|
2021-06-29 13:53:28 +01:00
|
|
|
IDs: []string{ip.String(), "1:2:3::4", "aa:aa:aa:aa:aa:aa", "2.2.2.0/24"},
|
2021-03-11 14:32:58 +00:00
|
|
|
Name: "client1",
|
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.True(t, ok)
|
2019-12-23 13:59:02 +00:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
// Now add an auto-client with the same IP.
|
2022-11-09 11:37:07 +00:00
|
|
|
ok = clients.AddHost(ip, "test", ClientSourceRDNS)
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.True(t, ok)
|
2019-12-23 13:59:02 +00:00
|
|
|
})
|
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
t.Run("complicated", func(t *testing.T) {
|
2022-01-25 16:47:02 +00:00
|
|
|
// TODO(a.garipov): Properly decouple the DHCP server from the client
|
|
|
|
// storage.
|
|
|
|
if runtime.GOOS == "windows" {
|
|
|
|
t.Skip("skipping dhcp test on windows")
|
|
|
|
}
|
|
|
|
|
2023-03-23 13:52:01 +00:00
|
|
|
ip := netip.MustParseAddr("1.2.3.4")
|
2019-12-23 14:12:50 +00:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
// First, init a DHCP server with a single static lease.
|
2022-01-25 16:47:02 +00:00
|
|
|
config := &dhcpd.ServerConfig{
|
2023-04-18 13:12:11 +01:00
|
|
|
Enabled: true,
|
|
|
|
DataDir: t.TempDir(),
|
2021-03-16 16:11:32 +00:00
|
|
|
Conf4: dhcpd.V4ServerConf{
|
|
|
|
Enabled: true,
|
2022-10-14 13:29:44 +01:00
|
|
|
GatewayIP: netip.MustParseAddr("1.2.3.1"),
|
|
|
|
SubnetMask: netip.MustParseAddr("255.255.255.0"),
|
|
|
|
RangeStart: netip.MustParseAddr("1.2.3.2"),
|
|
|
|
RangeEnd: netip.MustParseAddr("1.2.3.10"),
|
2021-03-16 16:11:32 +00:00
|
|
|
},
|
2021-03-11 14:32:58 +00:00
|
|
|
}
|
2021-03-16 16:11:32 +00:00
|
|
|
|
2022-09-13 21:45:35 +01:00
|
|
|
dhcpServer, err := dhcpd.Create(config)
|
2021-06-16 14:48:46 +01:00
|
|
|
require.NoError(t, err)
|
2021-03-11 14:32:58 +00:00
|
|
|
|
2022-09-13 21:45:35 +01:00
|
|
|
clients.dhcpServer = dhcpServer
|
|
|
|
|
|
|
|
err = dhcpServer.AddStaticLease(&dhcpd.Lease{
|
2021-03-11 14:32:58 +00:00
|
|
|
HWAddr: net.HardwareAddr{0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA},
|
2021-06-29 13:53:28 +01:00
|
|
|
IP: ip,
|
2021-03-11 14:32:58 +00:00
|
|
|
Hostname: "testhost",
|
|
|
|
Expiry: time.Now().Add(time.Hour),
|
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-03-11 14:32:58 +00:00
|
|
|
|
|
|
|
// Add a new client with the same IP as for a client with MAC.
|
|
|
|
ok, err := clients.Add(&Client{
|
2021-06-29 13:53:28 +01:00
|
|
|
IDs: []string{ip.String()},
|
2021-03-11 14:32:58 +00:00
|
|
|
Name: "client2",
|
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.True(t, ok)
|
|
|
|
|
2021-10-22 09:58:18 +01:00
|
|
|
// Add a new client with the IP from the first client's IP range.
|
2021-03-11 14:32:58 +00:00
|
|
|
ok, err = clients.Add(&Client{
|
|
|
|
IDs: []string{"2.2.2.2"},
|
|
|
|
Name: "client3",
|
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.True(t, ok)
|
|
|
|
})
|
2019-12-23 13:59:02 +00:00
|
|
|
}
|
2020-05-13 18:31:43 +01:00
|
|
|
|
|
|
|
func TestClientsCustomUpstream(t *testing.T) {
|
2023-06-27 16:03:07 +01:00
|
|
|
clients := newClientsContainer(t)
|
2020-05-13 18:31:43 +01:00
|
|
|
|
2021-03-11 14:32:58 +00:00
|
|
|
// Add client with upstreams.
|
|
|
|
ok, err := clients.Add(&Client{
|
2020-05-13 18:31:43 +01:00
|
|
|
IDs: []string{"1.1.1.1", "1:2:3::4", "aa:aa:aa:aa:aa:aa"},
|
|
|
|
Name: "client1",
|
|
|
|
Upstreams: []string{
|
|
|
|
"1.1.1.1",
|
|
|
|
"[/example.org/]8.8.8.8",
|
|
|
|
},
|
2021-03-11 14:32:58 +00:00
|
|
|
})
|
2021-05-28 11:02:59 +01:00
|
|
|
require.NoError(t, err)
|
2020-05-13 18:31:43 +01:00
|
|
|
assert.True(t, ok)
|
|
|
|
|
2021-05-28 11:02:59 +01:00
|
|
|
config, err := clients.findUpstreams("1.2.3.4")
|
2020-05-13 18:31:43 +01:00
|
|
|
assert.Nil(t, config)
|
2021-05-28 11:02:59 +01:00
|
|
|
assert.NoError(t, err)
|
2020-05-13 18:31:43 +01:00
|
|
|
|
2021-05-28 11:02:59 +01:00
|
|
|
config, err = clients.findUpstreams("1.1.1.1")
|
2021-03-11 14:32:58 +00:00
|
|
|
require.NotNil(t, config)
|
2021-05-28 11:02:59 +01:00
|
|
|
assert.NoError(t, err)
|
2021-03-11 14:32:58 +00:00
|
|
|
assert.Len(t, config.Upstreams, 1)
|
|
|
|
assert.Len(t, config.DomainReservedUpstreams, 1)
|
2020-05-13 18:31:43 +01:00
|
|
|
}
|