From 7cbc0ad8e85ae885d741e1a9de2b964589876867 Mon Sep 17 00:00:00 2001
From: Litr0 <gonzalo.sobarzoa@gmail.com>
Date: Wed, 17 Jul 2024 17:51:37 -0400
Subject: [PATCH] test from and to bytes in rdata for srv rdata.

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

diff --git a/src/message/rdata.rs b/src/message/rdata.rs
index 485986c2..c186c080 100644
--- a/src/message/rdata.rs
+++ b/src/message/rdata.rs
@@ -403,6 +403,7 @@ mod resolver_query_tests {
     use super::nsec3param_rdata::Nsec3ParamRdata;
     use super::tsig_rdata::TSigRdata;
     use super::aaaa_rdata::AAAARdata;
+    use super::srv_rdata::SrvRdata;
     use std::net::IpAddr;
     use std::vec;
 
@@ -760,6 +761,17 @@ mod resolver_query_tests {
         assert_eq!(bytes, expected_bytes);
     }
 
+    #[test]
+    fn to_bytes_srv_rdata(){
+        let srv_rdata = SrvRdata::new_with_values(1, 2, 3, DomainName::new_from_str("example.com"));
+
+        let expected_bytes = [0, 1, 0, 2, 0, 3, 7, 101, 120, 97, 109, 112, 108, 101, 3, 99, 111, 109, 0];
+
+        let bytes = Rdata::SRV(srv_rdata).to_bytes();
+
+        assert_eq!(bytes, expected_bytes);
+    }
+
     //from bytes tests
     #[test]
     fn from_bytes_a_ch_rdata(){
@@ -1108,6 +1120,21 @@ mod resolver_query_tests {
         } 
     }
 
+    #[test]
+    fn from_bytes_srv_rdata(){
+        let data_bytes = [0, 1, 0, 2, 0, 3, 7, 101, 120, 97, 109, 112, 108, 101, 3, 99, 111, 109, 0, 0, 33, 0, 1];
+        let rdata = Rdata::from_bytes(&data_bytes, &data_bytes).unwrap();
+        match rdata {
+            Rdata::SRV(val) => {
+                assert_eq!(val.get_priority(), 1);
+                assert_eq!(val.get_weight(), 2);
+                assert_eq!(val.get_port(), 3);
+                assert_eq!(val.get_target().get_name(), "example.com");
+            }
+            _ => {}
+        }
+    }
+
     #[test]
     #[should_panic]
     fn from_bytes_format_error(){