From a6988e55f6c550335b7362bc5183776071b04a65 Mon Sep 17 00:00:00 2001
From: Luna Yao <40349250+ZnqbuZ@users.noreply.github.com>
Date: Mon, 6 Apr 2026 02:01:23 +0200
Subject: [PATCH] zone: some tests

---
 easytier/src/dns/zone.rs | 55 +++++++++++++++++++++++++++++++---------
 1 file changed, 43 insertions(+), 12 deletions(-)

diff --git a/easytier/src/dns/zone.rs b/easytier/src/dns/zone.rs
index e812e9a3..dcf2d3a8 100644
--- a/easytier/src/dns/zone.rs
+++ b/easytier/src/dns/zone.rs
@@ -253,6 +253,21 @@ mod tests {
 
     "#;
 
+    fn query(name: &str, rtype: RecordType) -> anyhow::Result<MessageRequest> {
+        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(name)?, rtype));
+
+        let mut request = Vec::new();
+        let mut encoder = BinEncoder::new(&mut request);
+        query.emit(&mut encoder)?;
+
+        Ok(MessageRequest::from_bytes(&request)?)
+    }
+
     // #[tokio::test]
     #[tokio::test(flavor = "current_thread")]
     async fn test_config() -> anyhow::Result<()> {
@@ -343,19 +358,8 @@ mod tests {
         authorities.extend(zone.create_forward_authority().into_iter());
         catalog.upsert(zone.origin.clone(), authorities);
 
-        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("et.top.")?, 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)?,
+            query("et.top.", RecordType::A)?,
             SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
             Protocol::Udp,
         );
@@ -365,6 +369,33 @@ mod tests {
 
         assert_eq!(info.response_code(), ResponseCode::NoError);
 
+        catalog.upsert(
+            Name::root().into(),
+            vec![Zone::system().create_forward_authority().unwrap()],
+        );
+
+        let request = Request::new(
+            query("example.com", RecordType::A)?,
+            SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
+            Protocol::Udp,
+        );
+
+        let response = ResponseHandle::new(512);
+        let info = catalog.lookup(&request, None, response.clone()).await;
+
+        assert_eq!(info.response_code(), ResponseCode::NoError);
+
+        let request = Request::new(
+            query("example.invalid", RecordType::A)?,
+            SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
+            Protocol::Udp,
+        );
+
+        let response = ResponseHandle::new(512);
+        let info = catalog.lookup(&request, None, response.clone()).await;
+
+        assert_eq!(info.response_code(), ResponseCode::NXDomain);
+
         let socket = UdpSocket::bind("127.0.0.1:0").await?;
         let addr = socket.local_addr()?;
         println!("listening on {}", addr);