From 8b269836c2cc35e107f6f6d22cbea9108d3b285e Mon Sep 17 00:00:00 2001
From: Jeff Baskin <jeff.l.baskin@gmail.com>
Date: Mon, 10 Nov 2025 15:36:56 -0500
Subject: [PATCH] Added method of updating the name id.

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

diff --git a/src/message.rs b/src/message.rs
index d025059..3a6007e 100644
--- a/src/message.rs
+++ b/src/message.rs
@@ -291,6 +291,10 @@ impl Message {
         )
     }
 
+    fn reset_name_id<NT>(&mut self, name: NT) where NT: Into<NameType> {
+        self.document_id = name.into();
+    }
+
     fn response<A>(&self, action: A) -> Self
     where
         A: Into<MsgAction>,
@@ -414,6 +418,18 @@ mod messages {
             _ => unreachable!("got {:?}: should have received a reply", action2),
         }
     }
+
+    #[test]
+    fn can_reset_document_id() {
+        let mut msg = Message::new(Name::english("something"), Query::new());
+        let id = Uuid::new_v4();
+        msg.reset_name_id(&id);
+        let name_id = msg.get_document_id();
+        match name_id {
+            NameType::ID(data) => assert_eq!(data, &id),
+            _ => unreachable!("got {:?}, should have been an id", name_id),
+        }
+    }
 }
 
 #[derive(Clone, Debug, Eq, Hash)]