test: add test for connect request in udp::handler

Author: josecelano

src/udp/handlers.rs

@@ -236,3 +236,61 @@ fn handle_error(e: ServerError, transaction_id: TransactionId) -> Response {
         message: message.into(),
     })
 }
+
+#[cfg(test)]
+mod tests {
+    use std::{
+        net::{IpAddr, Ipv4Addr, SocketAddr},
+        sync::Arc,
+    };
+
+    use crate::{protocol::utils::get_connection_id, tracker::tracker::TorrentTracker, udp::handle_connect, Configuration};
+    use aquatic_udp_protocol::{ConnectRequest, ConnectResponse, Response, TransactionId};
+
+    fn initialized_tracker() -> Arc<TorrentTracker> {
+        let config = Arc::new(Configuration::default());
+        Arc::new(TorrentTracker::new(config).unwrap())
+    }
+
+    fn sample_remote_addr() -> SocketAddr {
+        SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 8080)
+    }
+
+    #[tokio::test]
+    async fn a_connect_response_should_contain_the_same_transaction_id_as_the_connect_request() {
+        let request = ConnectRequest {
+            transaction_id: TransactionId(0i32),
+        };
+
+        let response = handle_connect(sample_remote_addr(), &request, initialized_tracker())
+            .await
+            .unwrap();
+
+        assert_eq!(
+            response,
+            Response::Connect(ConnectResponse {
+                connection_id: get_connection_id(&sample_remote_addr()),
+                transaction_id: request.transaction_id
+            })
+        );
+    }
+
+    #[tokio::test]
+    async fn a_connect_response_should_contain_a_new_connection_id() {
+        let request = ConnectRequest {
+            transaction_id: TransactionId(0i32),
+        };
+
+        let response = handle_connect(sample_remote_addr(), &request, initialized_tracker())
+            .await
+            .unwrap();
+
+        assert_eq!(
+            response,
+            Response::Connect(ConnectResponse {
+                connection_id: get_connection_id(&sample_remote_addr()),
+                transaction_id: request.transaction_id
+            })
+        );
+    }
+}