milo: buildbucket builer view

appengine/cmd/milo/buildbucket/builder_test.go

+// Copyright 2016 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 buildbucket
+
+import (
+        "encoding/json"
+        "flag"
+        "fmt"
+        "io/ioutil"
+        "os"
+        "path/filepath"
+        "testing"
+        "time"
+
+        "github.com/luci/luci-go/common/clock/testclock"
+        "golang.org/x/net/context"
+
+        . "github.com/smartystreets/goconvey/convey"
+)
+
+var generate = flag.Bool("test.generate", false, "Generate expectations instead of running tests.")
+
+func TestBuilder(t *testing.T) {
+        t.Parallel()
+
+        testCases := []struct{ bucket, builder string }{
+                {"master.tryserver.infra", "InfraPresubmit"},
+                {"master.tryserver.infra", "InfraPresubmit(Swarming)"},
+        }
+
+        Convey("Builder", t, func() {
+                c := context.Background()
+                c, _ = testclock.UseTime(c, time.Date(2016, time.March, 14, 11, 0, 0, 0, time.UTC))
+
+                for _, tc := range testCases {
+                        tc := tc
+                        Convey(fmt.Sprintf("%s:%s", tc.bucket, tc.builder), func() {
+                                expectationFilePath := filepath.Join("expectations", tc.bucket, tc.builder+".json")
+                                err := os.MkdirAll(filepath.Dir(expectationFilePath), 0777)
+                                So(err, ShouldBeNil)
+
+                                actual, err := builderImpl(c, "debug", tc.bucket, tc.builder, 0)
+                                So(err, ShouldBeNil)
+                                actualJSON, err := json.MarshalIndent(actual, "", "  ")
+                                So(err, ShouldBeNil)
+
+                                if *generate {
+                                        err := ioutil.WriteFile(expectationFilePath, actualJSON, 0777)
+                                        So(err, ShouldBeNil)
+                                } else {
+                                        expectedJSON, err := ioutil.ReadFile(expectationFilePath)
+                                        So(err, ShouldBeNil)
+                                        So(string(actualJSON), ShouldEqual, string(expectedJSON))
+                                }
+                        })
+                }
+        })
+}