chore: add flags test (#6007)

Signed-off-by: yyzxw <1020938856@qq.com> Signed-off-by: xiaowu.zhu <xiaowu.zhu@daocloud.io> Co-authored-by: xiaowu.zhu <xiaowu.zhu@daocloud.io>
2026-02-14 18:10:21 +00:00 · 2023-05-18 17:48:05 +08:00
parent 1b9bdffd06
commit 933d85c735
1 changed files with 104 additions and 0 deletions
--- a/cmd/core/app/options/options_test.go
+++ b/cmd/core/app/options/options_test.go
@@ -0,0 +1,104 @@
+/*
+Copyright 2023 The KubeVela 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 (
+	"testing"
+	"time"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/spf13/pflag"
+
+	oamcontroller "github.com/oam-dev/kubevela/pkg/controller/core.oam.dev"
+)
+
+func TestCoreOptions_Flags(t *testing.T) {
+	fs := pflag.NewFlagSet("test", pflag.ContinueOnError)
+	opt := &CoreOptions{
+		ControllerArgs: &oamcontroller.Args{},
+	}
+
+	for _, f := range opt.Flags().FlagSets {
+		fs.AddFlagSet(f)
+	}
+
+	args := []string{
+		"--application-re-sync-period=5s",
+		"--cluster-metrics-interval=5s",
+		"--disable-caps=true",
+		"--enable-cluster-gateway=true",
+		"--enable-cluster-metrics=true",
+		"--enable-leader-election=true",
+		"--health-addr=/healthz",
+		"--informer-sync-period=3s",
+		"--kube-api-burst=500",
+		"--kube-api-qps=200",
+		"--leader-election-lease-duration=3s",
+		"--leader-election-namespace=test-namespace",
+		"--leader-election-renew-deadline=5s",
+		"--leader-election-resource-lock=/leases",
+		"--leader-election-retry-period=3s",
+		"--log-debug=true",
+		"--log-file-max-size=50",
+		"--log-file-path=/path/to/log",
+		"--max-dispatch-concurrent=5",
+		"--max-workflow-failed-backoff-time=30",
+		"--max-workflow-step-error-retry-times=5",
+		"--max-workflow-wait-backoff-time=5",
+		"--metrics-addr=/metrics",
+		"--perf-enabled=true",
+		"--pprof-addr=/debug/pprof",
+		"--use-webhook=true",
+		"--webhook-cert-dir=/path/to/cert",
+		"--webhook-port=8080",
+	}
+
+	if err := fs.Parse(args); err != nil {
+		t.Errorf("Failed to parse args: %v", err)
+	}
+
+	expected := &CoreOptions{
+		UseWebhook:                 true,
+		CertDir:                    "/path/to/cert",
+		WebhookPort:                8080,
+		MetricsAddr:                "/metrics",
+		EnableLeaderElection:       true,
+		LeaderElectionNamespace:    "test-namespace",
+		LogFilePath:                "/path/to/log",
+		LogFileMaxSize:             50,
+		LogDebug:                   true,
+		ControllerArgs:             &oamcontroller.Args{},
+		HealthAddr:                 "/healthz",
+		DisableCaps:                "true",
+		StorageDriver:              "",
+		InformerSyncPeriod:         3 * time.Second,
+		QPS:                        200,
+		Burst:                      500,
+		PprofAddr:                  "/debug/pprof",
+		LeaderElectionResourceLock: "/leases",
+		LeaseDuration:              3 * time.Second,
+		RenewDeadLine:              5 * time.Second,
+		RetryPeriod:                3 * time.Second,
+		EnableClusterGateway:       true,
+		EnableClusterMetrics:       true,
+		ClusterMetricsInterval:     5 * time.Second,
+	}
+
+	if !cmp.Equal(opt, expected, cmp.AllowUnexported(CoreOptions{})) {
+		t.Errorf("Flags() diff: %v", cmp.Diff(opt, expected, cmp.AllowUnexported(CoreOptions{})))
+	}
+}