milo: fix running steps

appengine/cmd/milo/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"
         "io/ioutil"
         "path"
         "strings"
         "testing"
+        "time"
 
         "github.com/luci/luci-go/common/clock/testclock"
         . "github.com/smartystreets/goconvey/convey"
         "golang.org/x/net/context"
 )
 
 var generate = flag.Bool("test.generate", false, "Generate expectations instead of running tests.")
 
 func load(name string) ([]byte, error) {
         filename := strings.Join([]string{"expectations", name}, "/")
@@ skipping 11 matching lines @@
 }
 
 func TestBuild(t *testing.T) {
         testCases := []struct {
                 input        string
                 expectations string
         }{
                 {"debug:build-exception", "build-exception.json"},
                 {"debug:build-patch-failure", "build-patch-failure.json"},
                 {"debug:build-pending", "build-pending.json"},
+                {"debug:build-running", "build-running.json"},
                 {"debug:build-timeout", "build-timeout.json"},
         }
 
+        c := context.Background()
+        c, _ = testclock.UseTime(c, time.Date(2016, time.March, 14, 11, 0, 0, 0, time.UTC))
+
         if *generate {
                 for _, tc := range testCases {
-                        c := context.Background()
-                        c, _ = testclock.UseTime(c, testclock.TestTimeUTC)
                         fmt.Printf("Generating expectations for %s\n", tc.input)
                         build, err := swarmingBuildImpl(c, "foo", "default", tc.input)
                         if err != nil {
                                 panic(fmt.Errorf("Could not run swarmingBuildImpl for %s: %s", tc.input, err))
                         }
                         buildJSON, err := json.MarshalIndent(build, "", "  ")
                         if err != nil {
                                 panic(fmt.Errorf("Could not JSON marshal %s: %s", tc.input, err))
                         }
                         filename := path.Join("expectations", tc.expectations)
                         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()
-                c, _ = testclock.UseTime(c, testclock.TestTimeUTC)
-
                 for _, tc := range testCases {
                         Convey(fmt.Sprintf("Test Case: %s", tc.input), func() {
                                 build, err := swarmingBuildImpl(c, "foo", "default", tc.input)
                                 So(err, ShouldBeNil)
                                 So(build, shouldMatchExpectationsFor, tc.expectations)
                         })
                 }
         })
 }