mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-02-18 00:17:39 +01:00
9072b8eaa8
Router & Middleware: - Add auth context middleware for user/token extraction - Add tenant middleware with authorization checking - Refactor middleware chain ordering for proper isolation - Add router helpers for common patterns Authentication & SSO: - Enhance auth with tenant-aware context - Update OIDC, SAML, and SSO handlers for multi-tenant - Add RBAC handler improvements - Add security enhancements New Test Coverage: - API foundation tests - Auth and authorization tests - Router state and general tests - SSO handler CRUD tests - WebSocket isolation tests - Resource handler tests
90 lines
2.6 KiB
Go
90 lines
2.6 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/rcourtman/pulse-go-rewrite/internal/resources"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestHandleGetResource(t *testing.T) {
|
|
// Setup
|
|
handlers := NewResourceHandlers()
|
|
|
|
// Add a dummy resource
|
|
res := resources.Resource{
|
|
ID: "test-node-1",
|
|
Type: resources.ResourceTypeNode,
|
|
Name: "Test Node",
|
|
}
|
|
handlers.Store().Upsert(res)
|
|
|
|
// Test Case 1: Resource Found
|
|
req := httptest.NewRequest("GET", "/api/resources/test-node-1", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
// We need to route this correctly or call the handler directly with context
|
|
// Handler expects path param parsing usually handled by router/mux.
|
|
// But HandleGetResource implementation says:
|
|
// path := strings.TrimPrefix(r.URL.Path, "/api/resources/")
|
|
// So we can call it directly.
|
|
|
|
handlers.HandleGetResource(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
var returned resources.Resource
|
|
err := json.NewDecoder(w.Body).Decode(&returned)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "test-node-1", returned.ID)
|
|
|
|
// Test Case 2: Resource Not Found
|
|
req = httptest.NewRequest("GET", "/api/resources/non-existent", nil)
|
|
w = httptest.NewRecorder()
|
|
|
|
handlers.HandleGetResource(w, req)
|
|
assert.Equal(t, http.StatusNotFound, w.Code)
|
|
}
|
|
|
|
func TestHandleGetResources(t *testing.T) {
|
|
handlers := NewResourceHandlers()
|
|
|
|
handlers.Store().Upsert(resources.Resource{ID: "vm-1", Type: resources.ResourceTypeVM, Status: resources.StatusRunning})
|
|
handlers.Store().Upsert(resources.Resource{ID: "node-1", Type: resources.ResourceTypeNode, Status: resources.StatusOnline})
|
|
|
|
// Case 1: List All
|
|
req := httptest.NewRequest("GET", "/api/resources", nil)
|
|
w := httptest.NewRecorder()
|
|
handlers.HandleGetResources(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
var resp ResourcesResponse
|
|
json.NewDecoder(w.Body).Decode(&resp)
|
|
assert.Equal(t, 2, resp.Count)
|
|
|
|
// Case 2: Filter by Type
|
|
req = httptest.NewRequest("GET", "/api/resources?type=vm", nil)
|
|
w = httptest.NewRecorder()
|
|
handlers.HandleGetResources(w, req)
|
|
json.NewDecoder(w.Body).Decode(&resp)
|
|
assert.Equal(t, 1, resp.Count)
|
|
assert.Equal(t, "vm-1", resp.Resources[0].ID)
|
|
}
|
|
|
|
func TestHandleGetResourceStats(t *testing.T) {
|
|
handlers := NewResourceHandlers()
|
|
handlers.Store().Upsert(resources.Resource{ID: "1", Type: resources.ResourceTypeVM})
|
|
|
|
req := httptest.NewRequest("GET", "/api/resources/stats", nil)
|
|
w := httptest.NewRecorder()
|
|
handlers.HandleGetResourceStats(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
var stats resources.StoreStats
|
|
json.NewDecoder(w.Body).Decode(&stats)
|
|
assert.Equal(t, 1, stats.TotalResources)
|
|
}
|