Implement Mail 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/mail"
         "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)
@@ skipping 126 matching lines @@
                 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)
         })
+
+        Convey("works for mail", t, func() {
+                c, fb := featureBreaker.FilterMail(memory.Use(context.Background()), nil)
+                c, ctr := FilterMail(c)
+                So(c, ShouldNotBeNil)
+                So(ctr, ShouldNotBeNil)
+
+                m := mail.Get(c)
+
+                err := m.Send(&mail.Message{
+                        Sender: "admin@example.com",
+                        To:     []string{"coolDood@example.com"},
+                        Body:   "hi",
+                })
+                die(err)
+
+                fb.BreakFeatures(nil, "Send")
+                err = m.Send(&mail.Message{
+                        Sender: "admin@example.com",
+                        To:     []string{"coolDood@example.com"},
+                        Body:   "hi",
+                })
+                So(err, ShouldErrLike, `"Send" is broken`)
+
+                So(ctr.Send, 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
 }