test: Add cmd tests (#619)

Author: bakito

.github/workflows/go.yml

@@ -22,7 +22,7 @@ jobs:
           go-version-file: "go.mod"
 
       - name: Lint
-        run: make lint
+        run: make lint-ci
 
   test:
     name: test

Makefile

@@ -18,7 +18,6 @@ test-ci:
 	@sed -i '/pkg\/mocks/d'              coverage.out
 	@sed -i '/pkg\/export\/archive.go/d' coverage.out
 	@sed -i '/pkg\/export\/export.go/d'  coverage.out
-	@sed -i '/cmd/d'                     coverage.out
 	@sed -i '/uor/d'                     coverage.out
 	@sed -i '/log/d'                     coverage.out
 	go tool cover -func coverage.out

cmd/cmd_suite_test.go

@@ -0,0 +1,13 @@
+package cmd_test
+
+import (
+	"testing"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+)
+
+func TestTypes(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "cmd Suite")
+}

cmd/decrypt_test.go

@@ -0,0 +1,60 @@
+package cmd
+
+import (
+	"os"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	"github.com/spf13/cobra"
+)
+
+var _ = Describe("Decrypt Command", func() {
+	var testFile string
+	var testContent string
+
+	BeforeEach(func() {
+		// Create a temporary test file
+		testFile = "temp-secret.yaml"
+
+		// Create test content with encrypted data
+		testContent = `apiVersion: v1
+kind: Secret
+metadata:
+  name: test-secret
+  namespace: default
+type: Opaque
+data:
+  username: KUBEXPORTER_AES@f6MMcnyTt4Tm4zGotyhAzPLjWSeV42ke8hu93AJG251W4Ew17RI6pA==
+  password: KUBEXPORTER_AES@f6MMcnyTt4Tm4zGosDdI2vK9CDmU2W0eNvcVMbvee0S/vlFO6+Vf+w==
+stringData:
+  api-key: KUBEXPORTER_AES@f6MMcnyTt4Tm4zGooBVt0vT6QS6H3RFI6AR8sM7D8/ElvOzygKsuM3ZT5nCfmg==
+  token: KUBEXPORTER_AES@f6MMcnyTt4Tm4zGovgkj0/TtHiqDmUhM5hg/X5Nsgw07ZtUsFnYpG4OnGg==`
+
+		err := os.WriteFile(testFile, []byte(testContent), 0o644)
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	AfterEach(func() {
+		os.Remove(testFile)
+	})
+
+	It("should decrypt the file successfully", func() {
+		cmd := &cobra.Command{}
+		cmd.AddCommand(decrypt)
+		cmd.SetArgs([]string{"decrypt", testFile, "--aes-key", "1234567890123456"})
+
+		err := cmd.Execute()
+		Expect(err).NotTo(HaveOccurred())
+
+		// Verify the file was decrypted
+		content, err := os.ReadFile(testFile)
+		Expect(err).NotTo(HaveOccurred())
+
+		// Check that encrypted fields were decrypted
+		Expect(string(content)).To(ContainSubstring("dXNlcm5hbWU=")) // base64 encoded "username"
+		Expect(string(content)).To(ContainSubstring("cGFzc3dvcmQ=")) // base64 encoded "password"
+		Expect(string(content)).To(ContainSubstring("secret-api-key-123"))
+		Expect(string(content)).To(ContainSubstring("my-secret-token"))
+		Expect(string(content)).NotTo(ContainSubstring("KUBEXPORTER_AES@"))
+	})
+})

cmd/encrypt_test.go

@@ -0,0 +1,60 @@
+package cmd
+
+import (
+	"os"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	"github.com/spf13/cobra"
+)
+
+var _ = Describe("Encrypt Command", func() {
+	var testFile string
+
+	BeforeEach(func() {
+		testFile = "temp-secret.yaml"
+		testContent := `apiVersion: v1
+kind: Secret
+metadata:
+  name: test-secret 
+  namespace: default
+type: Opaque
+data:
+  username: dXNlcm5hbWU=
+  password: cGFzc3dvcmQ=
+stringData:
+  api-key: "secret-api-key-123"
+  token: "my-secret-token"`
+
+		err := os.WriteFile(testFile, []byte(testContent), 0o644)
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	AfterEach(func() {
+		os.Remove(testFile)
+	})
+
+	It("should encrypt and decrypt a file", func() {
+		cmd := &cobra.Command{}
+		cmd.AddCommand(encrypt)
+		cmd.AddCommand(decrypt)
+		cmd.SetArgs([]string{"encrypt", testFile, "--aes-key", "1234567890123456"})
+
+		err := cmd.Execute()
+		Expect(err).NotTo(HaveOccurred())
+
+		content, err := os.ReadFile(testFile)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(content)).To(ContainSubstring("KUBEXPORTER_AES@"))
+
+		cmd.SetArgs([]string{"decrypt", testFile, "--aes-key", "1234567890123456"})
+		err = cmd.Execute()
+		Expect(err).NotTo(HaveOccurred())
+
+		content, err = os.ReadFile(testFile)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(content)).To(ContainSubstring("apiVersion: v1"))
+		Expect(string(content)).To(ContainSubstring("kind: Secret"))
+		Expect(string(content)).To(ContainSubstring("secret-api-key-123"))
+	})
+})

go.mod

@@ -16,7 +16,6 @@ require (
 	github.com/onsi/gomega v1.38.0
 	github.com/spf13/cobra v1.9.1
 	github.com/spf13/pflag v1.0.7
-	github.com/stretchr/testify v1.10.0
 	github.com/vardius/worker-pool/v2 v2.1.0
 	github.com/vbauerster/mpb/v8 v8.10.2
 	go.uber.org/mock v0.5.2
@@ -101,7 +100,6 @@ require (
 	github.com/peterbourgon/diskv v2.0.1+incompatible // indirect
 	github.com/philhofer/fwd v1.2.0 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
-	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/rs/xid v1.6.0 // indirect
 	github.com/russross/blackfriday/v2 v2.1.0 // indirect