one more test

elastic · Jun 20, 2024 · 2394028 · 2394028
1 parent 0769798
commit 2394028
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/internal/pkg/agent/cmd/cmd_test.go b/internal/pkg/agent/cmd/cmd_test.go
@@ -6,6 +6,9 @@ package cmd
 
 import (
 	"testing"
+
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/require"
 )
 
 func TestAgent(t *testing.T) {
@@ -30,3 +33,18 @@ func TestAgent(t *testing.T) {
 	// 	assert.True(t, strings.Contains(string(contents), "Hello I am running"))
 	// })
 }
+
+func TestAddCommandIfNotNil(t *testing.T) {
+	cmd := &cobra.Command{}
+
+	parent := &cobra.Command{}
+	addCommandIfNotNil(parent, cmd)
+	require.Equal(t, 1, len(parent.Commands()))
+
+	parent = &cobra.Command{}
+	addCommandIfNotNil(parent, nil)
+	require.Equal(t, 0, len(parent.Commands()))
+
+	// this should not panic
+	addCommandIfNotNil(nil, cmd)
+}