artofmission-heroku 1.6.3

data/spec/commands/app_spec.rb

@@ -0,0 +1,175 @@
+require File.dirname(__FILE__) + '/../base'
+
+module Heroku::Command
+	describe App do
+		before(:each) do
+			@cli = prepare_command(App)
+		end
+
+		it "shows app info, converting bytes to kbs/mbs" do
+			@cli.stub!(:args).and_return(['myapp'])
+			@cli.heroku.should_receive(:info).with('myapp').and_return({ :name => 'myapp', :collaborators => [], :addons => [], :repo_size => 2*1024, :database_size => 5*1024*1024 })
+			@cli.should_receive(:display).with('=== myapp')
+			@cli.should_receive(:display).with('Web URL:        http://myapp.heroku.com/')
+			@cli.should_receive(:display).with('Repo size:      2k')
+			@cli.should_receive(:display).with('Data size:      5M')
+			@cli.info
+		end
+
+		it "shows app info using the --app syntax" do
+			@cli.stub!(:args).and_return(['--app', 'myapp'])
+			@cli.heroku.should_receive(:info).with('myapp').and_return({ :collaborators => [], :addons => []})
+			@cli.info
+		end
+
+		it "shows app info reading app from current git dir" do
+			@cli.stub!(:args).and_return([])
+			@cli.stub!(:extract_app_in_dir).and_return('myapp')
+			@cli.heroku.should_receive(:info).with('myapp').and_return({ :collaborators => [], :addons => []})
+			@cli.info
+		end
+
+		it "creates without a name" do
+			@cli.heroku.should_receive(:create).with(nil, {}).and_return("untitled-123")
+			@cli.create
+		end
+
+		it "creates with a name" do
+			@cli.stub!(:args).and_return([ 'myapp' ])
+			@cli.heroku.should_receive(:create).with('myapp', {}).and_return("myapp")
+			@cli.create
+		end
+
+		it "renames an app" do
+			@cli.stub!(:args).and_return([ 'myapp2' ])
+			@cli.heroku.should_receive(:update).with('myapp', { :name => 'myapp2' })
+			@cli.rename
+		end
+
+		it "runs a rake command on the app" do
+			@cli.stub!(:args).and_return(([ 'db:migrate' ]))
+			@cli.heroku.should_receive(:start).
+				with('myapp', 'rake db:migrate', attached=true).
+				and_return(['foo', 'bar', 'baz'])
+			@cli.rake
+		end
+
+		it "runs a single console command on the app" do
+			@cli.stub!(:args).and_return([ '2+2' ])
+			@cli.heroku.should_receive(:console).with('myapp', '2+2')
+			@cli.console
+		end
+
+		it "offers a console, opening and closing the session with the client" do
+			@console = mock('heroku console')
+			@cli.stub!(:console_history_read)
+			@cli.stub!(:console_history_add)
+			@cli.heroku.should_receive(:console).with('myapp').and_yield(@console)
+			Readline.should_receive(:readline).and_return('exit')
+			@cli.console
+		end
+
+		it "asks to restart servers" do
+			@cli.heroku.should_receive(:restart).with('myapp')
+			@cli.restart
+		end
+
+		it "scales dynos" do
+			@cli.stub!(:args).and_return(['+4'])
+			@cli.heroku.should_receive(:set_dynos).with('myapp', '+4').and_return(7)
+			@cli.dynos
+		end
+
+		it "destroys the app specified with --app if user confirms" do
+			@cli.stub!(:ask).and_return('y')
+			@cli.stub!(:args).and_return(['--app', 'myapp'])
+			@cli.heroku.stub!(:info).and_return({})
+			@cli.heroku.should_receive(:destroy).with('myapp')
+			@cli.destroy
+		end
+
+		it "doesn't destroy the app if the user doesn't confirms" do
+			@cli.stub!(:ask).and_return('no')
+			@cli.stub!(:args).and_return(['--app', 'myapp'])
+			@cli.heroku.stub!(:info).and_return({})
+			@cli.heroku.should_not_receive(:destroy)
+			@cli.destroy
+		end
+
+		it "doesn't destroy the app in the current dir" do
+			@cli.stub!(:extract_app).and_return('myapp')
+			@cli.heroku.should_not_receive(:destroy)
+			@cli.destroy
+		end
+
+		context "Git Integration" do
+			include SandboxHelper
+			before(:all) do
+				# setup a git dir to serve as a remote
+				@git = "/tmp/git_spec_#{Process.pid}"
+				FileUtils.mkdir_p(@git)
+				FileUtils.cd(@git) { |d| `git --bare init` }
+			end
+
+			after(:all) do
+				FileUtils.rm_rf(@git)
+			end
+
+			# setup sandbox in /tmp
+			before(:each) do
+				@sandbox = "/tmp/app_spec_#{Process.pid}"
+				FileUtils.mkdir_p(@sandbox)
+				bash "git init"
+				Dir.stub!(:pwd).and_return(@sandbox)
+			end
+
+			after(:each) do
+				FileUtils.rm_rf(@sandbox)
+			end
+
+			it "creates adding heroku to git remote" do
+				@cli.heroku.should_receive(:create).and_return('myapp')
+				@cli.create
+				bash("git remote").strip.should == 'heroku'
+			end
+
+			it "creates adding a custom git remote" do
+				@cli.stub!(:args).and_return([ 'myapp', '--remote', 'myremote' ])
+				@cli.heroku.should_receive(:create).and_return('myapp')
+				@cli.create
+				bash("git remote").strip.should == 'myremote'
+			end
+
+			it "doesn't add a git remote if it already exists" do
+				@cli.heroku.should_receive(:create).and_return('myapp')
+				bash "git remote add heroku #{@git}"
+				@cli.create
+			end
+
+			it "renames updating the corresponding heroku git remote" do
+				bash "git remote add github     git@github.com:test/test.git"
+				bash "git remote add production git@heroku.com:myapp.git"
+				bash "git remote add staging    git@heroku.com:myapp-staging.git"
+
+				@cli.heroku.stub!(:update)
+				@cli.stub!(:args).and_return([ 'myapp2' ])
+				@cli.rename
+				remotes = bash("git remote -v")
+				remotes.should include('git@github.com:test/test.git')
+				remotes.should include('git@heroku.com:myapp-staging.git')
+				remotes.should include('git@heroku.com:myapp2.git')
+				remotes.should_not include('git@heroku.com:myapp.git')
+			end
+
+			it "destroys removing any remotes pointing to the app" do
+				bash("git remote add heroku git@heroku.com:myapp.git")
+				@cli.stub!(:args).and_return(['--app', 'myapp'])
+				@cli.heroku.stub!(:info).and_return({})
+				@cli.stub!(:ask).and_return('y')
+				@cli.heroku.should_receive(:destroy)
+				@cli.destroy
+				bash("git remote").strip.should == ''
+			end
+		end
+	end
+end

data/spec/commands/auth_spec.rb

@@ -0,0 +1,104 @@
+require File.dirname(__FILE__) + '/../base'
+
+module Heroku::Command
+	describe Auth do
+		before do
+			@cli = prepare_command(Auth)
+			@sandbox = "#{Dir.tmpdir}/cli_spec_#{Process.pid}"
+			File.open(@sandbox, "w") { |f| f.write "user\npass\n" }
+			@cli.stub!(:credentials_file).and_return(@sandbox)
+			@cli.stub!(:running_on_a_mac?).and_return(false)
+		end
+
+		after do
+			FileUtils.rm_rf(@sandbox)
+		end
+
+		it "reads credentials from the credentials file" do
+			@cli.read_credentials.should == %w(user pass)
+		end
+
+		it "takes the user from the first line and the password from the second line" do
+			@cli.user.should == 'user'
+			@cli.password.should == 'pass'
+		end
+
+		it "asks for credentials when the file doesn't exist" do
+			FileUtils.rm_rf(@sandbox)
+			@cli.should_receive(:ask_for_credentials).and_return([ 'u', 'p'])
+			@cli.should_receive(:save_credentials)
+			@cli.get_credentials.should == [ 'u', 'p' ]
+		end
+
+		it "writes the credentials to a file" do
+			@cli.stub!(:credentials).and_return(['one', 'two'])
+			@cli.should_receive(:set_credentials_permissions)
+			@cli.write_credentials
+			File.read(@sandbox).should == "one\ntwo\n"
+		end
+
+		it "sets ~/.heroku/credentials to be readable only by the user" do
+			unless RUBY_PLATFORM =~ /mswin32|mingw32/
+				sandbox = "#{Dir.tmpdir}/cli_spec_#{Process.pid}"
+				FileUtils.rm_rf(sandbox)
+				FileUtils.mkdir_p(sandbox)
+				fname = "#{sandbox}/file"
+				system "touch #{fname}"
+				@cli.stub!(:credentials_file).and_return(fname)
+				@cli.set_credentials_permissions
+				File.stat(sandbox).mode.should == 040700
+				File.stat(fname).mode.should == 0100600
+			end
+		end
+
+		it "writes credentials and uploads authkey when credentials are saved" do
+			@cli.stub!(:credentials)
+			@cli.should_receive(:write_credentials)
+			Heroku::Command.should_receive(:run_internal).with('keys:add', [])
+			@cli.save_credentials
+		end
+
+		it "preserves the args when running keys:add" do
+			@cli.stub!(:credentials)
+			@cli.stub!(:args).and_return(['mykey.pub'])
+			Heroku::Command.should_receive(:run_internal).with('keys:add', ['mykey.pub'])
+			@cli.save_credentials
+		end
+
+		it "save_credentials deletes the credentials when the upload authkey is unauthorized" do
+			@cli.stub!(:write_credentials)
+			@cli.stub!(:retry_login?).and_return(false)
+			Heroku::Command.should_receive(:run_internal).with('keys:add', []).and_raise(RestClient::Unauthorized)
+			@cli.should_receive(:delete_credentials)
+			lambda { @cli.save_credentials }.should raise_error(RestClient::Unauthorized)
+		end
+
+		it "save_credentials deletes the credentials when there's no authkey" do
+			@cli.stub!(:write_credentials)
+			Heroku::Command.should_receive(:run_internal).with('keys:add', []).and_raise(RuntimeError)
+			@cli.should_receive(:delete_credentials)
+			lambda { @cli.save_credentials }.should raise_error
+		end
+
+		it "save_credentials deletes the credentials when the authkey is weak" do
+			@cli.stub!(:write_credentials)
+			Heroku::Command.should_receive(:run_internal).with('keys:add', []).and_raise(RestClient::RequestFailed)
+			@cli.should_receive(:delete_credentials)
+			lambda { @cli.save_credentials }.should raise_error
+		end
+
+		it "asks for login again when not authorized, for three times" do
+			@cli.stub!(:read_credentials)
+			@cli.stub!(:write_credentials)
+			@cli.stub!(:delete_credentials)
+			Heroku::Command.stub!(:run_internal).with('keys:add', []).and_raise(RestClient::Unauthorized)
+			@cli.should_receive(:ask_for_credentials).exactly(4).times
+			lambda { @cli.save_credentials }.should raise_error(RestClient::Unauthorized)
+		end
+
+		it "deletes the credentials file" do
+			FileUtils.should_receive(:rm_f).with(@cli.credentials_file)
+			@cli.delete_credentials
+		end
+	end
+end

data/spec/commands/base_spec.rb

@@ -0,0 +1,114 @@
+require File.dirname(__FILE__) + '/../base'
+
+module Heroku::Command
+	describe Base do
+		before do
+			@args = [1, 2]
+			@base = Base.new(@args)
+			@base.stub!(:display)
+			@client = mock('heroku client', :host => 'heroku.com')
+		end
+
+		it "initializes the heroku client with the Auth command" do
+			Heroku::Command.should_receive(:run_internal).with('auth:client', @args)
+			@base.heroku
+		end
+
+		context "detecting the app" do
+			before do
+				@base.stub!(:heroku).and_return(@client)
+			end
+
+			it "attempts to find the app via the --app argument" do
+				@base.stub!(:args).and_return(['--app', 'myapp'])
+				@base.extract_app.should == 'myapp'
+			end
+
+			it "read remotes from git config" do
+				File.stub!(:exists?).with('/home/dev/myapp/.git/config').and_return(true)
+				File.stub!(:read).with('/home/dev/myapp/.git/config').and_return("
+[core]
+	repositoryformatversion = 0
+	filemode = true
+	bare = false
+	logallrefupdates = true
+[remote \"staging\"]
+	url = git@heroku.com:myapp-staging.git
+	fetch = +refs/heads/*:refs/remotes/staging/*
+[remote \"production\"]
+	url = git@heroku.com:myapp.git
+	fetch = +refs/heads/*:refs/remotes/production/*
+[remote \"github\"]
+	url = git@github.com:myapp.git
+	fetch = +refs/heads/*:refs/remotes/github/*
+				")
+				@base.git_remotes('/home/dev/myapp').should == { 'staging' => 'myapp-staging', 'production' => 'myapp' }
+			end
+
+			it "attempts to find the git config file in parent folders" do
+				File.should_receive(:exists?).with('/home/dev/myapp/app/models/.git/config').and_return(false)
+				File.should_receive(:exists?).with('/home/dev/myapp/app/.git/config').and_return(false)
+				File.should_receive(:exists?).with('/home/dev/myapp/.git/config').and_return(true)
+				File.should_receive(:read).with('/home/dev/myapp/.git/config').and_return('')
+				@base.git_remotes('/home/dev/myapp/app/models')
+			end
+
+			it "gets the app from remotes when there's only one app" do
+				@base.stub!(:git_remotes).and_return({ 'heroku' => 'myapp' })
+				@base.extract_app.should == 'myapp'
+			end
+
+			it "uses the remote named after the current folder name when there are multiple" do
+				@base.stub!(:git_remotes).and_return({ 'staging' => 'myapp-staging', 'production' => 'myapp' })
+				Dir.stub!(:pwd).and_return('/home/dev/myapp')
+				@base.extract_app.should == 'myapp'
+			end
+
+			it "accepts a --remote argument to choose the app from the remote name" do
+				@base.stub!(:git_remotes).and_return({ 'staging' => 'myapp-staging', 'production' => 'myapp' })
+				@base.stub!(:args).and_return(['--remote', 'staging'])
+				@base.extract_app.should == 'myapp-staging'
+			end
+
+			it "raises when cannot determine which app is it" do
+				@base.stub!(:git_remotes).and_return({ 'staging' => 'myapp-staging', 'production' => 'myapp' })
+				lambda { @base.extract_app }.should raise_error(Heroku::Command::CommandFailed)
+			end
+		end
+
+		context "option parsing" do
+			it "extracts options from args" do
+				@base.stub!(:args).and_return(%w( a b --something value c d ))
+				@base.extract_option('--something').should == 'value'
+			end
+
+			it "accepts options without value" do
+				@base.stub!(:args).and_return(%w( a b --something))
+				@base.extract_option('--something').should be_true
+			end
+
+			it "doesn't consider parameters as a value" do
+				@base.stub!(:args).and_return(%w( a b --something --something-else c d))
+				@base.extract_option('--something').should be_true
+			end
+
+			it "accepts a default value" do
+				@base.stub!(:args).and_return(%w( a b --something))
+				@base.extract_option('--something', 'default').should == 'default'
+			end
+
+			it "is not affected by multiple arguments with the same value" do
+				@base.stub!(:args).and_return(%w( --arg1 val --arg2 val ))
+				@base.extract_option('--arg1').should == 'val'
+				@base.args.should == ['--arg2', 'val']
+			end
+		end
+
+		describe "formatting" do
+			it "formats app urls (http and git), displayed as output on create and other commands" do
+				@base.stub!(:heroku).and_return(mock('heroku client', :host => 'example.com'))
+				@base.app_urls('test').should == "http://test.example.com/ | git@example.com:test.git"
+			end
+		end
+	end
+end

data/spec/commands/bundles_spec.rb

@@ -0,0 +1,48 @@
+require File.dirname(__FILE__) + '/../base'
+
+module Heroku::Command
+	describe Bundles do
+		before do
+			@bundles = prepare_command(Bundles)
+		end
+
+		it "lists bundles" do
+			@bundles.heroku.should_receive(:bundles).with('myapp').and_return([])
+			@bundles.list
+		end
+
+		it "captures a bundle with the specified name" do
+			@bundles.stub!(:args).and_return(['mybundle'])
+			@bundles.heroku.should_receive(:bundle_capture).with('myapp', 'mybundle')
+			@bundles.capture
+		end
+
+		it "captures a bundle with no name" do
+			@bundles.heroku.should_receive(:bundle_capture).with('myapp', nil)
+			@bundles.capture
+		end
+
+		it "destroys a bundle" do
+			@bundles.stub!(:args).and_return(['mybundle'])
+			@bundles.heroku.should_receive(:bundle_destroy).with('myapp', 'mybundle')
+			@bundles.destroy
+		end
+
+		it "downloads a bundle to appname.tar.gz" do
+			@bundles.stub!(:args).and_return(['mybundle'])
+			@bundles.heroku.should_receive(:bundle_url).with('myapp', 'mybundle').and_return('https://s3.amazonaws.com/herokubundles/123.tar.gz')
+			RestClient.should_receive(:get).with('https://s3.amazonaws.com/herokubundles/123.tar.gz').and_return('bundle contents')
+			@file = mock('bundle file')
+			File.should_receive(:open).with('myapp.tar.gz', 'wb').and_yield(@file)
+			@file.should_receive(:write).with('bundle contents')
+			File.stub!(:stat).and_return(mock('app stat', :size => 1234))
+			@bundles.download
+		end
+
+		it "animates a bundle" do
+			@bundles.stub!(:args).and_return(['mybundle'])
+			@bundles.heroku.should_receive(:create).with(nil, :origin_bundle_app => 'myapp', :origin_bundle => 'mybundle')
+			@bundles.animate
+		end
+	end
+end

data/spec/commands/config_spec.rb

@@ -0,0 +1,45 @@
+require File.dirname(__FILE__) + '/../base'
+
+module Heroku::Command
+	describe Config do
+		before do
+			@config = prepare_command(Config)
+		end
+
+		it "shows all configs" do
+			@config.heroku.should_receive(:config_vars).and_return({ 'A' => 'one', 'B' => 'two' })
+			@config.should_receive(:display).with('A => one')
+			@config.should_receive(:display).with('B => two')
+			@config.index
+		end
+
+		it "trims long values" do
+			@config.heroku.should_receive(:config_vars).and_return({ 'LONG' => 'A' * 60 })
+			@config.should_receive(:display).with('LONG => AAAAAAAAAAAAAAAA...AAAAAAAAAAAAAAAA')
+			@config.index
+		end
+
+		it "sets config vars" do
+			@config.stub!(:args).and_return(['a=1', 'b=2'])
+			@config.heroku.should_receive(:add_config_vars).with('myapp', {'a'=>'1','b'=>'2'})
+			@config.add
+		end
+
+		it "allows config vars with = in the value" do
+			@config.stub!(:args).and_return(['a=b=c'])
+			@config.heroku.should_receive(:add_config_vars).with('myapp', {'a'=>'b=c'})
+			@config.add
+		end
+
+		it "unsets config vars" do
+			@config.stub!(:args).and_return(['a'])
+			@config.heroku.should_receive(:remove_config_var).with('myapp', 'a')
+			@config.remove
+		end
+
+		it "resets config" do
+			@config.heroku.should_receive(:clear_config_vars).with('myapp')
+			@config.clear
+		end
+	end
+end