Add a exeflags kwarg to the get_connect_cmd() function (#14)

* Add a `exeflags` kwarg to the `get_connect_cmd()` function

Author: DilumAluthge

src/ElasticClusterManager.jl

@@ -14,12 +14,14 @@ using Distributed: launch, manage, kill, init_worker, connect
 export launch, manage, kill, init_worker, connect
 export ElasticManager, elastic_worker
 
-function worker_cookie()
-  Distributed.init_multi()
-  return Distributed.cluster_cookie()
-end
-
-worker_arg() = `--worker=$(worker_cookie())`
+# This seems to be currently unused:
+# function worker_cookie()
+#   Distributed.init_multi()
+#   return Distributed.cluster_cookie()
+# end
+
+# This seems to be currently unused:
+# worker_arg() = `--worker=$(worker_cookie())`
 
 include("elastic.jl")
 

src/elastic.jl

@@ -147,8 +147,7 @@ function elastic_worker(cookie, addr="127.0.0.1", port=9009; stdout_to_master=tr
     Distributed.start_worker(c, cookie)
 end
 
-function get_connect_cmd(em::ElasticManager; absolute_exename=true, same_project=true)
-
+function get_connect_cmd(em::ElasticManager; absolute_exename=true, same_project=true, exeflags::Tuple=())
     ip = string(em.sockname[1])
     port = convert(Int,em.sockname[2])
     cookie = Distributed.cluster_cookie()
@@ -157,6 +156,7 @@ function get_connect_cmd(em::ElasticManager; absolute_exename=true, same_project
 
     join([
         exename,
+        exeflags...,
         project...,
         "-e 'import ElasticClusterManager; ElasticClusterManager.elastic_worker(\"$cookie\",\"$ip\",$port)'"
     ]," ")

test/elastic.jl

@@ -5,7 +5,8 @@
     @test em isa ElasticManager
 
     # launch worker
-    old_cmd = ElasticClusterManager.get_connect_cmd(em)
+    exeflags = ("--code-coverage=user",)
+    old_cmd = ElasticClusterManager.get_connect_cmd(em; exeflags=exeflags)
     new_cmd = `$(old_cmd) --coverage=user`
     # run(`sh -c $(new_cmd)`) # comment out this line when you are finished debugging
     run(`sh -c $(new_cmd)`; wait=false) # uncomment this line when you are finished debugging