mirror/Easytier
1
0
Fork 0
mirror of https://github.com/EasyTier/EasyTier.git synced 2026-09-02 17:15:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: new_request

Browse Source
This commit is contained in:
Luna Yao
2026-04-06 11:55:25 +02:00
parent 9e8a4c94cb
commit 67108d14b0
1 changed files with 11 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
easytier/src/dns/zone.rs
+11 -20
View File
@@ -239,9 +239,9 @@ mod tests {
"#; "#;
fn query(name: &str, rtype: RecordType) -> anyhow::Result<MessageRequest> { fn new_request(name: &str, rtype: RecordType) -> anyhow::Result<Request> {
let mut query = Message::new(); let mut query = Message::new();
query.set_id(0x1234); query.set_id(0);
query.set_message_type(MessageType::Query); query.set_message_type(MessageType::Query);
query.set_op_code(OpCode::Query); query.set_op_code(OpCode::Query);
query.set_recursion_desired(true); query.set_recursion_desired(true);
@@ -251,11 +251,14 @@ mod tests {
let mut encoder = BinEncoder::new(&mut request); let mut encoder = BinEncoder::new(&mut request);
query.emit(&mut encoder)?; query.emit(&mut encoder)?;
Ok(MessageRequest::from_bytes(&request)?) Ok(Request::new(
MessageRequest::from_bytes(&request)?,
SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
Protocol::Udp,
))
} }
// #[tokio::test] #[tokio::test]
#[tokio::test(flavor = "current_thread")]
async fn test_config() -> anyhow::Result<()> { async fn test_config() -> anyhow::Result<()> {
log::tests::init(); log::tests::init();
@@ -344,11 +347,7 @@ mod tests {
authorities.extend(zone.create_forward_authority().into_iter()); authorities.extend(zone.create_forward_authority().into_iter());
catalog.upsert(zone.origin.clone(), authorities); catalog.upsert(zone.origin.clone(), authorities);
let request = Request::new( let request = new_request("et.top.", RecordType::A)?;
query("et.top.", RecordType::A)?,
SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
Protocol::Udp,
);
let response = ResponseHandle::new(512); let response = ResponseHandle::new(512);
let info = catalog.lookup(&request, None, response.clone()).await; let info = catalog.lookup(&request, None, response.clone()).await;
@@ -360,22 +359,14 @@ mod tests {
vec![Zone::system().create_forward_authority().unwrap()], vec![Zone::system().create_forward_authority().unwrap()],
); );
let request = Request::new( let request = new_request("example.com", RecordType::A)?;
query("example.com", RecordType::A)?,
SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
Protocol::Udp,
);
let response = ResponseHandle::new(512); let response = ResponseHandle::new(512);
let info = catalog.lookup(&request, None, response.clone()).await; let info = catalog.lookup(&request, None, response.clone()).await;
assert_eq!(info.response_code(), ResponseCode::NoError); assert_eq!(info.response_code(), ResponseCode::NoError);
let request = Request::new( let request = new_request("example.invalid", RecordType::A)?;
query("example.invalid", RecordType::A)?,
SocketAddrV4::new(Ipv4Addr::LOCALHOST, 0).into(),
Protocol::Udp,
);
let response = ResponseHandle::new(512); let response = ResponseHandle::new(512);
let info = catalog.lookup(&request, None, response.clone()).await; let info = catalog.lookup(&request, None, response.clone()).await;