From 6099e2fbb44743aebbff5a73b9a357cefd812c27 Mon Sep 17 00:00:00 2001
From: Benjamin Rupp <brupp@ciena.com>
Date: Wed, 25 Sep 2019 09:11:30 -0400
Subject: [PATCH] Adding unit tests for Query/Scan and Exec mocking, using
 custom []string data type

---
 expectations_test.go | 92 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)

diff --git a/expectations_test.go b/expectations_test.go
index 8d2f6d7..facf43c 100644
--- a/expectations_test.go
+++ b/expectations_test.go
@@ -1,8 +1,11 @@
 package sqlmock
 
 import (
+	"context"
 	"database/sql/driver"
+	"errors"
 	"fmt"
+	"reflect"
 	"testing"
 	"time"
 )
@@ -137,3 +140,92 @@ func TestBuildQuery(t *testing.T) {
 		t.Error(err)
 	}
 }
+
+type CustomConverter struct{}
+
+func (s CustomConverter) ConvertValue(v interface{}) (driver.Value, error) {
+	switch v.(type) {
+	case string:
+		return v.(string), nil
+	case []string:
+		return v.([]string), nil
+	case int:
+		return v.(int), nil
+	default:
+		return nil, errors.New(fmt.Sprintf("cannot convert %T with value %v", v, v))
+	}
+}
+func TestCustomValueConverterQueryScan(t *testing.T) {
+	db, mock, _ := New(ValueConverterOption(CustomConverter{}))
+	query := `
+		SELECT
+			name,
+			email,
+			address,
+			anotherfield
+		FROM user
+		where
+			name    = 'John'
+			and
+			address = 'Jakarta'
+
+	`
+	expectedStringValue := "ValueOne"
+	expectedIntValue := 2
+	expectedArrayValue := []string{"Three", "Four"}
+	mock.ExpectQuery(query).WillReturnRows(mock.NewRows([]string{"One", "Two", "Three"}).AddRow(expectedStringValue, expectedIntValue, []string{"Three", "Four"}))
+	row := db.QueryRow(query)
+	var stringValue string
+	var intValue int
+	var arrayValue []string
+	if e := row.Scan(&stringValue, &intValue, &arrayValue); e != nil {
+		t.Error(e)
+	}
+	if stringValue != expectedStringValue {
+		t.Errorf("Expectation %s does not met: %s", expectedStringValue, stringValue)
+	}
+	if intValue != expectedIntValue {
+		t.Errorf("Expectation %d does not met: %d", expectedIntValue, intValue)
+	}
+	if !reflect.DeepEqual(expectedArrayValue, arrayValue) {
+		t.Errorf("Expectation %v does not met: %v", expectedArrayValue, arrayValue)
+	}
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Error(err)
+	}
+}
+
+func TestCustomValueConverterExec(t *testing.T) {
+	db, mock, _ := New(ValueConverterOption(CustomConverter{}))
+	expectedQuery := "INSERT INTO tags \\(name,email,age,hobbies\\) VALUES \\(\\?,\\?,\\?,\\?\\)"
+	query := "INSERT INTO tags (name,email,age,hobbies) VALUES (?,?,?,?)"
+	name := "John"
+	email := "j@jj.j"
+	age := 12
+	hobbies := []string{"soccer", "netflix"}
+	mock.ExpectBegin()
+	mock.ExpectPrepare(expectedQuery)
+	mock.ExpectExec(expectedQuery).WithArgs(name, email, age, hobbies).WillReturnResult(NewResult(1, 1))
+	mock.ExpectCommit()
+
+	ctx := context.Background()
+	tx, e := db.BeginTx(ctx, nil)
+	if e != nil {
+		t.Error(e)
+		return
+	}
+	stmt, e := db.PrepareContext(ctx, query)
+	if e != nil {
+		t.Error(e)
+		return
+	}
+	_, e = stmt.Exec(name, email, age, hobbies)
+	if e != nil {
+		t.Error(e)
+		return
+	}
+	tx.Commit()
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Error(err)
+	}
+}