fix: SET snapshot semantics and parenthesised arithmetic

Author: LeeroyHannigan

crates/core/src/expression/update_evaluator.rs

@@ -27,10 +27,13 @@ pub fn apply_update(
     item: &mut BTreeMap<String, AttributeValue>,
     maps: &ExpressionMaps,
 ) -> Result<(), DynamoDbError> {
-    // DynamoDB applies actions in order: SET, REMOVE, ADD, DELETE
+    // DynamoDB evaluates all SET RHS values against the pre-update snapshot,
+    // then applies the results. This means `SET a = :v, b = a` assigns `b`
+    // the *original* value of `a`, not the value just written by the first clause.
+    let snapshot = item.clone();
     for action in actions {
         if let UpdateAction::Set { path, value } = action {
-            let resolved_value = evaluate_set_value(value, item, maps)?;
+            let resolved_value = evaluate_set_value(value, &snapshot, maps)?;
             set_path(item, path, resolved_value, maps)?;
         }
     }

crates/core/src/expression/update_evaluator_tests.rs

@@ -341,3 +341,26 @@ fn set_intermediate_map_path_missing_fails() {
     let result = apply("SET a.b.c = :v", &mut item, HashMap::new(), values);
     assert!(result.is_err());
 }
+
+#[test]
+fn set_snapshot_semantics_second_clause_reads_old_value() {
+    let mut item = BTreeMap::new();
+    item.insert("pk".into(), AttributeValue::S("key1".into()));
+    item.insert("a".into(), AttributeValue::S("OLD".into()));
+    let mut values = HashMap::new();
+    values.insert("v".into(), AttributeValue::S("NEW".into()));
+    apply("SET a = :v, b = a", &mut item, HashMap::new(), values).unwrap();
+    assert_eq!(item.get("a"), Some(&AttributeValue::S("NEW".into())));
+    assert_eq!(item.get("b"), Some(&AttributeValue::S("OLD".into())));
+}
+
+#[test]
+fn set_parenthesised_arithmetic() {
+    let mut item = BTreeMap::new();
+    item.insert("pk".into(), AttributeValue::S("key1".into()));
+    item.insert("c".into(), AttributeValue::N("10".into()));
+    let mut values = HashMap::new();
+    values.insert("v".into(), AttributeValue::N("3".into()));
+    apply("SET c = (c - :v)", &mut item, HashMap::new(), values).unwrap();
+    assert_eq!(item.get("c"), Some(&AttributeValue::N("7".into())));
+}

crates/core/src/expression/update_parser.rs

@@ -239,6 +239,17 @@ fn parse_set_operand(tokens: &[Token], pos: &mut usize) -> Result<Expr, DynamoDb
             }
         }
         Token::NameRef(_) => parser_common::parse_path(tokens, pos).map(Expr::Path),
+        Token::LParen => {
+            *pos += 1; // consume '('
+            let expr = parse_value_expr(tokens, pos)?;
+            if *pos >= tokens.len() || tokens[*pos] != Token::RParen {
+                return Err(validation_err(
+                    "Invalid UpdateExpression: expected closing ')'",
+                ));
+            }
+            *pos += 1; // consume ')'
+            Ok(expr)
+        }
         _ => Err(validation_err(
             "Invalid UpdateExpression: expected path, placeholder, or function",
         )),

tests/test_set_evaluation.py

@@ -0,0 +1,76 @@
+# Copyright 2026 ExtendDB contributors
+# SPDX-License-Identifier: Apache-2.0
+
+"""Tests for SET evaluation snapshot semantics and parenthesised arithmetic."""
+
+from __future__ import annotations
+
+import uuid
+
+import pytest
+
+
+@pytest.fixture()
+def hash_table(dynamodb_client):
+    name = f"set_eval_{uuid.uuid4().hex[:8]}"
+    dynamodb_client.create_table(
+        TableName=name,
+        KeySchema=[{"AttributeName": "pk", "KeyType": "HASH"}],
+        AttributeDefinitions=[{"AttributeName": "pk", "AttributeType": "S"}],
+        BillingMode="PAY_PER_REQUEST",
+    )
+    from conftest import wait_for_active
+    wait_for_active(dynamodb_client, name)
+    yield name
+    dynamodb_client.delete_table(TableName=name)
+
+
+class TestSetSnapshotSemantics:
+    """SET clauses evaluate RHS against the pre-update item snapshot."""
+
+    def test_second_set_reads_old_value(self, dynamodb_client, hash_table):
+        dynamodb_client.put_item(
+            TableName=hash_table,
+            Item={"pk": {"S": "x"}, "a": {"S": "OLD"}},
+        )
+        resp = dynamodb_client.update_item(
+            TableName=hash_table,
+            Key={"pk": {"S": "x"}},
+            UpdateExpression="SET a = :v, b = a",
+            ExpressionAttributeValues={":v": {"S": "NEW"}},
+            ReturnValues="ALL_NEW",
+        )
+        assert resp["Attributes"]["a"]["S"] == "NEW"
+        assert resp["Attributes"]["b"]["S"] == "OLD"
+
+
+class TestParenthesisedArithmetic:
+    """SET supports parenthesised arithmetic expressions."""
+
+    def test_subtraction_in_parens(self, dynamodb_client, hash_table):
+        dynamodb_client.put_item(
+            TableName=hash_table,
+            Item={"pk": {"S": "y"}, "c": {"N": "10"}},
+        )
+        resp = dynamodb_client.update_item(
+            TableName=hash_table,
+            Key={"pk": {"S": "y"}},
+            UpdateExpression="SET c = (c - :v)",
+            ExpressionAttributeValues={":v": {"N": "3"}},
+            ReturnValues="ALL_NEW",
+        )
+        assert resp["Attributes"]["c"]["N"] == "7"
+
+    def test_addition_in_parens(self, dynamodb_client, hash_table):
+        dynamodb_client.put_item(
+            TableName=hash_table,
+            Item={"pk": {"S": "z"}, "n": {"N": "5"}},
+        )
+        resp = dynamodb_client.update_item(
+            TableName=hash_table,
+            Key={"pk": {"S": "z"}},
+            UpdateExpression="SET n = (n + :v)",
+            ExpressionAttributeValues={":v": {"N": "2"}},
+            ReturnValues="ALL_NEW",
+        )
+        assert resp["Attributes"]["n"]["N"] == "7"