From 26fbc0b394ae618f055e9ae6c24faec7ee1e8e9c Mon Sep 17 00:00:00 2001 From: NegroDCC Date: Thu, 11 Jan 2024 11:57:38 -0300 Subject: [PATCH] add tests max values from bytes and to bytes in DnsKey --- src/message/rdata/dnskey_rdata.rs | 33 +++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) diff --git a/src/message/rdata/dnskey_rdata.rs b/src/message/rdata/dnskey_rdata.rs index 8c139145..b5606fa7 100644 --- a/src/message/rdata/dnskey_rdata.rs +++ b/src/message/rdata/dnskey_rdata.rs @@ -243,4 +243,37 @@ mod dnskey_rdata_test{ assert_eq!(Err("Format Error"), result); } + + #[test] + fn max_values_from_bytes_test(){ + let mut dnskey_rdata = DnskeyRdata::new(); + //Max value of 2 bytes is 65535 (16 ones in the 2 bytes) + dnskey_rdata.set_flags(65535); + dnskey_rdata.set_protocol(255); + dnskey_rdata.set_algorithm(255); + dnskey_rdata.set_public_key(vec![255, 255]); + + let bytes_test: Vec = vec![255, 255, 255, 255, 255, 255]; + + if let Ok(result)= DnskeyRdata::from_bytes(&bytes_test, &bytes_test) { + assert_eq!(dnskey_rdata, result); + } + else { + assert!(false, "Error"); + } + } + + #[test] + fn max_values_to_bytes_test(){ + let mut dnskey_rdata = DnskeyRdata::new(); + //Max value of 2 bytes is 65535 (16 ones in the 2 bytes) + dnskey_rdata.set_flags(65535); + dnskey_rdata.set_protocol(255); + dnskey_rdata.set_algorithm(255); + dnskey_rdata.set_public_key(vec![255, 255]); + + let bytes_test: Vec = vec![255, 255, 255, 255, 255, 255]; + + assert_eq!(dnskey_rdata.to_bytes(), bytes_test); + } } \ No newline at end of file