コード例 #1
0
ファイル: main.go プロジェクト: jtacoma/daycount
func getQuery() *views.Query {
	var err error
	var command views.Query
	flag.Parse()
	command.Count = *count
	command.Template = *template
	command.Start, err = days.Parse(*start)
	command.Out = os.Stdout
	if err != nil {
		fmt.Printf("error: failed to parse: %v\n", *start)
		os.Exit(1)
	}
	command.View = views.Resolve(*format)
	if command.View == nil {
		fmt.Printf("error: unsupported format: %v\n", *format)
		os.Exit(1)
	}
	return &command
}
コード例 #2
0
ファイル: views_test.go プロジェクト: jtacoma/daycount
func TestRange(t *testing.T) {
	q := new(Query)
	q.Start, _ = days.Parse("2006-01-02")
	q.Count = 3
	r := q.Range()
	if len(r) != 4 {
		t.Fatalf("expected 4 results, got %v", len(r))
	}
	if r[0] != q.Start {
		t.Fatal("expected %v got %v", q.Start, r[0])
	}
	if r[1] != q.Start.Add(1) {
		t.Fatal("expected %v got %v", q.Start.Add(1), r[1])
	}
	if r[2] != q.Start.Add(2) {
		t.Fatal("expected %v got %v", q.Start.Add(2), r[2])
	}
	if r[3] != q.Start.Add(3) {
		t.Fatal("expected %v got %v", q.Start.Add(3), r[3])
	}
}