lib/dk/test_runner.rb in dk-0.1.0 vs lib/dk/test_runner.rb in dk-0.1.1

@@ -21,17 +21,22 @@
     # don't run any sub-tasks, just track that a sub-task was run
     def run_task(task_class, params = nil)
       TaskRun.new(task_class, params).tap{ |tr| self.runs << tr }
     end
 
+    # track that a local cmd was started
+    def start(*args)
+      super(*args).tap{ |c| self.runs << c }
+    end
+
     # track that a local cmd was run
-    def cmd(task, cmd_str, input, given_opts)
-      super(task, cmd_str, input, given_opts).tap{ |c| self.runs << c }
+    def cmd(*args)
+      super(*args).tap{ |c| self.runs << c }
     end
 
     # track that a remote cmd was run
-    def ssh(task, cmd_str, input, given_opts, ssh_opts)
-      super(task, cmd_str, input, given_opts, ssh_opts).tap{ |c| self.runs << c }
+    def ssh(*args)
+      super(*args).tap{ |c| self.runs << c }
     end
 
     # test task API
 
     def task(params = nil)