From 599aad63a1d6960813dbee05bd2eba19d4329623 Mon Sep 17 00:00:00 2001
From: rankgice <209192670@qq.com>
Date: Mon, 30 Sep 2024 19:02:57 +0800
Subject: [PATCH] Added test file raw_test.go

---
 tests/raw_test.go | 110 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100644 tests/raw_test.go

diff --git a/tests/raw_test.go b/tests/raw_test.go
new file mode 100644
index 00000000..1e70f0ac
--- /dev/null
+++ b/tests/raw_test.go
@@ -0,0 +1,110 @@
+package tests_test
+
+import (
+	. "gorm.io/gorm/utils/tests"
+	"testing"
+)
+
+func TestRawSelect(t *testing.T) {
+	users := []User{
+		*GetUser("raw1", Config{}),
+		*GetUser("raw2", Config{}),
+		*GetUser("raw3", Config{}),
+		*GetUser("@name", Config{}),
+		*GetUser("@age", Config{}),
+	}
+
+	if err := DB.Create(&users).Error; err != nil {
+		t.Fatalf("errors happened when create users: %v", err)
+	}
+	tests := []struct {
+		TestName string
+		Sql      string
+		args     map[string]interface{}
+		Expect   []User
+		errFlag  bool
+	}{
+		{
+			"raw_test1",
+			`select * from users where name like @name and age = 18`,
+			map[string]interface{}{
+				"name": "raw1",
+			},
+			[]User{
+				users[0],
+			},
+			false,
+		},
+		{
+			"raw_test2",
+			`select * from users where name like @name and age = 18`,
+			map[string]interface{}{
+				"name": "@name",
+			},
+			[]User{
+				users[3],
+			},
+			false,
+		},
+		{
+			"raw_test3",
+			`select * from users where name like @name and age = 18`,
+			map[string]interface{}{
+				"name": "@age",
+			},
+			[]User{
+				users[4],
+			},
+			false,
+		},
+		{
+			"raw_test4",
+			`select * from users where name like "@name" and age = 18`,
+			map[string]interface{}{
+				"name": "raw1",
+			},
+			[]User{
+				users[3],
+			},
+			false,
+		},
+		{
+			"raw_test5",
+			`select * from users where name like "@name" and age = 18`,
+			map[string]interface{}{
+				"name": "@raw",
+			},
+			[]User{
+				users[3],
+			},
+			false,
+		},
+		{
+			"raw_test6",
+			`select * from users where name like "@name" and age = 18`,
+			map[string]interface{}{
+				"name": "@age",
+			},
+			[]User{
+				users[3],
+			},
+			false,
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.TestName, func(t *testing.T) {
+			var results []User
+			if err := DB.Raw(test.Sql, test.args).Scan(&results).Error; err != nil && !test.errFlag {
+				t.Errorf("errors %s: %v", test.TestName, err)
+			} else {
+				if len(results) != len(test.Expect) {
+					t.Errorf("errors %s: %v", test.TestName, err)
+				} else {
+					for i := 0; i < len(results); i++ {
+						CheckUser(t, results[i], test.Expect[i])
+					}
+				}
+			}
+		})
+	}
+}