initial import

core/stores/sqlx/tx_test.go

@@ -0,0 +1,76 @@
+package sqlx
+
+import (
+	"database/sql"
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+const (
+	mockCommit   = 1
+	mockRollback = 2
+)
+
+type mockTx struct {
+	status int
+}
+
+func (mt *mockTx) Commit() error {
+	mt.status |= mockCommit
+	return nil
+}
+
+func (mt *mockTx) Exec(q string, args ...interface{}) (sql.Result, error) {
+	return nil, nil
+}
+
+func (mt *mockTx) Prepare(query string) (StmtSession, error) {
+	return nil, nil
+}
+
+func (mt *mockTx) QueryRow(v interface{}, q string, args ...interface{}) error {
+	return nil
+}
+
+func (mt *mockTx) QueryRowPartial(v interface{}, q string, args ...interface{}) error {
+	return nil
+}
+
+func (mt *mockTx) QueryRows(v interface{}, q string, args ...interface{}) error {
+	return nil
+}
+
+func (mt *mockTx) QueryRowsPartial(v interface{}, q string, args ...interface{}) error {
+	return nil
+}
+
+func (mt *mockTx) Rollback() error {
+	mt.status |= mockRollback
+	return nil
+}
+
+func beginMock(mock *mockTx) beginnable {
+	return func(*sql.DB) (trans, error) {
+		return mock, nil
+	}
+}
+
+func TestTransactCommit(t *testing.T) {
+	mock := &mockTx{}
+	err := transactOnConn(nil, beginMock(mock), func(Session) error {
+		return nil
+	})
+	assert.Equal(t, mockCommit, mock.status)
+	assert.Nil(t, err)
+}
+
+func TestTransactRollback(t *testing.T) {
+	mock := &mockTx{}
+	err := transactOnConn(nil, beginMock(mock), func(Session) error {
+		return errors.New("rollback")
+	})
+	assert.Equal(t, mockRollback, mock.status)
+	assert.NotNil(t, err)
+}