transaction_test.go

package godb

import (
	"testing"

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

func TestBegin(t *testing.T) {
	Convey("Given an existing connection", t, func() {
		db := createInMemoryConnection(t)
		defer db.Close()

		Convey("Begin create a new transaction", func() {
			err := db.Begin()
			So(err, ShouldBeNil)
			So(db.sqlTx, ShouldNotBeNil)

			Convey("Begin fails is a transactions already exists", func() {
				err = db.Begin()
				So(err, ShouldNotBeNil)
			})
		})
	})
}

func TestCommit(t *testing.T) {
	Convey("Given an existing connexion", t, func() {
		db := createInMemoryConnection(t)
		defer db.Close()

		Convey("Commit end an existing transaction", func() {
			db.Begin()
			err := db.Commit()
			So(err, ShouldBeNil)
		})

		Convey("Commit fails if there is no transaction", func() {
			err := db.Commit()
			So(err, ShouldNotBeNil)
		})
	})
}

func TestRollback(t *testing.T) {
	Convey("Given an existing connexion", t, func() {
		db := createInMemoryConnection(t)
		defer db.Close()

		Convey("Commit end an existing transaction", func() {
			db.Begin()
			err := db.Rollback()
			So(err, ShouldBeNil)
		})

		Convey("Commit fails if there is no transaction", func() {
			err := db.Rollback()
			So(err, ShouldNotBeNil)
		})
	})
}

func TestCurrentTx(t *testing.T) {
	Convey("Given an existing connexion", t, func() {
		db := createInMemoryConnection(t)
		defer db.Close()

		Convey("CurrentTx returns nil there is no current transaction", func() {
			So(db.CurrentTx(), ShouldBeNil)
		})

		Convey("CurrentTx returns Tx if there is a current transaction", func() {
			db.Begin()
			So(db.CurrentTx(), ShouldEqual, db.sqlTx)
		})
	})
}