From a9842f9e3ee040b2035b606c5d469dde4ca10d65 Mon Sep 17 00:00:00 2001
From: Alex Emmet <111742636+Alex-Emmet@users.noreply.github.com>
Date: Fri, 26 Jun 2026 17:14:43 +0200
Subject: [PATCH] [Fix] Integration test

---
 transport/Cargo.toml           | 4 ++++
 transport/tests/integration.rs | 7 ++++---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/transport/Cargo.toml b/transport/Cargo.toml
index 7e4d972..6db802f 100644
--- a/transport/Cargo.toml
+++ b/transport/Cargo.toml
@@ -24,6 +24,10 @@ log = "0.4"
 [dev-dependencies]
 rcgen = "0.14"
 
+[[test]]
+name = "integration"
+required-features = ["host"]
+
 [features]
 default = []
 # Enables hosting a MTP server
diff --git a/transport/tests/integration.rs b/transport/tests/integration.rs
index d2ad8ac..db96e8f 100644
--- a/transport/tests/integration.rs
+++ b/transport/tests/integration.rs
@@ -1,8 +1,7 @@
 use std::net::{IpAddr, Ipv4Addr};
 
-use mtp_codec::{CommunicationType, DataType};
+use mtp_codec::{CommunicationType, DataType, TypeMap};
 use mtp_transport::{Policy, connect, host};
-use mtp_type_map::TypeMap;
 
 fn generate_self_signed_cert() -> (Vec<u8>, Vec<u8>) {
     let key_pair = rcgen::KeyPair::generate().unwrap();
@@ -81,7 +80,9 @@ async fn test_send_receive_roundtrip() {
         client_received.get_type(),
         CommunicationType::Pong.to_id(&tm)
     );
-    let client_val = client_received.get_data(DataType::PqSignature.to_id(&tm)).clone();
+    let client_val = client_received
+        .get_data(DataType::PqSignature.to_id(&tm))
+        .clone();
     assert_eq!(client_val, mtp_codec::DataValue::UnsignedNumber(99));
 
     // Close both sides