a61a2affe3
* Add stale job expiry and expire API * Add expire job button * Add test hook and coverage for ExpirePluginJobAPI * Document scheduler filtering side effect and reuse helper * Restore job spec proposal test * Regenerate plugin template output --------- Co-authored-by: Copilot <copilot@github.com>
100 lines
3.8 KiB
Go
100 lines
3.8 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/gorilla/sessions"
|
|
"github.com/seaweedfs/seaweedfs/weed/admin/dash"
|
|
)
|
|
|
|
func TestSetupRoutes_RegistersPluginSchedulerStatesAPI_NoAuth(t *testing.T) {
|
|
router := mux.NewRouter()
|
|
|
|
newRouteTestAdminHandlers().SetupRoutes(router, false, "", "", "", "", true)
|
|
|
|
if !hasRoute(router, http.MethodGet, "/api/plugin/scheduler-states") {
|
|
t.Fatalf("expected GET /api/plugin/scheduler-states to be registered in no-auth mode")
|
|
}
|
|
if !hasRoute(router, http.MethodGet, "/api/plugin/jobs/example/detail") {
|
|
t.Fatalf("expected GET /api/plugin/jobs/:jobId/detail to be registered in no-auth mode")
|
|
}
|
|
if !hasRoute(router, http.MethodPost, "/api/plugin/jobs/example/expire") {
|
|
t.Fatalf("expected POST /api/plugin/jobs/:jobId/expire to be registered in no-auth mode")
|
|
}
|
|
}
|
|
|
|
func TestSetupRoutes_RegistersPluginSchedulerStatesAPI_WithAuth(t *testing.T) {
|
|
router := mux.NewRouter()
|
|
|
|
newRouteTestAdminHandlers().SetupRoutes(router, true, "admin", "password", "", "", true)
|
|
|
|
if !hasRoute(router, http.MethodGet, "/api/plugin/scheduler-states") {
|
|
t.Fatalf("expected GET /api/plugin/scheduler-states to be registered in auth mode")
|
|
}
|
|
if !hasRoute(router, http.MethodGet, "/api/plugin/jobs/example/detail") {
|
|
t.Fatalf("expected GET /api/plugin/jobs/:jobId/detail to be registered in auth mode")
|
|
}
|
|
if !hasRoute(router, http.MethodPost, "/api/plugin/jobs/example/expire") {
|
|
t.Fatalf("expected POST /api/plugin/jobs/:jobId/expire to be registered in auth mode")
|
|
}
|
|
}
|
|
|
|
func TestSetupRoutes_RegistersPluginPages_NoAuth(t *testing.T) {
|
|
router := mux.NewRouter()
|
|
|
|
newRouteTestAdminHandlers().SetupRoutes(router, false, "", "", "", "", true)
|
|
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/configuration")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/queue")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/detection")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/execution")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/monitoring")
|
|
}
|
|
|
|
func TestSetupRoutes_RegistersPluginPages_WithAuth(t *testing.T) {
|
|
router := mux.NewRouter()
|
|
|
|
newRouteTestAdminHandlers().SetupRoutes(router, true, "admin", "password", "", "", true)
|
|
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/configuration")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/queue")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/detection")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/execution")
|
|
assertHasRoute(t, router, http.MethodGet, "/plugin/monitoring")
|
|
}
|
|
|
|
func newRouteTestAdminHandlers() *AdminHandlers {
|
|
adminServer := &dash.AdminServer{}
|
|
store := sessions.NewCookieStore([]byte("test-session-key"))
|
|
return &AdminHandlers{
|
|
adminServer: adminServer,
|
|
sessionStore: store,
|
|
authHandlers: &AuthHandlers{adminServer: adminServer, sessionStore: store},
|
|
clusterHandlers: &ClusterHandlers{adminServer: adminServer},
|
|
fileBrowserHandlers: &FileBrowserHandlers{adminServer: adminServer},
|
|
userHandlers: &UserHandlers{adminServer: adminServer},
|
|
policyHandlers: &PolicyHandlers{adminServer: adminServer},
|
|
pluginHandlers: &PluginHandlers{adminServer: adminServer},
|
|
mqHandlers: &MessageQueueHandlers{adminServer: adminServer},
|
|
serviceAccountHandlers: &ServiceAccountHandlers{adminServer: adminServer},
|
|
}
|
|
}
|
|
|
|
func hasRoute(router *mux.Router, method string, path string) bool {
|
|
req := httptest.NewRequest(method, path, nil)
|
|
var match mux.RouteMatch
|
|
return router.Match(req, &match)
|
|
}
|
|
|
|
func assertHasRoute(t *testing.T, router *mux.Router, method string, path string) {
|
|
t.Helper()
|
|
if !hasRoute(router, method, path) {
|
|
t.Fatalf("expected %s %s to be registered", method, path)
|
|
}
|
|
}
|