Sha256: bdf859ec4db62c29f6b21c1f926ac2d0d41b4be4c866e1a7132acfca3266c15a

Contents?: true

Size: 1.3 KB

Versions: 4

Compression:

Stored size: 1.3 KB

Contents

require 'serverspec/backend/exec'

module Serverspec
  module Backend
    class Ssh < Exec
      def do_check(cmd, opt={})
        cmd = "sudo #{cmd}" if not RSpec.configuration.ssh.options[:user] == 'root'
        ssh_exec!(cmd)
      end

      private
      def ssh_exec!(command)
        stdout_data = ''
        stderr_data = ''
        exit_code   = nil
        exit_signal = nil

        ssh = RSpec.configuration.ssh
        ssh.open_channel do |channel|
          channel.request_pty do |ch, success|
            abort "Could not obtain pty " if !success
          end
          channel.exec("#{command}") do |ch, success|
            abort "FAILED: couldn't execute command (ssh.channel.exec)" if !success
            channel.on_data do |ch,data|
              stdout_data += data
            end

            channel.on_extended_data do |ch,type,data|
              stderr_data += data
            end

            channel.on_request("exit-status") do |ch,data|
              exit_code = data.read_long
            end

            channel.on_request("exit-signal") do |ch, data|
              exit_signal = data.read_long
            end
          end
        end
        ssh.loop
        { :stdout => stdout_data, :stderr => stderr_data, :exit_code => exit_code, :exit_signal => exit_signal }
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version Path
serverspec-0.2.4 lib/serverspec/backend/ssh.rb
serverspec-0.2.3 lib/serverspec/backend/ssh.rb
serverspec-0.2.2 lib/serverspec/backend/ssh.rb
serverspec-0.2.1 lib/serverspec/backend/ssh.rb