impl ToSqlString for UPDATE stmt + tests

vendored/sqlite3-parser/src/to_sql_string/stmt/mod.rs

@@ -9,6 +9,7 @@ mod create_virtual_table;
 mod delete;
 mod insert;
 mod select;
+mod update;
 
 impl ToSqlString for ast::Stmt {
     fn to_sql_string<C: super::ToSqlContext>(&self, context: &C) -> String {
@@ -184,6 +185,7 @@ impl ToSqlString for ast::Stmt {
             ),
             Self::Savepoint(name) => format!("SAVEPOINT {};", name.0),
             Self::Select(select) => format!("{};", select.to_sql_string(context)),
+            Self::Update(update) => format!("{};", update.to_sql_string(context)),
             _ => todo!(),
         }
     }

vendored/sqlite3-parser/src/to_sql_string/stmt/update.rs

@@ -0,0 +1,144 @@
+use crate::{ast, to_sql_string::ToSqlString};
+
+impl ToSqlString for ast::Update {
+    fn to_sql_string<C: crate::to_sql_string::ToSqlContext>(&self, context: &C) -> String {
+        format!(
+            "{}UPDATE {}{}{} SET {}{}{}{}",
+            self.with.as_ref().map_or("".to_string(), |with| format!(
+                "{} ",
+                with.to_sql_string(context)
+            )),
+            self.or_conflict.map_or("".to_string(), |conflict| format!(
+                "OR {} ",
+                conflict.to_sql_string(context)
+            )),
+            self.tbl_name.to_sql_string(context),
+            self.indexed
+                .as_ref()
+                .map_or("".to_string(), |indexed| format!(
+                    " {}",
+                    indexed.to_sql_string(context)
+                )),
+            self.sets
+                .iter()
+                .map(|set| set.to_sql_string(context))
+                .collect::<Vec<_>>()
+                .join(", "),
+            self.from.as_ref().map_or("".to_string(), |from| format!(
+                " {}",
+                from.to_sql_string(context)
+            )),
+            self.where_clause
+                .as_ref()
+                .map_or("".to_string(), |expr| format!(
+                    " WHERE {}",
+                    expr.to_sql_string(context)
+                )),
+            self.returning
+                .as_ref()
+                .map_or("".to_string(), |returning| format!(
+                    " RETURNING {}",
+                    returning
+                        .iter()
+                        .map(|col| col.to_sql_string(context))
+                        .collect::<Vec<_>>()
+                        .join(", ")
+                ))
+        )
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::to_sql_string_test;
+
+    // Basic UPDATE with a single column
+    to_sql_string_test!(
+        test_update_single_column,
+        "UPDATE employees SET salary = 55000;"
+    );
+
+    // UPDATE with multiple columns
+    to_sql_string_test!(
+        test_update_multiple_columns,
+        "UPDATE employees SET salary = 60000, name = 'John Smith';"
+    );
+
+    // UPDATE with a WHERE clause
+    to_sql_string_test!(
+        test_update_with_where,
+        "UPDATE employees SET salary = 60000 WHERE id = 1;"
+    );
+
+    // UPDATE with multiple WHERE conditions
+    to_sql_string_test!(
+        test_update_with_multi_where,
+        "UPDATE employees SET salary = 65000 WHERE department_id = 3 AND salary < 50000;"
+    );
+
+    // UPDATE with a subquery in SET
+    to_sql_string_test!(
+        test_update_with_subquery_set,
+        "UPDATE employees SET department_id = (SELECT id FROM departments WHERE name = 'Sales') WHERE id = 1;"
+    );
+
+    // UPDATE with a subquery in WHERE
+    to_sql_string_test!(
+        test_update_with_subquery_where,
+        "UPDATE employees SET salary = 70000 WHERE department_id IN (SELECT id FROM departments WHERE name = 'Marketing');"
+    );
+
+    // UPDATE with EXISTS clause
+    to_sql_string_test!(
+        test_update_with_exists,
+        "UPDATE employees SET salary = 75000 WHERE EXISTS (SELECT 1 FROM orders WHERE orders.employee_id = employees.id AND orders.status = 'pending');"
+    );
+
+    // UPDATE with FROM clause (join-like behavior)
+    to_sql_string_test!(
+        test_update_with_from,
+        "UPDATE employees SET salary = 80000 FROM departments WHERE employees.department_id = departments.id AND departments.name = 'Engineering';"
+    );
+
+    // UPDATE with RETURNING clause
+    to_sql_string_test!(
+        test_update_with_returning,
+        "UPDATE employees SET salary = 60000 WHERE id = 1 RETURNING id, name, salary;"
+    );
+
+    // UPDATE with expression in SET
+    to_sql_string_test!(
+        test_update_with_expression,
+        "UPDATE employees SET salary = salary * 1.1 WHERE department_id = 2;"
+    );
+
+    // UPDATE with NULL value
+    to_sql_string_test!(
+        test_update_with_null,
+        "UPDATE employees SET department_id = NULL WHERE id = 1;"
+    );
+
+    // UPDATE with schema-qualified table
+    to_sql_string_test!(
+        test_update_schema_qualified,
+        "UPDATE main.employees SET salary = 65000 WHERE id = 1;"
+    );
+
+    // UPDATE with CASE expression
+    to_sql_string_test!(
+        test_update_with_case,
+        "UPDATE employees SET salary = CASE WHEN salary < 50000 THEN 55000 ELSE salary * 1.05 END WHERE department_id = 3;"
+    );
+
+    // UPDATE with LIKE clause in WHERE
+    to_sql_string_test!(
+        test_update_with_like,
+        "UPDATE employees SET name = 'Updated' WHERE name LIKE 'J%';"
+    );
+
+    // UPDATE with ON CONFLICT (upsert-like behavior)
+    to_sql_string_test!(
+        test_update_with_on_conflict,
+        "INSERT INTO employees (id, name, salary) VALUES (1, 'John Doe', 50000) ON CONFLICT(id) DO UPDATE SET name = excluded.name, salary = excluded.salary;"
+    );
+}