pogo 2.31.2

data/spec/heroku/command/logs_spec.rb

@@ -0,0 +1,60 @@
+require "spec_helper"
+require "heroku/command/logs"
+
+describe Heroku::Command::Logs do
+  describe "logs" do
+    it "runs with no options" do
+      stub_core.read_logs("myapp", [])
+      execute "logs"
+    end
+
+    it "runs with options" do
+      stub_core.read_logs("myapp", [
+        "tail=1",
+        "num=2",
+        "ps=ps.3",
+        "source=source.4"
+      ])
+      execute "logs --tail --num 2 --ps ps.3 --source source.4"
+    end
+
+    describe "with log output" do
+      before(:each) do
+        stub_core.read_logs("myapp", []).yields("2011-01-01T00:00:00+00:00 app[web.1]: test")
+      end
+
+      it "prettifies tty output" do
+        old_stdout_isatty = $stdout.isatty
+        stub($stdout).isatty.returns(true)
+        stderr, stdout = execute("logs")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+\e[36m2011-01-01T00:00:00+00:00 app[web.1]:\e[0m test
+STDOUT
+        stub($stdout).isatty.returns(old_stdout_isatty)
+      end
+
+      it "does not use ansi if stdout is not a tty" do
+        old_stdout_isatty = $stdout.isatty
+        stub($stdout).isatty.returns(false)
+        stderr, stdout = execute("logs")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+2011-01-01T00:00:00+00:00 app[web.1]: test
+STDOUT
+        stub($stdout).isatty.returns(old_stdout_isatty)
+      end
+
+      it "does not use ansi if TERM is not set" do
+        term = ENV.delete("TERM")
+        stderr, stdout = execute("logs")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+2011-01-01T00:00:00+00:00 app[web.1]: test
+STDOUT
+        ENV["TERM"] = term
+      end
+    end
+  end
+
+end

data/spec/heroku/command/maintenance_spec.rb

@@ -0,0 +1,51 @@
+require "spec_helper"
+require "heroku/command/maintenance"
+
+module Heroku::Command
+  describe Maintenance do
+
+    before(:each) do
+      stub_core
+      api.post_app("name" => "myapp", "stack" => "cedar")
+    end
+
+    after(:each) do
+      api.delete_app("myapp")
+    end
+
+    it "displays off for maintenance mode of an app" do
+      stderr, stdout = execute("maintenance")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+off
+STDOUT
+    end
+
+    it "displays on for maintenance mode of an app" do
+      api.post_app_maintenance('myapp', '1')
+
+      stderr, stdout = execute("maintenance")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+on
+STDOUT
+    end
+
+    it "turns on maintenance mode for the app" do
+      stderr, stdout = execute("maintenance:on")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+Enabling maintenance mode for myapp... done
+STDOUT
+    end
+
+    it "turns off maintenance mode for the app" do
+      stderr, stdout = execute("maintenance:off")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+Disabling maintenance mode for myapp... done
+STDOUT
+    end
+
+  end
+end

data/spec/heroku/command/pg_spec.rb

@@ -0,0 +1,223 @@
+require "spec_helper"
+require "heroku/command/pg"
+
+module Heroku::Command
+  describe Pg do
+    before do
+      stub_core
+
+      api.post_app "name" => "myapp"
+      api.put_config_vars "myapp", {
+        "DATABASE_URL" => "postgres://database_url",
+        "SHARED_DATABASE_URL" => "postgres://other_database_url",
+        "HEROKU_POSTGRESQL_RONIN_URL" => "postgres://ronin_database_url"
+      }
+    end
+
+    after do
+      api.delete_app "myapp"
+    end
+
+    it "resets the app's database if user confirms" do
+      stub_pg.reset
+
+      stderr, stdout = execute("pg:reset RONIN --confirm myapp")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+Resetting HEROKU_POSTGRESQL_RONIN... done
+STDOUT
+    end
+
+    it "resets shared databases" do
+      Heroku::Client.any_instance.should_receive(:database_reset).with('myapp')
+
+      stderr, stdout = execute("pg:reset SHARED_DATABASE --confirm myapp")
+      stderr.should == ''
+      stdout.should == <<-STDOUT
+Resetting SHARED_DATABASE... done
+STDOUT
+    end
+
+    it "doesn't reset the app's database if the user doesn't confirm" do
+      stub_pg.reset
+
+      stderr, stdout = execute("pg:reset RONIN")
+      stderr.should == <<-STDERR
+ !    Confirmation did not match myapp. Aborted.
+STDERR
+      stdout.should == "
+ !    WARNING: Destructive Action
+ !    This command will affect the app: myapp
+ !    To proceed, type \"myapp\" or re-run this command with --confirm myapp
+
+> "
+    end
+
+    context "index" do
+      it "requests the info from the server" do
+        stub_pg.get_database.returns(:info => [
+          {"name"=>"Plan", "values"=>["Ronin"]},
+          {"name"=>"Status", "values"=>["available"]},
+          {"name"=>"Data Size", "values"=>["1 MB"]},
+          {"name"=>"Tables", "values"=>[1]},
+          {"name"=>"PG Version", "values"=>["9.1.4"]},
+          {"name"=>"Fork/Follow", "values"=>["Available"]},
+          {"name"=>"Created", "values"=>["2011-12-13 00:00 UTC"]},
+          {"name"=>"Conn Info", "values"=>["[Deprecated] Please use `heroku pg:credentials HEROKU_POSTGRESQL_RONIN` to view connection info"]},
+          {"name"=>"Maintenance", "values"=>["not required"]}
+        ])
+
+        stderr, stdout = execute("pg")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+=== HEROKU_POSTGRESQL_RONIN
+Plan:        Ronin
+Status:      available
+Data Size:   1 MB
+Tables:      1
+PG Version:  9.1.4
+Fork/Follow: Available
+Created:     2011-12-13 00:00 UTC
+Conn Info:   [Deprecated] Please use `heroku pg:credentials HEROKU_POSTGRESQL_RONIN` to view connection info
+Maintenance: not required
+
+=== SHARED_DATABASE
+Data Size: (empty)
+
+STDOUT
+      end
+    end
+
+    context "info" do
+      it "requests the info from the server" do
+        stub_pg.get_database.returns(:info => [
+          {"name"=>"Plan", "values"=>["Ronin"]},
+          {"name"=>"Status", "values"=>["available"]},
+          {"name"=>"Data Size", "values"=>["1 MB"]},
+          {"name"=>"Tables", "values"=>[1]},
+          {"name"=>"PG Version", "values"=>["9.1.4"]},
+          {"name"=>"Fork/Follow", "values"=>["Available"]},
+          {"name"=>"Forked From", "values"=>["postgres://username:password@postgreshost.com:5432/database_name"], "resolve_db_name" => "true"},
+          {"name"=>"Created", "values"=>["2011-12-13 00:00 UTC"]},
+          {"name"=>"Conn Info", "values"=>["[Deprecated] Please use `heroku pg:credentials HEROKU_POSTGRESQL_RONIN` to view connection info"]},
+          {"name"=>"Maintenance", "values"=>["not required"]}
+        ])
+
+        stderr, stdout = execute("pg:info RONIN")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+=== HEROKU_POSTGRESQL_RONIN
+Plan:        Ronin
+Status:      available
+Data Size:   1 MB
+Tables:      1
+PG Version:  9.1.4
+Fork/Follow: Available
+Forked From: Database on postgreshost.com:5432/database_name
+Created:     2011-12-13 00:00 UTC
+Conn Info:   [Deprecated] Please use `heroku pg:credentials HEROKU_POSTGRESQL_RONIN` to view connection info
+Maintenance: not required
+
+STDOUT
+      end
+    end
+
+    context "promotion" do
+      it "promotes the specified database" do
+        stderr, stdout = execute("pg:promote RONIN --confirm myapp")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Promoting HEROKU_POSTGRESQL_RONIN to DATABASE_URL... done
+STDOUT
+        api.get_config_vars("myapp").body["DATABASE_URL"].should == "postgres://ronin_database_url"
+      end
+
+      it "promotes the specified database url case-sensitively" do
+        stderr, stdout = execute("pg:promote postgres://john:S3nsit1ve@my.example.com/db_name --confirm=myapp")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Promoting Custom URL to DATABASE_URL... done
+STDOUT
+      end
+
+      it "fails if no database is specified" do
+        stderr, stdout = execute("pg:promote")
+        stderr.should == <<-STDERR
+ !    Usage: heroku pg:promote DATABASE
+ !    Must specify DATABASE to promote.
+STDERR
+        stdout.should == ""
+      end
+    end
+
+    context "credential resets" do
+      before do
+        api.put_config_vars "myapp", {
+          "DATABASE_URL" => "postgres:///to_reset_credentials",
+          "HEROKU_POSTGRESQL_RESETME_URL" => "postgres:///to_reset_credentials"
+        }
+      end
+
+      it "resets credentials and promotes to DATABASE_URL if it's the main DB" do
+        stub_pg.rotate_credentials
+        stderr, stdout = execute("pg:credentials resetme --reset")
+        stderr.should be_empty
+        stdout.should == <<-STDOUT
+Resetting credentials for HEROKU_POSTGRESQL_RESETME (DATABASE_URL)... done
+Promoting HEROKU_POSTGRESQL_RESETME (DATABASE_URL)... done
+STDOUT
+      end
+
+      it "does not update DATABASE_URL if it's not the main db" do
+        stub_pg.rotate_credentials
+        api.put_config_vars "myapp", {
+          "DATABASE_URL" => "postgres://to_reset_credentials",
+          "HEROKU_POSTGRESQL_RESETME_URL" => "postgres://something_else"
+        }
+        stderr, stdout = execute("pg:credentials resetme --reset")
+        stderr.should be_empty
+        stdout.should_not include("Promoting")
+      end
+
+    end
+
+    context "unfollow" do
+      before do
+        api.put_config_vars "myapp", {
+          "DATABASE_URL" => "postgres://database_url",
+          "SHARED_DATABASE_URL" => "postgres://other_database_url",
+          "HEROKU_POSTGRESQL_RONIN_URL" => "postgres://ronin_database_url",
+          "HEROKU_POSTGRESQL_OTHER_URL" => "postgres://other_database_url"
+        }
+      end
+
+      it "sends request to unfollow" do
+        hpg_client = double('Heroku::Client::HerokuPostgresql')
+        Heroku::Client::HerokuPostgresql.should_receive(:new).twice.with('postgres://other_database_url').and_return(hpg_client)
+        hpg_client.should_receive(:unfollow)
+        hpg_client.should_receive(:get_database).and_return(
+          :following => 'postgresql://user:pass@roninhost/database',
+          :info => [
+            {"name"=>"Plan", "values"=>["Ronin"]},
+            {"name"=>"Status", "values"=>["available"]},
+            {"name"=>"Data Size", "values"=>["1 MB"]},
+            {"name"=>"Tables", "values"=>[1]},
+            {"name"=>"PG Version", "values"=>["9.1.4"]},
+            {"name"=>"Fork/Follow", "values"=>["Available"]},
+            {"name"=>"Created", "values"=>["2011-12-13 00:00 UTC"]},
+            {"name"=>"Conn Info", "values"=>["[Deprecated] Please use `heroku pg:credentials HEROKU_POSTGRESQL_RONIN` to view connection info"]},
+            {"name"=>"Maintenance", "values"=>["not required"]}
+          ]
+        )
+        stderr, stdout = execute("pg:unfollow HEROKU_POSTGRESQL_OTHER --confirm myapp")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+ !    HEROKU_POSTGRESQL_OTHER will become writable and no longer
+ !    follow Database on roninhost:5432/database. This cannot be undone.
+Unfollowing HEROKU_POSTGRESQL_OTHER... done
+STDOUT
+      end
+    end
+
+  end
+end

data/spec/heroku/command/pgbackups_spec.rb

@@ -0,0 +1,280 @@
+require "spec_helper"
+require "heroku/command/pgbackups"
+
+module Heroku::Command
+  describe Pgbackups do
+    before do
+      @pgbackups = prepare_command(Pgbackups)
+      @pgbackups.heroku.stub!(:info).and_return({})
+
+      api.post_app("name" => "myapp")
+      api.put_config_vars(
+        "myapp",
+        {
+          "DATABASE_URL"            => "postgres://database",
+          "HEROKU_POSTGRESQL_IVORY" => "postgres://database",
+          "PGBACKUPS_URL"           => "https://ip:password@pgbackups.heroku.com/client"
+        }
+      )
+    end
+
+    after do
+      api.delete_app("myapp")
+    end
+
+    it "requests a pgbackups transfer list for the index command" do
+      stub_core
+      stub_pgbackups.get_transfers.returns([{
+        "created_at"  => "2012-01-01 12:00:00 +0000",
+        "from_name"   => "DATABASE",
+        "size"        => "1024",
+        "to_name"     => "BACKUP",
+        "to_url"      => "s3://bucket/userid/b001.dump"
+      }])
+
+      stderr, stdout = execute("pgbackups")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+ID    Backup Time                Size  Database
+----  -------------------------  ----  --------
+b001  2012-01-01 12:00:00 +0000  1024  DATABASE
+STDOUT
+    end
+
+    describe "single backup" do
+      it "gets the url for the latest backup if nothing is specified" do
+        stub_core
+        stub_pgbackups.get_latest_backup.returns({"public_url" => "http://latest/backup.dump"})
+
+        old_stdout_isatty = STDOUT.isatty
+        $stdout.stub!(:isatty).and_return(true)
+        stderr, stdout = execute("pgbackups:url")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+http://latest/backup.dump
+STDOUT
+        $stdout.stub!(:isatty).and_return(old_stdout_isatty)
+      end
+
+      it "gets the url for the named backup if a name is specified" do
+        stub_pgbackups.get_backup.with("b001").returns({"public_url" => "http://latest/backup.dump" })
+
+        old_stdout_isatty = STDOUT.isatty
+        $stdout.stub!(:isatty).and_return(true)
+        stderr, stdout = execute("pgbackups:url b001")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+http://latest/backup.dump
+STDOUT
+        $stdout.stub!(:isatty).and_return(old_stdout_isatty)
+      end
+
+      it "should capture a backup when requested" do
+        from_name, from_url = "FROM_NAME", "postgres://from/bar"
+        backup_obj = {'to_url' => "s3://bucket/userid/b001.dump"}
+
+        @pgbackups.stub!(:args).and_return([])
+        @pgbackups.stub!(:hpg_resolve).and_return([from_name, from_url])
+        @pgbackups.stub!(:transfer!).with(from_url, from_name, nil, "BACKUP", {:expire => nil}).and_return(backup_obj)
+        @pgbackups.stub!(:poll_transfer!).with(backup_obj).and_return(backup_obj)
+
+        @pgbackups.capture
+      end
+
+      it "should send expiration flag to client if specified on args" do
+        from_name, from_url = "FROM_NAME", "postgres://from/bar"
+        backup_obj = {'to_url' => "s3://bucket/userid/b001.dump"}
+
+        @pgbackups.stub!(:options).and_return({:expire => true})
+        @pgbackups.stub!(:hpg_resolve).and_return([from_name, from_url])
+        @pgbackups.stub!(:transfer!).with(from_url, from_name, nil, "BACKUP", {:expire => true}).and_return(backup_obj)
+        @pgbackups.stub!(:poll_transfer!).with(backup_obj).and_return(backup_obj)
+
+        @pgbackups.capture
+      end
+
+      it "destroys no backup without a name" do
+        stub_core
+        stderr, stdout = execute("pgbackups:destroy")
+        stderr.should == <<-STDERR
+ !    Usage: heroku pgbackups:destroy BACKUP_ID
+ !    Must specify BACKUP_ID to destroy.
+STDERR
+        stdout.should == ""
+      end
+
+      it "destroys a backup" do
+        stub_core
+        stub_pgbackups.get_backup("b001").returns({})
+        stub_pgbackups.delete_backup("b001").returns({})
+
+        stderr, stdout = execute("pgbackups:destroy b001")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Destroying b001... done
+STDOUT
+      end
+
+      it "aborts if no database addon is present" do
+        api.delete_config_var("myapp", "DATABASE_URL")
+        stub_core
+        stderr, stdout = execute("pgbackups:capture")
+        stderr.should == <<-STDERR
+ !    Your app has no databases.
+STDERR
+        stdout.should == ""
+      end
+
+      context "on errors" do
+        def stub_failed_capture(log)
+          @backup_obj = {
+            "error_at"    => Time.now.to_s,
+            "finished_at" => Time.now.to_s,
+            "log"         => log,
+            'to_url'      => 'postgres://from/bar'
+          }
+          stub_core
+          stub_pgbackups.create_transfer.returns(@backup_obj)
+          stub_pgbackups.get_transfer.returns(@backup_obj)
+        end
+
+        it 'aborts on a generic error' do
+          stub_failed_capture "something generic"
+          stderr, stdout = execute("pgbackups:capture")
+          stderr.should == <<-STDERR
+ !    An error occurred and your backup did not finish.
+STDERR
+          stdout.should == <<-STDOUT
+
+HEROKU_POSTGRESQL_IVORY (DATABASE_URL)  ----backup--->  bar
+
+\r\e[0K... 0 -
+STDOUT
+        end
+
+        it 'aborts and informs when the database isnt up yet' do
+          stub_failed_capture 'could not translate host name "ec2-42-42-42-42.compute-1.amazonaws.com" to address: Name or service not known'
+          stderr, stdout = execute("pgbackups:capture")
+          stderr.should == <<-STDERR
+ !    An error occurred and your backup did not finish.
+ !    The database is not yet online. Please try again.
+STDERR
+          stdout.should == <<-STDOUT
+
+HEROKU_POSTGRESQL_IVORY (DATABASE_URL)  ----backup--->  bar
+
+\r\e[0K... 0 -
+STDOUT
+        end
+
+        it 'aborts and informs when the credentials are incorrect' do
+          stub_failed_capture 'psql: FATAL:  database "randomname" does not exist'
+          stderr, stdout = execute("pgbackups:capture")
+          stderr.should == <<-STDERR
+ !    An error occurred and your backup did not finish.
+ !    The database credentials are incorrect.
+STDERR
+          stdout.should == <<-STDOUT
+
+HEROKU_POSTGRESQL_IVORY (DATABASE_URL)  ----backup--->  bar
+
+\r\e[0K... 0 -
+STDOUT
+        end
+      end
+    end
+
+    context "restore" do
+      before do
+        from_name, from_url = "FROM_NAME", "postgres://fromhost/database"
+
+        @pgbackups_client = mock("pgbackups_client")
+        @pgbackups.stub!(:pgbackup_client).and_return(@pgbackups_client)
+      end
+
+      it "should receive a confirm_command on restore" do
+        @pgbackups_client.stub!(:get_latest_backup).and_return({"to_url" => "s3://bucket/user/bXXX.dump"})
+
+        @pgbackups.should_receive(:confirm_command).and_return(false)
+        @pgbackups_client.should_not_receive(:transfer!)
+
+        @pgbackups.restore
+      end
+
+      it "aborts if no database addon is present" do
+        @pgbackups.should_receive(:hpg_resolve).and_raise(SystemExit)
+        lambda { @pgbackups.restore }.should raise_error(SystemExit)
+      end
+
+      context "for commands which perform restores" do
+        before do
+          @backup_obj = {
+            "to_name" => "TO_NAME",
+            "to_url" => "s3://bucket/userid/bXXX.dump",
+            "from_url" => "FROM_NAME",
+            "from_name" => "postgres://databasehost/dbname"
+          }
+
+          @pgbackups.stub!(:confirm_command).and_return(true)
+          @pgbackups_client.should_receive(:create_transfer).and_return(@backup_obj)
+          @pgbackups.stub!(:poll_transfer!).and_return(@backup_obj)
+        end
+
+        it "should default to the latest backup" do
+          @pgbackups.stub(:args).and_return([])
+          @pgbackups_client.should_receive(:get_latest_backup).and_return(@backup_obj)
+          @pgbackups.restore
+        end
+
+        it "should restore the named backup" do
+          name = "backupname"
+          args = ['DATABASE', name]
+          @pgbackups.stub(:args).and_return(args)
+          @pgbackups.stub(:shift_argument).and_return(*args)
+          @pgbackups.stub(:hpg_resolve).and_return([name])
+          @pgbackups_client.should_receive(:get_backup).with(name).and_return(@backup_obj)
+          @pgbackups.restore
+        end
+
+        it "should handle external restores" do
+          args = ['db_name_gets_shifted_out_in_resolve_db', 'http://external/file.dump']
+          @pgbackups.stub(:args).and_return(args)
+          @pgbackups.stub(:shift_argument).and_return(*args)
+          @pgbackups.stub(:hpg_resolve).and_return(["name", "url"])
+          @pgbackups_client.should_not_receive(:get_backup)
+          @pgbackups_client.should_not_receive(:get_latest_backup)
+          @pgbackups.restore
+        end
+      end
+
+      context "on errors" do
+        before(:each) do
+          @pgbackups_client.stub!(:get_latest_backup).and_return({"to_url" => "s3://bucket/user/bXXX.dump"})
+          @pgbackups.stub!(:confirm_command).and_return(true)
+        end
+
+        def stub_error_backup_with_log(log)
+          @backup_obj = {
+            "error_at" => Time.now.to_s,
+            "log" => log
+          }
+
+          @pgbackups_client.should_receive(:create_transfer).and_return(@backup_obj)
+          @pgbackups.stub!(:poll_transfer!).and_return(@backup_obj)
+        end
+
+        it 'aborts for a generic error' do
+          stub_error_backup_with_log 'something generic'
+          @pgbackups.should_receive(:error).with("An error occurred and your restore did not finish.")
+          @pgbackups.restore
+        end
+
+        it 'aborts and informs for expired s3 urls' do
+          stub_error_backup_with_log 'Invalid dump format: /tmp/aDMyoXPrAX/b031.dump: XML  document text'
+          @pgbackups.should_receive(:error).with { |message| message.should =~ /backup url is invalid/ }
+          @pgbackups.restore
+        end
+      end
+    end
+  end
+end