Modification of Integration test in order to make them function with our
new solution (sockets, and not py4j)