test: postgre update

Refs: #242
phnx47 · Jun 1, 2023 · 12353c8 · 12353c8
1 parent be53749
commit 12353c8
Showing 1 changed file with 34 additions and 1 deletion.
diff --git a/tests/SqlGenerator.Tests/PostgresSqlGeneratorTests.cs b/tests/SqlGenerator.Tests/PostgresSqlGeneratorTests.cs
@@ -74,7 +74,8 @@ public void SelectOrderByWithTableIdentifier_QuoMarks()
         filterData.OrderInfo = data;
 
         var sqlQuery = sqlGenerator.GetSelectAll(x => x.Identifier == Guid.Empty, filterData);
-        Assert.Equal("SELECT \"Cities\".\"Identifier\", \"Cities\".\"Name\" FROM \"Cities\" WHERE \"Cities\".\"Identifier\" = @Identifier_p0 ORDER BY \"Cities\".\"Name\" ASC", sqlQuery.GetSql());
+        Assert.Equal("SELECT \"Cities\".\"Identifier\", \"Cities\".\"Name\" FROM \"Cities\" WHERE \"Cities\".\"Identifier\" = @Identifier_p0 ORDER BY \"Cities\".\"Name\" ASC",
+            sqlQuery.GetSql());
     }
 
     [Fact]
@@ -107,6 +108,38 @@ public void SelectFirst_QuoMarks()
         Assert.Equal("SELECT \"Cities\".\"Identifier\", \"Cities\".\"Name\" FROM \"Cities\" WHERE \"Cities\".\"Identifier\" = @Identifier_p0 LIMIT 1", sqlQuery.GetSql());
     }
 
+    [Fact]
+    public static void Update()
+    {
+        ISqlGenerator<User> userSqlGenerator = new SqlGenerator<User>(_sqlConnector);
+        var sqlQuery = userSqlGenerator.GetUpdate(new User());
+
+        Assert.Equal("UPDATE Users " +
+                     "SET Name = @UserName, " +
+                     "AddressId = @UserAddressId, " +
+                     "PhoneId = @UserPhoneId, " +
+                     "OfficePhoneId = @UserOfficePhoneId, " +
+                     "Deleted = @UserDeleted, " +
+                     "UpdatedAt = @UserUpdatedAt " +
+                     "WHERE Users.Id = @UserId", sqlQuery.GetSql());
+    }
+
+    [Fact]
+    public static void Update_QuoMarks()
+    {
+        ISqlGenerator<User> userSqlGenerator = new SqlGenerator<User>(_sqlConnector, true);
+        var sqlQuery = userSqlGenerator.GetUpdate(new User());
+
+        Assert.Equal("UPDATE \"Users\" " +
+                     "SET \"Name\" = @UserName, " +
+                     "\"AddressId\" = @UserAddressId, " +
+                     "\"PhoneId\" = @UserPhoneId, " +
+                     "\"OfficePhoneId\" = @UserOfficePhoneId, " +
+                     "\"Deleted\" = @UserDeleted, " +
+                     "\"UpdatedAt\" = @UserUpdatedAt " +
+                     "WHERE \"Users\".\"Id\" = @UserId", sqlQuery.GetSql());
+    }
+
     [Fact]
     public static void BulkUpdate()
     {