From d81d53c1f70446a51899b9e6ad3861008b9b0a27 Mon Sep 17 00:00:00 2001
From: joalopez1206 <joalopez@ug.uchile.cl>
Date: Tue, 1 Oct 2024 18:09:37 -0300
Subject: [PATCH] new test

---
 tests/edns_test.rs | 41 ++++++++++++++++++++++++++++++++++-------
 1 file changed, 34 insertions(+), 7 deletions(-)

diff --git a/tests/edns_test.rs b/tests/edns_test.rs
index 392d6032..ee9ad5ee 100644
--- a/tests/edns_test.rs
+++ b/tests/edns_test.rs
@@ -28,12 +28,12 @@ async fn query_from_ip_with_edns(domain_name: &str,
     response.map(|lookup_response| lookup_response.to_dns_msg())
 }
 
-async fn query_response_edns(domain_name: &str, 
-    protocol: &str, 
-    qtype: &str, 
-    max_payload: Option<u16>, 
-    version: u8, 
-    do_bit: bool, 
+async fn query_response_edns(domain_name: &str,
+    protocol: &str,
+    qtype: &str,
+    max_payload: Option<u16>,
+    version: u8,
+    do_bit: bool,
     option: Option<Vec<u16>>) -> Result<DnsMessage, ClientError> {
 
     let mut config = ResolverConfig::default();
@@ -152,5 +152,32 @@ async fn query_a_type_with_rrsig_edns() {
         assert_eq!(opt.get_rtype(), Rrtype::OPT);
         assert_eq!(opt.get_rclass(), Rclass::UNKNOWN(512));
         println!("{:?}", opt);
-    } 
+    }
+}
+
+#[tokio::test]
+async fn query_from_root() {
+    const ROOTSV1: [u8; 4] = [192,58,128,30];
+
+    let mut ip2req = ROOTSV1.into();
+    let response = query_from_ip_with_edns("example.com",
+                                           "UDP", "A", Some(1024), 0, true,
+                                           Some(vec![3]), ip2req).await;
+    let response = match response {
+        Ok(rrs) => rrs,
+        Err(e) => panic!("{:?}", e),
+    };
+
+    println!("{}", response);
+
+    let additional_rrs = response.get_additional();
+
+    let a_rrs: Vec<_> = additional_rrs.iter()
+        .filter(|arrs|
+            if let Rdata::A(_) = arrs.get_rdata() {true}
+            else {false}).collect();
+
+    if let Rdata::A(rdata) = a_rrs[4].get_rdata() {
+        ip2req = rdata.get_address();
+    }
 }
\ No newline at end of file