98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
package websvc_test
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/v1/websvc"
|
|
"github.com/AdguardTeam/golibs/netutil"
|
|
"github.com/AdguardTeam/golibs/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const testTimeout = 1 * time.Second
|
|
|
|
// testStart is the server start value for tests.
|
|
var testStart = time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
|
|
// newTestServer creates and starts a new web service instance as well as its
|
|
// sole address. It also registers a cleanup procedure, which shuts the
|
|
// instance down.
|
|
//
|
|
// TODO(a.garipov): Use svc or remove it.
|
|
func newTestServer(t testing.TB) (svc *websvc.Service, addr string) {
|
|
t.Helper()
|
|
|
|
c := &websvc.Config{
|
|
TLS: nil,
|
|
Addresses: []*netutil.IPPort{{
|
|
IP: net.IP{127, 0, 0, 1},
|
|
Port: 0,
|
|
}},
|
|
SecureAddresses: nil,
|
|
Timeout: testTimeout,
|
|
Start: testStart,
|
|
}
|
|
|
|
svc = websvc.New(c)
|
|
|
|
err := svc.Start()
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), testTimeout)
|
|
t.Cleanup(cancel)
|
|
|
|
err = svc.Shutdown(ctx)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
addrs := svc.Addrs()
|
|
require.Len(t, addrs, 1)
|
|
|
|
return svc, addrs[0]
|
|
}
|
|
|
|
// httpGet is a helper that performs an HTTP GET request and returns the body of
|
|
// the response as well as checks that the status code is correct.
|
|
//
|
|
// TODO(a.garipov): Add helpers for other methods.
|
|
func httpGet(t testing.TB, u *url.URL, wantCode int) (body []byte) {
|
|
t.Helper()
|
|
|
|
req, err := http.NewRequest(http.MethodGet, u.String(), nil)
|
|
require.NoErrorf(t, err, "creating req")
|
|
|
|
httpCli := &http.Client{
|
|
Timeout: testTimeout,
|
|
}
|
|
resp, err := httpCli.Do(req)
|
|
require.NoErrorf(t, err, "performing req")
|
|
require.Equal(t, wantCode, resp.StatusCode)
|
|
|
|
testutil.CleanupAndRequireSuccess(t, resp.Body.Close)
|
|
|
|
body, err = io.ReadAll(resp.Body)
|
|
require.NoErrorf(t, err, "reading body")
|
|
|
|
return body
|
|
}
|
|
|
|
func TestService_Start_getHealthCheck(t *testing.T) {
|
|
_, addr := newTestServer(t)
|
|
u := &url.URL{
|
|
Scheme: "http",
|
|
Host: addr,
|
|
Path: websvc.PathHealthCheck,
|
|
}
|
|
|
|
body := httpGet(t, u, http.StatusOK)
|
|
|
|
assert.Equal(t, []byte("OK"), body)
|
|
}
|