sablier/pkg/sabliercmd/cmd_test.go

package sabliercmd_test

import (
	"bufio"
	"bytes"
	"encoding/json"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/sablierapp/sablier/pkg/sabliercmd"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"github.com/stretchr/testify/require"
	"gotest.tools/v3/assert"
)

func TestDefault(t *testing.T) {
	testDir, err := os.Getwd()
	require.NoError(t, err, "error getting the current working directory")

	wantConfig, err := os.ReadFile(filepath.Join(testDir, "testdata", "config_default.json"))
	require.NoError(t, err, "error reading test config file")

	// CHANGE `startCmd` behavior to only print the config, this is for testing purposes only
	sabliercmd.SetStartCommand(mockStartCommand)
	defer sabliercmd.ResetStartCommand()

	t.Run("config file", func(t *testing.T) {
		sabliercmd.ResetConfig()
		cmd := sabliercmd.NewRootCommand()
		output := &bytes.Buffer{}
		cmd.SetOut(output)
		cmd.SetArgs([]string{
			"start",
		})
		_ = cmd.Execute()

		gotOutput := output.String()

		assert.Equal(t, string(wantConfig), gotOutput)
	})
}

func TestPrecedence(t *testing.T) {
	testDir, err := os.Getwd()
	require.NoError(t, err, "error getting the current working directory")

	// CHANGE `startCmd` behavior to only print the config, this is for testing purposes only
	sabliercmd.SetStartCommand(mockStartCommand)
	defer sabliercmd.ResetStartCommand()

	t.Run("config file", func(t *testing.T) {
		wantConfig, err := os.ReadFile(filepath.Join(testDir, "testdata", "config_yaml_wanted.json"))
		require.NoError(t, err, "error reading test config file")

		sabliercmd.ResetConfig()
		cmd := sabliercmd.NewRootCommand()
		output := &bytes.Buffer{}
		cmd.SetOut(output)
		cmd.SetArgs([]string{
			"--configFile", filepath.Join(testDir, "testdata", "config.yml"),
			"start",
		})
		_ = cmd.Execute()

		gotOutput := output.String()

		assert.Equal(t, string(wantConfig), gotOutput)
	})

	t.Run("env var", func(t *testing.T) {
		setEnvsFromFile(filepath.Join(testDir, "testdata", "config.env"))
		defer unsetEnvsFromFile(filepath.Join(testDir, "testdata", "config.env"))

		wantConfig, err := os.ReadFile(filepath.Join(testDir, "testdata", "config_env_wanted.json"))
		require.NoError(t, err, "error reading test config file")

		sabliercmd.ResetConfig()
		cmd := sabliercmd.NewRootCommand()
		output := &bytes.Buffer{}
		cmd.SetOut(output)
		cmd.SetArgs([]string{
			"--configFile", filepath.Join(testDir, "testdata", "config.yml"),
			"start",
		})
		_ = cmd.Execute()

		gotOutput := output.String()

		assert.Equal(t, string(wantConfig), gotOutput)
	})

	t.Run("flag", func(t *testing.T) {
		setEnvsFromFile(filepath.Join(testDir, "testdata", "config.env"))
		defer unsetEnvsFromFile(filepath.Join(testDir, "testdata", "config.env"))

		wantConfig, err := os.ReadFile(filepath.Join(testDir, "testdata", "config_cli_wanted.json"))
		require.NoError(t, err, "error reading test config file")

		cmd := sabliercmd.NewRootCommand()
		output := &bytes.Buffer{}
		sabliercmd.ResetConfig()
		cmd.SetOut(output)
		cmd.SetArgs([]string{
			"--configFile", filepath.Join(testDir, "testdata", "config.yml"),
			"start",
			"--provider.name", "cli",
			"--provider.kubernetes.qps", "256",
			"--provider.kubernetes.burst", "512",
			"--provider.kubernetes.delimiter", "_",
			"--provider.podman.uri", "unix:///run/podman/podman.sock.cli",
			"--server.port", "3333",
			"--server.base-path", "/cli/",
			"--storage.file", "/tmp/cli.json",
			"--sessions.default-duration", "3h",
			"--sessions.expiration-interval", "3h",
			"--logging.level", "info",
			"--strategy.dynamic.custom-themes-path", "/tmp/cli/themes",
			// Must use `=` see https://github.com/spf13/cobra/issues/613
			"--strategy.dynamic.show-details-by-default=false",
			"--strategy.dynamic.default-theme", "cli",
			"--strategy.dynamic.default-refresh-frequency", "3h",
			"--strategy.blocking.default-timeout", "3h",
			"--strategy.blocking.default-refresh-frequency", "3h",
		})
		_ = cmd.Execute()

		gotOutput := output.String()

		assert.Equal(t, string(wantConfig), gotOutput)
	})
}

func setEnvsFromFile(path string) {
	readFile, err := os.Open(path)
	if err != nil {
		panic(err)
	}

	defer func() { _ = readFile.Close() }()

	fileScanner := bufio.NewScanner(readFile)

	fileScanner.Split(bufio.ScanLines)

	for fileScanner.Scan() {
		split := strings.Split(fileScanner.Text(), "=")
		_ = os.Setenv(split[0], split[1])
	}
}

func unsetEnvsFromFile(path string) {
	readFile, err := os.Open(path)

	if err != nil {
		panic(err)
	}

	defer func() { _ = readFile.Close() }()

	fileScanner := bufio.NewScanner(readFile)

	fileScanner.Split(bufio.ScanLines)

	for fileScanner.Scan() {
		split := strings.Split(fileScanner.Text(), "=")
		_ = os.Unsetenv(split[0])
	}
}

func mockStartCommand() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "start",
		Short: "InstanceStart the Sablier server",
		Run: func(cmd *cobra.Command, args []string) {
			conf := sabliercmd.GetConfig()
			_ = viper.Unmarshal(conf)

			out := cmd.OutOrStdout()

			encoder := json.NewEncoder(out)

			encoder.SetIndent("", "  ")
			_ = encoder.Encode(conf)
		},
	}
	return cmd
}