initial commit

2025-06-14 23:55:00 +02:00 · 2014-02-05 16:21:07 +02:00
commit 3e67393335
10 changed files with 789 additions and 0 deletions
--- a/sqlmock_test.go
+++ b/sqlmock_test.go
@ -0,0 +1,268 @@
+package sqlmock
+
+import (
+	"database/sql"
+	"errors"
+	"testing"
+)
+
+func TestMockQuery(t *testing.T) {
+	db, err := sql.Open("mock", "")
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when opening a stub database connection", err)
+	}
+
+	rs := RowsFromCSVString([]string{"id", "title"}, "5,hello world")
+
+	ExpectQuery("SELECT (.+) FROM articles WHERE id = ?").
+		WithArgs(5).
+		WillReturnRows(rs)
+
+	rows, err := db.Query("SELECT (.+) FROM articles WHERE id = ?", 5)
+	if err != nil {
+		t.Errorf("Error '%s' was not expected while retrieving mock rows", err)
+	}
+	defer rows.Close()
+	if !rows.Next() {
+		t.Error("It must have had one row as result, but got empty result set instead")
+	}
+
+	var id int
+	var title string
+
+	err = rows.Scan(&id, &title)
+	if err != nil {
+		t.Errorf("Error '%s' was not expected while trying to scan row", err)
+	}
+
+	if id != 5 {
+		t.Errorf("Expected mocked id to be 5, but got %d instead", id)
+	}
+
+	if title != "hello world" {
+		t.Errorf("Expected mocked title to be 'hello world', but got '%s' instead", title)
+	}
+
+	if err = db.Close(); err != nil {
+		t.Errorf("Error '%s' was not expected while closing the database", err)
+	}
+}
+
+func TestTransactionExpectations(t *testing.T) {
+	db, err := sql.Open("mock", "")
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when opening a stub database connection", err)
+	}
+
+	// begin and commit
+	ExpectBegin()
+	ExpectCommit()
+
+	tx, err := db.Begin()
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when beginning a transaction", err)
+	}
+
+	err = tx.Commit()
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when commiting a transaction", err)
+	}
+
+	// begin and rollback
+	ExpectBegin()
+	ExpectRollback()
+
+	tx, err = db.Begin()
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when beginning a transaction", err)
+	}
+
+	err = tx.Rollback()
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when rolling back a transaction", err)
+	}
+
+	// begin with an error
+	ExpectBegin().WillReturnError(errors.New("Some err"))
+
+	tx, err = db.Begin()
+	if err == nil {
+		t.Error("An error was expected when beginning a transaction, but got none")
+	}
+
+	if err = db.Close(); err != nil {
+		t.Errorf("Error '%s' was not expected while closing the database", err)
+	}
+}
+
+func TestPreparedQueryExecutions(t *testing.T) {
+	db, err := sql.Open("mock", "")
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when opening a stub database connection", err)
+	}
+
+	rs1 := RowsFromCSVString([]string{"id", "title"}, "5,hello world")
+	ExpectQuery("SELECT (.+) FROM articles WHERE id = ?").
+		WithArgs(5).
+		WillReturnRows(rs1)
+
+	rs2 := RowsFromCSVString([]string{"id", "title"}, "2,whoop")
+	ExpectQuery("SELECT (.+) FROM articles WHERE id = ?").
+		WithArgs(2).
+		WillReturnRows(rs2)
+
+	stmt, err := db.Prepare("SELECT (.+) FROM articles WHERE id = ?")
+	if err != nil {
+		t.Errorf("Error '%s' was not expected while creating a prepared statement", err)
+	}
+
+	var id int
+	var title string
+
+	err = stmt.QueryRow(5).Scan(&id, &title)
+	if err != nil {
+		t.Errorf("Error '%s' was not expected querying row from statement and scanning", err)
+	}
+
+	if id != 5 {
+		t.Errorf("Expected mocked id to be 5, but got %d instead", id)
+	}
+
+	if title != "hello world" {
+		t.Errorf("Expected mocked title to be 'hello world', but got '%s' instead", title)
+	}
+
+	err = stmt.QueryRow(2).Scan(&id, &title)
+	if err != nil {
+		t.Errorf("Error '%s' was not expected querying row from statement and scanning", err)
+	}
+
+	if id != 2 {
+		t.Errorf("Expected mocked id to be 2, but got %d instead", id)
+	}
+
+	if title != "whoop" {
+		t.Errorf("Expected mocked title to be 'whoop', but got '%s' instead", title)
+	}
+
+	if err = db.Close(); err != nil {
+		t.Errorf("Error '%s' was not expected while closing the database", err)
+	}
+}
+
+func TestUnexpectedOperations(t *testing.T) {
+	db, err := sql.Open("mock", "")
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when opening a stub database connection", err)
+	}
+
+	stmt, err := db.Prepare("SELECT (.+) FROM articles WHERE id = ?")
+	if err != nil {
+		t.Errorf("Error '%s' was not expected while creating a prepared statement", err)
+	}
+
+	var id int
+	var title string
+
+	err = stmt.QueryRow(5).Scan(&id, &title)
+	if err == nil {
+		t.Error("Error was expected querying row, since there was no such expectation")
+	}
+
+	ExpectRollback()
+
+	err = db.Close()
+	if err == nil {
+		t.Error("Error was expected while closing the database, expectation was not fulfilled", err)
+	}
+}
+
+func TestWrongUnexpectations(t *testing.T) {
+	db, err := sql.Open("mock", "")
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when opening a stub database connection", err)
+	}
+
+	ExpectBegin()
+
+	rs1 := RowsFromCSVString([]string{"id", "title"}, "5,hello world")
+	ExpectQuery("SELECT (.+) FROM articles WHERE id = ?").
+		WithArgs(5).
+		WillReturnRows(rs1)
+
+	ExpectCommit().WillReturnError(errors.New("Deadlock occured"))
+	ExpectRollback() // won't be triggered
+
+	stmt, err := db.Prepare("SELECT (.+) FROM articles WHERE id = ? FOR UPDATE")
+	if err != nil {
+		t.Errorf("Error '%s' was not expected while creating a prepared statement", err)
+	}
+
+	var id int
+	var title string
+
+	err = stmt.QueryRow(5).Scan(&id, &title)
+	if err == nil {
+		t.Error("Error was expected while querying row, since there Begin transaction expectation is not fulfilled")
+	}
+
+	// lets go around and start transaction
+	tx, err := db.Begin()
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when beginning a transaction", err)
+	}
+
+	err = stmt.QueryRow(5).Scan(&id, &title)
+	if err != nil {
+		t.Errorf("Error '%s' was not expected while querying row, since transaction was started", err)
+	}
+
+	err = tx.Commit()
+	if err == nil {
+		t.Error("A deadlock error was expected when commiting a transaction", err)
+	}
+
+	err = db.Close()
+	if err == nil {
+		t.Error("Error was expected while closing the database, expectation was not fulfilled", err)
+	}
+}
+
+func TestExecExpectations(t *testing.T) {
+	db, err := sql.Open("mock", "")
+	if err != nil {
+		t.Errorf("An error '%s' was not expected when opening a stub database connection", err)
+	}
+
+	result := NewResult(1, 1)
+	ExpectExec("^INSERT INTO articles").
+		WithArgs("hello").
+		WillReturnResult(result)
+
+	res, err := db.Exec("INSERT INTO articles (title) VALUES (?)", "hello")
+	if err != nil {
+		t.Errorf("Error '%s' was not expected, while inserting a row", err)
+	}
+
+	id, err := res.LastInsertId()
+	if err != nil {
+		t.Errorf("Error '%s' was not expected, while getting a last insert id", err)
+	}
+
+	affected, err := res.RowsAffected()
+	if err != nil {
+		t.Errorf("Error '%s' was not expected, while getting affected rows", err)
+	}
+
+	if id != 1 {
+		t.Errorf("Expected last insert id to be 1, but got %d instead", id)
+	}
+
+	if affected != 1 {
+		t.Errorf("Expected affected rows to be 1, but got %d instead", affected)
+	}
+
+	if err = db.Close(); err != nil {
+		t.Errorf("Error '%s' was not expected while closing the database", err)
+	}
+}