main_test.go

package main

import (
	"bytes"
	"os"
	"testing"

	"github.com/fatih/color"
	. "github.com/jlandowner/helm-chartsnap/pkg/snap/gomega"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

func TestMain(t *testing.T) {
	RegisterFailHandler(Fail)
	RunSpecs(t, "Main Suite")
}

var _ = Describe("rootCmd", func() {
	BeforeEach(func() {
		initRootCmd()
	})
	Context("success", func() {
		Context("snapshot local chart with single values file", func() {
			It("should pass", func() {
				var output bytes.Buffer
				color.Output = &output
				DeferCleanup(func() {
					color.Output = os.Stdout
				})
				rootCmd.SetArgs([]string{"--chart", "example/app1", "-f", "example/app1/test_latest/test_ingress_enabled.yaml", "--namespace", "default"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
				Ω(output.String()).To(MatchSnapShot())
			})
		})

		Context("snapshot local chart with values directory", func() {
			It("should pass", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app1", "-f", "example/app1/test_latest/", "--namespace", "default"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("snapshot OCI chart", func() {
			It("should pass", func() {
				rootCmd.SetArgs([]string{"--chart", "oci://ghcr.io/nginxinc/charts/nginx-gateway-fabric", "-f", "example/remote/nginx-gateway-fabric.values.yaml", "--", "--namespace", "nginx-gateway"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("snapshot remote chart 1", func() {
			It("should pass", func() {
				rootCmd.SetArgs([]string{"--chart", "cilium", "-f", "example/remote/cilium.values.yaml", "--", "--namespace", "kube-system", "--repo", "https://helm.cilium.io"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("snapshot remote chart 2", func() {
			It("should pass", func() {
				rootCmd.SetArgs([]string{"--chart", "ingress-nginx", "-f", "example/remote/ingress-nginx.values.yaml", "--", "--namespace", "ingress-nginx", "--repo", "https://kubernetes.github.io/ingress-nginx", "--skip-tests"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("snapshot empty chart", func() {
			It("should pass", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app2", "--namespace", "default"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("snapshot helm error", func() {
			It("should pass", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app3", "--namespace", "default"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("snapshot empty chart with no config file", func() {
			It("should pass", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app2", "--namespace", "default", "--config-file", "notfound"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("snapshot empty chart with debug mode", func() {
			It("should pass", func() {
				os.Setenv("HELM_DEBUG", "true")
				defer os.Unsetenv("HELM_DEBUG")
				rootCmd.SetArgs([]string{"--chart", "example/app2", "--namespace", "default"})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
			})
		})

		Context("env FORCE_COLOR is enabled", func() {
			It("should force a colorized output", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app1", "-f", "example/app1/test_latest/test_ingress_enabled.yaml", "--namespace", "default"})
				var output bytes.Buffer
				color.Output = &output
				os.Setenv("FORCE_COLOR", "1")
				DeferCleanup(func() {
					color.Output = os.Stdout
					color.NoColor = true
					os.Unsetenv("FORCE_COLOR")
				})
				err := rootCmd.Execute()
				Expect(err).ShouldNot(HaveOccurred())
				Ω(output.String()).To(MatchSnapShot())
			})
		})
	})

	Context("fail", func() {
		Context("including dynamic outputs", func() {
			It("should fail", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app1", "--namespace", "default"})
				err := rootCmd.Execute()
				Expect(err).To(HaveOccurred())
				Ω(err.Error()).To(MatchSnapShot())
			})
		})

		Context("snapshot is different", func() {
			It("should fail", func() {
				var output bytes.Buffer
				color.Output = &output
				DeferCleanup(func() {
					color.Output = os.Stdout
				})
				rootCmd.SetArgs([]string{"--chart", "example/app1", "--namespace", "default", "-f", "example/app1/testfail/test_ingress_enabled.yaml"})
				err := rootCmd.Execute()
				Expect(err).To(HaveOccurred())
				Ω(err.Error()).To(MatchSnapShot())
				Ω(output.String()).To(MatchSnapShot())
			})
		})

		Context("values directory contains not matched snapshots", func() {
			It("should fail", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app1", "--namespace", "default", "-f", "example/app1/testfail"})
				err := rootCmd.Execute()
				Expect(err).To(HaveOccurred())
				Ω(err.Error()).To(MatchSnapShot())
			})
		})

		Context("values file not found", func() {
			It("should fail", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app1", "-f", "example/app1/test_latest/notfound.yaml", "--namespace", "default"})
				err := rootCmd.Execute()
				Expect(err).To(HaveOccurred())
				Ω(err.Error()).To(MatchSnapShot())
			})
		})

		Context("snapshot helm error with --fail-helm-error", func() {
			It("should fail", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app3", "--namespace", "default", "--fail-helm-error"})
				err := rootCmd.Execute()
				Expect(err).To(HaveOccurred())
				Ω(err.Error()).To(MatchSnapShot())
			})
		})

		Context("required flag is not set", func() {
			It("should fail", func() {
				rootCmd.SetArgs([]string{})
				err := rootCmd.Execute()
				Expect(err).To(HaveOccurred())
				Ω(err.Error()).To(MatchSnapShot())
			})
		})

		Context("invalid flag", func() {
			It("should fail", func() {
				rootCmd.SetArgs([]string{"--chart", "example/app1", "-f", "example/app1/test_latest/test_ingress_enabled.yaml", "--namespace", "default", "--invalid"})
				err := rootCmd.Execute()
				Expect(err).To(HaveOccurred())
				Ω(err.Error()).To(MatchSnapShot())
			})
		})
	})

	Context("--help", func() {
		It("should show help", func() {
			rootCmd.SetArgs([]string{"--help"})
			help := rootCmd.UsageString()
			Ω(help).To(MatchSnapShot())
		})
	})
})