From c630a4101756e8c2a68e784abb7f976ecf52cfd1 Mon Sep 17 00:00:00 2001
From: drinktee <zjsxzong89@gmail.com>
Date: Sun, 15 Oct 2017 10:29:46 +0800
Subject: [PATCH] add unit test for cloud-controller-manager

---
 .../app/options/BUILD                         |  13 +++
 .../app/options/options_test.go               | 101 ++++++++++++++++++
 2 files changed, 114 insertions(+)
 create mode 100644 cmd/cloud-controller-manager/app/options/options_test.go

diff --git a/cmd/cloud-controller-manager/app/options/BUILD b/cmd/cloud-controller-manager/app/options/BUILD
index 759babd29e0..3b0c3cdebd0 100644
--- a/cmd/cloud-controller-manager/app/options/BUILD
+++ b/cmd/cloud-controller-manager/app/options/BUILD
@@ -3,6 +3,7 @@ package(default_visibility = ["//visibility:public"])
 load(
     "@io_bazel_rules_go//go:def.bzl",
     "go_library",
+    "go_test",
 )
 
 go_library(
@@ -31,3 +32,15 @@ filegroup(
     srcs = [":package-srcs"],
     tags = ["automanaged"],
 )
+
+go_test(
+    name = "go_default_test",
+    srcs = ["options_test.go"],
+    library = ":go_default_library",
+    deps = [
+        "//pkg/apis/componentconfig:go_default_library",
+        "//vendor/github.com/spf13/pflag:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/util/diff:go_default_library",
+    ],
+)
diff --git a/cmd/cloud-controller-manager/app/options/options_test.go b/cmd/cloud-controller-manager/app/options/options_test.go
new file mode 100644
index 00000000000..ece3bf09d22
--- /dev/null
+++ b/cmd/cloud-controller-manager/app/options/options_test.go
@@ -0,0 +1,101 @@
+/*
+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 := NewCloudControllerManagerServer()
+	s.AddFlags(f)
+
+	args := []string{
+		"--address=192.168.4.10",
+		"--allocate-node-cidrs=true",
+		"--cloud-config=/cloud-config",
+		"--cloud-provider=gce",
+		"--cluster-cidr=1.2.3.4/24",
+		"--cluster-name=k8s",
+		"--configure-cloud-routes=false",
+		"--contention-profiling=true",
+		"--controller-start-interval=2m",
+		"--min-resync-period=5m",
+		"--kube-api-burst=100",
+		"--kube-api-content-type=application/vnd.kubernetes.protobuf",
+		"--kube-api-qps=50.0",
+		"--kubeconfig=/kubeconfig",
+		"--leader-elect=false",
+		"--leader-elect-lease-duration=30s",
+		"--leader-elect-renew-deadline=15s",
+		"--leader-elect-resource-lock=configmap",
+		"--leader-elect-retry-period=5s",
+		"--master=192.168.4.20",
+		"--min-resync-period=8h",
+		"--port=10000",
+		"--profiling=false",
+		"--node-status-update-frequency=10m",
+		"--route-reconciliation-period=30s",
+		"--min-resync-period=100m",
+		"--use-service-account-credentials=false",
+	}
+	f.Parse(args)
+
+	expected := &CloudControllerManagerServer{
+		KubeControllerManagerConfiguration: componentconfig.KubeControllerManagerConfiguration{
+			CloudProvider:             "gce",
+			CloudConfigFile:           "/cloud-config",
+			Port:                      10000,
+			Address:                   "192.168.4.10",
+			ConcurrentServiceSyncs:    1,
+			MinResyncPeriod:           metav1.Duration{Duration: 100 * time.Minute},
+			NodeMonitorPeriod:         metav1.Duration{Duration: 5 * time.Second},
+			ClusterName:               "k8s",
+			ConfigureCloudRoutes:      false,
+			AllocateNodeCIDRs:         true,
+			ContentType:               "application/vnd.kubernetes.protobuf",
+			EnableContentionProfiling: true,
+			KubeAPIQPS:                50.0,
+			KubeAPIBurst:              100,
+			LeaderElection: componentconfig.LeaderElectionConfiguration{
+				ResourceLock:  "configmap",
+				LeaderElect:   false,
+				LeaseDuration: metav1.Duration{Duration: 30 * time.Second},
+				RenewDeadline: metav1.Duration{Duration: 15 * time.Second},
+				RetryPeriod:   metav1.Duration{Duration: 5 * time.Second},
+			},
+			ControllerStartInterval:   metav1.Duration{Duration: 2 * time.Minute},
+			RouteReconciliationPeriod: metav1.Duration{Duration: 30 * time.Second},
+			ClusterCIDR:               "1.2.3.4/24",
+		},
+		Kubeconfig: "/kubeconfig",
+		Master:     "192.168.4.20",
+		NodeStatusUpdateFrequency: metav1.Duration{Duration: 10 * time.Minute},
+	}
+	if !reflect.DeepEqual(expected, s) {
+		t.Errorf("Got different run options than expected.\nDifference detected on:\n%s", diff.ObjectReflectDiff(expected, s))
+	}
+}