Index: appengine/logdog/coordinator/endpoints/logs/get_test.go |
diff --git a/appengine/logdog/coordinator/endpoints/logs/get_test.go b/appengine/logdog/coordinator/endpoints/logs/get_test.go |
index efc4b20b5b655ddda57cb55789aa1ea9232a224a..2b9b24e8a251a90f799d7fa9ecdbae81c8f8f3b8 100644 |
--- a/appengine/logdog/coordinator/endpoints/logs/get_test.go |
+++ b/appengine/logdog/coordinator/endpoints/logs/get_test.go |
@@ -30,6 +30,7 @@ import ( |
"github.com/luci/luci-go/server/auth" |
"github.com/luci/luci-go/server/auth/authtest" |
"github.com/luci/luci-go/server/logdog/storage" |
+ ast "github.com/luci/luci-go/server/logdog/storage/archive" |
memoryStorage "github.com/luci/luci-go/server/logdog/storage/memory" |
"golang.org/x/net/context" |
@@ -212,11 +213,15 @@ func TestGet(t *testing.T) { |
ms := memoryStorage.Storage{} |
gsc := testGSClient{} |
- s.StorageFunc = func(context.Context) (storage.Storage, error) { |
+ s.IntermediateStorageFunc = func(c context.Context) (storage.Storage, error) { |
return &ms, nil |
} |
- s.GSClientFunc = func(context.Context) (gs.Client, error) { |
- return gsc, nil |
+ s.ArchiveStorageFunc = func(c context.Context, ls *coordinator.LogStream) (storage.Storage, error) { |
+ return ast.New(c, ast.Options{ |
+ IndexURL: ls.ArchiveIndexURL, |
+ StreamURL: ls.ArchiveStreamURL, |
+ Client: &gsc, |
+ }) |
} |
Convey(`Testing Get requests (no logs)`, func() { |