databricks-cli/cmd/root/progress_logger_test.go

70 lines
2.1 KiB
Go
Raw Normal View History

2023-03-29 12:58:09 +00:00
package root
import (
"context"
"testing"
"github.com/databricks/cli/libs/cmdio"
"github.com/databricks/cli/libs/flags"
"github.com/spf13/cobra"
2023-03-29 12:58:09 +00:00
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
type progressLoggerTest struct {
*cobra.Command
*logFlags
*progressLoggerFlag
}
func initializeProgressLoggerTest(t *testing.T) (
*progressLoggerTest,
*flags.LogLevelFlag,
*flags.LogFileFlag,
*flags.ProgressLogFormat,
) {
plt := &progressLoggerTest{
Command: &cobra.Command{},
}
plt.logFlags = initLogFlags(plt.Command)
plt.progressLoggerFlag = initProgressLoggerFlag(plt.Command, plt.logFlags)
return plt, &plt.logFlags.level, &plt.logFlags.file, &plt.progressLoggerFlag.ProgressLogFormat
}
2023-03-29 12:58:09 +00:00
func TestInitializeErrorOnIncompatibleConfig(t *testing.T) {
plt, logLevel, logFile, progressFormat := initializeProgressLoggerTest(t)
2023-03-29 12:58:09 +00:00
logLevel.Set("info")
logFile.Set("stderr")
progressFormat.Set("inplace")
_, err := plt.progressLoggerFlag.initializeContext(context.Background())
2023-03-29 12:58:09 +00:00
assert.ErrorContains(t, err, "inplace progress logging cannot be used when log-file is stderr")
}
func TestNoErrorOnDisabledLogLevel(t *testing.T) {
plt, logLevel, logFile, progressFormat := initializeProgressLoggerTest(t)
2023-03-29 12:58:09 +00:00
logLevel.Set("disabled")
logFile.Set("stderr")
progressFormat.Set("inplace")
_, err := plt.progressLoggerFlag.initializeContext(context.Background())
2023-03-29 12:58:09 +00:00
assert.NoError(t, err)
}
func TestNoErrorOnNonStderrLogFile(t *testing.T) {
plt, logLevel, logFile, progressFormat := initializeProgressLoggerTest(t)
2023-03-29 12:58:09 +00:00
logLevel.Set("info")
logFile.Set("stdout")
progressFormat.Set("inplace")
_, err := plt.progressLoggerFlag.initializeContext(context.Background())
2023-03-29 12:58:09 +00:00
assert.NoError(t, err)
}
func TestDefaultLoggerModeResolution(t *testing.T) {
plt, _, _, progressFormat := initializeProgressLoggerTest(t)
require.Equal(t, *progressFormat, flags.ModeDefault)
ctx, err := plt.progressLoggerFlag.initializeContext(context.Background())
2023-03-29 12:58:09 +00:00
require.NoError(t, err)
logger, ok := cmdio.FromContext(ctx)
2023-03-29 12:58:09 +00:00
assert.True(t, ok)
assert.Equal(t, logger.Mode, flags.ModeAppend)
}