LogDog: Handle archive failures.

appengine/logdog/coordinator/endpoints/services/archiveStream_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"
 
         "github.com/luci/gae/filter/featureBreaker"
         "github.com/luci/gae/impl/memory"
         ds "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/appengine/logdog/coordinator"
         ct "github.com/luci/luci-go/appengine/logdog/coordinator/coordinatorTest"
         "github.com/luci/luci-go/common/api/logdog_coordinator/services/v1"
+        "github.com/luci/luci-go/common/logging/gologger"
         "github.com/luci/luci-go/common/proto/logdog/svcconfig"
         "github.com/luci/luci-go/server/auth"
         "github.com/luci/luci-go/server/auth/authtest"
         "golang.org/x/net/context"
+        "google.golang.org/grpc/codes"
 
         . "github.com/luci/luci-go/common/testing/assertions"
         . "github.com/smartystreets/goconvey/convey"
 )
 
 func TestArchiveStream(t *testing.T) {
         t.Parallel()
 
         Convey(`With a testing configuration`, t, func() {
                 c := memory.Use(context.Background())
+                c = gologger.Use(c)
                 be := Server{}
 
                 c = ct.UseConfig(c, &svcconfig.Coordinator{
                         ServiceAuthGroup: "test-services",
                 })
                 fs := authtest.FakeState{}
                 c = auth.WithState(c, &fs)
 
                 // Register a testing log stream (not archived).
                 ls := ct.TestLogStream(c, ct.TestLogStreamDescriptor(c, "foo"))
@@ skipping 65 matching lines @@
 
                                 _, err := be.ArchiveStream(c, req)
                                 So(err, ShouldBeNil)
 
                                 ls.TerminalIndex = -1 // To make sure it reloaded.
                                 So(ds.Get(c).Get(ls), ShouldBeNil)
                                 So(ls.Archived(), ShouldBeTrue)
                                 So(ls.TerminalIndex, ShouldEqual, 1337)
                         })
 
+                        Convey(`If the archive has failed`, func() {
+                                req.Error = true
+
+                                Convey(`If the stream is below error threshold, will increment error count and return FailedPrecondition.`, func() {
+                                        ls.ArchiveErrors = 0
+                                        So(ls.ArchiveErrors, ShouldBeLessThan, maxArchiveErrors)
+                                        So(ls.Put(ds.Get(c)), ShouldBeNil)
+
+                                        _, err := be.ArchiveStream(c, req)
+                                        So(err, ShouldHaveRPCCode, codes.FailedPrecondition)
+
+                                        So(ds.Get(c).Get(ls), ShouldBeNil)
+                                        So(ls.Archived(), ShouldBeFalse)
+                                        So(ls.ArchiveErrors, ShouldEqual, 1)
+                                })
+
+                                Convey(`If the stream is above error threshold, will succeed.`, func() {
+                                        ls.ArchiveErrors = maxArchiveErrors
+                                        So(ls.Put(ds.Get(c)), ShouldBeNil)
+
+                                        _, err := be.ArchiveStream(c, req)
+                                        So(err, ShouldBeNil)
+                                        So(ds.Get(c).Get(ls), ShouldBeNil)
+                                        So(ls.Archived(), ShouldBeTrue)
+                                        So(ls.TerminalIndex, ShouldEqual, 13)
+                                })
+                        })
+
                         Convey(`When datastore Get fails, returns internal error.`, func() {
                                 c, fb := featureBreaker.FilterRDS(c, nil)
                                 fb.BreakFeatures(errors.New("test error"), "GetMulti")
 
                                 _, err := be.ArchiveStream(c, req)
                                 So(err, ShouldBeRPCInternal)
                         })
 
                         Convey(`When datastore Put fails, returns internal error.`, func() {
                                 c, fb := featureBreaker.FilterRDS(c, nil)
                                 fb.BreakFeatures(errors.New("test error"), "PutMulti")
 
                                 _, err := be.ArchiveStream(c, req)
                                 So(err, ShouldBeRPCInternal)
                         })
                 })
         })
 }