pbox 1.17.2

data/spec/rhc/commands/snapshot_spec.rb

@@ -0,0 +1,179 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc/commands/snapshot'
+require 'rhc/config'
+require 'rhc/tar_gz'
+
+describe RHC::Commands::Snapshot do
+
+  APP_NAME = 'mockapp'
+
+  let!(:rest_client) { MockRestClient.new }
+  before do
+    user_config
+    @app = rest_client.add_domain("mockdomain").add_application APP_NAME, 'mock-1.0'
+    @ssh_uri = URI.parse @app.ssh_url
+    @targz_filename = APP_NAME + '.tar.gz'
+    FileUtils.cp(File.expand_path('../../assets/targz_sample.tar.gz', __FILE__), @targz_filename)
+    File.chmod 0644, @targz_filename unless File.executable? @targz_filename
+  end
+
+  after do
+    File.delete @targz_filename if File.exist? @targz_filename
+  end
+
+  describe 'snapshot without an action' do
+    let(:arguments) {['snapshot', '--trace', '--noprompt']}
+    it('should raise') { expect{ run }.to raise_error(ArgumentError, /Please specify an action to take/) }
+  end
+
+  describe 'snapshot save' do
+    let(:arguments) {['snapshot', 'save', '--app', 'mockapp', '-d']}
+
+    context 'when saving a snapshot' do
+      before do
+        subject.class.any_instance.should_receive(:exec).with("ssh #{@ssh_uri.user}@#{@ssh_uri.host} 'snapshot' > #{@app.name}.tar.gz").and_return([0, 'some save output'])
+      end
+      it { expect { run }.to exit_with_code(0) }
+      it { run_output.should_not match 'some save output' }
+    end
+
+    context 'when failing to save a snapshot' do
+      before do
+        subject.class.any_instance.should_receive(:has_ssh?).and_return(true)
+        subject.class.any_instance.should_receive(:exec).with("ssh #{@ssh_uri.user}@#{@ssh_uri.host} 'snapshot' > #{@app.name}.tar.gz").and_return([1, 'some save failures'])
+      end
+      it { expect { run }.to exit_with_code(130) }
+      it { run_output.should match 'some save failures' }
+    end
+
+    context 'when saving a snapshot on windows' do
+      before do
+        RHC::Helpers.stub(:windows?) do ; true; end
+        RHC::Helpers.stub(:jruby?) do ; false ; end
+        RHC::Helpers.stub(:linux?) do ; false ; end
+        ssh = double(Net::SSH)
+        Net::SSH.should_receive(:start).with(@ssh_uri.host, @ssh_uri.user).and_yield(ssh)
+        ssh.should_receive(:exec!).with("snapshot").and_yield(nil, :stdout, 'foo').and_yield(nil, :stderr, 'foo')
+      end
+      it { expect { run }.to exit_with_code(0) }
+      it { run_output.should match("Success") }
+    end
+
+    context 'when timing out on windows' do
+      before do
+        RHC::Helpers.stub(:windows?) do ; true; end
+        RHC::Helpers.stub(:jruby?) do ; false ; end
+        RHC::Helpers.stub(:linux?) do ; false ; end
+        ssh = double(Net::SSH)
+        Net::SSH.should_receive(:start).with(@ssh_uri.host, @ssh_uri.user).and_raise(Timeout::Error)
+      end
+      it { expect { run }.to exit_with_code(130) }
+    end
+
+    describe 'snapshot save deployment' do
+      let(:arguments) {['snapshot', 'save', '--app', 'mockapp', '--deployment', '-d']}
+
+      context 'when saving a deployment snapshot' do
+        before do
+          subject.class.any_instance.should_receive(:exec).with("ssh #{@ssh_uri.user}@#{@ssh_uri.host} 'gear archive-deployment' > #{@app.name}.tar.gz").and_return([0, 'some save output'])
+        end
+        it { expect { run }.to exit_with_code(0) }
+        it { run_output.should_not match 'some save output' }
+      end
+    end
+
+  end
+
+  describe 'snapshot save with invalid ssh executable' do
+    let(:arguments) {['snapshot', 'save', '--trace', '--noprompt', '-l', 'test@test.foo', '-p', 'password', '--app', 'mockapp', '--ssh', 'path_to_ssh']}
+    it('should raise') { expect{ run }.to raise_error(RHC::InvalidSSHExecutableException, /SSH executable 'path_to_ssh' does not exist./) }
+  end
+
+  describe 'snapshot save when ssh is not executable' do
+    let(:arguments) {['snapshot', 'save', '--trace', '--noprompt', '-l', 'test@test.foo', '-p', 'password', '--app', 'mockapp', '--ssh', @targz_filename]}
+    it('should raise') { expect{ run }.to raise_error(RHC::InvalidSSHExecutableException, /SSH executable '#{@targz_filename}' is not executable./) }
+  end
+
+  describe 'snapshot restore' do
+    let(:arguments) {['snapshot', 'restore', '--app', 'mockapp', '-d']}
+
+    context 'when restoring a snapshot' do
+      before do
+        File.stub(:exists?).and_return(true)
+        RHC::TarGz.stub(:contains).and_return(true)
+        subject.class.any_instance.should_receive(:exec).with("cat '#{@app.name}.tar.gz' | ssh #{@ssh_uri.user}@#{@ssh_uri.host} 'restore INCLUDE_GIT'").and_return([0, 'some restore output'])
+      end
+      it('should succeed') { expect { run }.to exit_with_code(0) }
+      it { run_output.should_not match 'some restore output' }
+    end
+
+    context 'when restoring a snapshot and failing to ssh' do
+      before do
+        File.stub(:exists?).and_return(true)
+        RHC::TarGz.stub(:contains).and_return(true)
+        subject.class.any_instance.should_receive(:has_ssh?).and_return(true)
+        subject.class.any_instance.should_receive(:exec).with("cat '#{@app.name}.tar.gz' | ssh #{@ssh_uri.user}@#{@ssh_uri.host} 'restore INCLUDE_GIT'").and_return([1, 'some restore failures'])
+      end
+      it { expect { run }.to exit_with_code(130) }
+      it { run_output.should match 'some restore failures' }
+    end
+
+    context 'when restoring a snapshot on windows' do
+      before do
+        RHC::Helpers.stub(:windows?) do ; true; end
+        RHC::Helpers.stub(:jruby?) do ; false ; end
+        RHC::Helpers.stub(:linux?) do ; false ; end
+        ssh = double(Net::SSH)
+        session = double(Net::SSH::Connection::Session)
+        channel = double(Net::SSH::Connection::Channel)
+        Net::SSH.should_receive(:start).with(@ssh_uri.host, @ssh_uri.user).and_return(session)
+        session.should_receive(:open_channel).and_yield(channel)
+        channel.should_receive(:exec).with("restore INCLUDE_GIT").and_yield(nil, nil)
+        channel.should_receive(:on_data).and_yield(nil, 'foo')
+        channel.should_receive(:on_extended_data).and_yield(nil, nil, 'foo')
+        channel.should_receive(:on_close).and_yield(nil)
+        lines = ''
+        File.open(File.expand_path('../../assets/targz_sample.tar.gz', __FILE__), 'rb') do |file|
+          file.chunk(1024) do |chunk|
+            lines << chunk
+          end
+        end
+        channel.should_receive(:send_data).with(lines)
+        channel.should_receive(:eof!)
+        session.should_receive(:loop)
+      end
+      it { expect { run }.to exit_with_code(0) }
+    end
+
+    context 'when timing out on windows' do
+      before do
+        RHC::Helpers.stub(:windows?) do ; true; end
+        RHC::Helpers.stub(:jruby?) do ; false ; end
+        RHC::Helpers.stub(:linux?) do ; false ; end
+        ssh = double(Net::SSH)
+        Net::SSH.should_receive(:start).with(@ssh_uri.host, @ssh_uri.user).and_raise(Timeout::Error)
+      end
+      it { expect { run }.to exit_with_code(130) }
+    end
+
+  end
+
+  describe 'snapshot restore file not found' do
+    let(:arguments) {['snapshot', 'restore', 'mockapp', '-f', 'foo.tar.gz']}
+    context 'when restoring a snapshot' do
+      it { expect { run }.to exit_with_code(130) }
+    end
+  end
+
+  describe 'snapshot restore with invalid ssh executable' do
+    let(:arguments) {['snapshot', 'restore', '--trace', '--noprompt', '-l', 'test@test.foo', '-p', 'password', '--app', 'mockapp', '--ssh', 'path_to_ssh']}
+    it('should raise') { expect{ run }.to raise_error(RHC::InvalidSSHExecutableException, /SSH executable 'path_to_ssh' does not exist./) }
+  end
+
+  describe 'snapshot save when ssh is not executable' do
+    let(:arguments) {['snapshot', 'restore', '--trace', '--noprompt', '-l', 'test@test.foo', '-p', 'password', '--app', 'mockapp', '--ssh', @targz_filename]}
+    it('should raise') { expect{ run }.to raise_error(RHC::InvalidSSHExecutableException, /SSH executable '#{@targz_filename}' is not executable./) }
+  end
+end
+

data/spec/rhc/commands/ssh_spec.rb

@@ -0,0 +1,163 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc/commands/ssh'
+
+describe RHC::Commands::Ssh do
+  let!(:rest_client){ MockRestClient.new }
+  let!(:config){ user_config }
+  before{ RHC::Config.stub(:home_dir).and_return('/home/mock_user') }
+  before{ Kernel.stub(:exec).and_raise(RuntimeError) }
+
+  describe 'ssh default' do
+    context 'ssh' do
+      let(:arguments) { ['ssh'] }
+      it { run_output.should match('Usage:') }
+    end
+  end
+
+  describe 'ssh without command' do
+    let(:arguments) { ['app', 'ssh', 'app1'] }
+
+    context 'when run' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        Kernel.should_receive(:exec).with("ssh", "fakeuuidfortestsapp1@127.0.0.1").and_return(0)
+      end
+      it { run_output.should match("Connecting to fakeuuidfortestsapp") }
+      it { expect{ run }.to exit_with_code(0) }
+    end
+  end
+
+  describe 'app ssh with command' do
+    let(:arguments) { ['app', 'ssh', 'app1', 'ls', '/tmp'] }
+
+    context 'when run' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        Kernel.should_receive(:exec).with("ssh", "fakeuuidfortestsapp1@127.0.0.1", "ls", "/tmp").and_return(0)
+      end
+      it { run_output.should_not match("Connecting to fakeuuidfortestsapp") }
+      it { expect { run }.to exit_with_code(0) }
+    end
+  end
+
+  describe 'app ssh with --gears' do
+    before{ rest_client.add_domain("mockdomain").add_application("app1", "mock_type", true) }
+
+    context 'with no command' do
+      let(:arguments) { ['app', 'ssh', 'app1', '--gears'] }
+      it('should display usage info') { run_output.should match("Usage:") }
+      it { expect { run }.to exit_with_code(1) }
+    end
+    context 'with a command' do
+      let(:arguments) { ['app', 'ssh', 'app1', '--gears', '--', 'command', '--trace'] }
+      before{ expect_multi_ssh('command', 'fakegearid0@fakesshurl.com' => 'foo', 'fakegearid1@fakesshurl.com' => 'bar') }
+      it('should print the ssh output') { run_output.should == "[fakegearid0 ] foo\n[fakegearid1 ] bar\n\n" }
+      it('should return successfully') { expect{ run }.to exit_with_code(0) }
+    end
+    context 'with an implicit app name' do
+      before{ subject.class.any_instance.stub(:git_config_get){ |key| 'app1' if key == "pbox.app-name" } }
+      let(:arguments) { ['app', 'ssh', '--gears', '--', 'command', '--trace'] }
+      before{ expect_multi_ssh('command', 'fakegearid0@fakesshurl.com' => 'foo', 'fakegearid1@fakesshurl.com' => 'bar') }
+      it('should print the ssh output') { run_output.should == "[fakegearid0 ] foo\n[fakegearid1 ] bar\n\n" }
+      it('should return successfully') { expect{ run }.to exit_with_code(0) }
+    end
+    context 'with an application id' do
+      let(:arguments) { ['app', 'ssh', '--application-id', rest_client.domains.first.applications.first.id, '--gears', 'command', '--trace'] }
+      before{ expect_multi_ssh('command', 'fakegearid0@fakesshurl.com' => 'foo', 'fakegearid1@fakesshurl.com' => 'bar') }
+      it('should print the ssh output') { run_output.should == "[fakegearid0 ] foo\n[fakegearid1 ] bar\n\n" }
+      it('should return successfully') { expect{ run }.to exit_with_code(0) }
+    end
+    context 'with --raw' do
+      let(:arguments) { ['app', 'ssh', 'app1', '--gears', 'command', '--raw'] }
+      before{ expect_multi_ssh('command', 'fakegearid0@fakesshurl.com' => 'foo', 'fakegearid1@fakesshurl.com' => 'bar') }
+      it('should print the ssh output') { run_output.should == "foo\nbar\n\n" }
+    end
+    context 'with --limit' do
+      let(:arguments) { ['app', 'ssh', 'app1', '--gears', 'command', '--limit', '1'] }
+      before{ expect_multi_ssh('command', 'fakegearid0@fakesshurl.com' => 'foo', 'fakegearid1@fakesshurl.com' => 'bar') }
+      it('should print the ssh output') { run_output.should == "[fakegearid0 ] foo\n[fakegearid1 ] bar\n\n" }
+    end
+    context 'with invalid --limit value' do
+      ['0','-10'].each do |value|
+        let(:arguments) { ['app', 'ssh', 'app1', '--gears', 'command', '--limit', value] }
+        it { run_output.should match('--limit must be an integer greater than zero') }
+      end
+    end
+    context 'with multiline output and --always-prefix' do
+      let(:arguments) { ['app', 'ssh', 'app1', '--gears', 'command', '--always-prefix'] }
+      before{ expect_multi_ssh('command', 'fakegearid0@fakesshurl.com' => "foo\ntest", 'fakegearid1@fakesshurl.com' => "bar\ntest") }
+      it('should print the ssh output') { run_output.should == "[fakegearid0 ] foo\n[fakegearid0 ] test\n[fakegearid1 ] bar\n[fakegearid1 ] test\n\n" }
+    end
+    context 'with multiline output' do
+      let(:arguments) { ['app', 'ssh', 'app1', '--gears', 'command'] }
+      before{ expect_multi_ssh('command', 'fakegearid0@fakesshurl.com' => "foo\ntest", 'fakegearid1@fakesshurl.com' => "bar\ntest") }
+      it('should print the ssh output') { run_output.should == "=== fakegearid0 \nfoo\ntest\n=== fakegearid1 \nbar\ntest\n\n" }
+    end
+  end
+
+  describe 'app ssh no system ssh' do
+    let(:arguments) { ['app', 'ssh', 'app1'] }
+
+    context 'when run' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        RHC::Commands::Ssh.any_instance.should_receive(:has_ssh?).and_return(false)
+      end
+      it { run_output.should match("No system SSH available. Please use the --ssh option to specify the path to your SSH executable, or install SSH.") }
+      it { expect { run }.to exit_with_code(1) }
+    end
+  end
+
+  describe 'app ssh custom ssh' do
+    let(:arguments) { ['app', 'ssh', 'app1', '--ssh', 'path_to_ssh'] }
+
+    context 'when custom ssh does not exist' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        RHC::Commands::Ssh.any_instance.should_not_receive(:has_ssh?)
+        File.should_receive(:exist?).with("path_to_ssh").once.and_return(false)
+      end
+      it { run_output.should match("SSH executable 'path_to_ssh' does not exist.") }
+      it { expect { run }.to exit_with_code(1) }
+    end
+
+    context 'when custom ssh is not executable' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        RHC::Commands::Ssh.any_instance.should_not_receive(:has_ssh?)
+        File.should_receive(:exist?).with("path_to_ssh").once.and_return(true)
+        File.should_receive(:executable?).with(/.*path_to_ssh/).at_least(1).and_return(false)
+      end
+      it { run_output.should match("SSH executable 'path_to_ssh' is not executable.") }
+      it { expect { run }.to exit_with_code(1) }
+    end
+
+    context 'when custom ssh exists' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        RHC::Commands::Ssh.any_instance.should_not_receive(:has_ssh?)
+        File.should_receive(:exist?).with("path_to_ssh").once.and_return(true)
+        File.should_receive(:executable?).with("path_to_ssh").once.and_return(true)
+        Kernel.should_receive(:exec).with("path_to_ssh", "fakeuuidfortestsapp1@127.0.0.1").once.times.and_return(0)
+      end
+      it { run_output.should match("Connecting to fakeuuidfortestsapp") }
+      it { expect { run }.to exit_with_code(0) }
+    end
+  end
+
+  describe 'ssh tests' do
+    let(:arguments) { ['app', 'ssh', 'app1', '-s /bin/blah'] }
+
+    context 'has_ssh?' do
+      before{ RHC::Commands::Ssh.any_instance.stub(:ssh_version){ raise "Fake Exception" } }
+      its(:has_ssh?) { should be_false }
+    end
+  end
+end

data/spec/rhc/commands/sshkey_spec.rb

@@ -0,0 +1,188 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc/commands/sshkey'
+require 'rhc/config'
+
+
+describe RHC::Commands::Sshkey do
+  let!(:rest_client){ MockRestClient.new }
+  before{ user_config }
+
+  describe 'list' do
+
+    context "when run with list command" do
+      let(:arguments) { %w[sshkey list --noprompt --config test.conf -l test@test.foo -p  password --trace] }
+
+      it { expect { run }.to exit_with_code(0) }
+      
+      it { run_output.should match(/mockkey1 \(type: ssh-rsa\)/) }
+      it { run_output.should match(/Fingerprint:.*0f:ce:86:80:df:a0:81:ca:db:f1:a7:0c:70:85:ce:00/) }
+      
+      it { run_output.should match(/mockkey3 \(type: krb5-principal\)/) }
+      it { run_output.should match(/Principal:.* mockuser@mockdomain/) }
+      it { run_output.should_not match(/Fingerprint:.*Invalid key/) }
+    end
+  end
+
+  describe 'show' do
+    context "when run with show command" do
+      let(:arguments) { %w[sshkey show mockkey1 --noprompt --config test.conf -l test@test.foo -p  password --trace] }
+
+      it { expect { run }.to exit_with_code(0) }
+      it { run_output.should match(/mockkey1 \(type: ssh-rsa\)/) }
+    end
+  end
+
+  describe "add" do
+    context "when adding a valid key" do
+      let(:arguments) { %w[sshkey add --noprompt --config test.conf -l test@test.foo -p password foobar id_rsa.pub] }
+
+      it 'adds the key' do
+        FakeFS do
+          keys = rest_client.sshkeys
+          num_keys = keys.length
+          File.open('id_rsa.pub', 'w') do |f|
+            f << 'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQCnCOqK7/mmvZ9AtCAerxjAasJ1rSpfuWT4vNm1+O/Fh0Di3chTWjY9a0M2hEnqkqnVG589L9CqCUeT0kdc3Vgw3JEcacSUr1z7tLr9kO+p/D5lSdQYzDGGRFOZ0H6lc/y8iNxWV1VO/sJvKx6cr5zvKIn8Q6GvhVNOxlai0IOb9FJxLGK95GLpZ+elzh8Tc9giy7KfwheAwhV2JoF9uRltE5JP/CNs7w/E29i1Z+jlueuu8RVotLmhSVNJm91Ey7OCtoI1iBE0Wv/SucFe32Qi08RWTM/MaGGz93KQNOVRGjNkosJjPmP1qU6WGBfliDkJAZXB0b6sEcnx1fbVikwZ'
+          end
+          expect { run }.to exit_with_code(0)
+          rest_client.sshkeys.length.should == num_keys + 1
+        end
+      end
+    end
+
+    context "when adding a valid key via command line arguments" do
+      let(:arguments) { %w[sshkey add --noprompt --config test.conf -l test@test.foo -p password foobar --type ssh-rsa --content AAAAB3NzaC1yc2EAAAADAQABAAABAQCnCOqK7/mmvZ9AtCAerxjAasJ1rSpfuWT4vNm1+O/Fh0Di3chTWjY9a0M2hEnqkqnVG589L9CqCUeT0kdc3Vgw3JEcacSUr1z7tLr9kO+p/D5lSdQYzDGGRFOZ0H6lc/y8iNxWV1VO/sJvKx6cr5zvKIn8Q6GvhVNOxlai0IOb9FJxLGK95GLpZ+elzh8Tc9giy7KfwheAwhV2JoF9uRltE5JP/CNs7w/E29i1Z+jlueuu8RVotLmhSVNJm91Ey7OCtoI1iBE0Wv/SucFe32Qi08RWTM/MaGGz93KQNOVRGjNkosJjPmP1qU6WGBfliDkJAZXB0b6sEcnx1fbVikwZ] }
+
+      it 'adds the key' do
+        keys = rest_client.sshkeys
+        num_keys = keys.length
+        expect { run }.to exit_with_code(0)
+        rest_client.sshkeys.length.should == num_keys + 1
+      end
+    end
+
+    context "when adding an invalid key" do
+      let(:arguments) { %w[sshkey add --noprompt --config test.conf -l test@test.foo -p password foobar id_rsa.pub] }
+
+      it "fails to add the key" do
+        FakeFS do
+          keys = rest_client.sshkeys
+          num_keys = keys.length
+          File.open('id_rsa.pub', 'w') do |f|
+            f << 'ssh-rsa AADAQABAAABAQCnCOqK7/mmvZ9AtCAerxjAasJ1rSpfuWT4vNm1+O/Fh0Di3chTWjY9a0M2hEnqkqnVG589L9CqCUeT0kdc3Vgw3JEcacSUr1z7tLr9kO+p/D5lSdQYzDGGRFOZ0H6lc/y8iNxWV1VO/sJvKx6cr5zvKIn8Q6GvhVNOxlai0IOb9FJxLGK95GLpZ+elzh8Tc9giy7KfwheAwhV2JoF9uRltE5JP/CNs7w/E29i1Z+jlueuu8RVotLmhSVNJm91Ey7OCtoI1iBE0Wv/SucFe32Qi08RWTM/MaGGz93KQNOVRGjNkosJjPmP1qU6WGBfliDkJAZXB0b6sEcnx1fbVikwZ'
+          end
+          expect { run }.to exit_with_code(128)
+          expect { run_output.should match(/Name:.* mockkey/) }
+          rest_client.sshkeys.length.should == num_keys
+        end
+      end
+
+      it "does not throw an error on an empty file" do
+        FakeFS do
+          keys = rest_client.sshkeys
+          num_keys = keys.length
+          File.open('id_rsa.pub', 'w') do |f|
+            f << ''
+          end
+          expect { run }.to exit_with_code(128)
+          expect { run_output.should match(/Name:.* mockkey/) }
+          rest_client.sshkeys.length.should == num_keys
+        end
+      end
+
+      it "exits with status code Errno::EACCES::Errno" do
+        IO.should_receive(:read).and_return("ssh_foo bar")
+        expect { run }.to exit_with_code(128)
+      end
+    end
+
+    context "when adding an invalid key via command line arguments" do
+      let(:arguments) { %w[sshkey add --noprompt --config test.conf -l test@test.foo -p password foobar --type ssh-rsa --content abcdef] }
+
+      it "fails to add the key" do
+        keys = rest_client.sshkeys
+        num_keys = keys.length
+        expect { run }.to exit_with_code(128)
+        expect { run_output.should match(/Name:.* mockkey/) }
+        rest_client.sshkeys.length.should == num_keys
+      end
+    end
+
+    context "when adding an invalid key with --confirm" do
+      let(:arguments) { %w[sshkey add --noprompt --confirm --config test.conf -l test@test.foo -p password foobar id_rsa.pub] }
+
+      it "warns and then adds the key" do
+        FakeFS do
+          keys = rest_client.sshkeys
+          num_keys = keys.length
+          File.open('id_rsa.pub', 'w') do |f|
+            f << 'ssh-rsa AADAQABAAABAQCnCOqK7/mmvZ9AtCAerxjAasJ1rSpfuWT4vNm1+O/Fh0Di3chTWjY9a0M2hEnqkqnVG589L9CqCUeT0kdc3Vgw3JEcacSUr1z7tLr9kO+p/D5lSdQYzDGGRFOZ0H6lc/y8iNxWV1VO/sJvKx6cr5zvKIn8Q6GvhVNOxlai0IOb9FJxLGK95GLpZ+elzh8Tc9giy7KfwheAwhV2JoF9uRltE5JP/CNs7w/E29i1Z+jlueuu8RVotLmhSVNJm91Ey7OCtoI1iBE0Wv/SucFe32Qi08RWTM/MaGGz93KQNOVRGjNkosJjPmP1qU6WGBfliDkJAZXB0b6sEcnx1fbVikwZ'
+          end
+          expect { run }.to exit_with_code(0)
+          expect { run_output.should match("key you are uploading is not recognized") }
+          rest_client.sshkeys.length.should == num_keys + 1
+        end
+      end
+    end
+
+    context "when adding a nonexistent key" do
+      let(:arguments) { %w[sshkey add --noprompt --config test.conf -l test@test.foo -p password foobar id_rsa.pub] }
+    
+      it "exits with status code Errno::ENOENT::Errno" do
+        expect { run }.to exit_with_code(128)
+      end
+    end
+    
+    context "when attempting to add an existing but inaccessible key" do
+      let(:arguments) { %w[sshkey add --noprompt --config test.conf -l test@test.foo -p password foobar inaccessible_key.pub] }
+      
+      #before :all do
+      #  @inaccessible_key = 'inaccessible_key.pub'
+      #  File.new(@inaccessible_key, 'w+')
+      #  File.chmod(0000, @inaccessible_key)
+      #end
+      
+      #after :all do
+      #  File.delete @inaccessible_key
+      #end
+    
+      it "exits with status code Errno::EACCES::Errno" do
+        IO.should_receive(:read).and_raise(Errno::EACCES)
+        expect { run }.to exit_with_code(128)
+      end
+
+    end
+
+    context "when adding a key without correct arguments" do
+      let(:arguments) { %w[sshkey add --noprompt --config test.conf -l test@test.foo -p password foobar] }
+  
+      it "exits with argument error" do
+        expect { run }.to exit_with_code(1)
+      end
+    end
+  end
+
+  describe "remove" do
+    context "when removing an existing key" do
+      let (:arguments) { %w[sshkey remove --noprompt --config test.conf -l test@test.foo -p password mockkey2] }
+
+      it 'deletes the key' do
+        keys = rest_client.sshkeys
+        num_keys = keys.length
+        expect{ run }.to exit_with_code(0)
+        rest_client.sshkeys.length.should == num_keys - 1
+      end
+    end
+
+    context "when removing a nonexistent key" do
+      let (:arguments) { %w[sshkey remove --noprompt --config test.conf -l test@test.foo -p password no_match] }
+      before{ @keys = rest_client.sshkeys }
+
+      it 'leaves keys untouched' do
+        num_keys = @keys.length
+        expect{ run }.to exit_with_code(0)
+        rest_client.sshkeys.length.should == num_keys
+      end
+    end
+  end
+end

data/spec/rhc/commands/tail_spec.rb

@@ -0,0 +1,81 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc/commands/tail'
+require 'rhc/config'
+
+describe RHC::Commands::Tail do
+  let!(:rest_client) { MockRestClient.new }
+  before(:each) do
+    user_config
+    domain = rest_client.add_domain("mock-domain-0")
+    @app = domain.add_application("mock-app-0", "ruby-1.8.7")
+    @app.stub(:ssh_url).and_return("ssh://user@test.domain.com")
+  end
+
+  describe 'help' do
+    let(:arguments) { ['tail', '--help'] }
+
+    context 'help is run' do
+      it "should display help" do
+        expect { run }.to exit_with_code(0)
+      end
+      it('should output usage') { run_output.should match("Usage: pbox tail") }
+    end
+  end
+
+  describe 'tail' do
+    let(:arguments) { ['tail', 'mock-app-0'] }
+
+    context 'when ssh connects' do
+      before (:each) {Net::SSH.should_receive(:start).with('test.domain.com', 'user', :compression => false) }
+      it { expect { run }.to exit_with_code(0) }
+    end
+
+    context 'is run on an unreachable domain' do
+      before (:each) {Net::SSH.should_receive(:start).and_raise(SocketError) }
+      it { expect { run }.to exit_with_code(1) }
+      it { run_output.should =~ /The connection to test.domain.com failed: / }
+    end
+
+    context 'is refused' do
+      before (:each) {Net::SSH.should_receive(:start).and_raise(Errno::ECONNREFUSED) }
+      it { expect { run }.to exit_with_code(1) }
+      it { run_output.should =~ /The server test.domain.com refused a connection with user user/ }
+    end
+
+    context 'succeeds and exits on Interrupt' do
+      before (:each) { rest_client.stub(:find_domain) { raise Interrupt } }
+      it { expect { run }.to raise_error(Interrupt) }
+    end
+
+    context 'succeeds when a gear is specified' do
+      before (:each) {Net::SSH.should_receive(:start).with('fakesshurl.com', 'fakegearid0', :compression => false) }
+      let(:arguments) { ['tail', 'mock-app-0', '--gear', 'fakegearid0' ] }
+
+      it { run_output.should_not =~ /Connecting to fakesshurl.com/ }
+      it { expect { run }.to exit_with_code(0) }
+    end
+
+    context 'fails when an invalid gear is specified' do
+      let(:arguments) { ['tail', 'mock-app-0', '--gear', 'gearthatdoesnotexist' ] }
+
+      it { expect { run }.to exit_with_code(1) }
+      it { run_output.should =~ /Gear gearthatdoesnotexist not found/ }
+    end
+
+    context 'fails when a gear with no ssh info is specified' do
+      let(:arguments) { ['tail', 'mock-app-0', '--gear', 'fakegearid' ] }
+
+      # Given - gears in gear group should not have ssh info
+      before(:each) do
+        gg = MockRestGearGroup.new(rest_client)
+        @app.stub(:gear_groups).and_return([gg])
+        gg.stub(:gears).and_return([{'state' => 'started', 'id' => 'fakegearid'}])
+      end
+
+      it { expect { run }.to exit_with_code(1) }
+      it { run_output.should =~ /The server does not support operations on individual gears./ }
+    end
+
+  end
+end