1 files changed, 8 insertions, 7 deletions
diff --git a/src/tests/fakes/fake_input.rs b/src/tests/fakes/fake_input.rs
index 146a85e..9f2fb80 100644
--- a/src/tests/fakes/fake_input.rs
+++ b/src/tests/fakes/fake_input.rs
@@ -16,6 +16,7 @@ use crate::{
         dns::{self, Lookup},
         Connection, Protocol,
     },
+    os::ProcessInfo,
     OpenSockets,
 };
 
@@ -96,7 +97,7 @@ pub fn get_open_sockets() -> OpenSockets {
             443,
             Protocol::Tcp,
         ),
-        String::from("1"),
+        ProcessInfo::new("1", 1),
     );
     open_sockets.insert(
         Connection::new(
@@ -105,7 +106,7 @@ pub fn get_open_sockets() -> OpenSockets {
             4434,
             Protocol::Tcp,
         ),
-        String::from("4"),
+        ProcessInfo::new("4", 4),
     );
     open_sockets.insert(
         Connection::new(
@@ -114,7 +115,7 @@ pub fn get_open_sockets() -> OpenSockets {
             4435,
             Protocol::Tcp,
         ),
-        String::from("5"),
+        ProcessInfo::new("5", 5),
     );
     open_sockets.insert(
         Connection::new(
@@ -123,7 +124,7 @@ pub fn get_open_sockets() -> OpenSockets {
             4432,
             Protocol::Tcp,
         ),
-        String::from("2"),
+        ProcessInfo::new("2", 2),
     );
     open_sockets.insert(
         Connection::new(
@@ -132,12 +133,12 @@ pub fn get_open_sockets() -> OpenSockets {
             443,
             Protocol::Tcp,
         ),
-        String::from("1"),
+        ProcessInfo::new("1", 1),
     );
     let mut local_socket_to_procs = HashMap::new();
     let mut connections = std::vec::Vec::new();
-    for (connection, process_name) in open_sockets {
-        local_socket_to_procs.insert(connection.local_socket, process_name);
+    for (connection, proc_info) in open_sockets {
+        local_socket_to_procs.insert(connection.local_socket, proc_info);
         connections.push(connection);
     }