mirror of https://github.com/coder/coder.git
137 lines
2.9 KiB
Go
137 lines
2.9 KiB
Go
package cliui_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/cli/clibase"
|
|
"github.com/coder/coder/cli/cliui"
|
|
)
|
|
|
|
type format struct {
|
|
id string
|
|
attachOptionsFn func(opts *clibase.OptionSet)
|
|
formatFn func(ctx context.Context, data any) (string, error)
|
|
}
|
|
|
|
var _ cliui.OutputFormat = &format{}
|
|
|
|
func (f *format) ID() string {
|
|
return f.id
|
|
}
|
|
|
|
func (f *format) AttachOptions(opts *clibase.OptionSet) {
|
|
if f.attachOptionsFn != nil {
|
|
f.attachOptionsFn(opts)
|
|
}
|
|
}
|
|
|
|
func (f *format) Format(ctx context.Context, data any) (string, error) {
|
|
if f.formatFn != nil {
|
|
return f.formatFn(ctx, data)
|
|
}
|
|
|
|
return "", nil
|
|
}
|
|
|
|
func Test_OutputFormatter(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("RequiresTwoFormatters", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Panics(t, func() {
|
|
cliui.NewOutputFormatter()
|
|
})
|
|
require.Panics(t, func() {
|
|
cliui.NewOutputFormatter(cliui.JSONFormat())
|
|
})
|
|
})
|
|
|
|
t.Run("NoMissingFormatID", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Panics(t, func() {
|
|
cliui.NewOutputFormatter(
|
|
cliui.JSONFormat(),
|
|
&format{id: ""},
|
|
)
|
|
})
|
|
})
|
|
|
|
t.Run("NoDuplicateFormats", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Panics(t, func() {
|
|
cliui.NewOutputFormatter(
|
|
cliui.JSONFormat(),
|
|
cliui.JSONFormat(),
|
|
)
|
|
})
|
|
})
|
|
|
|
t.Run("OK", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var called int64
|
|
f := cliui.NewOutputFormatter(
|
|
cliui.JSONFormat(),
|
|
&format{
|
|
id: "foo",
|
|
attachOptionsFn: func(opts *clibase.OptionSet) {
|
|
opts.Add(clibase.Option{
|
|
Name: "foo",
|
|
Flag: "foo",
|
|
FlagShorthand: "f",
|
|
Value: clibase.DiscardValue,
|
|
Description: "foo flag 1234",
|
|
})
|
|
},
|
|
formatFn: func(_ context.Context, _ any) (string, error) {
|
|
atomic.AddInt64(&called, 1)
|
|
return "foo", nil
|
|
},
|
|
},
|
|
)
|
|
|
|
cmd := &clibase.Cmd{}
|
|
f.AttachOptions(&cmd.Options)
|
|
|
|
fs := cmd.Options.FlagSet()
|
|
|
|
selected, err := fs.GetString("output")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "json", selected)
|
|
usage := fs.FlagUsages()
|
|
require.Contains(t, usage, "Available formats: json, foo")
|
|
require.Contains(t, usage, "foo flag 1234")
|
|
|
|
ctx := context.Background()
|
|
data := []string{"hi", "dean", "was", "here"}
|
|
out, err := f.Format(ctx, data)
|
|
require.NoError(t, err)
|
|
|
|
var got []string
|
|
require.NoError(t, json.Unmarshal([]byte(out), &got))
|
|
require.Equal(t, data, got)
|
|
require.EqualValues(t, 0, atomic.LoadInt64(&called))
|
|
|
|
require.NoError(t, fs.Set("output", "foo"))
|
|
out, err = f.Format(ctx, data)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "foo", out)
|
|
require.EqualValues(t, 1, atomic.LoadInt64(&called))
|
|
|
|
require.NoError(t, fs.Set("output", "bar"))
|
|
out, err = f.Format(ctx, data)
|
|
require.Error(t, err)
|
|
require.ErrorContains(t, err, "bar")
|
|
require.Equal(t, "", out)
|
|
require.EqualValues(t, 1, atomic.LoadInt64(&called))
|
|
})
|
|
}
|