Add unit tests

Signed-off-by: Jian Qiu <[email protected]>

pkg/common/options/agent.go

@@ -18,6 +18,13 @@ import (
 	"open-cluster-management.io/ocm/pkg/registration/spoke/registration"
 )
 
+const (
+	// spokeAgentNameLength is the length of the spoke agent name which is generated automatically
+	spokeAgentNameLength = 5
+	// defaultSpokeComponentNamespace is the default namespace in which the spoke agent is deployed
+	defaultSpokeComponentNamespace = "open-cluster-management-agent"
+)
+
 // AgentOptions is the common agent options
 type AgentOptions struct {
 	CommoOpts           *Options

pkg/common/options/options.go

@@ -8,13 +8,6 @@ import (
 	"k8s.io/apimachinery/pkg/version"
 )
 
-const (
-	// spokeAgentNameLength is the length of the spoke agent name which is generated automatically
-	spokeAgentNameLength = 5
-	// defaultSpokeComponentNamespace is the default namespace in which the spoke agent is deployed
-	defaultSpokeComponentNamespace = "open-cluster-management-agent"
-)
-
 type Options struct {
 	CmdConfig *controllercmd.ControllerCommandConfig
 	Burst     int

pkg/common/options/options_test.go

@@ -2,6 +2,9 @@ package options
 
 import (
 	"context"
+	"github.com/openshift/library-go/pkg/controller/controllercmd"
+	"github.com/spf13/cobra"
+	"open-cluster-management.io/ocm/pkg/version"
 	"os"
 	"path"
 	"testing"
@@ -92,7 +95,7 @@ func TestComplete(t *testing.T) {
 	for _, c := range cases {
 		t.Run(c.name, func(t *testing.T) {
 			// setup kube client
-			objects := []runtime.Object{}
+			var objects []runtime.Object
 			if c.secret != nil {
 				objects = append(objects, c.secret)
 			}
@@ -219,3 +222,34 @@ func TestGetOrGenerateClusterAgentNames(t *testing.T) {
 		})
 	}
 }
+
+func TestNewOptions(t *testing.T) {
+	opts := NewOptions()
+	cmd := &cobra.Command{
+		Use:   "test",
+		Short: "test Controller",
+		Run: func(cmd *cobra.Command, args []string) {
+			_ = cmd.Help()
+			os.Exit(1)
+		},
+	}
+
+	opts.NewControllerCommandConfig("test", version.Get(), func(ctx context.Context, controllerCtx *controllercmd.ControllerContext) error {
+		return nil
+	})
+
+	opts.AddFlags(cmd.Flags())
+	if err := cmd.Flags().Set("kube-api-qps", "10"); err != nil {
+		t.Fatal(err)
+	}
+	if err := cmd.Flags().Set("kube-api-burst", "20"); err != nil {
+		t.Fatal(err)
+
+	}
+	if err := cmd.Flags().Set("disable-leader-election", "true"); err != nil {
+		t.Fatal(err)
+	}
+	if err := cmd.Flags().Set("unsupported-flag", "true"); err == nil {
+		t.Errorf("Should return err")
+	}
+}