From ffa7ca38506b1225b1f77e0ecff5770f15f20413 Mon Sep 17 00:00:00 2001
From: Luna Yao <40349250+ZnqbuZ@users.noreply.github.com>
Date: Wed, 1 Apr 2026 01:15:36 +0200
Subject: [PATCH] node_mgr: test

---
 easytier/src/dns/node_mgr.rs | 77 ++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)

diff --git a/easytier/src/dns/node_mgr.rs b/easytier/src/dns/node_mgr.rs
index 5079df33..c56148e1 100644
--- a/easytier/src/dns/node_mgr.rs
+++ b/easytier/src/dns/node_mgr.rs
@@ -165,3 +165,80 @@ impl DnsNodeMgrRpc for DnsNodeMgr {
         Ok(HeartbeatResponse { resync })
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::dns::utils::response::ResponseHandle;
+    use hickory_proto::op::{Message, MessageType, OpCode, Query, ResponseCode};
+    use hickory_proto::rr::{rdata, Name, RData, RecordType};
+    use hickory_proto::serialize::binary::{BinDecodable, BinDecoder, BinEncodable, BinEncoder};
+    use hickory_proto::xfer::Protocol;
+    use hickory_server::authority::{MessageRequest, MessageResponse};
+    use hickory_server::server::Request;
+    use std::net::{Ipv4Addr, SocketAddrV4};
+    use crate::common::log;
+
+    #[tokio::test]
+    async fn test_force_insert_node_info_then_catalog_lookup() -> anyhow::Result<()> {
+        log::tests::init();
+
+        let mgr = DnsNodeMgr::new();
+
+        let zone = Zone::try_from(&crate::proto::dns::ZoneData {
+            id: Some(Uuid::new_v4().into()),
+            origin: "catalog.test".to_string(),
+            ttl: 60,
+            records: vec!["@ IN A 10.20.30.40".to_string()],
+            forwarders: vec![],
+        })?;
+
+        mgr.nodes
+            .insert(
+                Uuid::new_v4(),
+                DnsNodeInfo {
+                    digest: vec![],
+                    zones: vec![zone].into(),
+                    addresses: Default::default(),
+                    listeners: Default::default(),
+                },
+            )
+            .await;
+
+        let mut query = Message::new();
+        query.set_id(0x1234);
+        query.set_message_type(MessageType::Query);
+        query.set_op_code(OpCode::Query);
+        query.set_recursion_desired(true);
+        query.add_query(Query::query(
+            Name::from_ascii("catalog.test.")?,
+            RecordType::A,
+        ));
+
+        let mut request = Vec::new();
+        let mut encoder = BinEncoder::new(&mut request);
+        query.emit(&mut encoder)?;
+
+        let request = Request::new(
+            MessageRequest::from_bytes(&request)?,
+            SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
+            Protocol::Udp,
+        );
+
+        let response = ResponseHandle::new(512);
+        let info = mgr.catalog().lookup(&request, None, response.clone()).await;
+
+        assert_eq!(info.response_code(), ResponseCode::NoError);
+
+        let response = response.into_inner().unwrap();
+        let message = Message::from_vec(&response)?;
+        assert!(message.answers().iter().any(|record| {
+            matches!(
+                record.data(),
+                RData::A(addr) if *addr == rdata::a::A(Ipv4Addr::new(10, 20, 30, 40))
+            )
+        }));
+
+        Ok(())
+    }
+}