You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.4 KiB
100 lines
3.4 KiB
package pluginworker
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/plugin_pb"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
)
|
|
|
|
func TestAdminScriptDescriptorDefaults(t *testing.T) {
|
|
descriptor := NewAdminScriptHandler(nil).Descriptor()
|
|
if descriptor == nil {
|
|
t.Fatalf("expected descriptor")
|
|
}
|
|
if descriptor.AdminRuntimeDefaults == nil {
|
|
t.Fatalf("expected admin runtime defaults")
|
|
}
|
|
if descriptor.AdminRuntimeDefaults.DetectionIntervalSeconds != adminScriptDetectTickSecs {
|
|
t.Fatalf("unexpected detection interval seconds: got=%d want=%d",
|
|
descriptor.AdminRuntimeDefaults.DetectionIntervalSeconds, adminScriptDetectTickSecs)
|
|
}
|
|
if descriptor.AdminConfigForm == nil {
|
|
t.Fatalf("expected admin config form")
|
|
}
|
|
runInterval := readInt64Config(descriptor.AdminConfigForm.DefaultValues, "run_interval_minutes", 0)
|
|
if runInterval != defaultAdminScriptRunMins {
|
|
t.Fatalf("unexpected run_interval_minutes default: got=%d want=%d", runInterval, defaultAdminScriptRunMins)
|
|
}
|
|
script := readStringConfig(descriptor.AdminConfigForm.DefaultValues, "script", "")
|
|
if strings.TrimSpace(script) == "" {
|
|
t.Fatalf("expected non-empty default script")
|
|
}
|
|
}
|
|
|
|
func TestAdminScriptDetectSkipsByRunInterval(t *testing.T) {
|
|
handler := NewAdminScriptHandler(nil)
|
|
sender := &recordingDetectionSender{}
|
|
err := handler.Detect(context.Background(), &plugin_pb.RunDetectionRequest{
|
|
JobType: adminScriptJobType,
|
|
LastSuccessfulRun: timestamppb.New(time.Now().Add(-2 * time.Minute)),
|
|
AdminConfigValues: map[string]*plugin_pb.ConfigValue{
|
|
"script": {
|
|
Kind: &plugin_pb.ConfigValue_StringValue{StringValue: defaultAdminScript},
|
|
},
|
|
"run_interval_minutes": {
|
|
Kind: &plugin_pb.ConfigValue_Int64Value{Int64Value: 17},
|
|
},
|
|
},
|
|
}, sender)
|
|
if err != nil {
|
|
t.Fatalf("detect returned err = %v", err)
|
|
}
|
|
if sender.proposals == nil {
|
|
t.Fatalf("expected proposals message")
|
|
}
|
|
if len(sender.proposals.Proposals) != 0 {
|
|
t.Fatalf("expected zero proposals, got %d", len(sender.proposals.Proposals))
|
|
}
|
|
if sender.complete == nil || !sender.complete.Success {
|
|
t.Fatalf("expected successful completion message")
|
|
}
|
|
if len(sender.events) == 0 {
|
|
t.Fatalf("expected detector activity events")
|
|
}
|
|
if !strings.Contains(sender.events[0].Message, "run interval") {
|
|
t.Fatalf("unexpected skip message: %q", sender.events[0].Message)
|
|
}
|
|
}
|
|
|
|
func TestAdminScriptDetectCreatesProposalWhenIntervalElapsed(t *testing.T) {
|
|
handler := NewAdminScriptHandler(nil)
|
|
sender := &recordingDetectionSender{}
|
|
err := handler.Detect(context.Background(), &plugin_pb.RunDetectionRequest{
|
|
JobType: adminScriptJobType,
|
|
LastSuccessfulRun: timestamppb.New(time.Now().Add(-20 * time.Minute)),
|
|
AdminConfigValues: map[string]*plugin_pb.ConfigValue{
|
|
"script": {
|
|
Kind: &plugin_pb.ConfigValue_StringValue{StringValue: defaultAdminScript},
|
|
},
|
|
"run_interval_minutes": {
|
|
Kind: &plugin_pb.ConfigValue_Int64Value{Int64Value: 17},
|
|
},
|
|
},
|
|
}, sender)
|
|
if err != nil {
|
|
t.Fatalf("detect returned err = %v", err)
|
|
}
|
|
if sender.proposals == nil {
|
|
t.Fatalf("expected proposals message")
|
|
}
|
|
if len(sender.proposals.Proposals) != 1 {
|
|
t.Fatalf("expected one proposal, got %d", len(sender.proposals.Proposals))
|
|
}
|
|
if sender.complete == nil || !sender.complete.Success || sender.complete.TotalProposals != 1 {
|
|
t.Fatalf("unexpected completion message: %+v", sender.complete)
|
|
}
|
|
}
|