tugboat 0.0.5 → 0.0.6

data/spec/cli/snapshot_cli_spec.rb

@@ -5,14 +5,10 @@ describe Tugboat::CLI do
 
   let(:snapshot_name)     { "foo-snapshot" }
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "snapshots a droplet" do
     it "with a fuzzy name" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
-           to_return(:status => 200, :body => fixture("show_droplets"))
+           to_return(:status => 200, :body => fixture("show_droplets_inactive"))
 
       stub_request(:get, "https://api.digitalocean.com/droplets/100823/snapshot?api_key=#{api_key}&client_id=#{client_key}&name=#{snapshot_name}").
            to_return(:status => 200, :body => fixture("show_droplet"))
@@ -31,7 +27,7 @@ Queuing snapshot 'foo-snapshot' for 100823 (foo)...done
 
     it "with an id" do
       stub_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}").
-           to_return(:status => 200, :body => fixture("show_droplet"))
+           to_return(:status => 200, :body => fixture("show_droplet_inactive"))
 
       stub_request(:get, "https://api.digitalocean.com/droplets/100823/snapshot?api_key=#{api_key}&client_id=#{client_key}&name=#{snapshot_name}").
            to_return(:status => 200, :body => fixture("show_droplet"))
@@ -52,7 +48,7 @@ Queuing snapshot 'foo-snapshot' for 100823 (foo)...done
 
     it "with a name" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
-           to_return(:status => 200, :body => fixture("show_droplets"))
+           to_return(:status => 200, :body => fixture("show_droplets_inactive"))
 
       stub_request(:get, "https://api.digitalocean.com/droplets/100823/snapshot?api_key=#{api_key}&client_id=#{client_key}&name=#{snapshot_name}").
            to_return(:status => 200, :body => fixture("show_droplet"))
@@ -70,6 +66,21 @@ Queuing snapshot 'foo-snapshot' for 100823 (foo)...done
       expect(a_request(:get, "https://api.digitalocean.com/droplets/100823/snapshot?api_key=#{api_key}&client_id=#{client_key}&name=#{snapshot_name}")).to have_been_made
     end
 
+    it "does not snaphshot a droplet that is active" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      expect {@cli.snapshot(snapshot_name)}.to raise_error(SystemExit)
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Droplet must be off for this operation to be successful.
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
   end
 
 end

data/spec/cli/ssh_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "ssh" do
     it "tries to fetch the droplet's IP from the API" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
@@ -18,5 +14,24 @@ describe Tugboat::CLI do
       expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).
         to have_been_made
     end
+
+    it "does not allow ssh into a droplet that is inactive" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplets_inactive"))
+
+      Kernel.stub(:exec)
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+
+      expect {@cli.ssh("test222")}.to raise_error(SystemExit)
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Droplet must be on for this operation to be successful.
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
   end
 end

data/spec/cli/start_cli_spec.rb

@@ -0,0 +1,76 @@
+require 'spec_helper'
+
+describe Tugboat::CLI do
+  include_context "spec"
+
+  describe "start" do
+    it "starts the droplet with a fuzzy name" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => fixture("show_droplets_inactive"))
+      stub_request(:put, "https://api.digitalocean.com/droplets/100823/power_on?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.start("foo")
+
+      expect($stdout.string).to eq <<-eos
+Droplet fuzzy name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing start for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:put, "https://api.digitalocean.com/droplets/100823/power_on?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "starts the droplet with an id" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}").
+       to_return(:status => 200, :body => fixture("show_droplet_inactive"))
+      stub_request(:put, "https://api.digitalocean.com/droplets/100823/power_on?api_key=#{api_key}&client_id=#{client_key}").
+       to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:id => droplet_id)
+      @cli.start
+
+      expect($stdout.string).to eq <<-eos
+Droplet id provided. Finding Droplet...done\e[0m, 100823 (foo)
+Queuing start for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:put, "https://api.digitalocean.com/droplets/100823/power_on?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+
+    it "starts the droplet with a name" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => fixture("show_droplets_inactive"))
+      stub_request(:put, "https://api.digitalocean.com/droplets/100823/power_on?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      @cli.start
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing start for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:put, "https://api.digitalocean.com/droplets/100823/power_on?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "does not start a droplet that is inactive" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      expect {@cli.start}.to raise_error(SystemExit)
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Droplet must be off for this operation to be successful.
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+  end
+end

data/spec/cli/version_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "version" do
     it "shows the correct version" do
 

data/spec/cli/wait_cli_spec.rb

@@ -0,0 +1,66 @@
+require 'spec_helper'
+
+describe Tugboat::CLI do
+  include_context "spec"
+
+  describe "wait" do
+    it "waits for a droplet with a fuzzy name" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:state => "active")
+      @cli.wait("foo")
+
+      expect($stdout.string).to eq <<-eos
+Droplet fuzzy name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Waiting for droplet to become active..done\e[0m (0s)
+eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "waits for a droplet with an id" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:id => droplet_id, :state => "active")
+      @cli.wait
+
+      expect($stdout.string).to eq <<-eos
+Droplet id provided. Finding Droplet...done\e[0m, 100823 (foo)
+Waiting for droplet to become active..done\e[0m (0s)
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "waits for a droplet with a name" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name, :state => "active")
+      @cli.wait
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Waiting for droplet to become active..done\e[0m (0s)
+eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+  end
+
+end

data/spec/config_spec.rb

@@ -1,6 +1,8 @@
 require 'spec_helper'
 
 describe Tugboat::Configuration do
+  include_context "spec"
+
   let(:tmp_path)             { project_path + "/tmp/tugboat" }
 
   after :each do

data/spec/fixtures/show_droplet_inactive.json

@@ -0,0 +1,13 @@
+{
+  "status": "OK",
+  "droplet": {
+    "backups_active": null,
+    "id": 100823,
+    "image_id": 420,
+    "name": "foo",
+    "ip_address": "33.33.33.10",
+    "region_id": 1,
+    "size_id": 33,
+    "status": "off"
+  }
+}

data/spec/fixtures/show_droplets.json

@@ -29,7 +29,7 @@
       "name": "foo",
       "region_id": 1,
       "size_id": 33,
-      "status": "off"
+      "status": "active"
     }
   ]
 }

data/spec/fixtures/show_droplets_inactive.json

@@ -0,0 +1,35 @@
+{
+  "status": "OK",
+  "droplets": [
+    {
+      "ip_address": "33.33.33.10",
+      "backups_active": null,
+      "id": 100823,
+      "image_id": 420,
+      "name": "test222",
+      "region_id": 1,
+      "size_id": 33,
+      "status": "off"
+    },
+    {
+      "ip_address": "33.33.33.10",
+      "backups_active": null,
+      "id": 100823,
+      "image_id": 420,
+      "name": "test223",
+      "region_id": 1,
+      "size_id": 33,
+      "status": "off"
+    },
+    {
+      "ip_address": "33.33.33.10",
+      "backups_active": null,
+      "id": 100823,
+      "image_id": 420,
+      "name": "foo",
+      "region_id": 1,
+      "size_id": 33,
+      "status": "off"
+    }
+  ]
+}

data/spec/middleware/check_configuration_spec.rb

@@ -3,9 +3,6 @@ require 'spec_helper'
 describe Tugboat::Middleware::CheckConfiguration do
   include_context "spec"
 
-  let(:app) { lambda { |env| } }
-  let(:env) { {} }
-
   describe ".call" do
     it "raises SystemExit with no configuration" do
 

data/spec/middleware/check_credentials_spec.rb

@@ -3,9 +3,6 @@ require 'spec_helper'
 describe Tugboat::Middleware::CheckCredentials do
   include_context "spec"
 
-  let(:app) { lambda { |env| } }
-  let(:env) { {} }
-
   describe ".call" do
     it "raises SystemExit with no configuration" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").

data/spec/middleware/check_droplet_active_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+describe Tugboat::Middleware::CheckDropletActive do
+  include_context "spec"
+
+  describe ".call" do
+    it "raises an error when droplet is not active" do
+
+      env["droplet_status"] = "off"
+
+      expect {described_class.new(app).call(env) }.to raise_error(SystemExit)
+    end
+  end
+
+end

data/spec/middleware/check_droplet_inactive_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+describe Tugboat::Middleware::CheckDropletInactive do
+  include_context "spec"
+
+  describe ".call" do
+    it "raises an error when droplet is active" do
+
+      env["droplet_status"] = "active"
+
+      expect {described_class.new(app).call(env) }.to raise_error(SystemExit)
+    end
+  end
+
+end

data/spec/middleware/find_droplet_spec.rb

@@ -3,9 +3,6 @@ require 'spec_helper'
 describe Tugboat::Middleware::FindDroplet do
   include_context "spec"
 
-  let(:app) { lambda { |env| } }
-  let(:env) { {} }
-
   describe ".call" do
     it "raises SystemExit with no droplet data" do
       expect {described_class.new(app).call(env) }.to raise_error(SystemExit)

data/spec/middleware/inject_configuration_spec.rb

@@ -3,9 +3,6 @@ require 'spec_helper'
 describe Tugboat::Middleware::InjectConfiguration do
   include_context "spec"
 
-  let(:app) { lambda { |env| } }
-  let(:env) { {} }
-
   describe ".call" do
 
     it "loads the configuration into the environment" do

data/spec/middleware/ssh_droplet_spec.rb

@@ -3,9 +3,6 @@ require 'spec_helper'
 describe Tugboat::Middleware::SSHDroplet do
   include_context "spec"
 
-  let(:app) { lambda { |env| } }
-  let(:env) { {} }
-
   before do
     Kernel.stub!(:exec)
   end
@@ -28,6 +25,25 @@ describe Tugboat::Middleware::SSHDroplet do
       described_class.new(app).call(env)
     end
 
+    it "executes ssh with custom options" do
+      Kernel.should_receive(:exec).with("ssh",
+                        "-o", "IdentitiesOnly=yes",
+                        "-o", "LogLevel=ERROR",
+                        "-o", "StrictHostKeyChecking=no",
+                        "-o", "UserKnownHostsFile=/dev/null",
+                        "-i", ssh_key_path,
+                        "-p", ssh_port,
+                        "-q",
+                        "-X",
+                        "#{ssh_user}@#{droplet_ip}")
+
+      env["droplet_ip"] = droplet_ip
+      env["config"] = config
+      env["user_droplet_ssh_opts"] = "-q -X"
+
+      described_class.new(app).call(env)
+    end
+
   end
 
 end

data/spec/shared/environment.rb

@@ -12,11 +12,14 @@ shared_context "spec" do
   let(:droplet_ip)       { "33.33.33.10" }
   let(:droplet_id)       { 1234 }
   let(:ocean)            { DigitalOcean::API.new :client_id => client_key, :api_key =>api_key }
+  let(:app)              { lambda { |env| } }
+  let(:env)              { {} }
 
   before(:each) do
     $stdout.sync = true
     $stderr.sync = true
 
+    @cli = Tugboat::CLI.new
 
     # Set a temprary project path and create fake config.
     config.create_config_file(client_key, api_key, ssh_key_path, ssh_user, ssh_port)

data/spec/spec_helper.rb

@@ -1,3 +1,6 @@
+require 'coveralls'
+Coveralls.wear! { add_filter '/spec/' }
+
 require 'tugboat'
 require 'webmock/rspec'
 require 'digital_ocean'
@@ -6,6 +9,8 @@ require "shared/environment"
 RSpec.configure do |config|
   # Pretty tests
   config.color_enabled = true
+
+  config.order = :random
 end
 
 def project_path