From ebb97fb7367fb983cffc1935a8fb57e4b80f5249 Mon Sep 17 00:00:00 2001 From: Dave Henderson Date: Thu, 25 Jan 2024 20:11:31 -0500 Subject: Move funcs package to internal (#1977) Signed-off-by: Dave Henderson --- internal/funcs/env_test.go | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 internal/funcs/env_test.go (limited to 'internal/funcs/env_test.go') diff --git a/internal/funcs/env_test.go b/internal/funcs/env_test.go new file mode 100644 index 00000000..a6ee43ac --- /dev/null +++ b/internal/funcs/env_test.go @@ -0,0 +1,37 @@ +package funcs + +import ( + "context" + "os" + "strconv" + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestCreateEnvFuncs(t *testing.T) { + t.Parallel() + + for i := 0; i < 10; i++ { + // Run this a bunch to catch race conditions + t.Run(strconv.Itoa(i), func(t *testing.T) { + t.Parallel() + + ctx := context.Background() + fmap := CreateEnvFuncs(ctx) + actual := fmap["env"].(func() interface{}) + + assert.Equal(t, ctx, actual().(*EnvFuncs).ctx) + }) + } +} + +func TestEnvGetenv(t *testing.T) { + t.Parallel() + + ef := &EnvFuncs{} + expected := os.Getenv("USER") + assert.Equal(t, expected, ef.Getenv("USER")) + + assert.Equal(t, "foo", ef.Getenv("bogusenvvar", "foo")) +} -- cgit v1.2.3