// Copyright © 2016 Abcum Ltd // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package db import ( "testing" "net/http/httptest" "github.com/abcum/fibre" "github.com/abcum/surreal/cnf" "github.com/abcum/surreal/util/data" . "github.com/smartystreets/goconvey/convey" ) func setupDB() { cnf.Settings = &cnf.Options{} cnf.Settings.DB.Path = "memory" cnf.Settings.DB.Base = "*" workerCount = 1 Setup(cnf.Settings) } func setupKV() *fibre.Context { auth := &cnf.Auth{} auth.Kind = cnf.AuthKV auth.Possible.NS = "*" auth.Selected.NS = "*" auth.Possible.DB = "*" auth.Selected.DB = "*" req := &fibre.Request{Request: httptest.NewRequest("GET", "/", nil)} res := &fibre.Response{} ctx := fibre.NewContext(req, res, nil) ctx.Set("auth", auth) return ctx } func setupSC() *fibre.Context { auth := &cnf.Auth{} auth.Kind = cnf.AuthSC auth.Possible.NS = "*" auth.Selected.NS = "*" auth.Possible.DB = "*" auth.Selected.DB = "*" req := &fibre.Request{Request: httptest.NewRequest("GET", "/", nil)} res := &fibre.Response{} ctx := fibre.NewContext(req, res, nil) ctx.Set("auth", auth) return ctx } func TestYield(t *testing.T) { Convey("Yield different responses when modifying a record", t, func() { setupDB() txt := ` USE NS test DB test; CREATE person:test SET test=1 RETURN AFTER; UPDATE person:test SET test=2 RETURN BEFORE; UPDATE person:test SET test=3 RETURN BOTH; UPDATE person:test SET test=4 RETURN DIFF; UPDATE person:test SET test=5 RETURN NONE; DELETE person:test RETURN BEFORE; ` res, err := Execute(setupKV(), txt, nil) So(err, ShouldBeNil) So(res[1].Result, ShouldHaveLength, 1) So(data.Consume(res[1].Result[0]).Get("test").Data(), ShouldEqual, 1) So(res[2].Result, ShouldHaveLength, 1) So(data.Consume(res[2].Result[0]).Get("test").Data(), ShouldEqual, 1) So(res[3].Result, ShouldHaveLength, 1) So(data.Consume(res[3].Result[0]).Get("before.test").Data(), ShouldEqual, 2) So(data.Consume(res[3].Result[0]).Get("after.test").Data(), ShouldEqual, 3) So(res[4].Result, ShouldHaveLength, 1) So(res[4].Result[0], ShouldHaveLength, 1) So(data.Consume(res[4].Result[0]).Get("0.op").Data(), ShouldEqual, "replace") So(data.Consume(res[4].Result[0]).Get("0.path").Data(), ShouldEqual, "/test") So(data.Consume(res[4].Result[0]).Get("0.value").Data(), ShouldEqual, 4) So(res[5].Result, ShouldHaveLength, 0) So(res[6].Result, ShouldHaveLength, 1) So(data.Consume(res[6].Result[0]).Get("test").Data(), ShouldEqual, 5) }) }