test: add test for options env

options_test.go

@@ -1,40 +1,90 @@
 package envbuilder_test
 
 import (
+	"bytes"
 	"testing"
 
 	"github.com/coder/envbuilder"
 	"github.com/coder/serpent"
 	"github.com/stretchr/testify/require"
 )
 
-type testEnvCase struct {
-	Env      string
-	EnvValue string
-	Expected interface{}
-	Actual   func() interface{}
-}
+// TestEnvOptionParsing tests that given environment variables of different types are handled as expected.
+func TestTestEnvOptionParsing(t *testing.T) {
+	t.Run("string", func(t *testing.T) {
+		t.Setenv("SETUP_SCRIPT", "setup.sh")
+		t.Setenv("INIT_COMMAND", "sleep infinity")
+
+		var o envbuilder.Options
+		err := runCLI(&o)
+		require.NoError(t, err)
+
+		require.Equal(t, o.SetupScript, "setup.sh")
+		require.Equal(t, o.InitCommand, "sleep infinity")
+	})
+
+	t.Run("int", func(t *testing.T) {
+		t.Setenv("CACHE_TTL_DAYS", "7")
+
+		var o envbuilder.Options
+		err := runCLI(&o)
+		require.NoError(t, err)
+
+		require.Equal(t, o.CacheTTLDays, int64(7))
+	})
+
+	t.Run("string array", func(t *testing.T) {
+		t.Setenv("IGNORE_PATHS", "/var,/temp")
+
+		var o envbuilder.Options
+		err := runCLI(&o)
+		require.NoError(t, err)
 
-func TestEnvOptions(t *testing.T) {
-	t.Setenv("SETUP_SCRIPT", "setup.sh")
-	t.Setenv("CACHE_TTL_DAYS", "7")
-	t.Setenv("IGNORE_PATHS", "/var,/tmp")
-	t.Setenv("SKIP_REBUILD", "true")
-	t.Setenv("GIT_CLONE_SINGLE_BRANCH", "")
+		require.Equal(t, o.IgnorePaths, []string{"/var", "/temp"})
+	})
 
-	var o = envbuilder.Options{}
+	t.Run("bool", func(t *testing.T) {
+		t.Setenv("SKIP_REBUILD", "true")
+		t.Setenv("GIT_CLONE_SINGLE_BRANCH", "false")
+		t.Setenv("EXIT_ON_BUILD_FAILURE", "true")
+		t.Setenv("FORCE_SAFE", "false")
+		t.Setenv("INSECURE", "true")
+
+		var o envbuilder.Options
+		err := runCLI(&o)
+		require.NoError(t, err)
+
+		require.True(t, o.SkipRebuild)
+		require.False(t, o.GitCloneSingleBranch)
+		require.True(t, o.ExitOnBuildFailure)
+		require.False(t, o.ForceSafe)
+		require.True(t, o.Insecure)
+	})
+}
+
+func runCLI(o *envbuilder.Options) error {
 	cmd := serpent.Command{
 		Options: o.CLI(),
 		Handler: func(inv *serpent.Invocation) error {
 			return nil
 		},
 	}
-	err := cmd.Invoke().WithOS().Run()
-	require.NoError(t, err)
-
-	require.Equal(t, o.SetupScript, "setup.sh")
-	require.Equal(t, o.CacheTTLDays, int64(7))
-	require.Equal(t, o.IgnorePaths, []string{"/var", "/tmp"})
-	require.True(t, o.SkipRebuild)
-	require.False(t, o.GitCloneSingleBranch)
+	i := cmd.Invoke().WithOS()
+	fakeIO(i)
+	err := i.Run()
+	return err
+}
+
+type ioBufs struct {
+	Stdin  bytes.Buffer
+	Stdout bytes.Buffer
+	Stderr bytes.Buffer
+}
+
+func fakeIO(i *serpent.Invocation) *ioBufs {
+	var b ioBufs
+	i.Stdout = &b.Stdout
+	i.Stderr = &b.Stderr
+	i.Stdin = &b.Stdin
+	return &b
 }