Merge branch 'mitchellh/default_process_class'

mitchellh · Jan 7, 2011 · 1d36aea · 1d36aea
2 parents 1f4bc83 + 2566f50
commit 1d36aea
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/net/ssh/shell.rb b/lib/net/ssh/shell.rb
@@ -11,13 +11,15 @@ class Shell
       attr_reader :state
       attr_reader :shell
       attr_reader :processes
+      attr_accessor :default_process_class
 
       def initialize(session, shell=:default)
         @session = session
         @shell = shell
         @state = :closed
         @processes = []
         @when_open = []
+        @default_process_class = Net::SSH::Shell::Process
         open
       end
 
@@ -60,7 +62,8 @@ def opening?
         !open? && !closed?
       end
 
-      def execute(command, klass=Net::SSH::Shell::Process, &callback)
+      def execute(command, klass=nil, &callback)
+        klass ||= default_process_class
         process = klass.new(self, command, callback)
         process.run if processes.empty?
         processes << process