242 lines
7.1 KiB
Go
242 lines
7.1 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/DigitalTolk/wireguard-ui/model"
|
|
)
|
|
|
|
func TestAPIListWolHosts(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
req, rec := jsonRequest(http.MethodGet, "/api/v1/wol-hosts", nil)
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APIListWolHosts(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, rec.Code)
|
|
|
|
var hosts []model.WakeOnLanHost
|
|
parseJSON(t, rec, &hosts)
|
|
assert.Len(t, hosts, 0)
|
|
}
|
|
|
|
func TestAPISaveWolHost(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
body := map[string]string{
|
|
"name": "TestHost",
|
|
"mac_address": "AA:BB:CC:DD:EE:FF",
|
|
}
|
|
|
|
req, rec := jsonRequest(http.MethodPost, "/api/v1/wol-hosts", body)
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APISaveWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, rec.Code)
|
|
|
|
// verify it was saved
|
|
hosts, _ := env.db.GetWakeOnLanHosts()
|
|
assert.Len(t, hosts, 1)
|
|
}
|
|
|
|
func TestAPISaveWolHost_InvalidMac(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
body := map[string]string{
|
|
"name": "Bad",
|
|
"mac_address": "not-a-mac",
|
|
}
|
|
|
|
req, rec := jsonRequest(http.MethodPost, "/api/v1/wol-hosts", body)
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APISaveWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusBadRequest, rec.Code)
|
|
}
|
|
|
|
func TestAPIDeleteWolHost(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
env.db.SaveWakeOnLanHost(model.WakeOnLanHost{MacAddress: "AA:BB:CC:DD:EE:FF", Name: "Del"})
|
|
|
|
req, rec := jsonRequest(http.MethodDelete, "/api/v1/wol-hosts/AA-BB-CC-DD-EE-FF", nil)
|
|
c := env.echo.NewContext(req, rec)
|
|
c.SetParamNames("mac")
|
|
c.SetParamValues("AA-BB-CC-DD-EE-FF")
|
|
err := APIDeleteWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusNoContent, rec.Code)
|
|
}
|
|
|
|
func TestAPISaveWolHost_UpdateMac(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
env.db.SaveWakeOnLanHost(model.WakeOnLanHost{MacAddress: "AA:BB:CC:DD:EE:FF", Name: "Original"})
|
|
|
|
body := map[string]string{
|
|
"name": "Updated",
|
|
"mac_address": "11:22:33:44:55:66",
|
|
"old_mac_address": "AA:BB:CC:DD:EE:FF",
|
|
}
|
|
|
|
req, rec := jsonRequest(http.MethodPost, "/api/v1/wol-hosts", body)
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APISaveWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, rec.Code)
|
|
}
|
|
|
|
func TestAPISaveWolHost_PreserveLatestUsed(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
env.db.SaveWakeOnLanHost(model.WakeOnLanHost{MacAddress: "AA:BB:CC:DD:EE:FF", Name: "Keep"})
|
|
|
|
body := map[string]string{
|
|
"name": "Keep Updated",
|
|
"mac_address": "AA:BB:CC:DD:EE:FF",
|
|
}
|
|
|
|
req, rec := jsonRequest(http.MethodPost, "/api/v1/wol-hosts", body)
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APISaveWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, rec.Code)
|
|
|
|
var result model.WakeOnLanHost
|
|
parseJSON(t, rec, &result)
|
|
assert.Equal(t, "Keep Updated", result.Name)
|
|
}
|
|
|
|
func TestAPIWakeHost_NotFound(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
req, rec := jsonRequest(http.MethodPost, "/api/v1/wol-hosts/XX-XX-XX-XX-XX-XX/wake", nil)
|
|
c := env.echo.NewContext(req, rec)
|
|
c.SetParamNames("mac")
|
|
c.SetParamValues("XX-XX-XX-XX-XX-XX")
|
|
err := APIWakeHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusNotFound, rec.Code)
|
|
}
|
|
|
|
func TestAPISaveWolHost_InvalidBody(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/wol-hosts", strings.NewReader("{invalid"))
|
|
req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
|
|
rec := httptest.NewRecorder()
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APISaveWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusBadRequest, rec.Code)
|
|
}
|
|
|
|
func TestAPISaveWolHost_MissingName(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
body := map[string]string{
|
|
"name": " ",
|
|
"mac_address": "AA:BB:CC:DD:EE:FF",
|
|
}
|
|
|
|
req, rec := jsonRequest(http.MethodPost, "/api/v1/wol-hosts", body)
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APISaveWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusBadRequest, rec.Code)
|
|
assert.Contains(t, rec.Body.String(), "Name is required")
|
|
}
|
|
|
|
func TestAPIListWolHosts_WithHosts(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
env.db.SaveWakeOnLanHost(model.WakeOnLanHost{MacAddress: "AA:BB:CC:DD:EE:01", Name: "Host1"})
|
|
env.db.SaveWakeOnLanHost(model.WakeOnLanHost{MacAddress: "AA:BB:CC:DD:EE:02", Name: "Host2"})
|
|
env.db.SaveWakeOnLanHost(model.WakeOnLanHost{MacAddress: "AA:BB:CC:DD:EE:03", Name: "Host3"})
|
|
|
|
req, rec := jsonRequest(http.MethodGet, "/api/v1/wol-hosts", nil)
|
|
c := env.echo.NewContext(req, rec)
|
|
err := APIListWolHosts(env.db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, rec.Code)
|
|
|
|
var hosts []model.WakeOnLanHost
|
|
parseJSON(t, rec, &hosts)
|
|
assert.Len(t, hosts, 3)
|
|
}
|
|
|
|
func TestAPIDeleteWolHost_NonExistent(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
// Deleting a non-existent host should still return NoContent (DELETE is idempotent)
|
|
req, rec := jsonRequest(http.MethodDelete, "/api/v1/wol-hosts/XX-XX-XX-XX-XX-XX", nil)
|
|
c := env.echo.NewContext(req, rec)
|
|
c.SetParamNames("mac")
|
|
c.SetParamValues("XX-XX-XX-XX-XX-XX")
|
|
err := APIDeleteWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
// The DeleteWakeOnHostLanHost will fail because XX-XX-XX-XX-XX-XX isn't a valid MAC
|
|
// Let's also test with a valid but nonexistent MAC
|
|
}
|
|
|
|
func TestAPIDeleteWolHost_ValidMacNotFound(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
// Use a valid MAC that doesn't exist in the DB
|
|
req, rec := jsonRequest(http.MethodDelete, "/api/v1/wol-hosts/AA:BB:CC:DD:EE:99", nil)
|
|
c := env.echo.NewContext(req, rec)
|
|
c.SetParamNames("mac")
|
|
c.SetParamValues("AA:BB:CC:DD:EE:99")
|
|
err := APIDeleteWolHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
// Delete of non-existent row succeeds silently
|
|
assert.Equal(t, http.StatusNoContent, rec.Code)
|
|
}
|
|
|
|
func TestAPIListWolHosts_DBError(t *testing.T) {
|
|
db := &errStore{}
|
|
e := echo.New()
|
|
|
|
req, rec := jsonRequest(http.MethodGet, "/api/v1/wol-hosts", nil)
|
|
c := e.NewContext(req, rec)
|
|
err := APIListWolHosts(db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusInternalServerError, rec.Code)
|
|
}
|
|
|
|
func TestAPIDeleteWolHost_DBError(t *testing.T) {
|
|
db := &errStore{}
|
|
e := echo.New()
|
|
|
|
req, rec := jsonRequest(http.MethodDelete, "/api/v1/wol-hosts/AA:BB:CC:DD:EE:FF", nil)
|
|
c := e.NewContext(req, rec)
|
|
c.SetParamNames("mac")
|
|
c.SetParamValues("AA:BB:CC:DD:EE:FF")
|
|
err := APIDeleteWolHost(db)(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusInternalServerError, rec.Code)
|
|
}
|
|
|
|
func TestAPIWakeHost_ExistingHost(t *testing.T) {
|
|
env := setupTestEnv(t)
|
|
|
|
env.db.SaveWakeOnLanHost(model.WakeOnLanHost{MacAddress: "AA:BB:CC:DD:EE:FF", Name: "WakeMe"})
|
|
|
|
req, rec := jsonRequest(http.MethodPost, "/api/v1/wol-hosts/AA:BB:CC:DD:EE:FF/wake", nil)
|
|
c := env.echo.NewContext(req, rec)
|
|
c.SetParamNames("mac")
|
|
c.SetParamValues("AA:BB:CC:DD:EE:FF")
|
|
err := APIWakeHost(env.db)(c)
|
|
require.NoError(t, err)
|
|
// On systems where UDP broadcast to 255.255.255.255:0 works, returns 200
|
|
// On systems where it's blocked, returns 500
|
|
assert.Contains(t, []int{http.StatusOK, http.StatusInternalServerError}, rec.Code)
|
|
}
|