From 8e2fda467fd98e58839eeaa9cde955196bfb1d28 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Wed, 25 Oct 2023 15:37:00 +0800
Subject: [PATCH] enable for update test for postgres

---
 internal/statements/query.go | 4 ++--
 tests/tests.go               | 1 -
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/internal/statements/query.go b/internal/statements/query.go
index 8a9e59e4..674f8712 100644
--- a/internal/statements/query.go
+++ b/internal/statements/query.go
@@ -233,8 +233,8 @@ func (statement *Statement) writeForUpdate(w *builder.BytesWriter) error {
 		return nil
 	}
 
-	if statement.dialect.URI().DBType != schemas.MYSQL {
-		return errors.New("only support mysql for update")
+	if statement.dialect.URI().DBType != schemas.MYSQL && statement.dialect.URI().DBType != schemas.POSTGRES {
+		return errors.New("only support mysql and postgres for update")
 	}
 	_, err := fmt.Fprint(w, " FOR UPDATE")
 	return err
diff --git a/tests/tests.go b/tests/tests.go
index 220e1c67..e0a760e4 100644
--- a/tests/tests.go
+++ b/tests/tests.go
@@ -88,7 +88,6 @@ func createEngine(dbType, connStr string) error {
 					}
 				}
 				db.Close()
-				*ignoreSelectUpdate = true
 			case schemas.MYSQL:
 				db, err := sql.Open(dbType, strings.ReplaceAll(connStr, "xorm_test", "mysql"))
 				if err != nil {