plugins/pkg/skel/skel_test.go

package skel

import (
	"os"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("Skel", func() {
	var (
		fNoop = func(_ *CmdArgs) error { return nil }
		// fErr    = func(_ *CmdArgs) error { return errors.New("dummy") }
		envVars = []struct {
			name string
			val  string
		}{
			{"CNI_CONTAINERID", "dummy"},
			{"CNI_NETNS", "dummy"},
			{"CNI_IFNAME", "dummy"},
			{"CNI_ARGS", "dummy"},
			{"CNI_PATH", "dummy"},
		}
	)

	It("Must be possible to set the env vars", func() {
		for _, v := range envVars {
			err := os.Setenv(v.name, v.val)
			Expect(err).NotTo(HaveOccurred())
		}
	})

	Context("When dummy environment variables are passed", func() {

		It("should not fail with ADD and noop callback", func() {
			err := os.Setenv("CNI_COMMAND", "ADD")
			Expect(err).NotTo(HaveOccurred())
			PluginMain(fNoop, nil)
		})

		// TODO: figure out howto mock printing and os.Exit()
		// It("should fail with ADD and error callback", func() {
		// 	err := os.Setenv("CNI_COMMAND", "ADD")
		// 	Expect(err).NotTo(HaveOccurred())
		// 	PluginMain(fErr, nil)
		// })

		It("should not fail with DEL and noop callback", func() {
			err := os.Setenv("CNI_COMMAND", "DEL")
			Expect(err).NotTo(HaveOccurred())
			PluginMain(nil, fNoop)
		})

		// TODO: figure out howto mock printing and os.Exit()
		// It("should fail with DEL and error callback", func() {
		// 	err := os.Setenv("CNI_COMMAND", "DEL")
		// 	Expect(err).NotTo(HaveOccurred())
		// 	PluginMain(fErr, nil)
		// })
	})
})