py gateway setup

Author: sonalgoyal

test/ExampleApplication.java

@@ -2,7 +2,7 @@
 public class TestPy4JGateway {
 
     public static void main(String[] args) {
-        GatewayServer gatewayServer = new GatewayServer();
+        GatewayServer gatewayServer = new GatewayServer(new Object());
         gatewayServer.start();
         System.out.println("Gateway Server Started");
     }

test/TestPy4JGateway.class

@@ -9,34 +9,60 @@
 from py4j.java_gateway import JavaGateway
 from py4j.protocol import Py4JNetworkError
 from time import sleep
+from multiprocessing import Process
+from py4j.java_gateway import JavaGateway, GatewayParameters
 
 PY4J_JAVA_PATH='.:../thirdParty/lib//py4j0.10.9.jar:$ZINGG_HOME/common/client/target/zingg-common-client-0.4.0-SNAPSHOT.jar'
-def start_example_server():
+def compileGatewayEntry():
     subprocess.call([
         "javac", "-cp", PY4J_JAVA_PATH,
         "TestPy4JGateway.java"])
+    
+def startGatewayEntry():
     subprocess.call([
         "java", "-Xmx512m", "-cp", PY4J_JAVA_PATH,
         "TestPy4JGateway"])
+    
+def start_example_app_process():
+    # XXX DO NOT FORGET TO KILL THE PROCESS IF THE TEST DOES NOT SUCCEED
+    p = Process(target=startGatewayEntry)
+    p.start()
+    sleep(2)
+    return p
 
-def check_connection(gateway):
+def check_connection(gateway_parameters=None):
+    test_gateway = JavaGateway(gateway_parameters=gateway_parameters)
     try:
-        gateway.jvm.System.currentTimeMillis()
+        # Call a dummy method just to make sure we can connect to the JVM
+        test_gateway.jvm.System.currentTimeMillis()
     except Py4JNetworkError:
+        # We could not connect. Let"s wait a long time.
+        # If it fails after that, there is a bug with our code!
         sleep(2)
+    finally:
+        test_gateway.close()
+
+def safe_shutdown(instance):
+    if hasattr(instance, 'gateway'):
+        try:
+            instance.gateway.shutdown()
+        except Exception:
+            print("exception")
 
-class MyJavaClass:
-    def addition(self, a, b):
-        return a + b
+
 
 class MyJavaIntegrationTest(unittest.TestCase):
     def setUp(self):
-        start_example_server()
-        self.gateway = JavaGateway()
-        check_connection(self.gateway)
+        compileGatewayEntry()
+        self.p = start_example_app_process()
+        self.gateway = JavaGateway(
+            gateway_parameters=GatewayParameters(auto_convert=True))
 
     def tearDown(self):
-        self.gateway.close()
+        safe_shutdown(self)
+        self.p.join()
+        sleep(2)
+   
 
     def test_jvm_access(self):
         print("Accessing the JVM...")