Add run_on_studio powershell command fuction

Signed-off-by: Vitalii Chulak <[email protected]>

Author: Jedoku

lib/engines/hcktest/tools.rb

@@ -29,6 +29,9 @@ class RestartMachineError < AutoHCKError; end
     # A custom RunOnMachine error exception
     class RunOnMachineError < AutoHCKError; end
 
+    # A custom RunOnStudio error exception
+    class RunOnStudioError < AutoHCKError; end
+
     # A custom UploadToMachineError error exception
     class UploadToMachineError < AutoHCKError; end
 
@@ -160,6 +163,23 @@ def delete_machine(machine, pool)
       end
     end
 
+    def run_on_studio(command)
+      retries ||= 0
+      ret = act_with_tools { _1.run_on_studio(command) }
+
+      return ret if ret
+
+      e_message = "Running command (#{command}) on studio failed"
+      raise RunOnStudioError, e_message
+    rescue RunOnStudioError => e
+      @logger.warn(e.message)
+      raise unless (retries += 1) < ACTION_RETRIES
+
+      sleep ACTION_RETRY_SLEEP
+      @logger.info("Trying again to run command (#{command}) on studio")
+      retry
+    end
+
     def run_on_machine(machine, desc, cmd)
       retries ||= 0
       ret = act_with_tools { _1.run_on_machine(machine, cmd) }