OLD | NEW |
1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
3 // that can be found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
4 | 4 |
5 package featureBreaker | 5 package featureBreaker |
6 | 6 |
7 import ( | 7 import ( |
8 "testing" | 8 "testing" |
9 | 9 |
10 "github.com/luci/gae/impl/memory" | 10 "github.com/luci/gae/impl/memory" |
11 "github.com/luci/gae/service/datastore" | 11 "github.com/luci/gae/service/datastore" |
12 "github.com/luci/luci-go/common/errors" | 12 "github.com/luci/luci-go/common/errors" |
13 . "github.com/smartystreets/goconvey/convey" | 13 . "github.com/smartystreets/goconvey/convey" |
14 "golang.org/x/net/context" | 14 "golang.org/x/net/context" |
15 ) | 15 ) |
16 | 16 |
17 func TestBrokenFeatures(t *testing.T) { | 17 func TestBrokenFeatures(t *testing.T) { |
18 t.Parallel() | 18 t.Parallel() |
19 | 19 |
20 e := errors.New("default err") | 20 e := errors.New("default err") |
21 | 21 |
22 Convey("BrokenFeatures", t, func() { | 22 Convey("BrokenFeatures", t, func() { |
23 c := memory.Use(context.Background()) | 23 c := memory.Use(context.Background()) |
24 | 24 |
25 Convey("Can break ds", func() { | 25 Convey("Can break ds", func() { |
26 Convey("without a default", func() { | 26 Convey("without a default", func() { |
27 c, bf := FilterRDS(c, nil) | 27 c, bf := FilterRDS(c, nil) |
28 ds := datastore.Get(c) | 28 ds := datastore.Get(c) |
29 vals := []datastore.PropertyMap{{ | 29 vals := []datastore.PropertyMap{{ |
30 » » » » » "$key": {datastore.MkPropertyNI(ds.NewKe
y("Wut", "", 1, nil))}, | 30 » » » » » "$key": datastore.MkPropertyNI(ds.NewKey
("Wut", "", 1, nil)), |
31 }} | 31 }} |
32 | 32 |
33 Convey("by specifying an error", func() { | 33 Convey("by specifying an error", func() { |
34 bf.BreakFeatures(e, "GetMulti", "PutMult
i") | 34 bf.BreakFeatures(e, "GetMulti", "PutMult
i") |
35 So(ds.GetMulti(vals), ShouldEqual, e) | 35 So(ds.GetMulti(vals), ShouldEqual, e) |
36 | 36 |
37 Convey("and you can unbreak them as well
", func() { | 37 Convey("and you can unbreak them as well
", func() { |
38 bf.UnbreakFeatures("GetMulti") | 38 bf.UnbreakFeatures("GetMulti") |
39 | 39 |
40 So(errors.SingleError(ds.GetMult
i(vals)), ShouldEqual, datastore.ErrNoSuchEntity) | 40 So(errors.SingleError(ds.GetMult
i(vals)), ShouldEqual, datastore.ErrNoSuchEntity) |
41 | 41 |
42 Convey("no broken features at al
l is a shortcut", func() { | 42 Convey("no broken features at al
l is a shortcut", func() { |
43 bf.UnbreakFeatures("PutM
ulti") | 43 bf.UnbreakFeatures("PutM
ulti") |
44 So(errors.SingleError(ds
.GetMulti(vals)), ShouldEqual, datastore.ErrNoSuchEntity) | 44 So(errors.SingleError(ds
.GetMulti(vals)), ShouldEqual, datastore.ErrNoSuchEntity) |
45 }) | 45 }) |
46 }) | 46 }) |
47 }) | 47 }) |
48 | 48 |
49 Convey("Not specifying an error gets you a gener
ic error", func() { | 49 Convey("Not specifying an error gets you a gener
ic error", func() { |
50 bf.BreakFeatures(nil, "GetMulti") | 50 bf.BreakFeatures(nil, "GetMulti") |
51 So(ds.GetMulti(vals).Error(), ShouldCont
ainSubstring, `feature "GetMulti" is broken`) | 51 So(ds.GetMulti(vals).Error(), ShouldCont
ainSubstring, `feature "GetMulti" is broken`) |
52 }) | 52 }) |
53 }) | 53 }) |
54 | 54 |
55 Convey("with a default", func() { | 55 Convey("with a default", func() { |
56 c, bf := FilterRDS(c, e) | 56 c, bf := FilterRDS(c, e) |
57 ds := datastore.Get(c) | 57 ds := datastore.Get(c) |
58 vals := []datastore.PropertyMap{{ | 58 vals := []datastore.PropertyMap{{ |
59 » » » » » "$key": {datastore.MkPropertyNI(ds.NewKe
y("Wut", "", 1, nil))}, | 59 » » » » » "$key": datastore.MkPropertyNI(ds.NewKey
("Wut", "", 1, nil)), |
60 }} | 60 }} |
61 bf.BreakFeatures(nil, "GetMulti") | 61 bf.BreakFeatures(nil, "GetMulti") |
62 So(ds.GetMulti(vals), ShouldEqual, e) | 62 So(ds.GetMulti(vals), ShouldEqual, e) |
63 }) | 63 }) |
64 }) | 64 }) |
65 }) | 65 }) |
66 } | 66 } |
OLD | NEW |