Add filters for user service.

filter/count/count_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 count
 
 import (
         "fmt"
         "testing"
 
         "github.com/luci/gae/filter/featureBreaker"
         "github.com/luci/gae/impl/memory"
         "github.com/luci/gae/service/datastore"
         "github.com/luci/gae/service/info"
         "github.com/luci/gae/service/memcache"
         "github.com/luci/gae/service/taskqueue"
+        "github.com/luci/gae/service/user"
         . "github.com/luci/luci-go/common/testing/assertions"
         . "github.com/smartystreets/goconvey/convey"
         "golang.org/x/net/context"
 )
 
 func shouldHaveSuccessesAndErrors(actual interface{}, expected ...interface{}) string {
         a := actual.(Entry)
         if len(expected) != 2 {
                 panic("Invalid number of expected, should be 2 (successes, errors).")
         }
@@ skipping 106 matching lines @@
                 gi := info.Get(c)
 
                 _, err := gi.Namespace("foo")
                 die(err)
                 fb.BreakFeatures(nil, "Namespace")
                 _, err = gi.Namespace("boom")
                 So(err, ShouldErrLike, `"Namespace" is broken`)
 
                 So(ctr.Namespace, shouldHaveSuccessesAndErrors, 1, 1)
         })
+
+        Convey("works for user", t, func() {
+                c, fb := featureBreaker.FilterUser(memory.Use(context.Background()), nil)
+                c, ctr := FilterUser(c)
+                So(c, ShouldNotBeNil)
+                So(ctr, ShouldNotBeNil)
+
+                u := user.Get(c)
+
+                _, err := u.CurrentOAuth("foo")
+                die(err)
+                fb.BreakFeatures(nil, "CurrentOAuth")
+                _, err = u.CurrentOAuth("foo")
+                So(err, ShouldErrLike, `"CurrentOAuth" is broken`)
+
+                So(ctr.CurrentOAuth, shouldHaveSuccessesAndErrors, 1, 1)
+        })
 }
 
 func ExampleFilterRDS() {
         // Set up your context using a base service implementation (memory or prod)
         c := memory.Use(context.Background())
 
         // Apply the counter.FilterRDS
         c, counter := FilterRDS(c)
 
         // functions use ds from the context like normal... they don't need to know
@@ skipping 11 matching lines @@
 
         // Using the other function.
         someCalledFunc(c)
         someCalledFunc(c)
 
         // Then we can see what happened!
         fmt.Printf("%d\n", counter.PutMulti.Successes())
         // Output:
         // 2
 }