Add test for the TCP socket connection