From ab864e894fd40febfd41cb4ee15ce446fe278e3d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?G=C3=A1bor=20Lipt=C3=A1k?= <gliptak@gmail.com>
Date: Sat, 26 Nov 2016 15:38:28 -0500
Subject: [PATCH] Additional tests for prepare

---
 sqlmock_test.go | 74 ++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 73 insertions(+), 1 deletion(-)

diff --git a/sqlmock_test.go b/sqlmock_test.go
index 3cd8b66..aaa1b0a 100644
--- a/sqlmock_test.go
+++ b/sqlmock_test.go
@@ -6,6 +6,7 @@ import (
 	"sync"
 	"testing"
 	"time"
+	"github.com/golang/go/src/pkg/strconv"
 )
 
 func cancelOrder(db *sql.DB, orderID int) error {
@@ -861,9 +862,80 @@ func TestUnexpectedRollbackOrder(t *testing.T) {
 		return
 	}
 	mock.ExpectBegin()
-	mock.ExpectCommit()
+
 	tx, _ := db.Begin()
 	if err := tx.Rollback(); err == nil {
 		t.Error("an error was expected when calling rollback, but got none")
 	}
+}
+
+func TestPrepareExec(t *testing.T) {
+	// Open new mock database
+	db, mock, err := New()
+	if err != nil {
+		fmt.Println("error creating mock database")
+		return
+	}
+	defer db.Close()
+	mock.ExpectBegin()
+	ep := mock.ExpectPrepare("INSERT INTO ORDERS\\(ID, STATUS\\) VALUES \\(\\?, \\?\\)")
+	for i:=0; i < 3; i++ {
+		ep.ExpectExec().WillReturnResult(NewResult(1,1))
+	}
+	mock.ExpectCommit()
+	tx, _ := db.Begin()
+	stmt, err := tx.Prepare("INSERT INTO ORDERS(ID, STATUS) VALUES (?, ?)")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer stmt.Close()
+	for i:=0; i < 3; i++ {
+		_, err := stmt.Exec(i, "Hello" + strconv.Itoa(i))
+		if err != nil {
+			t.Fatal(err)
+		}
+	}
+	tx.Commit()
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Errorf("there were unfulfilled expections: %s", err)
+	}
+}
+
+func TestPrepareQuery(t *testing.T) {
+	// Open new mock database
+	db, mock, err := New()
+	if err != nil {
+		fmt.Println("error creating mock database")
+		return
+	}
+	defer db.Close()
+	mock.ExpectBegin()
+	ep := mock.ExpectPrepare("SELECT ID, STATUS FROM ORDERS WHERE ID = \\?")
+	ep.ExpectQuery().WithArgs(101).WillReturnRows(NewRows([]string{"ID", "STATUS"}).AddRow(101, "Hello"))
+	mock.ExpectCommit()
+	tx, _ := db.Begin()
+	stmt, err := tx.Prepare("SELECT ID, STATUS FROM ORDERS WHERE ID = ?")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer stmt.Close()
+	rows, err := stmt.Query(101)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var(
+			id int
+			status string
+		)
+		if rows.Scan(&id, &status); id != 101 || status != "Hello" {
+			t.Fatal("wrong query results")
+		}
+
+	}
+	tx.Commit()
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Errorf("there were unfulfilled expections: %s", err)
+	}
 }
\ No newline at end of file