Add TCP socket tests