use crate::{Resolver, Class, RecordType, Section};
use crate::record::MX;
use crate::record::TLSA;
#[test]
fn basic_test_query() {
let mut resolver = Resolver::new().unwrap();
let mut response = resolver.query(b"gmail.com", Class::IN, RecordType::MX).unwrap();
let flags = response.get_flags();
assert_eq!(flags.question_response(), true);
assert_eq!(flags.recursion_desired(), true);
assert!(response.get_section_count(Section::Question) > 0);
assert!(response.get_section_count(Section::Answer) > 0);
let mut count: usize = 0;
for answer in response.answers::<MX>() {
count += 1;
println!("{:?}", answer);
}
assert_eq!(response.get_section_count(Section::Answer), count);
}
#[test]
fn test_tlsa() {
let mut resolver = Resolver::new().unwrap();
let mut response = resolver.query(b"_443._tcp.www.middlebox-dane.org", Class::IN, RecordType::TLSA).unwrap();
let flags = response.get_flags();
assert_eq!(flags.question_response(), true);
assert_eq!(flags.recursion_desired(), true);
assert!(response.get_section_count(Section::Question) > 0);
assert!(response.get_section_count(Section::Answer) > 0);
let mut count: usize = 0;
for answer in response.answers::<TLSA>() {
count += 1;
println!("{:?}", answer);
}
assert_eq!(response.get_section_count(Section::Answer), count);
}