AdGuardHome/internal/dhcpd/http_unix_test.go

//go:build darwin || freebsd || linux || openbsd

package dhcpd

import (
	"bytes"
	"encoding/json"
	"net"
	"net/http"
	"net/http/httptest"
	"net/netip"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestServer_handleDHCPStatus(t *testing.T) {
	const staticName = "static-client"

	staticIP := netip.MustParseAddr("192.168.10.10")
	staticMAC := net.HardwareAddr{0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA}

	staticLease := &Lease{
		Expiry:   time.Unix(leaseExpireStatic, 0),
		Hostname: staticName,
		HWAddr:   staticMAC,
		IP:       staticIP,
	}

	s, err := Create(&ServerConfig{
		Enabled:        true,
		Conf4:          *defaultV4ServerConf(),
		WorkDir:        t.TempDir(),
		DBFilePath:     dbFilename,
		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 returs the response with default
	// configuration.
	defaultResponse := func() *dhcpStatusResponse {
		conf4 := defaultV4ServerConf()
		conf4.LeaseDuration = 86400

		resp := &dhcpStatusResponse{
			V4:           *conf4,
			V6:           V6ServerConf{},
			Leases:       []*Lease{},
			StaticLeases: []*Lease{},
			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 = []*Lease{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(&Lease{})
		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)
}