2022-10-04 16:02:55 +03:00
|
|
|
package websvc_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"encoding/json"
|
|
|
|
"net/http"
|
|
|
|
"net/netip"
|
|
|
|
"net/url"
|
|
|
|
"sync/atomic"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
2023-08-29 20:03:40 +03:00
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghhttp"
|
2022-10-04 16:02:55 +03:00
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
|
2022-10-10 14:05:24 +03:00
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/next/agh"
|
2022-10-04 16:02:55 +03:00
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/next/dnssvc"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/next/websvc"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestService_HandlePatchSettingsDNS(t *testing.T) {
|
|
|
|
wantDNS := &websvc.HTTPAPIDNSSettings{
|
2023-06-28 12:48:53 +03:00
|
|
|
Addresses: []netip.AddrPort{netip.MustParseAddrPort("127.0.1.1:53")},
|
|
|
|
BootstrapServers: []string{"1.0.0.1"},
|
|
|
|
UpstreamServers: []string{"1.1.1.1"},
|
|
|
|
DNS64Prefixes: []netip.Prefix{netip.MustParsePrefix("1234::/64")},
|
2023-08-29 20:03:40 +03:00
|
|
|
UpstreamTimeout: aghhttp.JSONDuration(2 * time.Second),
|
2023-06-28 12:48:53 +03:00
|
|
|
BootstrapPreferIPv6: true,
|
|
|
|
UseDNS64: true,
|
2022-10-04 16:02:55 +03:00
|
|
|
}
|
|
|
|
|
2023-02-08 13:39:04 +03:00
|
|
|
var started atomic.Bool
|
2022-10-04 16:02:55 +03:00
|
|
|
confMgr := newConfigManager()
|
2022-10-10 14:05:24 +03:00
|
|
|
confMgr.onDNS = func() (s agh.ServiceWithConfig[*dnssvc.Config]) {
|
2022-10-04 16:02:55 +03:00
|
|
|
return &aghtest.ServiceWithConfig[*dnssvc.Config]{
|
|
|
|
OnStart: func() (err error) {
|
2023-02-08 13:39:04 +03:00
|
|
|
started.Store(true)
|
2022-10-04 16:02:55 +03:00
|
|
|
|
|
|
|
return nil
|
|
|
|
},
|
|
|
|
OnShutdown: func(_ context.Context) (err error) { panic("not implemented") },
|
|
|
|
OnConfig: func() (c *dnssvc.Config) { panic("not implemented") },
|
|
|
|
}
|
|
|
|
}
|
|
|
|
confMgr.onUpdateDNS = func(ctx context.Context, c *dnssvc.Config) (err error) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
_, addr := newTestServer(t, confMgr)
|
|
|
|
u := &url.URL{
|
|
|
|
Scheme: "http",
|
|
|
|
Host: addr.String(),
|
|
|
|
Path: websvc.PathV1SettingsDNS,
|
|
|
|
}
|
|
|
|
|
|
|
|
req := jobj{
|
2023-06-28 12:48:53 +03:00
|
|
|
"addresses": wantDNS.Addresses,
|
|
|
|
"bootstrap_servers": wantDNS.BootstrapServers,
|
|
|
|
"upstream_servers": wantDNS.UpstreamServers,
|
|
|
|
"dns64_prefixes": wantDNS.DNS64Prefixes,
|
|
|
|
"upstream_timeout": wantDNS.UpstreamTimeout,
|
|
|
|
"bootstrap_prefer_ipv6": wantDNS.BootstrapPreferIPv6,
|
|
|
|
"use_dns64": wantDNS.UseDNS64,
|
2022-10-04 16:02:55 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
respBody := httpPatch(t, u, req, http.StatusOK)
|
|
|
|
resp := &websvc.HTTPAPIDNSSettings{}
|
|
|
|
err := json.Unmarshal(respBody, resp)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
2023-02-08 13:39:04 +03:00
|
|
|
assert.True(t, started.Load())
|
2022-10-04 16:02:55 +03:00
|
|
|
assert.Equal(t, wantDNS, resp)
|
|
|
|
assert.Equal(t, wantDNS, resp)
|
|
|
|
}
|