flags_test.go

package cmd

import (
	"reflect"
	"strings"
	"testing"
	"time"
)

func TestUsage(t *testing.T) {
	f := newFlags()
	f.Flag("-x", new(bool), "")
	f.Flag("-y", new(bool), "")
	got := f.usage()
	want := "[OPTION]..."
	if got != want {
		t.Errorf("usage returned %v, want %v", got, want)
	}
}

func TestParse(t *testing.T) {
	var (
		size     int
		timeout  time.Duration
		v        bool
		percent  float64
		count    int
		distance int
		name     string
	)
	f := newFlags()
	f.Bytes("-m --max-size", &size, "SIZE", "")
	f.Duration("-timeout", &timeout, "D", "")
	f.Flag("-v", &v, "")
	f.Float("--percent", &percent, "P", "")
	f.Int("--count", &count, "N", "")
	f.Metric("-d", &distance, "DISTANCE", "")
	f.String("-name", &name, "NAME", "")

	args := "-m 2k -timeout 5m -v --percent 99.5 --count 7 -d 150G -name moon"
	f.parse(strings.Split(args, " "))
	wantSize := 2048
	wantTimeout := 5 * time.Minute
	wantV := true
	wantPercent := 99.5
	wantCount := 7
	wantDistance := 150 * 1000000000
	wantName := "moon"
	if size != wantSize {
		t.Errorf("parse set size = %v, want %v", size, wantSize)
	}
	if timeout != wantTimeout {
		t.Errorf("parse set timeout = %v, want %v", timeout, wantTimeout)
	}
	if v != wantV {
		t.Errorf("parse set v = %v, want %v", v, wantV)
	}
	if percent != wantPercent {
		t.Errorf("parse set percent = %v, want %v", percent, wantPercent)
	}
	if count != wantCount {
		t.Errorf("parse set count = %v, want %v", count, wantCount)
	}
	if distance != wantDistance {
		t.Errorf("parse set distance = %v, want %v", distance, wantDistance)
	}
	if name != wantName {
		t.Errorf("parse set name = %v, want %v", name, wantName)
	}
}

func TestSplitSpec(t *testing.T) {
	cases := []struct {
		spec      string
		want      []string
		wantError bool
	}{
		{"-v", []string{"-v"}, false},
		{"-v --verbose -d --debug", []string{"-v", "--verbose", "-d", "--debug"}, false},
		{"", nil, true},
		{"---verbose", nil, true},
		{"hello", nil, true},
	}
	for _, c := range cases {
		got, err := splitSpec(c.spec)
		if err != nil && !c.wantError {
			t.Errorf("splitSpec(%v) returned error", c.spec)
			continue
		}
		if err == nil && c.wantError {
			t.Errorf("splitSpec(%v) didn't return error", c.spec)
			continue
		}
		if !reflect.DeepEqual(got, c.want) {
			t.Errorf("splitSpec(%v) returned %v, want %v", c.spec, got, c.want)
		}
	}
}