LogDog: Implement RegisterPrefix RPC.

appengine/logdog/coordinator/endpoints/services/terminateStream_test.go

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package services
 
 import (
         "errors"
         "testing"
         "time"
@@ skipping 14 matching lines @@
 )
 
 func TestTerminateStream(t *testing.T) {
         t.Parallel()
 
         Convey(`With a testing configuration`, t, func() {
                 c, env := ct.Install()
 
                 // Set our archival delays. The project delay is smaller than the service
                 // delay, so it should be used.
-                env.ModServiceConfig(c, func(cfg *svcconfig.Coordinator) {
-                        cfg.ArchiveTopic = "projects/test/topics/archive"
-                        cfg.ArchiveSettleDelay = google.NewDuration(10 * time.Second)
-                        cfg.ArchiveDelayMax = google.NewDuration(24 * time.Hour)
+                env.ModServiceConfig(c, func(cfg *svcconfig.Config) {
+                        coord := cfg.Coordinator
+                        coord.ArchiveTopic = "projects/test/topics/archive"
+                        coord.ArchiveSettleDelay = google.NewDuration(10 * time.Second)
+                        coord.ArchiveDelayMax = google.NewDuration(24 * time.Hour)
                 })
                 env.ModProjectConfig("proj-foo", func(pcfg *svcconfig.ProjectConfig) {
                         pcfg.MaxStreamAge = google.NewDuration(time.Hour)
                 })
 
                 svr := New()
 
                 tls := ct.MakeStream(c, "proj-foo", "testing/+/foo/bar")
 
                 req := logdog.TerminateStreamRequest{
@@ skipping 114 matching lines @@
 
                                                 _, err := svr.TerminateStream(c, &req)
                                                 So(err, ShouldBeRPCOK)
 
                                                 So(env.ArchivalPublisher.Hashes(), ShouldResemble, []string{string(tls.Stream.ID)})
                                                 So(len(env.ArchivalPublisher.Tasks()), ShouldEqual, 1)
                                                 So(env.ArchivalPublisher.Tasks()[0].CompletePeriod.Duration(), ShouldEqual, 24*time.Hour)
                                         })
 
                                         Convey(`When there is no service or project config delay.`, func() {
-                                                env.ModServiceConfig(c, func(cfg *svcconfig.Coordinator) {
-                                                        cfg.ArchiveDelayMax = nil
+                                                env.ModServiceConfig(c, func(cfg *svcconfig.Config) {
+                                                        cfg.Coordinator.ArchiveDelayMax = nil
                                                 })
                                                 env.ModProjectConfig("proj-foo", func(pcfg *svcconfig.ProjectConfig) {
                                                         pcfg.MaxStreamAge = nil
                                                 })
 
                                                 _, err := svr.TerminateStream(c, &req)
                                                 So(err, ShouldBeRPCOK)
 
                                                 So(env.ArchivalPublisher.Hashes(), ShouldResemble, []string{string(tls.Stream.ID)})
                                                 So(len(env.ArchivalPublisher.Tasks()), ShouldEqual, 1)
@@ skipping 28 matching lines @@
                                 So(err, ShouldBeRPCInvalidArgument, "Invalid ID")
                         })
 
                         Convey(`Will fail if the stream is not registered.`, func() {
                                 _, err := svr.TerminateStream(c, &req)
                                 So(err, ShouldBeRPCNotFound, "is not registered")
                         })
                 })
         })
 }