milo/appengine/buildbot/console_test.go

// Copyright 2017 The LUCI Authors. All rights reserved.
// Use of this source code is governed under the Apache License, Version 2.0
// that can be found in the LICENSE file.

package buildbot

import (
	"context"
	"strconv"
	"testing"

	"github.com/luci/gae/impl/memory"
	"github.com/luci/gae/service/datastore"
	"github.com/luci/luci-go/common/clock/testclock"
	"github.com/luci/luci-go/milo/api/resp"

	. "github.com/smartystreets/goconvey/convey"
)

func TestConsole(t *testing.T) {
	c := memory.UseWithAppID(context.Background(), "dev~luci-milo")
	c, _ = testclock.UseTime(c, testclock.TestTimeUTC)
	fakeTime := float64(123)
	fakeResult := int(0)

	// Seed a builder with 8 builds.
	for i := 1; i < 10; i++ {
		datastore.Put(c, &buildbotBuild{
			Master:      "fake",
			Buildername: "fake",
			Number:      i,
			Internal:    false,
			Times:       []*float64{&fakeTime, &fakeTime},
			Sourcestamp: &buildbotSourceStamp{
				Revision: strconv.Itoa(i),
			},
			Results:  &fakeResult,
			Finished: true,
		})
	}
	putDSMasterJSON(c, &buildbotMaster{
		Name:     "fake",
		Builders: map[string]*buildbotBuilder{"fake": {}},
	}, false)
	datastore.GetTestable(c).Consistent(true)
	datastore.GetTestable(c).AutoIndex(true)
	datastore.GetTestable(c).CatchupIndexes()

	Convey(`Console tests for buildbot`, t, func() {
		Convey(`Empty request`, func() {
			cb, err := GetConsoleBuilds(c, []resp.BuilderRef{}, []string{})
			So(err, ShouldBeNil)
			So(cb, ShouldResemble, [][]*resp.ConsoleBuild{})
		})
		Convey(`Basic`, func() {
			ref := []resp.BuilderRef{
				{
					Module:    "buildbot",
					Name:      "fake/fake",
					Category:  []string{"np"},
					ShortName: "np",
				},
			}
			cb, err := GetConsoleBuilds(c, ref, []string{"2", "3", "5"})
			So(err, ShouldBeNil)
			So(len(cb), ShouldEqual, 3)
			So(len(cb[0]), ShouldEqual, 1)
		})
	})
}