-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
0 additions
and
116 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,116 +0,0 @@ | ||
|
||
use std::{net::IpAddr, str::FromStr}; | ||
use dns_rust::{async_resolver::{config::ResolverConfig, AsyncResolver}, client::client_error::ClientError, domain_name::DomainName, message::{rclass::Rclass, rdata::Rdata, resource_record::{ResourceRecord, ToBytes}, rrtype::Rrtype, DnsMessage}}; | ||
|
||
|
||
|
||
|
||
// TODO: Change params type to intoDomainName | ||
async fn query_response(domain_name: &str, protocol: &str, qtype: &str) -> Result<Vec<ResourceRecord>, ClientError> { | ||
|
||
let config = ResolverConfig::default(); | ||
let mut resolver = AsyncResolver::new(config); | ||
|
||
let response = resolver.lookup( | ||
domain_name, | ||
protocol, | ||
qtype, | ||
|
||
"IN").await; | ||
|
||
response.map(|lookup_response| lookup_response.to_vec_of_rr()) | ||
} | ||
|
||
/// 6.2.1 Query test Qtype = A | ||
#[tokio::test] | ||
async fn query_a_type() { | ||
let response = query_response("example.com", "UDP", "A").await; | ||
|
||
if let Ok(rrs) = response { | ||
assert_eq!(rrs.iter().count(), 1); | ||
let rdata = rrs[0].get_rdata(); | ||
if let Rdata::A(ip) = rdata { | ||
assert_eq!(ip.get_address(), IpAddr::from_str("93.184.215.14").unwrap()); | ||
} else { | ||
panic!("No ip address"); | ||
} | ||
} | ||
} | ||
|
||
/// 6.2.2 Query normal Qtype = * | ||
#[tokio::test] | ||
/// Ignored due to halting problem | ||
#[ignore] | ||
async fn query_any_type() { | ||
let udp_response = query_response("example.com", "UDP", "ANY").await; | ||
let tcp_response = query_response("example.com", "TCP", "ANY").await; | ||
assert!(udp_response.is_err()); | ||
assert!(tcp_response.is_err()); | ||
} | ||
|
||
/// 6.2.3 Query Qtype = MX | ||
#[tokio::test] | ||
async fn query_mx_type() { | ||
let response = query_response("example.com", "UDP", "MX").await; | ||
|
||
if let Ok(rrs) = response { | ||
assert_eq!(rrs.len(), 1); | ||
|
||
if let Rdata::MX(mxdata) = rrs[0].get_rdata() { | ||
assert_eq!( | ||
mxdata.get_exchange(), | ||
DomainName::new_from_str("")); | ||
|
||
assert_eq!( | ||
mxdata.get_preference(), | ||
0 | ||
) | ||
} else { | ||
panic!("Record is not MX type"); | ||
} | ||
} | ||
} | ||
|
||
|
||
// 6.2.4 Query Qtype = NS | ||
#[tokio::test] | ||
async fn query_ns_type() { | ||
let response = query_response("example.com", "UDP", "NS").await; | ||
if let Ok(rrs) = response { | ||
assert_eq!(rrs.len(), 2); | ||
|
||
if let Rdata::NS(ns1) = rrs[0].get_rdata() { | ||
assert!( | ||
ns1.get_nsdname() == DomainName::new_from_str("a.iana-servers.net") || | ||
ns1.get_nsdname() == DomainName::new_from_str("b.iana-servers.net") | ||
) | ||
} else { | ||
panic!("First record is not NS"); | ||
} | ||
|
||
if let Rdata::NS(ns) = rrs[1].get_rdata() { | ||
assert!( | ||
ns.get_nsdname() == DomainName::new_from_str("a.iana-servers.net") || | ||
ns.get_nsdname() == DomainName::new_from_str("b.iana-servers.net") | ||
) | ||
} else { | ||
panic!("Second record is not NS"); | ||
} | ||
} | ||
} | ||
|
||
/// 6.2.5 Mistyped host name Qtype = A | ||
#[tokio::test] | ||
async fn mistyped_host_name() { | ||
let response = query_response("exampllee.com", "UDP", "A").await; | ||
assert!(response.is_err()); | ||
} | ||
|
||
/// No record test | ||
#[tokio::test] | ||
async fn no_resource_available() { | ||
let response = query_response("example.com", "UDP", "CNAME").await; | ||
println!("{:?}", response); | ||
assert!(response.is_err()); | ||
} | ||
|
||