| 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 mutate | 5 package mutate |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "testing" | 8 "testing" |
| 9 | 9 |
| 10 "github.com/luci/gae/filter/featureBreaker" | 10 "github.com/luci/gae/filter/featureBreaker" |
| 11 "github.com/luci/gae/service/datastore" | 11 "github.com/luci/gae/service/datastore" |
| 12 "github.com/luci/luci-go/appengine/cmd/dm/model" | 12 "github.com/luci/luci-go/appengine/cmd/dm/model" |
| 13 "github.com/luci/luci-go/appengine/tumble" | 13 "github.com/luci/luci-go/appengine/tumble" |
| 14 "github.com/luci/luci-go/common/api/dm/service/v1" | 14 "github.com/luci/luci-go/common/api/dm/service/v1" |
| 15 . "github.com/luci/luci-go/common/testing/assertions" | 15 . "github.com/luci/luci-go/common/testing/assertions" |
| 16 . "github.com/smartystreets/goconvey/convey" | 16 . "github.com/smartystreets/goconvey/convey" |
| 17 ) | 17 ) |
| 18 | 18 |
| 19 func TestMergeQuest(t *testing.T) { | 19 func TestMergeQuest(t *testing.T) { |
| 20 t.Parallel() | 20 t.Parallel() |
| 21 | 21 |
| 22 Convey("MergeQuest", t, func() { | 22 Convey("MergeQuest", t, func() { |
| 23 ttest := &tumble.Testing{} | 23 ttest := &tumble.Testing{} |
| 24 c := ttest.Context() | 24 c := ttest.Context() |
| 25 ds := datastore.Get(c) | 25 ds := datastore.Get(c) |
| 26 | 26 |
| 27 » » desc := dm.NewQuestDesc("distributor", `{"data":"yes"}`) | 27 » » desc := dm.NewQuestDesc("distributor", `{"data":"yes"}`, nil) |
| 28 » » qst, err := model.NewQuest(c, desc) | 28 » » So(desc.Normalize(), ShouldBeNil) |
| 29 » » So(err, ShouldBeNil) | 29 » » qst := model.NewQuest(c, desc) |
| 30 qst.BuiltBy = append(qst.BuiltBy, *dm.NewTemplateSpec("a", "b",
"c", "d")) | 30 qst.BuiltBy = append(qst.BuiltBy, *dm.NewTemplateSpec("a", "b",
"c", "d")) |
| 31 | 31 |
| 32 » » mq := &MergeQuest{qst} | 32 » » mq := &MergeQuest{qst, nil} |
| 33 | 33 |
| 34 Convey("root", func() { | 34 Convey("root", func() { |
| 35 So(mq.Root(c), ShouldResemble, ds.MakeKey("Quest", qst.I
D)) | 35 So(mq.Root(c), ShouldResemble, ds.MakeKey("Quest", qst.I
D)) |
| 36 }) | 36 }) |
| 37 | 37 |
| 38 Convey("quest doesn't exist", func() { | 38 Convey("quest doesn't exist", func() { |
| 39 muts, err := mq.RollForward(c) | 39 muts, err := mq.RollForward(c) |
| 40 So(err, ShouldBeNil) | 40 So(err, ShouldBeNil) |
| 41 So(muts, ShouldBeEmpty) | 41 So(muts, ShouldBeEmpty) |
| 42 | 42 |
| (...skipping 27 matching lines...) Expand all Loading... |
| 70 Convey("datastore fail", func() { | 70 Convey("datastore fail", func() { |
| 71 c, fb := featureBreaker.FilterRDS(c, nil) | 71 c, fb := featureBreaker.FilterRDS(c, nil) |
| 72 fb.BreakFeatures(nil, "GetMulti") | 72 fb.BreakFeatures(nil, "GetMulti") |
| 73 _, err := mq.RollForward(c) | 73 _, err := mq.RollForward(c) |
| 74 So(err, ShouldBeRPCUnknown) | 74 So(err, ShouldBeRPCUnknown) |
| 75 }) | 75 }) |
| 76 }) | 76 }) |
| 77 | 77 |
| 78 }) | 78 }) |
| 79 } | 79 } |
| OLD | NEW |