Milo: Use custom config caching layer

milo/appengine/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/gae/impl/memory"
         "github.com/luci/luci-go/common/clock/testclock"
         memcfg "github.com/luci/luci-go/common/config/impl/memory"
         "github.com/luci/luci-go/luci_config/server/cfgclient/backend/testconfig"
+        "github.com/luci/luci-go/milo/appengine/common"
         "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 := memory.UseWithAppID(context.Background(), "dev~luci-milo")
                 c, _ = testclock.UseTime(c, time.Date(2016, time.March, 14, 11, 0, 0, 0, time.UTC))
                 c = testconfig.WithCommonClient(c, memcfg.New(bktConfigFull))
+                // Update the service config so that the settings are loaded.
+                err := common.UpdateServiceConfig(c)
+                So(err, ShouldBeNil)
 
                 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,
                                         builderQuery{
@@ skipping 22 matching lines @@
 buildbucket: {
         host: "debug"
 }
 `
 
 var bktConfigFull = map[string]memcfg.ConfigSet{
         "services/luci-milo": {
                 "settings.cfg": bktConfig,
         },
 }