tx.go

package sqlmock

import (
	"errors"
	"fmt"
)

// Tx implements driver.Tx behavior on top of queued mock expectations.
type Tx struct {
	conn *Conn
}

// Commit validates and consumes the next commit expectation.
//
// Commit fails if there is no active transaction or the next queued
// expectation is not a commit.
func (t *Tx) Commit() error {
	c := t.conn
	m := c.mock

	m.mu.Lock()
	defer m.mu.Unlock()

	if !c.txStarted {
		return errors.New("commit without active transaction")
	}

	op, err := m.next()
	if err != nil {
		return err
	}

	e, ok := op.(*ExpectedCommit)
	if !ok {
		return fmt.Errorf("expected %T, got COMMIT", op)
	}

	if err := e.match(runtimeOp{kind: "commit"}); err != nil {
		return err
	}

	m.consume()
	c.txStarted = false
	return nil
}

// Rollback validates and consumes the next rollback expectation.
//
// Rollback fails if there is no active transaction or the next queued
// expectation is not a rollback.
func (t *Tx) Rollback() error {
	c := t.conn
	m := c.mock

	m.mu.Lock()
	defer m.mu.Unlock()

	if !c.txStarted {
		return errors.New("rollback without active transaction")
	}

	op, err := m.next()
	if err != nil {
		return err
	}

	e, ok := op.(*ExpectedRollback)
	if !ok {
		return fmt.Errorf("expected %T, got ROLLBACK", op)
	}

	if err := e.match(runtimeOp{kind: "rollback"}); err != nil {
		return err
	}

	m.consume()
	c.txStarted = false
	return nil
}