databricks-cli/libs/flags/output_test.go

38 lines
657 B
Go

package flags
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestOutputFlag(t *testing.T) {
var f Output
var err error
// Invalid
err = f.Set("foo")
assert.Error(t, err)
assert.ErrorContains(t, err, "accepted arguments are text and json")
// Lowercase
err = f.Set("text")
assert.NoError(t, err)
assert.Equal(t, "text", f.String())
// Uppercase
err = f.Set("TEXT")
assert.NoError(t, err)
assert.Equal(t, "text", f.String())
// Lowercase
err = f.Set("json")
assert.NoError(t, err)
assert.Equal(t, "json", f.String())
// Uppercase
err = f.Set("JSON")
assert.NoError(t, err)
assert.Equal(t, "json", f.String())
}