From 645bbfbb7d8482db2b660b8ee976b5f22d59dfc5 Mon Sep 17 00:00:00 2001
From: Neenad Ingole <needna78@gmail.com>
Date: Mon, 12 Aug 2024 20:36:12 +0200
Subject: [PATCH] Add test for WillReturnRows function returns Nil

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

diff --git a/expectations_test.go b/expectations_test.go
index afda582..d99f1ab 100644
--- a/expectations_test.go
+++ b/expectations_test.go
@@ -101,3 +101,26 @@ func TestCustomValueConverterQueryScan(t *testing.T) {
 		t.Error(err)
 	}
 }
+
+func TestQueryWillReturnsNil(t *testing.T) {
+	t.Parallel()
+
+	db, mock, err := New()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+
+	defer func() {
+		if err := recover(); err != nil {
+			t.Error(err)
+		}
+	}()
+
+	mock.ExpectQuery("SELECT (.+) FROM users WHERE (.+)").WithArgs("test").WillReturnRows(nil)
+	query := "SELECT name, email FROM users WHERE name = ?"
+	_, err = mock.(*sqlmock).Query(query, []driver.Value{"test"})
+	if err != nil {
+		t.Error(err)
+	}
+}