Add some more tests for socket