arikawa/utils/bot/extras/arguments/flag_test.go

package arguments

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

func TestFlagSet(t *testing.T) {
	fs := NewFlagSet()

	var b bool
	fs.BoolVar(&b, "b", false, "Test boolean")

	if usage := fs.Usage(); !strings.Contains(usage, "Test boolean") {
		t.Fatal("Unexpected usage:", usage)
	}

	if err := fs.Parse([]string{"-b", "asdasd"}); err != nil {
		t.Fatal("Failed to parse:", err)
	}

	if !b {
		t.Fatal("Test boolean is false")
	}
}

func TestFlag(t *testing.T) {
	f := Flag{}

	if err := f.ParseContent([]string{"--now", "1m4s"}); err != nil {
		t.Fatal("Failed to parse:", err)
	}

	if args := f.Args(); !reflect.DeepEqual(args, []string{"--now", "1m4s"}) {
		t.Fatal("Unexpected arguments:", args)
	}

	fs := NewFlagSet()

	var now bool
	fs.BoolVar(&now, "now", false, "Now")

	if err := f.With(fs.FlagSet); err != nil {
		t.Fatal("Failed to parse:", err)
	}

	if !now {
		t.Fatal("now is false")
	}

	if arg := fs.FlagSet.Arg(0); arg != "1m4s" {
		t.Fatal("Unexpected argument:", arg)
	}
}