do_snapshot 0.0.6 → 0.0.7

data/spec/do_snapshot/command_spec.rb

@@ -0,0 +1,121 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+describe DoSnapshot::Command do
+  include_context 'spec'
+  include_context 'uri_helpers'
+
+  subject(:log)     { DoSnapshot::Log }
+
+  describe '.snap' do
+    context 'when success' do
+      it 'sends success message' do
+        expect { snap_runner }
+          .not_to raise_error
+        expect(log.buffer)
+          .to include 'All operations has been finished.'
+      end
+    end
+
+    context 'when snapshot not cleanup' do
+      it 'sends cleanup message' do
+        stub_image_destroy_fail(image_id)
+        stub_image_destroy_fail(image_id2)
+
+        expect { snap_runner }
+          .not_to raise_error
+      end
+    end
+
+    context 'when droplet not found' do
+      it 'raised by exception' do
+        stub_droplet_fail(droplet_id)
+
+        expect { snap_runner }
+          .to raise_error(DoSnapshot::DropletFindError)
+      end
+    end
+
+    context 'when failed to list droplets' do
+      it 'raised with droplet list error' do
+        stub_droplets_fail
+
+        expect { snap_runner }
+          .to raise_error(DoSnapshot::DropletListError)
+      end
+    end
+
+    # TODO: MUST HAVE! Now when this two works others can fail...
+    # context 'when droplet failed for shutdown' do
+    #   it 'raised with shutdown error' do
+    #     fail = stub_droplet_stop_fail(droplet_id)
+    #
+    #     expect { snap_runner }
+    #       .to raise_error(DoSnapshot::DropletShutdownError)
+    #
+    #     remove_request_stub(fail)
+    #   end
+    # end
+    #
+    # context 'when no snapshot created' do
+    #   it 'raised with snapshot create error' do
+    #     no_snapshot = stub_droplet_snapshot_fail(droplet_id, snapshot_name)
+    #
+    #     expect { snap_runner }
+    #       .to raise_error(DoSnapshot::SnapshotCreateError)
+    #
+    #     remove_request_stub(no_snapshot)
+    #   end
+    # end
+  end
+
+  describe  '.fail_power_off' do
+    it 'when success' do
+      stub_droplet_inactive(droplet_id)
+
+      expect { @cmd.fail_power_off(DoSnapshot::DropletShutdownError.new(droplet_id)) }
+        .not_to raise_error
+      expect(log.buffer)
+        .to include 'Power On has been requested.'
+    end
+
+    it 'with request error' do
+      stub_droplet_fail(droplet_id)
+
+      expect { @cmd.fail_power_off(DoSnapshot::DropletShutdownError.new(droplet_id)) }
+        .to raise_error
+      expect(log.buffer)
+        .to include 'Droplet id: 100823 is Failed to Power Off.'
+      expect(log.buffer)
+      .to include 'Droplet Not Found'
+    end
+
+    it 'with start error' do
+      stub_droplet_inactive(droplet_id)
+      stub_droplet_start_fail(droplet_id)
+      stub_event_fail(event_id)
+
+      expect { @cmd.fail_power_off(DoSnapshot::DropletShutdownError.new(droplet_id)) }
+        .not_to raise_error
+      expect(log.buffer)
+        .to include 'Power On failed to request.'
+    end
+  end
+
+  after(:each) do
+    # WebMock.reset!
+    stub_cleanup
+  end
+
+  before(:each) do
+    @cmd = DoSnapshot::Command.dup
+    stub_all_api(nil, true)
+    log.buffer = %w()
+    log.quiet = true
+  end
+
+  def snap_runner(options = nil)
+    options ||= default_options
+    @cmd.snap(options, [:log, :trace, :digital_ocean_client_id, :digital_ocean_api_key])
+  end
+end

data/spec/do_snapshots_spec.rb

@@ -0,0 +1,58 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+describe DoSnapshot do
+  include_context 'spec'
+
+  subject(:log)         { DoSnapshot::Log }
+
+  describe DoSnapshot::DropletFindError do
+    subject(:error) { described_class }
+
+    it 'should work' do
+      error.new
+      expect(log.buffer)
+        .to include 'Droplet Not Found'
+    end
+  end
+
+  describe DoSnapshot::DropletListError do
+    subject(:error) { described_class }
+
+    it 'should work' do
+      error.new
+      expect(log.buffer)
+        .to include 'Droplet Listing is failed to retrieve'
+    end
+  end
+
+  describe DoSnapshot::SnapshotCleanupError do
+    subject(:error) { described_class }
+
+    it 'should be' do
+
+      expect { error.new }
+        .not_to raise_error
+    end
+  end
+
+  describe DoSnapshot::DropletShutdownError do
+    subject(:error) { described_class }
+
+    it 'should work' do
+      error.new(droplet_id)
+      expect(log.buffer)
+        .to include "Droplet id: #{droplet_id} is Failed to Power Off."
+    end
+  end
+
+  describe DoSnapshot::SnapshotCreateError do
+    subject(:error) { described_class }
+
+    it 'should work' do
+      error.new(droplet_id)
+      expect(log.buffer)
+        .to include "Droplet id: #{droplet_id} is Failed to Snapshot."
+    end
+  end
+end

data/spec/fixtures/error_message.json

@@ -0,0 +1,4 @@
+{
+    "status": "ERROR",
+    "message": "Some Message"
+}

data/spec/fixtures/response_event.json

@@ -0,0 +1,4 @@
+{
+    "status": "OK",
+    "event_id": 7501
+}

data/spec/fixtures/show_droplet.json

@@ -0,0 +1,39 @@
+{
+    "status": "OK",
+    "droplet": {
+        "backups_active": null,
+        "id": 100823,
+        "image_id": 420,
+        "name": "foo",
+        "ip_address": "33.33.33.10",
+        "snapshots": [
+            {
+                "id": 5019770,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            },
+            {
+                "id": 5019903,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            },
+            {
+                "id": 5020783,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            },
+            {
+                "id": 5030783,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            }
+        ],
+        "region_id": 1,
+        "size_id": 33,
+        "status": "active"
+    }
+}

data/spec/fixtures/show_droplet_inactive.json

@@ -0,0 +1,39 @@
+{
+    "status": "OK",
+    "droplet": {
+        "backups_active": null,
+        "id": 100823,
+        "image_id": 420,
+        "name": "foo",
+        "ip_address": "33.33.33.10",
+        "snapshots": [
+            {
+                "id": 5019770,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            },
+            {
+                "id": 5019903,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            },
+            {
+                "id": 5020783,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            },
+            {
+                "id": 5030783,
+                "name": "mrcr.ru_2014_07_19",
+                "slug": null,
+                "distribution": "CentOS"
+            }
+        ],
+        "region_id": 1,
+        "size_id": 33,
+        "status": "off"
+    }
+}

data/spec/fixtures/show_droplets.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": "active"
+        },
+        {
+            "ip_address": "33.33.33.10",
+            "backups_active": null,
+            "id": 100824,
+            "image_id": 420,
+            "name": "test223",
+            "region_id": 1,
+            "size_id": 33,
+            "status": "active"
+        },
+        {
+            "ip_address": "33.33.33.10",
+            "backups_active": null,
+            "id": 100825,
+            "image_id": 420,
+            "name": "foo",
+            "region_id": 1,
+            "size_id": 33,
+            "status": "active"
+        }
+    ]
+}

data/spec/fixtures/show_droplets_empty.json

@@ -0,0 +1,4 @@
+{
+    "status":"OK",
+    "droplets":[]
+}

data/spec/fixtures/show_event_done.json

@@ -0,0 +1,10 @@
+{
+    "status": "OK",
+    "event": {
+        "id": 7499,
+        "action_status": "done",
+        "droplet_id": 100823,
+        "event_type_id": 1,
+        "percentage": "100"
+    }
+}

data/spec/fixtures/show_event_start.json

@@ -0,0 +1,10 @@
+{
+    "status": "OK",
+    "event": {
+        "id": 7499,
+        "action_status": "done",
+        "droplet_id": 100823,
+        "event_type_id": 1,
+        "percentage": "3"
+    }
+}

data/spec/shared/api_helpers.rb

@@ -0,0 +1,117 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+shared_context 'api_helpers' do
+
+  # List of droplets
+  #
+  def stub_droplets
+    stub_without_id(droplets_uri, 'show_droplets')
+  end
+
+  def stub_droplets_empty
+    stub_without_id(droplets_uri, 'show_droplets_empty')
+  end
+
+  def stub_droplets_fail
+    stub_without_id(droplets_uri, 'error_message')
+  end
+
+  # Droplet data
+  #
+  def stub_droplet(id)
+    stub_with_id(droplet_find_uri, id, 'show_droplet')
+  end
+
+  def stub_droplet_fail(id)
+    stub_with_id(droplet_find_uri, id, 'error_message')
+  end
+
+  def stub_droplet_inactive(id)
+    stub_with_id(droplet_find_uri, id, 'show_droplet_inactive')
+  end
+
+  # Droplet actions
+  #
+  def stub_droplet_stop(id)
+    stub_with_id(droplet_stop_uri, id, 'response_event')
+  end
+
+  def stub_droplet_stop_fail(id)
+    stub_with_id(droplet_stop_uri, id, 'error_message')
+  end
+
+  def stub_droplet_start(id)
+    stub_with_id(droplet_start_uri, id, 'response_event')
+  end
+
+  def stub_droplet_start_fail(id)
+    stub_with_id(droplet_start_uri, id, 'error_message')
+  end
+
+  # Snapshot
+  #
+  def stub_droplet_snapshot(id, name)
+    stub_with_id_name(snapshot_uri, id, name, 'response_event')
+  end
+
+  def stub_droplet_snapshot_fail(id, name)
+    stub_with_id_name(snapshot_uri, id, name, 'error_message')
+  end
+
+  # Event status
+  #
+  def stub_event_done(id)
+    stub_with_id(event_find_uri, id, 'show_event_done')
+  end
+
+  def stub_event_fail(id)
+    stub_with_id(event_find_uri, id, 'error_message')
+  end
+
+  def stub_event_running(id)
+    stub_with_id(event_find_uri, id, 'show_event_running')
+  end
+
+  # Image actions
+  #
+  def stub_image_destroy(id)
+    stub_with_id(image_destroy_uri, id, 'response_event')
+  end
+
+  def stub_image_destroy_fail(id)
+    stub_with_id(image_destroy_uri, id, 'error_message')
+  end
+
+  # Stub helpers
+  #
+  def stub_with_id(request, id, fixture, status = 200)
+    return unless request && fixture && id
+    stub_request(:get, url_with_id(request, id))
+      .to_return(status: status, body: fixture(fixture))
+  end
+
+  def stub_without_id(request, fixture, status = 200)
+    return unless request && fixture
+    stub_request(:get, request)
+      .to_return(status: status, body: fixture(fixture))
+  end
+
+  def stub_with_id_name(request, id, name, fixture, status = 200)
+    return unless request && fixture && id && name
+    stub_request(:get, url_with_id_name(request, id, name))
+      .to_return(status: status, body: fixture(fixture))
+  end
+
+  # Url helpers
+  #
+  def url_with_id(request, id)
+    return unless request && id
+    request.sub('[id]', id.to_s)
+  end
+
+  def url_with_id_name(request, id, name)
+    return unless request && id && name
+    request.sub('[id]', id.to_s).sub('[name]', name)
+  end
+end