OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 package datastore | 5 package datastore |
6 | 6 |
7 import ( | 7 import ( |
8 "testing" | 8 "testing" |
9 | 9 |
10 "github.com/luci/gae/service/info" | 10 "github.com/luci/gae/service/info" |
11 . "github.com/smartystreets/goconvey/convey" | 11 . "github.com/smartystreets/goconvey/convey" |
12 "golang.org/x/net/context" | 12 "golang.org/x/net/context" |
13 ) | 13 ) |
14 | 14 |
15 type fakeInfo struct{ info.Interface } | 15 type fakeInfo struct{ info.Interface } |
16 | 16 |
17 func (fakeInfo) GetNamespace() string { return "ns" } | 17 func (fakeInfo) GetNamespace() string { return "ns" } |
18 func (fakeInfo) AppID() string { return "aid" } | 18 func (fakeInfo) AppID() string { return "aid" } |
19 | 19 |
20 type fakeService struct{ Interface } | 20 type fakeService struct{ RawInterface } |
21 | 21 |
22 type fakeFilt struct{ Interface } | 22 type fakeFilt struct{ RawInterface } |
23 | 23 |
24 func (fakeFilt) NewKey(kind, stringID string, intID int64, parent Key) Key { | 24 func (fakeFilt) NewKey(kind, stringID string, intID int64, parent Key) Key { |
25 return NewKey("aid", "ns", kind, stringID, intID, parent) | 25 return NewKey("aid", "ns", kind, stringID, intID, parent) |
26 } | 26 } |
27 | 27 |
28 func TestServices(t *testing.T) { | 28 func TestServices(t *testing.T) { |
29 t.Parallel() | 29 t.Parallel() |
30 | 30 |
31 Convey("Test service interfaces", t, func() { | 31 Convey("Test service interfaces", t, func() { |
32 c := context.Background() | 32 c := context.Background() |
33 Convey("without adding anything", func() { | 33 Convey("without adding anything", func() { |
34 » » » So(Get(c), ShouldBeNil) | 34 » » » So(GetRaw(c), ShouldBeNil) |
35 }) | 35 }) |
36 | 36 |
37 Convey("adding a basic implementation", func() { | 37 Convey("adding a basic implementation", func() { |
38 » » » c = Set(info.Set(c, fakeInfo{}), fakeService{}) | 38 » » » c = SetRaw(info.Set(c, fakeInfo{}), fakeService{}) |
39 | 39 |
40 Convey("lets you pull them back out", func() { | 40 Convey("lets you pull them back out", func() { |
41 » » » » So(Get(c), ShouldResemble, &checkFilter{fakeServ
ice{}, "aid", "ns"}) | 41 » » » » So(GetRaw(c), ShouldResemble, &checkFilter{fakeS
ervice{}, "aid", "ns"}) |
42 }) | 42 }) |
43 | 43 |
44 Convey("and lets you add filters", func() { | 44 Convey("and lets you add filters", func() { |
45 » » » » c = AddFilters(c, func(ic context.Context, rds I
nterface) Interface { | 45 » » » » c = AddRawFilters(c, func(ic context.Context, rd
s RawInterface) RawInterface { |
46 return fakeFilt{rds} | 46 return fakeFilt{rds} |
47 }) | 47 }) |
48 | 48 |
49 k := Get(c).NewKey("Kind", "", 1, nil) | 49 k := Get(c).NewKey("Kind", "", 1, nil) |
50 So(KeysEqual(k, NewKey("aid", "ns", "Kind", "",
1, nil)), ShouldBeTrue) | 50 So(KeysEqual(k, NewKey("aid", "ns", "Kind", "",
1, nil)), ShouldBeTrue) |
51 }) | 51 }) |
52 }) | 52 }) |
53 Convey("adding zero filters does nothing", func() { | 53 Convey("adding zero filters does nothing", func() { |
54 » » » So(AddFilters(c), ShouldEqual, c) | 54 » » » So(AddRawFilters(c), ShouldEqual, c) |
55 }) | 55 }) |
56 }) | 56 }) |
57 } | 57 } |
OLD | NEW |