From 7849ef3b8f6f9a3bd84cef3887ca6db598622e54 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sat, 22 Jul 2023 10:25:08 +0800
Subject: [PATCH] Fix test

---
 internal/statements/order_by.go  | 2 +-
 internal/statements/statement.go | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/internal/statements/order_by.go b/internal/statements/order_by.go
index 237f4a3d..feab0b80 100644
--- a/internal/statements/order_by.go
+++ b/internal/statements/order_by.go
@@ -28,7 +28,7 @@ func (statement *Statement) writeOrderBy(w builder.Writer, orderBy orderBy) erro
 		w.Append(t.Args()...)
 		return nil
 	case string:
-		if _, err := fmt.Fprint(w, t); err != nil {
+		if _, err := fmt.Fprint(w, statement.ReplaceQuote(t)); err != nil {
 			return err
 		}
 		w.Append(orderBy.orderArgs...)
diff --git a/internal/statements/statement.go b/internal/statements/statement.go
index 4d0d3761..2ca78182 100644
--- a/internal/statements/statement.go
+++ b/internal/statements/statement.go
@@ -167,15 +167,15 @@ func (statement *Statement) Reset() {
 
 // SQL adds raw sql statement
 func (statement *Statement) SQL(query interface{}, args ...interface{}) *Statement {
-	switch query.(type) {
+	switch t := query.(type) {
 	case (*builder.Builder):
 		var err error
-		statement.RawSQL, statement.RawParams, err = query.(*builder.Builder).ToSQL()
+		statement.RawSQL, statement.RawParams, err = t.ToSQL()
 		if err != nil {
 			statement.LastError = err
 		}
 	case string:
-		statement.RawSQL = query.(string)
+		statement.RawSQL = t
 		statement.RawParams = args
 	default:
 		statement.LastError = ErrUnSupportedSQLType