ridley-connectors 1.0.1

data/spec/unit/ridley-connectors/host_connector/winrm/command_uploader_spec.rb

@@ -0,0 +1,67 @@
+require 'spec_helper'
+
+describe Ridley::HostConnector::WinRM::CommandUploader do
+  let(:winrm_stub) {
+    double('WinRM',
+      run_cmd: run_cmd_data,
+      powershell: nil
+    )
+  }
+
+  subject { command_uploader }
+
+  let(:command_uploader) { described_class.new(winrm_stub) }
+  let(:command_string) { "a" * 2048 }
+  let(:run_cmd_data) { { data: [{ stdout: "abc123" }] } }
+  let(:command_file_name) { "my_command.bat" }
+
+  its(:winrm) { should eq(winrm_stub) }
+
+  before do
+    command_uploader.stub(:get_file_path).and_return("")
+  end
+
+  describe "#upload" do
+    let(:upload) { command_uploader.upload(command_string) }
+
+    it "calls winrm to upload and convert the command" do
+      winrm_stub.should_receive(:run_cmd).and_return(
+        run_cmd_data,
+        nil,
+        run_cmd_data
+      )
+      winrm_stub.should_receive(:powershell)
+
+      upload
+    end
+  end
+
+  describe "#command" do
+    subject { command }
+    let(:command) { command_uploader.command }
+
+    before do
+      command_uploader.stub command_file_name: command_file_name
+    end
+
+    it { should eq("cmd.exe /C #{command_file_name}") }
+  end
+
+  describe "#cleanup" do
+    subject { cleanup }
+
+    let(:cleanup) { command_uploader.cleanup }
+    let(:base64_file_name) { "my_base64_file" }
+
+    before do
+      command_uploader.stub command_file_name: command_file_name
+      command_uploader.stub base64_file_name: base64_file_name
+    end
+
+    it "cleans up the windows temp dir" do
+      winrm_stub.should_receive(:run_cmd).with("del #{base64_file_name} /F /Q")
+      winrm_stub.should_receive(:run_cmd).with("del #{command_file_name} /F /Q")
+      cleanup
+    end
+  end
+end

data/spec/unit/ridley-connectors/host_connector/winrm_spec.rb

@@ -0,0 +1,145 @@
+require 'spec_helper'
+
+describe Ridley::HostConnector::WinRM do
+  subject { connector }
+  let(:connector) { described_class.new }
+  let(:host) { 'fake.riotgames.com' }
+  let(:options) do
+    {
+      server_url: double('server_url'),
+      validator_path: fixtures_path.join('my-fake.pem'),
+      validator_client: double('validator_client'),
+      encrypted_data_bag_secret: 'encrypted_data_bag_secret',
+      winrm: Hash.new,
+      chef_version: double('chef_version')
+    }
+  end
+
+  before { described_class::CommandUploader.stub(:new).and_return(double('command_uploader')) }
+
+  describe "#get_command" do
+    subject(:get_command) { connector.get_command(command, command_uploader_stub) }
+
+    let(:command) { "echo %TEMP%" }
+    let(:command_uploader_stub) { double('CommandUploader') }
+
+    it { should eq(command) }
+
+    context "when a command is more than 2047 characters" do
+      let(:command) { "a" * 2048 }
+
+      it "uploads and returns a command" do
+        described_class::CommandUploader.stub(new: command_uploader_stub)
+
+        command_uploader_stub.should_receive :upload
+        command_uploader_stub.stub command: "my command"
+        command_uploader_stub.stub(:cleanup)
+
+        get_command.should eq("my command")
+      end
+    end
+  end
+
+  describe "#run" do
+    subject(:result) { connector.run(host, command, options) }
+    let(:command) { "dir" }
+    let(:command_uploader_stub) { double('CommandUploader', cleanup: true) }
+    let(:stdout) { "stdout" }
+    let(:stderr) { nil }
+    let(:winrm_stub) { double }
+
+    before do
+      described_class::CommandUploader.stub(:new).and_return(command_uploader_stub)
+      connector.stub(:winrm).and_return(winrm_stub)
+      winrm_stub.stub(:run_cmd).and_yield(stdout, stderr).and_return(exitcode: 0)
+    end
+
+    context "when the exit_code is 0" do
+      it "returns a non-error HostConnector::Response" do
+        expect(result).to be_a(Ridley::HostConnector::Response)
+        expect(result).to_not be_error
+      end
+
+      it "sets the response's stdout message" do
+        expect(result.stdout).to eq("stdout")
+      end
+    end
+
+    context "when the exit_code is not 0" do
+      let(:stderr) { "stderr" }
+
+      before do
+        winrm_stub.stub(:run_cmd).and_yield(stdout, stderr).and_return(exitcode: 1)
+      end
+
+      it "returns an error HostConnector::Response with an error" do
+        expect(result).to be_a(Ridley::HostConnector::Response)
+        expect(result).to be_error
+      end
+
+      it "sets the response's stderr message" do
+        expect(result.stderr).to eq("stderr")
+      end
+    end
+
+    context "when a WinRM::WinRMHTTPTransportError error is raised" do
+      let(:stderr) { "error" }
+      before { winrm_stub.stub(:run_cmd).and_yield(stdout, stderr).and_raise(::WinRM::WinRMHTTPTransportError) }
+
+      it "returns an error HostConnector::Response with an error" do
+        expect(result).to be_a(Ridley::HostConnector::Response)
+        expect(result).to be_error
+      end
+
+      it "sets the response's stderr message to the exception's message" do
+        expect(result.stderr).to eql("WinRM::WinRMHTTPTransportError")
+      end
+    end
+  end
+
+  describe "#bootstrap" do
+    it "sends a #run message to self to bootstrap a node" do
+      connector.should_receive(:run).with(host, anything, options)
+      connector.bootstrap(host, options)
+    end
+  end
+
+  describe "#chef_client" do
+    subject(:chef_client) { connector.chef_client(host, options) }
+
+    it "receives a command to run chef-client" do
+      connector.should_receive(:run).with(host, "chef-client", options)
+
+      chef_client
+    end
+  end
+
+  describe "#put_secret" do
+    subject(:put_secret) { connector.put_secret(host, secret, options) }
+    let(:encrypted_data_bag_secret_path) { fixtures_path.join("encrypted_data_bag_secret").to_s }
+    let(:secret) { File.read(encrypted_data_bag_secret_path).chomp }
+
+    it "receives a command to copy the secret" do
+      connector.should_receive(:run).with(host,
+        "echo #{secret} > C:\\chef\\encrypted_data_bag_secret",
+        options
+      )
+
+      put_secret
+    end
+  end
+
+  describe "#ruby_script" do
+    subject(:ruby_script) { connector.ruby_script(host, command_lines, options) }
+    let(:command_lines) { ["puts 'hello'", "puts 'there'"] }
+
+    it "receives a ruby call with the command" do
+      connector.should_receive(:run).with(host,
+        "#{described_class::EMBEDDED_RUBY_PATH} -e \"puts 'hello';puts 'there'\"",
+        options
+      )
+
+      ruby_script
+    end
+  end
+end

data/spec/unit/ridley-connectors/host_connector_spec.rb

@@ -0,0 +1,50 @@
+require 'spec_helper'
+
+describe Ridley::HostConnector::Base do
+  subject { Class.new(Ridley::HostConnector::Base).new }
+
+  let(:host) { double('host') }
+  let(:options) { Hash.new }
+
+  describe "#run" do
+    let(:command) { double('command') }
+
+    it "raises a RuntimeError" do
+      expect { subject.run(host, command, options) }.to raise_error(RuntimeError)
+    end
+  end
+
+  describe "#bootstrap" do
+    it "raises a RuntimeError" do
+      expect { subject.bootstrap(host, options) }.to raise_error(RuntimeError)
+    end
+  end
+
+  describe "#chef_client" do
+    it "raises a RuntimeError" do
+      expect { subject.chef_client(host, options) }.to raise_error(RuntimeError)
+    end
+  end
+
+  describe "#put_secret" do
+    let(:secret) { double('secret') }
+
+    it "raises a RuntimeError" do
+      expect { subject.put_secret(host, secret, options) }.to raise_error(RuntimeError)
+    end
+  end
+
+  describe "#ruby_script" do
+    let(:command_lines) { double('command-lines') }
+
+    it "raises a RuntimeError" do
+      expect { subject.ruby_script(host, command_lines, options) }.to raise_error(RuntimeError)
+    end
+  end
+
+  describe "#uninstall_chef" do
+    it "raises a RuntimeError" do
+      expect { subject.uninstall_chef(host, options) }.to raise_error(RuntimeError)
+    end
+  end
+end

data/spec/unit/ridley-connectors/resources/node_resource_spec.rb

@@ -0,0 +1,139 @@
+require 'spec_helper'
+
+describe Ridley::NodeResource do
+  let(:host) { "33.33.33.10" }
+  let(:host_commander) { double('host-commander') }
+  let(:options) do
+    {
+      server_url: double('server_url'),
+      validator_path: double('validator_path'),
+      validator_client: double('validator_client'),
+      encrypted_data_bag_secret: double('encrypted_data_bag_secret'),
+      ssh: double('ssh'),
+      winrm: double('winrm'),
+      chef_version: double('chef_version')
+    }
+  end
+  let(:instance) do
+    inst = described_class.new(double, options)
+    inst.stub(host_commander: host_commander)
+    inst
+  end
+
+  describe "#bootstrap" do
+    it "sends the message #bootstrap to the instance's host_commander" do
+      host_commander.should_receive(:bootstrap).with(host, options)
+      instance.bootstrap(host)
+    end
+
+    it "passes pre-configured options to #bootstrap" do
+      host_commander.should_receive(:bootstrap).with(host, options)
+      instance.bootstrap(host)
+    end
+  end
+
+  describe "#chef_run" do
+    it "sends the message #chef_client to the instance's host_commander" do
+      host_commander.should_receive(:chef_client).with(host, ssh: instance.ssh, winrm: instance.winrm)
+      instance.chef_run(host)
+    end
+  end
+
+  describe "#put_secret" do
+    let(:secret) { options[:encrypted_data_bag_secret] }
+
+    it "sends the message #put_secret to the instance's host_commander" do
+      host_commander.should_receive(:put_secret).with(host, secret, options.slice(:ssh, :winrm))
+      instance.put_secret(host)
+    end
+  end
+
+  describe "#ruby_script" do
+    let(:command_lines) { ["puts 'hello'", "puts 'there'"] }
+
+    it "sends the message #ruby_script to the instance's host_commander" do
+      host_commander.should_receive(:ruby_script).with(host, command_lines, ssh: instance.ssh, winrm: instance.winrm)
+      instance.ruby_script(host, command_lines)
+    end
+  end
+
+  describe "#run" do
+    let(:command) { "echo 'hello winrm_connectorld'" }
+
+    it "sends the message #run to the instance's host_commander" do
+      host_commander.should_receive(:run).with(host, command, ssh: instance.ssh, winrm: instance.winrm)
+      instance.run(host, command)
+    end
+  end
+
+  describe "#platform_specific_run" do
+    let(:ssh_command) { "hostname -f" }
+    let(:winrm_command) { "echo %COMPUTERNAME%" }
+    let(:ssh_connector) { Ridley::HostConnector::SSH.new }
+    let(:winrm_connector) { Ridley::HostConnector::WinRM.new }
+    let(:unsupported_connector) { Object.new }
+
+    describe "expecting the ssh connector" do
+      before do
+        host_commander.stub(:connector_for).and_return ssh_connector
+      end
+      it "sends the ssh command" do
+        instance.should_receive(:run).with(host, ssh_command)
+        instance.platform_specific_run(host, ssh: ssh_command, winrm: winrm_command)
+      end
+
+      it "raises an error if no command is provided for the ssh connector when the ssh connector is used" do
+        expect {
+          instance.platform_specific_run(host, winrm: winrm_command)
+        }.to raise_error(Ridley::Errors::CommandNotProvided)
+      end
+
+      it "raises an error if an empty command is provided for the ssh connector when the ssh connector is used" do
+        expect {
+          instance.platform_specific_run(host, ssh: "", winrm: winrm_command)
+        }.to raise_error(Ridley::Errors::CommandNotProvided)
+      end
+
+      it "raises an error if a nil command is provided for the ssh connector when the ssh connector is used" do
+        expect {
+          instance.platform_specific_run(host, ssh: nil, winrm: winrm_command)
+        }.to raise_error(Ridley::Errors::CommandNotProvided)
+      end
+    end
+
+    describe "expecting the winrm connector" do
+      before do
+        host_commander.stub(:connector_for).and_return winrm_connector
+      end
+      it "sends the ssh command if the connector is winrm" do
+        instance.should_receive(:run).with(host, winrm_command)
+        instance.platform_specific_run(host, ssh: ssh_command, winrm: winrm_command)
+      end
+
+      it "raises an error if no command is provided for the winrm connector when the winrm connector is used" do
+        expect {
+          instance.platform_specific_run(host, ssh: ssh_command)
+        }.to raise_error(Ridley::Errors::CommandNotProvided)
+      end
+
+      it "raises an error if an empty is provided for the winrm connector when the winrm connector is used" do
+        expect {
+          instance.platform_specific_run(host, ssh: ssh_command, winrm: "")
+        }.to raise_error(Ridley::Errors::CommandNotProvided)
+      end
+
+      it "raises a nil command is provided for the winrm connector when the winrm connector is used" do
+        expect {
+          instance.platform_specific_run(host, ssh: ssh_command, winrm: nil)
+        }.to raise_error(Ridley::Errors::CommandNotProvided)
+      end
+    end
+
+    it "raises a RuntimeError if an unsupported connector is used" do
+      host_commander.stub(:connector_for).and_return unsupported_connector
+      expect {
+        instance.platform_specific_run(host, ssh: ssh_command, winrm: winrm_command)
+      }.to raise_error(RuntimeError)
+    end
+  end  
+end

metadata

@@ -0,0 +1,178 @@
+--- !ruby/object:Gem::Specification
+name: ridley-connectors
+version: !ruby/object:Gem::Version
+  version: 1.0.1
+platform: ruby
+authors:
+- Jamie Winsor
+- Kyle Allan
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-11-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: celluloid
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.15'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.15'
+- !ruby/object:Gem::Dependency
+  name: celluloid-io
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.15'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.15'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ridley
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  name: winrm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+- !ruby/object:Gem::Dependency
+  name: buff-ruby_engine
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.1'
+description: A Connector API for talking to nodes managed by Chef
+email:
+- jamie@vialstudios.com
+- kallan@riotgames.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .ruby-version
+- .travis.yml
+- CHANGELOG.md
+- Gemfile
+- Guardfile
+- LICENSE
+- README.md
+- Thorfile
+- lib/ridley-connectors.rb
+- lib/ridley-connectors/chef_objects/node_object.rb
+- lib/ridley-connectors/client.rb
+- lib/ridley-connectors/host_commander.rb
+- lib/ridley-connectors/host_connector.rb
+- lib/ridley-connectors/host_connector/response.rb
+- lib/ridley-connectors/host_connector/ssh.rb
+- lib/ridley-connectors/host_connector/winrm.rb
+- lib/ridley-connectors/host_connector/winrm/command_uploader.rb
+- lib/ridley-connectors/resources/node_resource.rb
+- lib/ridley-connectors/version.rb
+- ridley-connectors.gemspec
+- spec/fixtures/encrypted_data_bag_secret
+- spec/fixtures/my-fake.pem
+- spec/spec_helper.rb
+- spec/support/actor_mocking.rb
+- spec/support/spec_helpers.rb
+- spec/unit/ridley-connectors/chef_objects/node_object_spec.rb
+- spec/unit/ridley-connectors/client_spec.rb
+- spec/unit/ridley-connectors/host_commander_spec.rb
+- spec/unit/ridley-connectors/host_connector/ssh_spec.rb
+- spec/unit/ridley-connectors/host_connector/winrm/command_uploader_spec.rb
+- spec/unit/ridley-connectors/host_connector/winrm_spec.rb
+- spec/unit/ridley-connectors/host_connector_spec.rb
+- spec/unit/ridley-connectors/resources/node_resource_spec.rb
+homepage: https://github.com/RiotGames/ridley-connectors
+licenses:
+- Apache 2.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.1
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: A Connector API for talking to nodes managed by Chef
+test_files:
+- spec/fixtures/encrypted_data_bag_secret
+- spec/fixtures/my-fake.pem
+- spec/spec_helper.rb
+- spec/support/actor_mocking.rb
+- spec/support/spec_helpers.rb
+- spec/unit/ridley-connectors/chef_objects/node_object_spec.rb
+- spec/unit/ridley-connectors/client_spec.rb
+- spec/unit/ridley-connectors/host_commander_spec.rb
+- spec/unit/ridley-connectors/host_connector/ssh_spec.rb
+- spec/unit/ridley-connectors/host_connector/winrm/command_uploader_spec.rb
+- spec/unit/ridley-connectors/host_connector/winrm_spec.rb
+- spec/unit/ridley-connectors/host_connector_spec.rb
+- spec/unit/ridley-connectors/resources/node_resource_spec.rb
+has_rdoc: