mobilize-ssh 1.0.5 → 1.0.6

data/lib/mobilize-ssh/handlers/ssh.rb

@@ -126,26 +126,36 @@ module Mobilize
       return tmp_file_path
     end
 
-    def Ssh.get_file_hash(gsheet_paths,gdrive_slot)
-      file_hash = {}
-      gsheet_paths.map do |gpath|
-        string = Gsheet.find_by_path(gpath,gdrive_slot).to_tsv
-        fname = gpath.split("/").last
-        {fname => string}
-      end.each do |f|
-        file_hash = f.merge(file_hash)
+    def Ssh.file_hash_by_task_path(task_path)
+      #this is not meant to be called directly
+      #from the Runner -- it's used by run_by_task_path
+      t = Task.where(:path=>task_path).first
+      params = t.params
+      gsheet_paths = if params['sources']
+                       params['sources']
+                     elsif params['source']
+                       [params['source']]
+                     end
+      if gsheet_paths.length>0
+        gdrive_slot = Gdrive.slot_worker_by_path(task_path)
+        file_hash = {}
+        gsheet_paths.map do |gpath|
+          string = Gsheet.find_by_path(gpath,gdrive_slot).to_tsv
+          fname = gpath.split("/").last
+          {fname => string}
+        end.each do |f|
+          file_hash = f.merge(file_hash)
+        end
+        Gdrive.unslot_worker_by_path(task_path)
+        return file_hash
       end
-      file_hash
     end
 
     def Ssh.run_by_task_path(task_path)
       t = Task.where(:path=>task_path).first
       params = t.params
       node, command = [params['node'],params['cmd']]
-      file_hash = if params['sources']
-                    gdrive_slot = Gdrive.slot_worker_by_path(task_path)
-                    Ssh.get_file_hash(params['sources'],gdrive_slot)
-                  end
+      file_hash = Ssh.file_hash_by_task_path(task_path)
       su_user = t.params['su_user']
       Ssh.run(node,command,file_hash,su_user)
     end

data/lib/mobilize-ssh/version.rb

@@ -1,5 +1,5 @@
 module Mobilize
   module Ssh
-    VERSION = "1.0.5"
+    VERSION = "1.0.6"
   end
 end

data/test/mobilize-ssh_test.rb

@@ -40,9 +40,12 @@ describe "Mobilize" do
     r.update_gsheet(gdrive_slot)
 
     puts "jobtracker posted data to test sheet"
-    ssh_target_sheet = Mobilize::Gsheet.find_by_path("#{r.path.split("/")[0..-2].join("/")}/test_ssh.out",gdrive_slot)
+    ssh_target_sheet_1 = Mobilize::Gsheet.find_by_path("#{r.path.split("/")[0..-2].join("/")}/test_ssh_1.out",gdrive_slot)
+    ssh_target_sheet_2 = Mobilize::Gsheet.find_by_path("#{r.path.split("/")[0..-2].join("/")}/test_ssh_2.out",gdrive_slot)
+
+    assert ssh_target_sheet_1.to_tsv.length > 100
+    assert ssh_target_sheet_2.to_tsv.length > 100
 
-    assert ssh_target_sheet.to_tsv.length > 100
   end
 
 end

data/test/ssh_job_rows.yml

@@ -1,6 +1,12 @@
-- name: test_ssh
+- name: test_ssh_1
   active: true
   trigger: once
   status: ""
   task1: 'ssh.run node:"test_node", cmd:"ruby code.rb", su_user:"root", sources:["Runner_mobilize(test)/code.rb", "Runner_mobilize(test)/code.sh"]'
-  task2: 'gsheet.write source:"task1", target:"Runner_mobilize(test)/test_ssh.out"'
+  task2: 'gsheet.write source:"task1", target:"Runner_mobilize(test)/test_ssh_1.out"'
+- name: test_ssh_2
+  active: true
+  trigger: once
+  status: ""
+  task1: 'ssh.run node:"test_node", cmd:"sh code.sh", su_user:"root", source:"Runner_mobilize(test)/code.sh"'
+  task2: 'gsheet.write source:"task1", target:"Runner_mobilize(test)/test_ssh_2.out"'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mobilize-ssh
 version: !ruby/object:Gem::Version
-  version: 1.0.5
+  version: 1.0.6
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-14 00:00:00.000000000 Z
+date: 2012-12-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mobilize-base