From cda6cc1c8c66fd8a7c9552c240909aacd4044e32 Mon Sep 17 00:00:00 2001
From: Jeff Baskin <jeff.l.baskin@gmail.com>
Date: Thu, 28 Aug 2025 10:37:42 -0400
Subject: [PATCH] Update only matches query.

---
 src/message.rs | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/src/message.rs b/src/message.rs
index 9ff1425..bb2d1fa 100644
--- a/src/message.rs
+++ b/src/message.rs
@@ -2410,6 +2410,51 @@ mod document_files {
             _ => unreachable!("got {:?}: should have gotten a reply", action),
         }
     }
+
+    #[test]
+    fn changes_only_the_queried() {
+        let (mut docdef, doc_name) =
+            create_docdef([FieldType::Uuid, FieldType::StaticString].to_vec());
+        let (mut queue, rx) = test_doc(doc_name.as_str(), docdef, standard_routes());
+        let id1 = Uuid::new_v4();
+        let id2 = Uuid::new_v4();
+        let old = "old";
+        let new = "new";
+        for id in [id1.clone(), id2.clone()].into_iter() {
+            let mut addition = Addition::new();
+            addition.add_field("field0".to_string(), id);
+            addition.add_field("field1".to_string(), old);
+            let msg = Message::new(doc_name.clone(), addition);
+            queue.send(msg).unwrap();
+            rx.recv_timeout(TIMEOUT).unwrap();
+        }
+        let mut update = Update::new();
+        update
+            .get_query_mut()
+            .add_specifier("field0".to_string(), Operand::Equal, id1.clone());
+        update.get_values_mut().add_field("field1".to_string(), new);
+        let msg = Message::new(doc_name.clone(), update);
+        queue.send(msg).unwrap();
+        rx.recv_timeout(TIMEOUT).unwrap();
+        let mut query = Query::new();
+        query.add_specifier("field0".to_string(), Operand::Equal, id2.clone());
+        let msg = Message::new(doc_name, query);
+        queue.send(msg).unwrap();
+        let result = rx.recv_timeout(TIMEOUT).unwrap();
+        let action = result.get_action();
+        match action {
+            MsgAction::Reply(docs) => {
+                assert_eq!(docs.len(), 1);
+                let expected_id: Field = id2.into();
+                let output: Field = old.into();
+                for doc in docs.iter() {
+                    assert_eq!(doc.get_field("field0").unwrap(), &expected_id);
+                    assert_eq!(doc.get_field("field1").unwrap(), &output);
+                }
+            }
+            _ => unreachable!("got {:?}: should have gotten a reply", action),
+        }
+    }
 }
 
 #[cfg(test)]