Buildbucket: Save buildbucket build info and summary on pubsub push

milo/buildsource/swarming/build_test.go

 // Copyright 2015 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 swarming
 
 import (
         "encoding/json"
         "flag"
         "fmt"
@@ skipping 41 matching lines @@
                 // This is one hour after the start timestamp in the sample test data.
                 c, _ = testclock.UseTime(c, time.Date(2016, time.March, 14, 11, 0, 0, 0, time.UTC))
                 c = memory.UseWithAppID(c, "dev~luci-milo")
                 c = testconfig.WithCommonClient(c, memcfg.New(aclConfgs))
                 c = auth.WithState(c, &authtest.FakeState{
                         Identity:       "user:alicebob@google.com",
                         IdentityGroups: []string{"all", "googlers"},
                 })
 
                 for _, tc := range getTestCases(".") {
-                        bl := buildLoader{
+                        bl := BuildLoader{
                                 logDogClientFunc: logDogClientFunc(tc),
                         }
                         svc := debugSwarmingService{tc}
 
                         fmt.Printf("Generating expectations for %s\n", tc)
 
-                        build, err := bl.swarmingBuildImpl(c, svc, "foo", tc.name)
+                        build, err := bl.SwarmingBuildImpl(c, svc, "foo", tc.name)
                         if err != nil {
                                 panic(fmt.Errorf("Could not run swarmingBuildImpl for %s: %s", tc, err))
                         }
                         buildJSON, err := json.MarshalIndent(build, "", "  ")
                         if err != nil {
                                 panic(fmt.Errorf("Could not JSON marshal %s: %s", tc.name, err))
                         }
                         filename := filepath.Join("expectations", tc.name+".json")
                         err = ioutil.WriteFile(filename, []byte(buildJSON), 0644)
                         if err != nil {
                                 panic(fmt.Errorf("Encountered error while trying to write to %s: %s", filename, err))
                         }
                 }
                 return
         }
 
         Convey(`A test Environment`, t, func() {
                 c := context.Background()
                 // This is one hour after the start timestamp in the sample test data.
                 c, _ = testclock.UseTime(c, time.Date(2016, time.March, 14, 11, 0, 0, 0, time.UTC))
                 c = memory.UseWithAppID(c, "dev~luci-milo")
                 c = testconfig.WithCommonClient(c, memcfg.New(aclConfgs))
                 c = auth.WithState(c, &authtest.FakeState{
                         Identity:       "user:alicebob@google.com",
                         IdentityGroups: []string{"all", "googlers"},
                 })
 
                 for _, tc := range getTestCases(".") {
                         Convey(fmt.Sprintf("Test Case: %s", tc.name), func() {
-                                bl := buildLoader{
+                                bl := BuildLoader{
                                         logDogClientFunc: logDogClientFunc(tc),
                                 }
                                 svc := debugSwarmingService{tc}
 
                                 // Special case: The build-internal test case to check that ACLs should fail.
                                 if tc.name == "build-internal" {
                                         Convey("Should fail", func() {
                                                 c := auth.WithState(c, &authtest.FakeState{
                                                         Identity:       identity.AnonymousIdentity,
                                                         IdentityGroups: []string{"all"},
                                                 })
-                                                _, err := bl.swarmingBuildImpl(c, svc, "foo", tc.name)
+                                                _, err := bl.SwarmingBuildImpl(c, svc, "foo", tc.name)
                                                 So(err.Error(), ShouldResemble, "Not a Milo Job or access denied")
                                         })
                                 }
 
-                                build, err := bl.swarmingBuildImpl(c, svc, "foo", tc.name)
+                                build, err := bl.SwarmingBuildImpl(c, svc, "foo", tc.name)
                                 So(err, ShouldBeNil)
                                 So(build, shouldMatchExpectationsFor, tc.name+".json")
                         })
                 }
         })
 }