aboutsummaryrefslogtreecommitdiffstats
path: root/client/go/internal/admin/vespa-wrapper/logfmt/formatflags_test.go
blob: 53c47d2420811cc89f2b44bd73111888650c5e54 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package logfmt

import (
	"fmt"
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestOutputFormat(t *testing.T) {
	type args struct {
		val string
	}
	tests := []struct {
		expected OutputFormat
		arg      string
		wantErr  assert.ErrorAssertionFunc
	}{
		{FormatVespa, "vespa", assert.NoError},
		{FormatRaw, "raw", assert.NoError},
		{FormatJSON, "json", assert.NoError},
		{-1, "foo", assert.Error},
	}
	for _, tt := range tests {
		t.Run(tt.arg, func(t *testing.T) {
			var v OutputFormat = -1
			tt.wantErr(t, v.Set(tt.arg), fmt.Sprintf("Set(%v)", tt.arg))
			assert.Equal(t, v, tt.expected)
		})
	}
}