X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2d7b22fd7d905e4fa44452775d9db6a06a50da8c..a4f061a5bf48f0335238da72b8d03ca349fa7553:/apps/workbench/app/models/job_work_unit.rb

diff --git a/apps/workbench/app/models/job_work_unit.rb b/apps/workbench/app/models/job_work_unit.rb
index 8246e8ac87..6bef50f4ca 100644
--- a/apps/workbench/app/models/job_work_unit.rb
+++ b/apps/workbench/app/models/job_work_unit.rb
@@ -1,7 +1,8 @@
 class JobWorkUnit < ProxyWorkUnit
   def children
     # Job tasks
-    tasks = JobTask.filter([['job_uuid', '=', self.proxied.uuid]]).results
+    uuid = (self.proxied.uuid if self.proxied.respond_to?(:uuid)) || self.proxied[:uuid]
+    tasks = JobTask.filter([['job_uuid', '=', uuid]]).results
     items = []
     tasks.each do |t|
       items << t.work_unit("task #{items.size}")
@@ -13,14 +14,16 @@ class JobWorkUnit < ProxyWorkUnit
   end
 
   def progress
-    if self.proxied.state == 'Complete'
+    state = (self.proxied.state if self.proxied.respond_to?(:state)) || self.proxied[:state]
+    if state == 'Complete'
       return 1.0
     end
 
-    failed = self.proxied.tasks_summary[:failed] || 0 rescue 0
-    done = self.proxied.tasks_summary[:done] || 0 rescue 0
-    running = self.proxied.tasks_summary[:running] || 0 rescue 0
-    todo = self.proxied.tasks_summary[:todo] || 0 rescue 0
+    tasks_summary = (self.proxied.tasks_summary if self.proxied.respond_to?(:tasks_summary)) || self.proxied[:tasks_summary]
+    failed = tasks_summary[:failed] || 0 rescue 0
+    done = tasks_summary[:done] || 0 rescue 0
+    running = tasks_summary[:running] || 0 rescue 0
+    todo = tasks_summary[:todo] || 0 rescue 0
     if done + running + failed + todo > 0
       total_tasks = done + running + failed + todo
       (done+failed).to_f / total_tasks
@@ -30,22 +33,35 @@ class JobWorkUnit < ProxyWorkUnit
   end
 
   def docker_image
-    self.proxied[:docker_image_locator]
+    (self.proxied.docker_image_locator if self.proxied.respond_to?(:docker_image_locator)) || self.proxied[:docker_image_locator]
   end
 
   def nondeterministic
-    self.proxied[:nondeterministic]
+    (self.proxied.nondeterministic if self.proxied.respond_to?(:nondeterministic)) || self.proxied[:nondeterministic]
   end
 
   def priority
-    self.proxied[:priority]
+    (self.proxied.priority if self.proxied.respond_to?(:priority)) || self.proxied[:priority]
   end
 
   def log_collection
-    self.proxied.log
+    (self.proxied.log if self.proxied.respond_to?(:log)) || self.proxied[:log]
   end
 
   def output
-    self.proxied.output
+    (self.proxied.output if self.proxied.respond_to?(:output)) || self.proxied[:output]
+  end
+
+  def uri
+    uuid = (self.proxied.uuid if self.proxied.respond_to?(:uuid)) || self.proxied[:uuid]
+    "/jobs/#{uuid}"
+  end
+
+  def child_summary
+    (self.proxied.tasks_summary if self.proxied.respond_to?(:tasks_summary)) || self.proxied[:tasks_summary]
+  end
+
+  def title
+    "job"
   end
 end