rows.go

package sqlmock

import (
	"database/sql/driver"
	"io"
)

// Rows implements driver.Rows with in-memory row data.
type Rows struct {
	columns []string
	data    [][]driver.Value
	index   int
}

// NewRows creates a row container with the provided column names.
func NewRows(columns []string) *Rows {
	return &Rows{columns: columns}
}

// AddRows appends one row of values and returns the same Rows for chaining.
func (r *Rows) AddRows(values ...any) *Rows {
	row := make([]driver.Value, len(values))
	for i, v := range values {
		row[i] = v
	}

	r.data = append(r.data, row)
	return r
}

// Columns returns the configured column names.
func (r *Rows) Columns() []string {
	return r.columns
}

// Next copies the next row into dest or returns io.EOF when no rows remain.
func (r *Rows) Next(dest []driver.Value) error {
	if r.index >= len(r.data) {
		return io.EOF
	}

	copy(dest, r.data[r.index])
	r.index++

	return nil
}

// Close closes the rows iterator.
func (r *Rows) Close() error {
	return nil
}