package args

import (
	"os"
	"testing"
	"time"
)

func TestNewArgSet(t *testing.T) {
	NewArgSet()
}

func TestAppend(t *testing.T) {
	as := NewArgSet()
	as.Append(INT, "key", "help", nil)
	if len(as.args) != 1 || *as.args[0] != *NewArg(INT, "key", "help", nil) {
		t.Errorf("argset.Append failed: got %v, want %v", *as.args[0], *NewArg(INT, "key", "help", nil))
	}
}

func TestParse(t *testing.T) {
	osArgs := append([]string{}, os.Args[:]...)
	defer func() {
		os.Args = osArgs
	}()
	os.Args = []string{"nothing", "-flag", "5"}
	os.Setenv("ENV", "9")
	as := NewArgSet()
	as.Append(INT, "flag", "help", 1)
	as.Append(INT, "env", "help", 2)
	as.Append(INT, "default", "help", 3)
	if err := as.Parse(); err != nil {
		t.Fatalf("cannot parse legal: %v", err)
	} else if arg := as.Get("flag"); arg == nil || arg.Value != 5 {
		t.Errorf("cannot set from flag: got %v", arg)
	} else if arg := as.Get("env"); arg == nil || arg.Value != 9 {
		t.Errorf("cannot set from env: got %v", arg)
	} else if arg := as.Get("default"); arg == nil || arg.Value != 3 {
		t.Errorf("cannot set from default: got %v", arg)
	}

	os.Args = []string{"nothing", "-testkey5", "bad"}
	as = NewArgSet()
	as.Append(INT, "testkey4", "help", "4")
	if err := as.Parse(); err == nil {
		t.Fatal(err)
	}
}

func TestParseFloatTime(t *testing.T) {
	osArgs := append([]string{}, os.Args[:]...)
	defer func() {
		os.Args = osArgs
	}()
	os.Args = []string{"nothing", "-flag", "5.5"}
	os.Setenv("ENV", "2019-01-01")
	as := NewArgSet()
	as.Append(FLOAT, "flag", "help", 1.1)
	as.Append(TIME, "env", "help", "2019-03-22")
	wantTime := time.Date(2019, 1, 1, 0, 0, 0, 0, time.UTC)
	if err := as.Parse(); err != nil {
		t.Fatalf("cannot parse legal: %v", err)
	} else if arg := as.Get("flag"); arg == nil || arg.Value != float32(5.5) {
		t.Errorf("cannot set from flag: got %v (%v / %T != %v / %T)", arg, arg.Value, arg.Value, 5.5, 5.5)
	} else if arg := as.Get("env"); arg == nil || arg.Value != wantTime {
		t.Errorf("cannot set from env: got %v (%v / %T != %v / %T)", arg, arg.Value, arg.Value, wantTime, wantTime)
	}
}

func TestGet(t *testing.T) {
	as := NewArgSet()
	as.Append(INT, "testkey1", "help", 1)
	as.Append(INT, "testkey2", "help", 2)
	as.Append(INT, "testkey3", "help", 3)
	if err := as.Parse(); err != nil {
		t.Fatal(err)
	}
	if as.Get("testkey1") == nil {
		t.Errorf("cannot get parsed at %q", "testkey1")
	}
}