mirror of
https://github.com/NginxProxyManager/nginx-proxy-manager.git
synced 2025-07-04 17:06:49 +00:00
Add more unit tests
This commit is contained in:
89
backend/internal/api/middleware/list_query_test.go
Normal file
89
backend/internal/api/middleware/list_query_test.go
Normal file
@ -0,0 +1,89 @@
|
||||
package middleware_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
c "npm/internal/api/context"
|
||||
"npm/internal/api/middleware"
|
||||
"npm/internal/entity/user"
|
||||
"npm/internal/model"
|
||||
"npm/internal/tags"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestListQuery(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
queryParams string
|
||||
expectedStatus int
|
||||
}{
|
||||
{
|
||||
name: "valid query params",
|
||||
queryParams: "?name:contains=John&sort=name.desc",
|
||||
expectedStatus: http.StatusOK,
|
||||
},
|
||||
{
|
||||
name: "invalid sort field",
|
||||
queryParams: "?name:contains=John&sort=invalid_field",
|
||||
expectedStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "invalid filter value",
|
||||
queryParams: "?name=123",
|
||||
expectedStatus: http.StatusOK,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
req, err := http.NewRequest("GET", "/test"+tt.queryParams, nil)
|
||||
assert.NoError(t, err)
|
||||
|
||||
testObj := user.Model{}
|
||||
|
||||
ctx := context.Background()
|
||||
ctx = context.WithValue(ctx, c.FiltersCtxKey, tags.GetFilterSchema(testObj))
|
||||
|
||||
rr := httptest.NewRecorder()
|
||||
handler := middleware.ListQuery(testObj)(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}))
|
||||
|
||||
handler.ServeHTTP(rr, req.WithContext(ctx))
|
||||
|
||||
assert.Equal(t, tt.expectedStatus, rr.Code)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetFiltersFromContext(t *testing.T) {
|
||||
req, err := http.NewRequest("GET", "/test", nil)
|
||||
assert.NoError(t, err)
|
||||
|
||||
filters := []model.Filter{
|
||||
{Field: "name", Modifier: "contains", Value: []string{"test"}},
|
||||
}
|
||||
ctx := context.WithValue(req.Context(), c.FiltersCtxKey, filters)
|
||||
req = req.WithContext(ctx)
|
||||
|
||||
result := middleware.GetFiltersFromContext(req)
|
||||
assert.Equal(t, filters, result)
|
||||
}
|
||||
|
||||
func TestGetSortFromContext(t *testing.T) {
|
||||
req, err := http.NewRequest("GET", "/test", nil)
|
||||
assert.NoError(t, err)
|
||||
|
||||
sorts := []model.Sort{
|
||||
{Field: "name", Direction: "asc"},
|
||||
}
|
||||
ctx := context.WithValue(req.Context(), c.SortCtxKey, sorts)
|
||||
req = req.WithContext(ctx)
|
||||
|
||||
result := middleware.GetSortFromContext(req)
|
||||
assert.Equal(t, sorts, result)
|
||||
}
|
Reference in New Issue
Block a user