azuki 0.0.1

data/spec/azuki/command/pgbackups_spec.rb

@@ -0,0 +1,307 @@
+require "spec_helper"
+require "azuki/command/pgbackups"
+
+module Azuki::Command
+  describe Pgbackups do
+    before do
+      @pgbackups = prepare_command(Pgbackups)
+      @pgbackups.azuki.stub!(:info).and_return({})
+
+      api.post_app("name" => "example")
+      api.put_config_vars(
+        "example",
+        {
+          "DATABASE_URL"            => "postgres://database",
+          "AZUKI_POSTGRESQL_IVORY" => "postgres://database",
+          "PGBACKUPS_URL"           => "https://ip:password@pgbackups.azukiapp.com/client"
+        }
+      )
+      @pgbackups.stub :app_attachments => mock_attachments
+    end
+
+    let(:mock_attachments) {
+      [
+        Azuki::Helpers::AzukiPostgresql::Attachment.new({
+          'config_var' => 'AZUKI_POSTGRESQL_IVORY',
+          'resource' => {'name'  => 'softly-mocking-123',
+                         'value' => 'postgres://database',
+                         'type'  => 'azuki-postgresql:baku' }})
+      ]
+    }
+
+    after do
+      api.delete_app("example")
+    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",
+        "started_at"  => "2012-01-01 12:00:01 +0000",
+        "from_name"   => "DATABASE",
+        "size"        => "1024",
+        "progress"    => "dump 2048",
+        "to_name"     => "BACKUP",
+        "to_url"      => "s3://bucket/userid/b001.dump"
+      }])
+
+      stderr, stdout = execute("pgbackups")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+ID    Backup Time                Status     Size  Database
+----  -------------------------  ---------  ----  --------
+b001  2012-01-01 12:00:00 +0000  Capturing  1024  DATABASE
+STDOUT
+    end
+
+    describe "single backup" do
+      let(:from_name)  { "FROM_NAME" }
+      let(:from_url)   { "postgres://from/bar" }
+      let(:attachment) { double('attachment', :display_name => from_name, :url => from_url ) }
+      before do
+        @pgbackups.stub!(:hpg_resolve).and_return(attachment)
+      end
+
+      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
+        backup_obj = {'to_url' => "s3://bucket/userid/b001.dump"}
+
+        @pgbackups.stub!(:args).and_return([])
+        @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
+        backup_obj = {'to_url' => "s3://bucket/userid/b001.dump"}
+
+        @pgbackups.stub!(:options).and_return({:expire => true})
+        @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: azuki 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("example", "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)
+
+          any_instance_of(Azuki::Command::Pgbackups) do |pgbackups|
+            stub(pgbackups).app_attachments.returns(
+              mock_attachments
+            )
+          end
+        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.
+ !    Please run `azuki logs --ps pgbackups` for details.
+STDERR
+          stdout.should == <<-STDOUT
+
+AZUKI_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.
+ !    Please run `azuki logs --ps pgbackups` for details.
+ !    The database is not yet online. Please try again.
+STDERR
+          stdout.should == <<-STDOUT
+
+AZUKI_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.
+ !    Please run `azuki logs --ps pgbackups` for details.
+ !    The database credentials are incorrect.
+STDERR
+          stdout.should == <<-STDOUT
+
+AZUKI_POSTGRESQL_IVORY (DATABASE_URL)  ----backup--->  bar
+
+\r\e[0K... 0 -
+STDOUT
+        end
+      end
+    end
+
+    context "restore" do
+      let(:attachment) { double('attachment', :display_name => 'someconfigvar', :url => 'postgres://fromhost/database') }
+      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(attachment)
+          @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(attachment)
+          @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.\nPlease run `azuki logs --ps pgbackups` for details.")
+          @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

data/spec/azuki/command/plugins_spec.rb

@@ -0,0 +1,104 @@
+require "spec_helper"
+require "azuki/command/plugins"
+
+module Azuki::Command
+  include SandboxHelper
+
+  describe Plugins do
+
+    before do
+      @plugin = Azuki::Plugin.new("git://github.com/azuki/Plugin.git")
+    end
+
+    context("install") do
+
+      before do
+        Azuki::Plugin.should_receive(:new).with('git://github.com/azuki/Plugin.git').and_return(@plugin)
+        @plugin.should_receive(:install).and_return(true)
+      end
+
+      it "installs plugins" do
+        Azuki::Plugin.should_receive(:load_plugin).and_return(true)
+        stderr, stdout = execute("plugins:install git://github.com/azuki/Plugin.git")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Installing Plugin... done
+STDOUT
+      end
+
+      it "does not install plugins that do not load" do
+        Azuki::Plugin.should_receive(:load_plugin).and_return(false)
+        @plugin.should_receive(:uninstall).and_return(true)
+        stderr, stdout = execute("plugins:install git://github.com/azuki/Plugin.git")
+        stderr.should == '' # normally would have error, but mocks/stubs don't allow
+        stdout.should == "Installing Plugin... " # also inaccurate, would end in ' failed'
+      end
+
+    end
+
+    context("uninstall") do
+
+      before do
+        Azuki::Plugin.should_receive(:new).with('Plugin').and_return(@plugin)
+      end
+
+      it "uninstalls plugins" do
+        @plugin.should_receive(:uninstall).and_return(true)
+        stderr, stdout = execute("plugins:uninstall Plugin")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Uninstalling Plugin... done
+STDOUT
+      end
+
+      it "does not uninstall plugins that do not exist" do
+        stderr, stdout = execute("plugins:uninstall Plugin")
+        stderr.should == <<-STDERR
+ !    Plugin plugin not found.
+STDERR
+        stdout.should == <<-STDOUT
+Uninstalling Plugin... failed
+STDOUT
+      end
+
+    end
+
+    context("update") do
+
+      before do
+        Azuki::Plugin.should_receive(:new).with('Plugin').and_return(@plugin)
+      end
+
+      it "updates plugin by name" do
+        @plugin.should_receive(:update).and_return(true)
+        stderr, stdout = execute("plugins:update Plugin")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Updating Plugin... done
+STDOUT
+      end
+
+      it "updates all plugins" do
+        Azuki::Plugin.should_receive(:list).and_return([], [], ['Plugin'])
+        @plugin.should_receive(:update).and_return(true)
+        stderr, stdout = execute("plugins:update")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Updating Plugin... done
+STDOUT
+      end
+
+      it "does not update plugins that do not exist" do
+        stderr, stdout = execute("plugins:update Plugin")
+        stderr.should == <<-STDERR
+ !    Plugin plugin not found.
+STDERR
+        stdout.should == <<-STDOUT
+Updating Plugin... failed
+STDOUT
+      end
+
+    end
+
+  end
+end

data/spec/azuki/command/ps_spec.rb

@@ -0,0 +1,195 @@
+require "spec_helper"
+require "azuki/command/ps"
+
+describe Azuki::Command::Ps do
+
+  before(:each) do
+    stub_core
+  end
+
+  context("cedar") do
+
+    before(:each) do
+      api.post_app("name" => "example", "stack" => "cedar")
+    end
+
+    after(:each) do
+      api.delete_app("example")
+    end
+
+    it "ps:dynos errors out on cedar apps" do
+      lambda { execute("ps:dynos") }.should raise_error(Azuki::Command::CommandFailed, "For Cedar apps, use `azuki ps`")
+    end
+
+    it "ps:workers errors out on cedar apps" do
+      lambda { execute("ps:workers") }.should raise_error(Azuki::Command::CommandFailed, "For Cedar apps, use `azuki ps`")
+    end
+
+    describe "ps" do
+
+      it "displays processes" do
+        Azuki::Command::Ps.any_instance.should_receive(:time_ago).exactly(10).times.and_return("2012/09/11 12:34:56 (~ 0s ago)")
+        api.post_ps_scale('example', 'web', 10)
+        stderr, stdout = execute("ps")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+=== web: `bundle exec thin start -p $PORT`
+web.1: created 2012/09/11 12:34:56 (~ 0s ago)
+web.2: created 2012/09/11 12:34:56 (~ 0s ago)
+web.3: created 2012/09/11 12:34:56 (~ 0s ago)
+web.4: created 2012/09/11 12:34:56 (~ 0s ago)
+web.5: created 2012/09/11 12:34:56 (~ 0s ago)
+web.6: created 2012/09/11 12:34:56 (~ 0s ago)
+web.7: created 2012/09/11 12:34:56 (~ 0s ago)
+web.8: created 2012/09/11 12:34:56 (~ 0s ago)
+web.9: created 2012/09/11 12:34:56 (~ 0s ago)
+web.10: created 2012/09/11 12:34:56 (~ 0s ago)
+
+STDOUT
+      end
+
+      it "displays one-off processes" do
+        Azuki::Command::Ps.any_instance.should_receive(:time_ago).and_return('2012/09/11 12:34:56 (~ 0s ago)', '2012/09/11 12:34:56 (~ 0s ago)')
+        api.post_ps "example", "bash"
+
+        stderr, stdout = execute("ps")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+=== run: one-off processes
+run.1: created 2012/09/11 12:34:56 (~ 0s ago): `bash`
+
+=== web: `bundle exec thin start -p $PORT`
+web.1: created 2012/09/11 12:34:56 (~ 0s ago)
+
+STDOUT
+      end
+
+    end
+
+    describe "ps:restart" do
+
+      it "restarts all processes with no args" do
+        stderr, stdout = execute("ps:restart")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Restarting processes... done
+STDOUT
+      end
+
+      it "restarts one process" do
+        stderr, stdout = execute("ps:restart web.1")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Restarting web.1 process... done
+STDOUT
+      end
+
+      it "restarts a type of process" do
+        stderr, stdout = execute("ps:restart web")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Restarting web processes... done
+STDOUT
+      end
+
+    end
+
+    describe "ps:scale" do
+
+      it "can scale using key/value format" do
+        stderr, stdout = execute("ps:scale web=5")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Scaling web processes... done, now running 5
+STDOUT
+      end
+
+      it "can scale relative amounts" do
+        stderr, stdout = execute("ps:scale web+2")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Scaling web processes... done, now running 3
+STDOUT
+      end
+
+    end
+
+    describe "ps:stop" do
+
+      it "restarts one process" do
+        stderr, stdout = execute("ps:restart ps.1")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Restarting ps.1 process... done
+STDOUT
+      end
+
+      it "restarts a type of process" do
+        stderr, stdout = execute("ps:restart ps")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+Restarting ps processes... done
+STDOUT
+      end
+
+    end
+
+  end
+
+  context("non-cedar") do
+
+    before(:each) do
+      api.post_app("name" => "example")
+    end
+
+    after(:each) do
+      api.delete_app("example")
+    end
+
+    describe "ps:dynos" do
+
+      it "displays the current number of dynos" do
+        stderr, stdout = execute("ps:dynos")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+~ `azuki ps:dynos QTY` has been deprecated and replaced with `azuki ps:scale dynos=QTY`
+example is running 0 dynos
+STDOUT
+      end
+
+      it "sets the number of dynos" do
+        stderr, stdout = execute("ps:dynos 5")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+~ `azuki ps:dynos QTY` has been deprecated and replaced with `azuki ps:scale dynos=QTY`
+Scaling dynos... done, now running 5
+STDOUT
+      end
+
+    end
+
+    describe "ps:workers" do
+
+      it "displays the current number of workers" do
+        stderr, stdout = execute("ps:workers")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+~ `azuki ps:workers QTY` has been deprecated and replaced with `azuki ps:scale workers=QTY`
+example is running 0 workers
+STDOUT
+      end
+
+      it "sets the number of workers" do
+        stderr, stdout = execute("ps:workers 5")
+        stderr.should == ""
+        stdout.should == <<-STDOUT
+~ `azuki ps:workers QTY` has been deprecated and replaced with `azuki ps:scale workers=QTY`
+Scaling workers... done, now running 5
+STDOUT
+      end
+
+    end
+
+  end
+
+end