99 lines
2.8 KiB
Go
99 lines
2.8 KiB
Go
//go:build test
|
|
// +build test
|
|
|
|
package admin
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"crussell/db"
|
|
"crussell/mw"
|
|
"crussell/testutils/testdb"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
)
|
|
|
|
// resetTestData truncates tables to clean up data between tests
|
|
func resetTestData(t *testing.T) {
|
|
t.Helper()
|
|
testdb.TruncateTables(t, db.DB)
|
|
}
|
|
|
|
// makeAdminRequest creates a request with admin context
|
|
// Note: Using 12-char IDs to match CHAR(12) columns in schema (e.g., created_by)
|
|
func makeAdminRequest(handler http.Handler, method, path string, body interface{}) *httptest.ResponseRecorder {
|
|
return makeRequestWithContext(handler, method, path, body, "admin001", "admin")
|
|
}
|
|
|
|
// makeUserRequest creates a request with regular user context
|
|
// Note: Using 12-char IDs to match CHAR(12) columns in schema (e.g., created_by)
|
|
func makeUserRequest(handler http.Handler, method, path string, body interface{}) *httptest.ResponseRecorder {
|
|
return makeRequestWithContext(handler, method, path, body, "user001", "verified_email")
|
|
}
|
|
|
|
// makeRequestWithContext creates a request with specific user context
|
|
|
|
func makeRequestWithContext(handler http.Handler, method, path string, body interface{}, userID, role string) *httptest.ResponseRecorder {
|
|
var req *http.Request
|
|
if body != nil {
|
|
bodyBytes, _ := json.Marshal(body)
|
|
req = httptest.NewRequest(method, path, bytes.NewReader(bodyBytes))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
} else {
|
|
req = httptest.NewRequest(method, path, nil)
|
|
}
|
|
|
|
rctx := chi.NewRouteContext()
|
|
if method == "GET" || method == "PUT" || method == "POST" || method == "DELETE" || method == "PATCH" {
|
|
if id, paramName := extractIDFromPath(path); id != "" {
|
|
rctx.URLParams.Add(paramName, id)
|
|
}
|
|
}
|
|
|
|
ctx := context.WithValue(req.Context(), chi.RouteCtxKey, rctx)
|
|
ctx = context.WithValue(ctx, mw.UserIDKey, userID)
|
|
ctx = context.WithValue(ctx, mw.UserRoleKey, role)
|
|
req = req.WithContext(ctx)
|
|
|
|
w := httptest.NewRecorder()
|
|
handler.ServeHTTP(w, req)
|
|
return w
|
|
}
|
|
|
|
func extractIDFromPath(path string) (string, string) {
|
|
patterns := []struct {
|
|
prefix string
|
|
paramName string
|
|
}{
|
|
{"/api/admin/bookings/user/", "user_id"},
|
|
{"/api/admin/users/", "id"},
|
|
{"/api/admin/bookings/", "id"},
|
|
{"/api/admin/services/", "id"},
|
|
{"/api/admin/patch-tests/", "id"},
|
|
{"/api/bookings/", "id"},
|
|
{"/api/services/eligible-for/", "userId"},
|
|
{"/api/services/", "id"},
|
|
}
|
|
|
|
for _, p := range patterns {
|
|
if strings.HasPrefix(path, p.prefix) {
|
|
suffix := path[len(p.prefix):]
|
|
if slashIdx := strings.Index(suffix, "/"); slashIdx >= 0 {
|
|
return suffix[:slashIdx], p.paramName
|
|
}
|
|
return suffix, p.paramName
|
|
}
|
|
}
|
|
return "", ""
|
|
}
|
|
|
|
func parseResponseBody(w *httptest.ResponseRecorder, dest interface{}) error {
|
|
return json.Unmarshal(w.Body.Bytes(), dest)
|
|
}
|