Fixed integration test.
Added socket.
Added integration test for socket.