Added more unit tests for kube-scheduler.

2025-09-09 21:21:14 +00:00 · 2017-10-12 17:17:15 +08:00
parent 4ff6ef4a37
commit d5febacc91
1 changed files with 103 additions and 0 deletions
--- a/plugin/cmd/kube-scheduler/app/options/options_test.go
+++ b/plugin/cmd/kube-scheduler/app/options/options_test.go
@@ -0,0 +1,103 @@
+/*
+Copyright 2017 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package options
+
+import (
+	"reflect"
+	"testing"
+	"time"
+
+	"github.com/spf13/pflag"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/util/diff"
+	"k8s.io/kubernetes/pkg/apis/componentconfig"
+)
+
+func TestAddFlags(t *testing.T) {
+	f := pflag.NewFlagSet("addflagstest", pflag.ContinueOnError)
+	s := NewSchedulerServer()
+	s.AddFlags(f)
+
+	args := []string{
+		"--address=192.168.4.20",
+		"--algorithm-provider=FooProvider",
+		"--contention-profiling=true",
+		"--failure-domains=kubernetes.io/hostname",
+		"--hard-pod-affinity-symmetric-weight=0",
+		"--kube-api-burst=80",
+		"--kube-api-content-type=application/vnd.kubernetes.protobuf",
+		"--kube-api-qps=40.0",
+		"--kubeconfig=/foo/bar",
+		"--leader-elect=true",
+		"--leader-elect-lease-duration=20s",
+		"--leader-elect-renew-deadline=15s",
+		"--leader-elect-resource-lock=endpoints",
+		"--leader-elect-retry-period=3s",
+		"--lock-object-name=foo",
+		"--lock-object-namespace=bar",
+		"--master=192.168.4.20",
+		"--policy-config-file=/foo/bar",
+		"--policy-configmap=foo",
+		"--policy-configmap-namespace=bar",
+		"--port=10000",
+		"--profiling=false",
+		"--scheduler-name=foo",
+		"--use-legacy-policy-config=true",
+	}
+
+	f.Parse(args)
+
+	expected := &SchedulerServer{
+		KubeSchedulerConfiguration: componentconfig.KubeSchedulerConfiguration{
+			Port:                      10000,
+			Address:                   "192.168.4.20",
+			AlgorithmProvider:         "FooProvider",
+			PolicyConfigFile:          "/foo/bar",
+			EnableContentionProfiling: true,
+			EnableProfiling:           false,
+
+			ContentType:   "application/vnd.kubernetes.protobuf",
+			KubeAPIQPS:    40.0,
+			KubeAPIBurst:  80,
+			SchedulerName: "foo",
+			LeaderElection: componentconfig.LeaderElectionConfiguration{
+				ResourceLock:  "endpoints",
+				LeaderElect:   true,
+				LeaseDuration: metav1.Duration{Duration: 20 * time.Second},
+				RenewDeadline: metav1.Duration{Duration: 15 * time.Second},
+				RetryPeriod:   metav1.Duration{Duration: 3 * time.Second},
+			},
+
+			LockObjectNamespace: "bar",
+			LockObjectName:      "foo",
+
+			PolicyConfigMapName:      "foo",
+			PolicyConfigMapNamespace: "bar",
+			UseLegacyPolicyConfig:    true,
+
+			HardPodAffinitySymmetricWeight: 0,
+			FailureDomains:                 "kubernetes.io/hostname",
+		},
+		Kubeconfig: "/foo/bar",
+		Master:     "192.168.4.20",
+	}
+
+	if !reflect.DeepEqual(expected, s) {
+		t.Errorf("Got different run options than expected.\nDifference detected on:\n%s", diff.ObjectReflectDiff(expected, s))
+	}
+}