160 lines
3.4 KiB
Go
160 lines
3.4 KiB
Go
//go:build darwin || freebsd || linux || openbsd
|
|
|
|
package dhcpd
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/netip"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestServer_handleDHCPStatus(t *testing.T) {
|
|
const (
|
|
staticName = "static-client"
|
|
staticMAC = "aa:aa:aa:aa:aa:aa"
|
|
)
|
|
|
|
staticIP := netip.MustParseAddr("192.168.10.10")
|
|
|
|
staticLease := &leaseStatic{
|
|
HWAddr: staticMAC,
|
|
IP: staticIP,
|
|
Hostname: staticName,
|
|
}
|
|
|
|
s, err := Create(&ServerConfig{
|
|
Enabled: true,
|
|
Conf4: *defaultV4ServerConf(),
|
|
DataDir: t.TempDir(),
|
|
ConfigModified: func() {},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// checkStatus is a helper that asserts the response of
|
|
// [*server.handleDHCPStatus].
|
|
checkStatus := func(t *testing.T, want *dhcpStatusResponse) {
|
|
w := httptest.NewRecorder()
|
|
var req *http.Request
|
|
req, err = http.NewRequest(http.MethodGet, "", nil)
|
|
require.NoError(t, err)
|
|
|
|
b := &bytes.Buffer{}
|
|
err = json.NewEncoder(b).Encode(&want)
|
|
require.NoError(t, err)
|
|
|
|
s.handleDHCPStatus(w, req)
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
assert.JSONEq(t, b.String(), w.Body.String())
|
|
}
|
|
|
|
// defaultResponse is a helper that returns the response with default
|
|
// configuration.
|
|
defaultResponse := func() *dhcpStatusResponse {
|
|
conf4 := defaultV4ServerConf()
|
|
conf4.LeaseDuration = 86400
|
|
|
|
resp := &dhcpStatusResponse{
|
|
V4: *conf4,
|
|
V6: V6ServerConf{},
|
|
Leases: []*leaseDynamic{},
|
|
StaticLeases: []*leaseStatic{},
|
|
Enabled: true,
|
|
}
|
|
|
|
return resp
|
|
}
|
|
|
|
ok := t.Run("status", func(t *testing.T) {
|
|
resp := defaultResponse()
|
|
|
|
checkStatus(t, resp)
|
|
})
|
|
require.True(t, ok)
|
|
|
|
ok = t.Run("add_static_lease", func(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
|
|
b := &bytes.Buffer{}
|
|
err = json.NewEncoder(b).Encode(staticLease)
|
|
require.NoError(t, err)
|
|
|
|
var r *http.Request
|
|
r, err = http.NewRequest(http.MethodPost, "", b)
|
|
require.NoError(t, err)
|
|
|
|
s.handleDHCPAddStaticLease(w, r)
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
resp := defaultResponse()
|
|
resp.StaticLeases = []*leaseStatic{staticLease}
|
|
|
|
checkStatus(t, resp)
|
|
})
|
|
require.True(t, ok)
|
|
|
|
ok = t.Run("add_invalid_lease", func(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
|
|
b := &bytes.Buffer{}
|
|
|
|
err = json.NewEncoder(b).Encode(&leaseStatic{})
|
|
require.NoError(t, err)
|
|
|
|
var r *http.Request
|
|
r, err = http.NewRequest(http.MethodPost, "", b)
|
|
require.NoError(t, err)
|
|
|
|
s.handleDHCPAddStaticLease(w, r)
|
|
assert.Equal(t, http.StatusBadRequest, w.Code)
|
|
})
|
|
require.True(t, ok)
|
|
|
|
ok = t.Run("remove_static_lease", func(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
|
|
b := &bytes.Buffer{}
|
|
err = json.NewEncoder(b).Encode(staticLease)
|
|
require.NoError(t, err)
|
|
|
|
var r *http.Request
|
|
r, err = http.NewRequest(http.MethodPost, "", b)
|
|
require.NoError(t, err)
|
|
|
|
s.handleDHCPRemoveStaticLease(w, r)
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
resp := defaultResponse()
|
|
|
|
checkStatus(t, resp)
|
|
})
|
|
require.True(t, ok)
|
|
|
|
ok = t.Run("set_config", func(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
|
|
resp := defaultResponse()
|
|
resp.Enabled = false
|
|
|
|
b := &bytes.Buffer{}
|
|
err = json.NewEncoder(b).Encode(&resp)
|
|
require.NoError(t, err)
|
|
|
|
var r *http.Request
|
|
r, err = http.NewRequest(http.MethodPost, "", b)
|
|
require.NoError(t, err)
|
|
|
|
s.handleDHCPSetConfig(w, r)
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
checkStatus(t, resp)
|
|
})
|
|
require.True(t, ok)
|
|
}
|