use https_dns::utils::{build_request_message, build_test_listener};
use tokio::{net::UdpSocket, test};
use trust_dns_proto::{
    op::Message,
    rr::{Name, RData, RecordType},
};

#[test]
async fn txt_record() {
    let udp_listener = build_test_listener().await;
    tokio::spawn(async move {
        udp_listener.listen().await;
    });

    for host in ["facebook.com", "cloudflare.com", "stripe.com"] {
        let request_name: Name = host.parse().unwrap();
        let request_message = build_request_message(request_name, RecordType::TXT);
        let raw_request_message = request_message.to_vec().unwrap();

        let udp_socket = UdpSocket::bind("127.0.0.1:10054").await.unwrap();
        udp_socket.connect("127.0.0.1:10053").await.unwrap();

        udp_socket.send(&raw_request_message).await.unwrap();
        let mut buffer = [0; 4096];
        udp_socket.recv(&mut buffer).await.unwrap();

        let response_message = Message::from_vec(&buffer).unwrap();
        let record_data = &response_message.answers()[0].data().unwrap();
        if let RData::TXT(txt) = record_data {
            assert!(!txt.txt_data().is_empty());
        } else {
            panic!("the record type is not TXT")
        }
    }
}