startapp 0.1.6

data/spec/rhc/commands/logout_spec.rb

@@ -0,0 +1,86 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc'
+require 'rhc/commands/logout'
+
+describe RHC::Commands::Logout do
+
+  describe '#run' do
+    let(:arguments) { ['logout'] }
+    let(:username) { 'foo' }
+    let(:password) { nil }
+    let(:supports_auth) { false }
+    let(:server) { mock_uri }
+    let!(:token_store) { RHC::Auth::TokenStore.new(Dir.mktmpdir) }
+    before{ user_config }
+    before do
+      stub_api(false, supports_auth)
+      challenge{ stub_user }
+      RHC::Auth::TokenStore.stub(:new).and_return(token_store)
+    end
+
+    context "when calling from the alias" do
+      let(:arguments){ ['account', 'logout', '-h'] }
+      it("should print usage"){ run_output.should match "Usage: app logout" }
+    end
+
+    it("should clear the token cache"){ expect{ run }.to call(:clear).on(token_store) }
+    it("should exit with success"){ expect{ run }.to exit_with_code(0) }
+    it("should display a message"){ run_output.should match("All local sessions removed.") }
+
+    context "when --all is requested" do
+      let(:arguments) { ['account', 'logout', '--all'] }
+
+      context "if the server does not implement authorizations" do
+        it("should display a message"){ run_output.should match(/Deleting all authorizations associated with your account.*not supported/) }
+        it("should exit with success"){ expect{ run }.to exit_with_code(0) }
+      end
+
+      context "if the server implements authorizations" do
+        let(:supports_auth) { true }
+        before{ stub_delete_authorizations }
+
+        it("should display a message"){ run_output.should match(/Deleting all authorizations associated with your account.*done/) }
+        it("should exit with success"){ expect{ run }.to exit_with_code(0) }
+      end
+    end
+
+    context "when --token is provided" do
+      let(:arguments) { ['account', 'logout', '--token', 'foo'] }
+      def user_auth; { :token => 'foo' }; end
+
+      context "if the server does not implement authorizations" do
+        it("should display a message"){ run_output.should match(/Ending session on server.*not supported/) }
+        it("should exit with success"){ expect{ run }.to exit_with_code(0) }
+      end
+
+      context "if the server implements authorizations" do
+        let(:supports_auth) { true }
+
+        context "if the server returns successfully" do
+          before{ stub_delete_authorization('foo') }
+
+          it("should display a message"){ run_output.should match(/Ending session on server.*deleted/) }
+          it("should exit with success"){ expect{ run }.to exit_with_code(0) }
+          it("should clear the token cache"){ expect{ run }.to call(:clear).on(token_store) }
+        end
+
+        context "if the server rejects the token" do
+          before{ stub_request(:delete, mock_href('broker/rest/user/authorizations/foo', false)).to_return(:status => 401, :body => {}.to_json) }
+
+          it("should display a message"){ run_output.should match(/Ending session on server.*already closed/) }
+          it("should exit with success"){ expect{ run }.to exit_with_code(0) }
+          it("should clear the token cache"){ expect{ run }.to call(:clear).on(token_store) }
+        end
+
+        context "if the server returns an unexpected error" do
+          before{ stub_request(:delete, mock_href('broker/rest/user/authorizations/foo', false)).to_return(:status => 500, :body => {}.to_json) }
+
+          it("should display a message"){ run_output.should match(/Ending session on server.*The server did not respond/) }
+          it("should exit with success"){ expect{ run }.to exit_with_code(0) }
+          it("should clear the token cache"){ expect{ run }.to call(:clear).on(token_store) }
+        end
+      end
+    end
+  end
+end

data/spec/rhc/commands/member_spec.rb

@@ -0,0 +1,247 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc/commands/member'
+
+describe RHC::Commands::Member do
+
+  before{ user_config }
+
+  describe 'help' do
+    let(:arguments) { ['member', '--help'] }
+
+    it "should display help" do
+      expect { run }.to exit_with_code(0)
+    end
+    it('should output usage') { run_output.should match "Usage: app member" }
+    it('should output info about roles') { run_output.should match "Teams of developers can collaborate" }
+  end
+
+  let(:username){ 'test_user' }
+  let(:password){ 'test_password' }
+  let(:server){ 'test.domain.com' }
+
+  def with_mock_rest_client
+    @rest_client ||= MockRestClient.new
+  end
+  def with_mock_domain
+    @domain ||= with_mock_rest_client.add_domain("mock-domain-0")
+  end
+  def with_mock_app
+    @app ||= begin
+      app = with_mock_domain.add_application("mock-app-0", "ruby-1.8.7")
+      app.stub(:ssh_url).and_return("ssh://user@test.domain.com")
+      app.stub(:supports_members?).and_return(supports_members)
+      app
+    end
+  end
+
+  let(:owner){ RHC::Rest::Membership::Member.new(:id => '1', :role => 'admin', :owner => true, :login => 'alice') }
+  let(:other_admin){ RHC::Rest::Membership::Member.new(:id => '2', :role => 'admin', :login => 'Bob') }
+  let(:other_editor){ RHC::Rest::Membership::Member.new(:id => '3', :role => 'edit', :name => 'Carol', :login => 'carol') }
+  let(:other_viewer){ RHC::Rest::Membership::Member.new(:id => '4', :role => 'view', :name => 'Doug', :login => 'doug@doug.com') }
+  let(:other_viewer2){ RHC::Rest::Membership::Member.new(:id => '5', :role => 'view', :name => 'ViewerC', :login => 'viewerc@viewer.com') }
+  let(:other_viewer3){ RHC::Rest::Membership::Member.new(:id => '6', :role => 'view', :name => 'ViewerB', :login => 'viewerb@viewer.com') }
+  let(:other_viewer4){ RHC::Rest::Membership::Member.new(:id => '7', :role => 'view', :name => 'ViewerA', :login => 'viewera@viewer.com') }
+
+  describe 'show-domain' do
+    context 'with members' do
+      let(:arguments) { ['domain', 'show', 'mock-domain-0'] }
+      let(:supports_members){ true }
+
+      before{ with_mock_domain.add_member(owner).add_member(other_editor).add_member(other_admin).add_member(other_viewer).add_member(other_viewer2).add_member(other_viewer3).add_member(other_viewer4) }
+      it { expect { run }.to exit_with_code(0) }
+      it { run_output.should =~ /owned by alice/ }
+      it { run_output.should =~ /Bob\s+\(admin\)/ }
+      it { run_output.should =~ /<carol>\s+\(edit\)/ }
+      it { run_output.should =~ /<doug@doug\.com>\s+\(view\)/ }
+      it("should order the members by role then by name") { run_output.should =~ /Bob.*admin.*Admins.*Carol.*Editors.*ViewerA.*ViewerB.*ViewerC.*Viewers/m }
+      it("should include the login value") { run_output.should =~ /alice.*Bob.*carol.*doug@doug\.com/m }
+    end
+  end
+
+  describe 'list-member' do
+    context 'on a domain' do
+      let(:arguments) { ['members', '-n', 'mock-domain-0'] }
+      let(:supports_members){ true }
+      before{ with_mock_domain.add_member(owner) }
+      it { expect { run }.to exit_with_code(0) }
+      it { run_output.should =~ /alice\s+admin \(owner\)/ }
+      it("should not show the name column") { run_output.should =~ /^Login\s+Role$/ }
+    end
+
+    context 'on an application' do
+      let(:arguments) { ['members', 'mock-domain-0/mock-app-0'] }
+      let(:supports_members){ false }
+      before{ with_mock_app }
+
+      it { expect { run }.to exit_with_code(1) }
+      it { run_output.should =~ /The server does not support adding or removing members/ }
+
+      context "with only owner" do
+        let(:supports_members){ true }
+        before{ with_mock_app.add_member(owner) }
+        it { expect { run }.to exit_with_code(0) }
+        it { run_output.should =~ /alice\s+admin \(owner\)/ }
+        it("should not show the name column") { run_output.should =~ /^Login\s+Role$/ }
+
+        context "with ids" do
+          let(:arguments) { ['members', 'mock-domain-0/mock-app-0', '--ids'] }
+          it { expect { run }.to exit_with_code(0) }
+          it { run_output.should =~ /alice\s+admin \(owner\) 1/ }
+          it("should not show the name column") { run_output.should =~ /^Login\s+Role\s+ID$/ }
+        end
+      end
+
+      context "with several members" do
+        let(:supports_members){ true }
+        before{ with_mock_app.add_member(owner).add_member(other_editor).add_member(other_admin).add_member(other_viewer) }
+        it { expect { run }.to exit_with_code(0) }
+        it { run_output.should =~ /alice\s+admin \(owner\)/ }
+        it { run_output.should =~ /Bob\s+admin/ }
+        it { run_output.should =~ /carol\s+edit/ }
+        it { run_output.should =~ /doug\.com\s+view/ }
+        it("should order the members by role") { run_output.should =~ /admin.*owner.*admin.*edit.*view/m }
+        it("should include the login value") { run_output.should =~ /alice.*Bob.*carol.*doug@doug\.com/m }
+        it("should show the name column") { run_output.should =~ /^Name\s+Login\s+Role$/ }
+      end
+    end
+  end
+
+  describe 'add-member' do
+    before do
+      stub_api
+      challenge{ stub_one_domain('test', nil, mock_user_auth) }
+    end
+
+    context "when the resource doesn't support membership changes" do
+      let(:arguments) { ['add-member', 'testuser', '-n', 'test'] }
+      it { expect { run }.to exit_with_code(1) }
+      it { run_output.should =~ /Adding 1 editor to domain .*The server does not support adding or removing members/ }
+    end
+
+    context "with supported membership" do
+      let(:supports_members?){ true }
+
+      context 'with a valid user' do
+        let(:arguments) { ['add-member', 'testuser', '-n', 'test'] }
+        before do
+          stub_api_request(:patch, "broker/rest/domains/test/members").
+            with(:body => {:members => [{'login' => 'testuser', 'role' => 'edit'}]}).
+            to_return({:body => {:type => 'members', :data => [], :messages => [{:exit_code => 0, :field => 'login', :index => 0, :severity => 'info', :text => 'Added 1 member'},]}.to_json, :status => 200})
+        end
+        it { expect { run }.to exit_with_code(0) }
+        it { run_output.should =~ /Adding 1 editor to domain .*done/ }
+      end
+
+      context 'with an invalid role' do
+        let(:arguments) { ['add-member', 'testuser', '-n', 'test', '--role', 'missing'] }
+        it { expect { run }.to exit_with_code(1) }
+        it { run_output.should =~ /The provided role 'missing' is not valid\. Supported values: .*admin/ }
+      end
+
+      context 'with a missing user' do
+        let(:arguments) { ['add-member', 'testuser', '-n', 'test'] }
+        before do
+          stub_api_request(:patch, "broker/rest/domains/test/members").
+            with(:body => {:members => [{'login' => 'testuser', 'role' => 'edit'}]}).
+            to_return({:body => {:messages => [{:exit_code => 132, :field => 'login', :index => 0, :severity => 'error', :text => 'There is no user with a login testuser'},]}.to_json, :status => 422})
+        end
+        it { expect { run }.to exit_with_code(1) }
+        it { run_output.should =~ /Adding 1 editor to domain.*There is no user with a login testuser/ }
+      end
+
+      context 'with a missing user id and role' do
+        let(:arguments) { ['add-member', '123', '-n', 'test', '--ids', '--role', 'admin'] }
+        before do
+          stub_api_request(:patch, "broker/rest/domains/test/members").
+            with(:body => {:members => [{'id' => '123', 'role' => 'admin'}]}).
+            to_return({:body => {:messages => [{:exit_code => 132, :field => 'id', :index => 0, :severity => 'error', :text => 'There is no user with the id 123'},]}.to_json, :status => 422})
+        end
+        it { expect { run }.to exit_with_code(1) }
+        it { run_output.should =~ /Adding 1 administrator to domain.*There is no user with the id 123/ }
+      end
+    end
+  end
+
+  describe 'remove-member' do
+    context "when the resource doesn't support membership changes" do
+      before{ stub_api }
+
+      context "when adjusting a domain" do
+        let(:arguments) { ['remove-member', 'testuser', '-n', 'test'] }
+        before{ challenge{ stub_one_domain('test', nil, mock_user_auth) } }
+        it { expect { run }.to exit_with_code(1) }
+        it { run_output.should =~ /Removing 1 member from domain .*The server does not support adding or removing members/ }
+      end
+    end
+
+    context "with supported membership" do
+      let(:supports_members?){ true }
+      before do
+        stub_api
+        challenge{ stub_one_domain('test', nil, mock_user_auth) }
+      end
+=begin Scenario removed
+      context "when adjusting an app" do
+        let(:arguments) { ['remove-member', 'testuser', '-n', 'test', '-a', 'app'] }
+        before{ challenge{ stub_one_application('test', 'app') } }
+        it { expect { run }.to exit_with_code(1) }
+        it { run_output.should =~ /You can only add or remove members on a domain/ }
+      end
+=end
+      context 'with a valid member' do
+        let(:arguments) { ['remove-member', 'testuser', '-n', 'test'] }
+        before do
+          stub_api_request(:patch, "broker/rest/domains/test/members").
+            with(:body => {:members => [{'login' => 'testuser', 'role' => 'none'}]}).
+            to_return({:body => {:type => 'members', :data => [], :messages => [{:exit_code => 0, :field => 'login', :index => 0, :severity => 'info', :text => 'Removed 1 member'},]}.to_json, :status => 200})
+        end
+        it { expect { run }.to exit_with_code(0) }
+        it { run_output.should =~ /Removing 1 member from domain .*done/ }
+      end
+
+      context 'with --all' do
+        let(:arguments) { ['remove-member', '--all', '-n', 'test'] }
+        before do
+          stub_api_request(:delete, "broker/rest/domains/test/members").
+            to_return({:body => {:type => 'members', :data => [], :messages => [{:exit_code => 0, :field => 'login', :index => 0, :severity => 'info', :text => 'Removed everyone except owner.'},]}.to_json, :status => 200})
+        end
+        it { expect { run }.to exit_with_code(0) }
+        it { run_output.should =~ /Removing all members from domain .*done/ }
+      end
+
+      context 'with a missing user' do
+        let(:arguments) { ['remove-member', 'testuser', '-n', 'test'] }
+        before do
+          stub_api_request(:patch, "broker/rest/domains/test/members").
+            with(:body => {:members => [{'login' => 'testuser', 'role' => 'none'}]}).
+            to_return({:body => {:messages => [{:exit_code => 132, :field => 'login', :index => 0, :severity => 'error', :text => 'There is no user with a login testuser'},]}.to_json, :status => 422})
+        end
+        it { expect { run }.to exit_with_code(1) }
+        it { run_output.should =~ /Removing 1 member from domain.*There is no user with a login testuser/ }
+      end
+
+      context 'with a missing user id and role' do
+        let(:arguments) { ['remove-member', '123', '-n', 'test', '--ids'] }
+        before do
+          stub_api_request(:patch, "broker/rest/domains/test/members").
+            with(:body => {:members => [{'id' => '123', 'role' => 'none'}]}).
+            to_return({:body => {:messages => [{:exit_code => 132, :field => 'id', :index => 0, :severity => 'error', :text => 'There is no user with the id 123'},]}.to_json, :status => 422})
+        end
+        it { expect { run }.to exit_with_code(1) }
+        it { run_output.should =~ /Removing 1 member from domain.*There is no user with the id 123/ }
+      end
+
+      context 'when the user isn''t a member' do
+        let(:arguments) { ['remove-member', 'testuser', '-n', 'test'] }
+        before do
+          stub_api_request(:patch, "broker/rest/domains/test/members").
+            with(:body => {:members => [{'login' => 'testuser', 'role' => 'none'}]}).
+            to_return({:body => {:type => 'members', :data => [], :messages => [{:exit_code => 0, :field => 'login', :index => 0, :severity => 'warning', :text => 'testuser is not a member of this domain.'},]}.to_json, :status => 200})
+        end
+        it { expect { run }.to exit_with_code(0) }
+        it { run_output.should =~ /Removing 1 member from domain.*testuser is not a member of this domain.*done/m }
+      end
+    end
+  end
+end

data/spec/rhc/commands/port_forward_spec.rb

@@ -0,0 +1,217 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc/commands/port_forward'
+
+describe RHC::Commands::PortForward do
+
+  let!(:rest_client){ MockRestClient.new }
+  before{ user_config }
+
+  describe 'run' do
+    let(:arguments) { ['port-forward', '--noprompt', '--config', 'test.conf', '-l', 'test@test.foo', '-p', 'password', '--app', 'mockapp'] }
+
+    before :each do
+      @domain = rest_client.add_domain("mockdomain")
+      @app = @domain.add_application 'mockapp', 'mock-1.0'
+      @uri = URI.parse @app.ssh_url
+      @ssh = double(Net::SSH)
+    end
+
+    context 'when port forwarding for a down appl' do
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh)
+        @ssh.should_receive(:exec!).with("rhc-list-ports").and_yield(nil, :stderr, '127.0.0.1:3306')
+        @gg = MockRestGearGroup.new(rest_client)
+        @app.should_receive(:gear_groups).and_return([@gg])
+        @gg.should_receive(:gears).and_return([{'state' => 'stopped', 'id' => 'fakegearid'}])
+      end
+      it "should error out and suggest restarting the application" do
+        expect { run }.to exit_with_code(1)
+      end
+      it { run_output.should match(/none.*The application is stopped\..*restart/m) }
+    end
+
+    context 'when port forwarding an app without ports to forward' do
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh)
+        @ssh.should_receive(:exec!).with("rhc-list-ports").and_yield(nil, :stderr, '127.0.0.1:3306')
+      end
+      it "should error out as no ports to forward" do
+        expect { run }.to exit_with_code(102)
+        rest_client.domains[0].name.should == 'mockdomain'
+        rest_client.domains[0].applications.size.should == 1
+        rest_client.domains[0].applications[0].name.should == 'mockapp'
+      end
+      it("should report no ports") { run_output.should match("no available ports to forward.") }
+    end
+
+    context 'when port forwarding an app with permission denied ports' do
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh)
+        @ssh.should_receive(:exec!).with("rhc-list-ports").and_yield(nil, :stderr, 'permission denied')
+      end
+      it "should error out as permission denied" do
+        expect { run }.to exit_with_code(129)
+        rest_client.domains[0].name.should == 'mockdomain'
+        rest_client.domains[0].applications.size.should == 1
+        rest_client.domains[0].applications[0].name.should == 'mockapp'
+      end
+      it { run_output.should match("Permission denied") }
+    end
+
+    context 'when port forwarding an app with ports to forward' do
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh).twice
+        @ssh.should_receive(:exec!).with("rhc-list-ports").and_yield(nil, :stderr, 'mysql -> 127.0.0.1:3306')
+        forward = double(Net::SSH::Service::Forward)
+        @ssh.should_receive(:forward).and_return(forward)
+        forward.should_receive(:local).with(3306, '127.0.0.1', 3306)
+        @ssh.should_receive(:loop)
+      end
+      it "should run successfully" do
+        expect { run }.to exit_with_code(0)
+        rest_client.domains[0].name.should == 'mockdomain'
+        rest_client.domains[0].applications.size.should == 1
+        rest_client.domains[0].applications[0].name.should == 'mockapp'
+      end
+      it { run_output.should match(/Forwarding ports.*Press CTRL-C/m) }
+    end
+
+    context 'when host is unreachable' do
+      before(:each) do
+        Net::SSH.should_receive(:start).and_raise(Errno::EHOSTUNREACH)
+      end
+      it "should error out" do
+        expect { run }.to exit_with_code(1)
+        rest_client.domains[0].name.should == 'mockdomain'
+        rest_client.domains[0].applications.size.should == 1
+        rest_client.domains[0].applications[0].name.should == 'mockapp'
+      end
+      it { run_output.should include("Error trying to forward ports.") }
+    end
+
+    context 'when REST client connection times out' do
+      before(:each) do
+        rest_client.should_receive(:find_domain).and_raise(RHC::Rest::ConnectionException)
+      end
+      it("should error out") { expect { run }.to exit_with_code(1) }
+      it{ run_output.should match("Connection.*failed:") }
+    end
+
+    context 'when port forwarding an app with ports to forward' do
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh).twice
+        @ssh.should_receive(:exec!).with("rhc-list-ports").and_yield(nil, :stderr, 'mysql -> 127.0.0.1:3306')
+        forward = double(Net::SSH::Service::Forward)
+        @ssh.should_receive(:forward).and_return(forward)
+        forward.should_receive(:local).with(3306, '127.0.0.1', 3306)
+        @ssh.should_receive(:loop).and_raise(Interrupt.new)
+      end
+      it "should exit when user interrupts" do
+        expect { run }.to exit_with_code(0)
+        rest_client.domains[0].name.should == 'mockdomain'
+        rest_client.domains[0].applications.size.should == 1
+        rest_client.domains[0].applications[0].name.should == 'mockapp'
+      end
+      it { run_output.should include("Ending port forward") }
+    end
+
+    context 'when local port is already bound' do
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh).twice
+        @ssh.should_receive(:exec!).with("rhc-list-ports").and_yield(nil, :stderr, 'mysql -> 127.0.0.1:3306')
+        forward = double(Net::SSH::Service::Forward)
+        @ssh.should_receive(:forward).at_least(2).and_return(forward)
+        forward.should_receive(:local).with(3306, '127.0.0.1', 3306).and_raise(Errno::EACCES)
+        forward.should_receive(:local).with(3307, '127.0.0.1', 3306)
+        @ssh.should_receive(:loop).and_raise(Interrupt.new)
+      end
+      it 'should bind to a higher port' do
+        run_output.should include("3307")
+      end
+    end
+
+    context 'when host refuses connection' do
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh).twice
+        @ssh.should_receive(:exec!).with("rhc-list-ports").and_yield(nil, :stderr, 'mysql -> 127.0.0.1:3306')
+        forward = double(Net::SSH::Service::Forward)
+        @ssh.should_receive(:forward).and_raise(Errno::ECONNREFUSED)
+      end
+      it "should error out" do
+        expect { run }.to exit_with_code(0)
+      end
+      it { run_output.should include("ssh -N") }
+      it { run_output.should include("Error forwarding") }
+    end
+
+    context 'when port forwarding a scaled app with ports to forward' do
+      let(:haproxy_host_1) { '127.0.0.1' }
+      let(:haproxy_host_2) { '127.0.0.2' }
+      let(:mongo_host) { '51125bb94a-test907742.dev.rhcloud.com' }
+      let(:ipv6_host) { '::1' }
+      before(:each) do
+        Net::SSH.should_receive(:start).with(@uri.host, @uri.user).and_yield(@ssh).twice
+        @ssh.should_receive(:exec!).with("rhc-list-ports").
+          and_yield(nil, :stderr, "httpd -> #{haproxy_host_1}:8080\nhttpd -> #{haproxy_host_2}:8080\nmongodb -> #{mongo_host}:35541\nmysqld -> #{ipv6_host}:3306")
+        forward = double(Net::SSH::Service::Forward)
+        @ssh.should_receive(:forward).at_least(3).times.and_return(forward)
+        forward.should_receive(:local).with(8080, haproxy_host_1, 8080)
+        forward.should_receive(:local).with(8080, haproxy_host_2, 8080).and_raise(Errno::EADDRINUSE)
+        forward.should_receive(:local).with(8081, haproxy_host_2, 8080)
+        forward.should_receive(:local).with(35541, mongo_host, 35541)
+        forward.should_receive(:local).with(3306, ipv6_host, 3306)
+        @ssh.should_receive(:loop).and_raise(Interrupt.new)
+      end
+      it "should exit when user interrupts" do
+        expect { run }.to exit_with_code(0)
+        rest_client.domains[0].name.should == 'mockdomain'
+        rest_client.domains[0].applications.size.should == 1
+        rest_client.domains[0].applications[0].name.should == 'mockapp'
+      end
+      it { run_output.should include("Ending port forward") }
+    end
+
+    context 'when port forwarding a single gear on a scaled app' do
+      let(:gear_host) { 'fakesshurl.com' }
+      let(:gear_user) { 'fakegearid0' }
+      let(:arguments) { ['port-forward', '--noprompt', '--config', 'test.conf', '-l', 'test@test.foo', '-p', 'password', '--app', 'mockapp', '--gear', @gear_id] }
+
+      it 'should run successfully' do
+        @gear_id = 'fakegearid0'
+        Net::SSH.should_receive(:start).with(gear_host, gear_user).and_yield(@ssh).twice
+
+        @ssh.should_receive(:exec!).with("rhc-list-ports --exclude-remote").
+          and_yield(nil, :stderr, "mongodb -> #{gear_host}:35541")
+        forward = double(Net::SSH::Service::Forward)
+        @ssh.should_receive(:forward).and_return(forward)
+        forward.should_receive(:local).with(35541, gear_host, 35541)
+        @ssh.should_receive(:loop).and_raise(Interrupt.new)
+
+        expect { run }.to exit_with_code(0)
+      end
+
+      it 'should fail if the specified gear is missing' do
+        @gear_id = 'notarealgearxxxxx'
+
+        expect { run }.to exit_with_code(1)
+        run_output.should include('Gear notarealgearxxxxx not found')
+      end
+
+      it 'should fail if the specified gear has no ssh info' do
+        @gear_id = 'fakegearid0'
+
+        # Given - gears in gear group should not have ssh info
+        gg = MockRestGearGroup.new(rest_client)
+        @app.stub(:gear_groups).and_return([gg])
+        gg.stub(:gears).and_return([{'state' => 'started', 'id' => 'fakegearid0'}])
+
+        expect { run }.to exit_with_code(1)
+        run_output.should match('The server does not support operations on individual gears.')
+      end
+
+    end
+
+  end
+end
+

data/spec/rhc/commands/scp_spec.rb

@@ -0,0 +1,77 @@
+require 'spec_helper'
+require 'rest_spec_helper'
+require 'rhc/commands/scp'
+
+describe RHC::Commands::Scp 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 'scp default' do
+    context 'scp' do
+      let(:arguments) { ['scp'] }
+      it { run_output.should match('Usage:') }
+    end
+  end
+
+  describe 'scp with invalid option' do
+    let (:arguments) {['app', 'scp', 'app1', 'invalid_command', 'file.txt', 'app-root/data']}
+
+    context 'when run' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+      end
+      it { run_output.should match("'invalid_command' is not a valid argument for this command.  Please use upload or download.") }
+    end
+  end
+
+  describe 'local file or path does not exist' do
+    let (:arguments) {['app', 'scp', 'app1', 'upload', 'file.txt', 'app-root/data']}
+
+    context 'when run' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        File.should_receive(:exist?).with("file.txt").once.and_return(false)
+      end
+      it { run_output.should match("Local file, file_path, or directory could not be found.") }
+    end
+  end
+
+  describe 'scp connections' do
+    let (:arguments) {['app', 'scp', 'app1', 'upload', 'file.txt', 'app-root/data']}
+
+    context 'connection refused' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        File.should_receive(:exist?).with("file.txt").once.and_return(true)
+        Net::SCP.should_receive("upload!".to_sym).with("127.0.0.1", "fakeuuidfortestsapp1","file.txt","app-root/data").and_raise(Errno::ECONNREFUSED)
+      end
+      it { run_output.should match("The server fakeuuidfortestsapp1 refused a connection with user 127.0.0.1.  The application may be unavailable.") }
+    end
+
+    context 'socket error' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        File.should_receive(:exist?).with("file.txt").once.and_return(true)
+        Net::SCP.should_receive("upload!".to_sym).with("127.0.0.1", "fakeuuidfortestsapp1","file.txt","app-root/data").and_raise(SocketError)
+      end
+      it { run_output.should match("The connection to 127.0.0.1 failed: SocketError") }
+    end
+
+    context 'remote file not found' do
+      before(:each) do
+        @domain = rest_client.add_domain("mockdomain")
+        @domain.add_application("app1", "mock_type")
+        File.should_receive(:exist?).with("file.txt").once.and_return(true)
+        Net::SCP.should_receive("upload!".to_sym).with("127.0.0.1", "fakeuuidfortestsapp1","file.txt","app-root/data").and_raise(Net::SCP::Error)
+      end
+      it { run_output.should match("Remote file, file_path, or directory could not be found.") }
+    end
+  end
+
+end