Modified tests for socket server