e1e5b4a8a6
* admin: add plugin lock coordination * shell: allow bypassing lock checks * plugin worker: add admin script handler * mini: include admin_script in plugin defaults * admin script UI: drop name and enlarge text * admin script: add default script * admin_script: make run interval configurable * plugin: gate other jobs during admin_script runs * plugin: use last completed admin_script run * admin: backfill plugin config defaults * templ Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * comparable to default version Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * default to run Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * format Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * shell: respect pre-set noLock for fix.replication * shell: add force no-lock mode for admin scripts * volume balance worker already exists Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * admin: expose scheduler status JSON * shell: add sleep command * shell: restrict sleep syntax * Revert "shell: respect pre-set noLock for fix.replication" This reverts commit 2b14e8b82602a740d3a473c085e3b3a14f1ddbb3. * templ Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * fix import Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * less logs Co-Authored-By: Copilot <223556219+Copilot@users.noreply.github.com> * Reduce master client logs on canceled contexts * Update mini default job type count --------- Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package plugin
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/plugin_pb"
|
|
)
|
|
|
|
func TestGetSchedulerStatusIncludesInProcessJobs(t *testing.T) {
|
|
pluginSvc, err := New(Options{})
|
|
if err != nil {
|
|
t.Fatalf("New: %v", err)
|
|
}
|
|
defer pluginSvc.Shutdown()
|
|
|
|
pluginSvc.trackExecutionStart("req-1", "worker-a", &plugin_pb.JobSpec{
|
|
JobId: "job-1",
|
|
JobType: "vacuum",
|
|
}, 1)
|
|
|
|
status := pluginSvc.GetSchedulerStatus()
|
|
if len(status.InProcessJobs) != 1 {
|
|
t.Fatalf("expected one in-process job, got %d", len(status.InProcessJobs))
|
|
}
|
|
if status.InProcessJobs[0].JobID != "job-1" {
|
|
t.Fatalf("unexpected job id: %s", status.InProcessJobs[0].JobID)
|
|
}
|
|
}
|
|
|
|
func TestGetSchedulerStatusIncludesLastDetectionCount(t *testing.T) {
|
|
pluginSvc, err := New(Options{})
|
|
if err != nil {
|
|
t.Fatalf("New: %v", err)
|
|
}
|
|
defer pluginSvc.Shutdown()
|
|
|
|
const jobType = "vacuum"
|
|
pluginSvc.registry.UpsertFromHello(&plugin_pb.WorkerHello{
|
|
WorkerId: "worker-a",
|
|
Capabilities: []*plugin_pb.JobTypeCapability{
|
|
{JobType: jobType, CanDetect: true},
|
|
},
|
|
})
|
|
|
|
pluginSvc.recordSchedulerDetectionSuccess(jobType, 3)
|
|
|
|
status := pluginSvc.GetSchedulerStatus()
|
|
found := false
|
|
for _, jt := range status.JobTypes {
|
|
if jt.JobType != jobType {
|
|
continue
|
|
}
|
|
found = true
|
|
if jt.LastDetectedCount != 3 {
|
|
t.Fatalf("unexpected last detected count: got=%d want=3", jt.LastDetectedCount)
|
|
}
|
|
if jt.LastDetectedAt == nil {
|
|
t.Fatalf("expected last detected at to be set")
|
|
}
|
|
}
|
|
if !found {
|
|
t.Fatalf("expected job type status for %s", jobType)
|
|
}
|
|
}
|