struct_delete_test.go

package godb

import (
	"testing"

	. "github.com/smartystreets/goconvey/convey"
)

func TestDeleteDo(t *testing.T) {
	Convey("Given a test database", t, func() {
		db := fixturesSetup(t)
		defer db.Close()

		Convey("Delete delete a record", func() {
			dummy := &Dummy{}
			err := db.Select(dummy).Where("an_integer = ?", 11).Do()
			So(err, ShouldBeNil)
			count, err := db.Delete(dummy).Do()
			So(err, ShouldBeNil)

			Convey("The data isn't in database", func() {
				found, err := db.SelectFrom("dummies").Where("id = ?", dummy.ID).Count()
				So(err, ShouldBeNil)
				So(found, ShouldEqual, 0)
			})

			Convey("Delete returns the count of affected rows", func() {
				So(count, ShouldEqual, 1)
			})
		})

		Convey("Delete returns error if optimistic locking fails", func() {
			dummy := &Dummy{}
			err := db.Select(dummy).Where("an_integer = ?", 11).Do()
			So(err, ShouldBeNil)

			// Simulate another update of the record.
			_, err = db.UpdateTable("dummies").Set("version", 1).Where("id = ?", dummy.ID).Do()
			So(err, ShouldBeNil)

			count, err := db.Delete(dummy).Do()
			So(count, ShouldEqual, 0)
			So(err, ShouldEqual, ErrOpLock)
		})
	})
}