Addressing reviewer comments

Author: raoulstrackx

fortanix-vme/aws-nitro-enclaves/tests/incoming_connection/out.expected

@@ -1,11 +1,11 @@
-Server run #1
+Running server on 127.0.0.1:3400
 Bind TCP socket to port 3400
 Listening for incoming connections...
 Waiting for connection 1
 Connection 1: Connected
 Waiting for connection 2
 Connection 2: Connected
-Server run #2
+Running server on localhost:3400
 Bind TCP socket to port 3400
 Listening for incoming connections...
 Waiting for connection 1

fortanix-vme/aws-nitro-enclaves/tests/incoming_connection/src/main.rs

@@ -1,15 +1,11 @@
 #![feature(io_error_uncategorized)]
 use std::io::{ErrorKind, Read, Write};
-use std::net::{IpAddr, Ipv4Addr, Shutdown, SocketAddr, TcpListener, TcpStream};
+use std::net::{IpAddr, Ipv4Addr, Shutdown, SocketAddr, TcpListener, TcpStream, ToSocketAddrs};
 use std::os::unix::io::{AsRawFd, FromRawFd};
 
-fn server_run(run: u32) {
+fn server_run<A: ToSocketAddrs>(addr: A) {
     println!("Bind TCP socket to port 3400");
-    let listener = if run == 1 {
-        TcpListener::bind("127.0.0.1:3400").expect("Bind failed")
-    } else {
-        TcpListener::bind("localhost:3400").expect("Bind failed")
-    };
+    let listener = TcpListener::bind(addr).expect("Bind failed");
     assert_eq!(listener.local_addr().unwrap(), SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 3400));
 
     let fd = listener.as_raw_fd();
@@ -51,9 +47,9 @@ fn server_run(run: u32) {
 }
 
 fn main() {
-    for run in 1..=2 {
-        println!("Server run #{}", run);
-        server_run(run)
+    for addr in &["127.0.0.1:3400", "localhost:3400"] {
+        println!("Running server on {}", addr);
+        server_run(addr)
     }
     println!("Bye bye");
 }