do_snapshot 0.3.0 → 0.3.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0c4e8fd9920e0fb822c5f84dadf54aa6cf0a90b5
-  data.tar.gz: 26fa16b5c5e5a1ebb5d474e31467488fca1a3c07
+  metadata.gz: 4af572f41b36fbe4252628e2508404b789afca71
+  data.tar.gz: d31302ccbc40a5a24491ecefa788ea89b6f1881e
 SHA512:
-  metadata.gz: 8368fd230ea50254df7456a01a7e20199b8ee0d7a6937b7e036c11a8902206b7c0034f79ea7cb2f0b80d4621d48b6715c2f02b62b254e5bc01978fb54ae07b8a
-  data.tar.gz: 8139cdbee917477ecc0dea28a63c5d2fb07485e2f4f47781340959ba97fc94317b401c99f632685b8c37177cbaef35e585de39adad0818a5a4f0b12e996e1757
+  metadata.gz: dcffac80be5524e1e54a6d2754aec6bde8c991bafe84d93b830d52b984f2830df03399333c687c5bc88d27786339e91e1077987741080b953922a7eb10d15f27
+  data.tar.gz: 7983504785434f31b8169280923c506f1581b63ee6a34132edaab245a853061246fdbe6a3e6e0eef2eeff43e9810d62ad22283f27b93d57fd9163d4440594325

data/README.md

@@ -185,6 +185,10 @@ For working mailer you need to set e-mail settings via run options.
     
       You can optionally specify parameters to select or exclude some droplets.   
 
+## Donating:
+Support this project and others by [merqlove](https://gratipay.com/~merqlove/) via [gratipay](https://gratipay.com/~merqlove/).  
+[![Support via Gratipay](https://cdn.rawgit.com/gratipay/gratipay-badge/2.3.0/dist/gratipay.png)](https://gratipay.com/merqlove/)
+
 ## Dependencies:
 
 - [Thor](https://github.com/erikhuda/thor) for CLI.

data/lib/do_snapshot/adapter/abstract.rb

@@ -25,10 +25,20 @@ module DoSnapshot
       def check_keys; end
 
       # Waiting for event exit
-      def wait_event(id)
-        logger.debug "Event Id: #{id}"
+      def wait_wrap(id, message = "Event Id: #{id}", &status_block)
+        logger.debug message
         time = Time.now
-        sleep(delay) until get_event_status(id, time)
+        sleep(delay) until status_block.call(id, time)
+      end
+
+      # Waiting for event exit
+      def wait_event(event_id)
+        wait_wrap(event_id) { |id, time| get_event_status(id, time) }
+      end
+
+      # Waiting for event exit
+      def wait_shutdown(droplet_id)
+        wait_wrap(droplet_id, "Droplet Id: #{droplet_id} shutting down") { |id, time| get_shutdown_status(id, time) }
       end
 
       def after_cleanup(droplet_id, droplet_name, snapshot, event)
@@ -40,6 +50,25 @@ module DoSnapshot
           logger.debug "Snapshot name: #{snapshot.name} delete requested."
         end
       end
+
+      def timeout?(id, time, message = "Event #{id} finished by timeout #{time}")
+        return false unless (Time.now - time) > @timeout
+        logger.debug message
+        true
+      end
+
+      def droplet_timeout?(id, time)
+        timeout? id, time, "Droplet id: #{id} shutdown event closed by timeout #{time}"
+      end
+
+      # Looking for event status.
+      # Before snapshot we to know that machine has powered off.
+      #
+      def get_shutdown_status(id, time)
+        fail "Droplet #{id} not responding for shutdown!" if droplet_timeout?(id, time)
+
+        inactive?(id)
+      end
     end
   end
 end

data/lib/do_snapshot/adapter/digitalocean.rb

@@ -35,11 +35,9 @@ module DoSnapshot
         # noinspection RubyResolve
         instance = droplet(id)
 
-        if instance.status.include? 'active'
-          logger.error 'Droplet is still running.'
-        else
-          power_on id
-        end
+        return power_on(id) unless instance.status.include? 'active'
+
+        logger.error "Droplet #{id} is still running. Skipping."
       end
 
       # Power Off request for Droplet
@@ -51,7 +49,7 @@ module DoSnapshot
         fail event.message unless event.status.include? 'OK'
 
         # noinspection RubyResolve
-        wait_event(event.event_id)
+        wait_shutdown(id)
       rescue => e
         raise DropletShutdownError.new(id), e.message, e.backtrace
       end
@@ -63,15 +61,23 @@ module DoSnapshot
         event = ::DigitaloceanC::Droplet.snapshot(id, name: name)
 
         if !event
-          fail 'Something wrong with DigitalOcean or with your connection :)'
+          fail DoSnapshot::SnapshotCreateError.new(id), 'Something wrong with DigitalOcean or with your connection :)'
         elsif event && !event.status.include?('OK')
-          fail event.message
+          fail DoSnapshot::SnapshotCreateError.new(id), event.message
         end
 
         # noinspection RubyResolve
         wait_event(event.event_id)
       end
 
+      # Checking if droplet is powered off.
+      #
+      def inactive?(id)
+        instance = droplet(id)
+
+        instance.status.include?('off')
+      end
+
       # Cleanup our snapshots.
       #
       def cleanup_snapshots(instance, size)
@@ -107,17 +113,11 @@ module DoSnapshot
         return true if timeout?(id, time)
 
         event = ::DigitaloceanC::Event.find(id)
-        fail event.message unless event.status.include?('OK')
+        fail DoSnapshot::EventError.new(id), event.message unless event.status.include?('OK')
         # noinspection RubyResolve,RubyResolve
         event.event.percentage && event.event.percentage.include?('100') ? true : false
       end
 
-      def timeout?(id, time)
-        return false unless (Time.now - time) > @timeout
-        logger.debug "Event #{id} finished by timeout #{time}"
-        true
-      end
-
       # Request Power On for droplet
       #
       def power_on(id)

data/lib/do_snapshot/adapter/digitalocean_v2.rb

@@ -41,21 +41,19 @@ module DoSnapshot
         # noinspection RubyResolve
         instance = droplet(id)
 
-        if instance.status && instance.status.include?('active')
-          logger.error 'Droplet is still running.'
-        else
-          power_on id
-        end
+        return power_on(id) unless instance.status && instance.status.include?('active')
+
+        logger.error "Droplet #{id} is still running. Skipping."
       end
 
       # Power Off request for Droplet
       #
       def stop_droplet(id)
         # noinspection RubyResolve,RubyResolve
-        event = client.droplet_actions.power_off(droplet_id: id)
+        client.droplet_actions.power_off(droplet_id: id)
 
         # noinspection RubyResolve
-        wait_event(event.id)
+        wait_shutdown(id)
       rescue => e
         raise DropletShutdownError.new(id), e.message, e.backtrace
       end
@@ -66,10 +64,18 @@ module DoSnapshot
         # noinspection RubyResolve,RubyResolve
         event = client.droplet_actions.snapshot(droplet_id: id, name: name)
 
+        fail DoSnapshot::SnapshotCreateError.new(id), name unless event && event.respond_to?(:id)
+
         # noinspection RubyResolve
         wait_event(event.id)
-      rescue => e
-        raise e.message, e.backtrace
+      end
+
+      # Checking if droplet is powered off.
+      #
+      def inactive?(id)
+        instance = droplet(id)
+
+        instance.status.include?('off')
       end
 
       # Cleanup our snapshots.
@@ -117,16 +123,14 @@ module DoSnapshot
       # Before snapshot we to know that machine has powered off.
       #
       def get_event_status(id, time)
-        if (Time.now - time) > @timeout
-          logger.debug "Event #{id} finished by timeout #{time}"
-          return true
-        end
+        return true if timeout?(id, time)
 
         action = client.actions.find(id: id)
+
+        fail DoSnapshot::EventError.new(id), 'Check your connection' unless action && action.respond_to?(:status)
+
         # noinspection RubyResolve,RubyResolve
         action.status.include?('completed') ? true : false
-      rescue => e
-        raise e.message, e.backtrace
       end
 
       # Request Power On for droplet

data/lib/do_snapshot/cli.rb

@@ -203,7 +203,7 @@ module DoSnapshot
           config.logger_level = Logger::DEBUG if config.verbose
           config.verbose = options['trace']
           config.quiet = options['quiet']
-          config.mailer = Mail.new(opts: options['mail'], smtp: options['smtp'])
+          config.mailer = Mail.new(opts: options['mail'], smtp: options['smtp']) if options['mail']
         end
       end
 

data/lib/do_snapshot/command.rb

@@ -42,11 +42,17 @@ module DoSnapshot
     def stop_droplet(droplet)
       logger.debug 'Shutting down droplet.'
       api.stop_droplet(droplet.id) unless droplet.status.include? 'off'
+      true
+    rescue => e
+      logger.error e.message
+      false
     end
 
     # Trying to create a snapshot.
     #
     def create_snapshot(droplet) # rubocop:disable MethodLength,Metrics/AbcSize
+      fail DropletPowerError.new(droplet.id), droplet.name unless api.inactive?(droplet.id)
+
       logger.info "Start creating snapshot for droplet id: #{droplet.id} name: #{droplet.name}."
 
       today         = DateTime.now
@@ -69,6 +75,8 @@ module DoSnapshot
       case e.class.to_s
       when 'DoSnapshot::SnapshotCleanupError'
         raise e.class, e.message, e.backtrace
+      when 'DoSnapshot::DropletPowerError'
+        return
       else
         raise SnapshotCreateError.new(droplet.id), e.message, e.backtrace
       end
@@ -132,8 +140,7 @@ module DoSnapshot
     #
     def thread_runner(droplet)
       threads << Thread.new do
-        stop_droplet droplet
-        create_snapshot droplet
+        create_snapshot droplet if stop_droplet(droplet)
       end
     end
 
@@ -141,7 +148,7 @@ module DoSnapshot
     # Droplet instance must be powered off first!
     #
     def prepare_droplet(id, name)
-      logger.debug "Droplet id: #{id} name: #{name} "
+      logger.debug "Droplet id: #{id} name: #{name}\n"
       droplet = api.droplet id
 
       return unless droplet

data/lib/do_snapshot/version.rb

@@ -2,5 +2,5 @@
 # Current version
 #
 module DoSnapshot
-  VERSION = '0.3.0'
+  VERSION = '0.3.4'
 end

data/lib/do_snapshot.rb

@@ -1,4 +1,6 @@
 # -*- encoding : utf-8 -*-
+require 'active_support/multibyte' # ActiveSupport 3.2 & Mail gem fix to work together.
+
 require_relative 'do_snapshot/version'
 require_relative 'do_snapshot/configuration'
 
@@ -80,6 +82,24 @@ module DoSnapshot
     end
   end
 
+  # When Droplet not Powered Off!
+  #
+  class DropletPowerError < RequestError
+    def initialize(*args)
+      DoSnapshot.logger.error "Droplet id: #{args[0]} must be Powered Off!"
+      super
+    end
+  end
+
+  # When Event is failed!
+  #
+  class EventError < RequestError
+    def initialize(*args)
+      DoSnapshot.logger.error "Event id: #{args[0]} is failed!"
+      super
+    end
+  end
+
   # When Digital Ocean API cannot retrieve list of droplets.
   # Sometimes it connection problem or DigitalOcean API maintenance.
   #

data/spec/do_snapshot/adapter/abstract_spec.rb

@@ -1,7 +1,7 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot::Adapter::Abstract do
+RSpec.describe DoSnapshot::Adapter::Abstract do
   include_context 'spec'
 
   subject(:api) { described_class }

data/spec/do_snapshot/adapter/digitalocean_spec.rb

@@ -1,7 +1,7 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot::Adapter::Digitalocean do
+RSpec.describe DoSnapshot::Adapter::Digitalocean do
   include_context 'spec'
   include_context 'api_v1_helpers'
 
@@ -40,7 +40,7 @@ describe DoSnapshot::Adapter::Digitalocean do
         stub_droplet_fail(droplet_id)
 
         expect { instance.droplet(droplet_id) }
-          .to raise_error
+          .to raise_error(DoSnapshot::DropletFindError)
         expect(DoSnapshot.logger.buffer)
           .to include 'Droplet Not Found'
 
@@ -62,7 +62,7 @@ describe DoSnapshot::Adapter::Digitalocean do
       it 'with error' do
         stub_droplets_fail
 
-        expect { instance.droplets }.to raise_error
+        expect { instance.droplets }.to raise_error(DoSnapshot::DropletListError)
         expect(DoSnapshot.logger.buffer)
           .to include 'Droplet Listing is failed to retrieve'
 
@@ -91,7 +91,7 @@ describe DoSnapshot::Adapter::Digitalocean do
         expect { instance.start_droplet(droplet_id) }
           .not_to raise_error
         expect(DoSnapshot.logger.buffer)
-          .to include 'Droplet is still running.'
+          .to include "Droplet #{droplet_id} is still running. Skipping."
 
         expect(a_request(:get, droplet_url))
           .to have_been_made
@@ -101,7 +101,7 @@ describe DoSnapshot::Adapter::Digitalocean do
         stub_droplet_fail(droplet_id)
 
         expect { instance.start_droplet(droplet_id) }
-          .to raise_error
+          .to raise_error(DoSnapshot::DropletFindError)
 
         expect(a_request(:get, droplet_url))
           .to have_been_made
@@ -109,23 +109,27 @@ describe DoSnapshot::Adapter::Digitalocean do
     end
 
     describe '.stop_droplet' do
-      it 'with event' do
+      it 'with success' do
         stub_event_done(event_id)
         stub_droplet_stop(droplet_id)
+        stub_droplet_inactive(droplet_id)
 
         instance.stop_droplet(droplet_id)
 
         expect(a_request(:get, droplet_stop_url))
           .to have_been_made
-        expect(a_request(:get, event_find_url))
+        expect(a_request(:get, droplet_url))
           .to have_been_made
       end
 
       it 'with error' do
         stub_droplet_stop_fail(droplet_id)
+        stub_droplet(droplet_id)
 
+        instance.timeout = 1
         expect { instance.stop_droplet(droplet_id) }
-          .to raise_error
+          .to raise_error(DoSnapshot::DropletShutdownError)
+        instance.timeout = timeout
         expect(DoSnapshot.logger.buffer)
           .to include 'Droplet id: 100823 is Failed to Power Off.'
 
@@ -152,7 +156,7 @@ describe DoSnapshot::Adapter::Digitalocean do
         stub_droplet_snapshot_fail(droplet_id, snapshot_name)
 
         expect { instance.create_snapshot(droplet_id, snapshot_name) }
-          .to raise_error
+          .to raise_error(DoSnapshot::SnapshotCreateError)
 
         expect(a_request(:get, droplet_snapshot_url))
           .to have_been_made
@@ -163,7 +167,7 @@ describe DoSnapshot::Adapter::Digitalocean do
         stub_event_fail(event_id)
 
         expect { instance.create_snapshot(droplet_id, snapshot_name) }
-          .to raise_error
+          .to raise_error(DoSnapshot::EventError)
 
         expect(a_request(:get, droplet_snapshot_url))
           .to have_been_made
@@ -172,6 +176,22 @@ describe DoSnapshot::Adapter::Digitalocean do
       end
     end
 
+    describe '.inactive?' do
+      it 'when inactive' do
+        stub_droplet_inactive(droplet_id)
+
+        expect(instance.inactive?(droplet_id))
+          .to be_truthy
+      end
+
+      it 'when active' do
+        stub_droplet(droplet_id)
+
+        expect(instance.inactive?(droplet_id))
+          .to be_falsey
+      end
+    end
+
     describe '.cleanup_snapshots' do
       it 'with success' do
         stub_droplet(droplet_id)

data/spec/do_snapshot/adapter/digitalocean_v2_spec.rb

@@ -1,7 +1,7 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot::Adapter::DigitaloceanV2 do
+RSpec.describe DoSnapshot::Adapter::DigitaloceanV2 do
   include_context 'spec'
   include_context 'api_v2_helpers'
 
@@ -42,7 +42,7 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
         stub_droplet_fail(droplet_id)
 
         expect { instance.droplet(droplet_id) }
-          .to raise_error
+          .to raise_error(DoSnapshot::DropletFindError)
         expect(DoSnapshot.logger.buffer)
           .to include 'Droplet Not Found'
 
@@ -64,7 +64,7 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
       it 'with error' do
         stub_droplets_fail
 
-        expect { instance.droplets }.to raise_error
+        expect { instance.droplets }.to raise_error(DoSnapshot::DropletListError)
         expect(DoSnapshot.logger.buffer)
           .to include 'Droplet Listing is failed to retrieve'
 
@@ -93,7 +93,7 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
         expect { instance.start_droplet(droplet_id) }
           .not_to raise_error
         expect(DoSnapshot.logger.buffer)
-          .to include 'Droplet is still running.'
+          .to include "Droplet #{droplet_id} is still running. Skipping."
 
         expect(a_request(:get, droplet_url))
           .to have_been_made
@@ -103,7 +103,7 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
         stub_droplet_fail(droplet_id)
 
         expect { instance.start_droplet(droplet_id) }
-          .to raise_error
+          .to raise_error(DoSnapshot::DropletFindError)
 
         expect(a_request(:get, droplet_url))
           .to have_been_made
@@ -111,23 +111,26 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
     end
 
     describe '.stop_droplet' do
-      it 'with event' do
+      it 'with success' do
         stub_event_done(event_id)
         stub_droplet_stop(droplet_id)
+        stub_droplet_inactive(droplet_id)
 
         instance.stop_droplet(droplet_id)
 
         expect(a_request(:post, droplet_stop_url))
           .to have_been_made
-        expect(a_request(:get, action_find_url))
+        expect(a_request(:get, droplet_url))
           .to have_been_made
       end
 
       it 'with error' do
         stub_droplet_stop_fail(droplet_id)
-
+        stub_droplet(droplet_id)
+        instance.timeout = 1
         expect { instance.stop_droplet(droplet_id) }
-          .to raise_error
+          .to raise_error(DoSnapshot::DropletShutdownError)
+        instance.timeout = timeout
         expect(DoSnapshot.logger.buffer)
           .to include 'Droplet id: 100823 is Failed to Power Off.'
 
@@ -154,7 +157,7 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
         stub_droplet_snapshot_fail(droplet_id, snapshot_name)
 
         expect { instance.create_snapshot(droplet_id, snapshot_name) }
-          .to raise_error
+          .to raise_error(DoSnapshot::SnapshotCreateError)
 
         expect(a_request(:post, droplet_snapshot_url))
           .to have_been_made
@@ -165,7 +168,7 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
         stub_event_fail(event_id)
 
         expect { instance.create_snapshot(droplet_id, snapshot_name) }
-          .to raise_error
+          .to raise_error(DoSnapshot::EventError)
 
         expect(a_request(:post, droplet_snapshot_url))
           .to have_been_made
@@ -174,6 +177,22 @@ describe DoSnapshot::Adapter::DigitaloceanV2 do
       end
     end
 
+    describe '.inactive?' do
+      it 'when inactive' do
+        stub_droplet_inactive(droplet_id)
+
+        expect(instance.inactive?(droplet_id))
+          .to be_truthy
+      end
+
+      it 'when active' do
+        stub_droplet(droplet_id)
+
+        expect(instance.inactive?(droplet_id))
+          .to be_falsey
+      end
+    end
+
     describe '.cleanup_snapshots' do
       it 'with success' do
         stub_droplet(droplet_id)

data/spec/do_snapshot/cli_spec.rb

@@ -1,7 +1,7 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot::CLI do
+RSpec.describe DoSnapshot::CLI do
   include_context 'spec'
   include_context 'api_v1_helpers'
 

data/spec/do_snapshot/command_spec.rb

@@ -1,137 +1,179 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot::Command do
+RSpec.describe DoSnapshot::Command do
   include_context 'spec'
   include_context 'uri_helpers'
-  include_context 'api_v1_helpers'
 
   subject(:cmd)     { DoSnapshot::Command.new }
   subject(:log)     { DoSnapshot::Log }
 
-  describe '.snap' do
-    context 'when success' do
-      it 'sends message' do
-        expect { snap_runner }
-          .not_to raise_error
-        expect(DoSnapshot.logger.buffer)
-          .to include 'All operations has been finished.'
+  describe 'V2' do
+    include_context 'api_v2_helpers'
+  end
+
+  describe 'V1' do
+    include_context 'api_v1_helpers'
+
+    describe '.snap' do
+      context 'when success' do
+        it 'sends message' do
+          stub_droplet_inactive(droplet_id)
+          expect { snap_runner }
+            .not_to raise_error
+          expect(DoSnapshot.logger.buffer)
+            .to include 'All operations has been finished.'
+        end
       end
-    end
 
-    context 'when snapshot not cleanup' do
-      it 'sends message' do
-        stub_image_destroy_fail(image_id)
-        stub_image_destroy_fail(image_id2)
+      context 'when snapshot not cleanup' do
+        it 'sends message' do
+          stub_droplet_inactive(droplet_id)
+          stub_image_destroy_fail(image_id)
+          stub_image_destroy_fail(image_id2)
 
-        expect { snap_runner }
-          .not_to raise_error
+          expect { snap_runner }
+            .not_to raise_error
+        end
       end
-    end
 
-    context 'when droplet not found' do
-      it 'raised by exception' do
-        stub_droplet_fail(droplet_id)
+      context 'when droplet not found' do
+        it 'raised by exception' do
+          stub_droplet_fail(droplet_id)
 
-        expect { snap_runner }
-          .to raise_error(DoSnapshot::DropletFindError)
+          expect { snap_runner }
+            .to raise_error(DoSnapshot::DropletFindError)
+        end
       end
-    end
 
-    context 'when failed to list droplets' do
-      it 'raised with error' do
-        stub_droplets_fail
+      context 'when failed to list droplets' do
+        it 'raised with error' do
+          stub_droplets_fail
 
-        expect { snap_runner }
-          .to raise_error(DoSnapshot::DropletListError)
+          expect { snap_runner }
+            .to raise_error(DoSnapshot::DropletListError)
+        end
       end
-    end
 
-    context 'when droplet failed for shutdown' do
-      it 'raised with error' do
-        stub_droplet_stop_fail(droplet_id)
+      context 'when droplet failed for shutdown' do
+        it 'raised with error' do
+          stub_droplet_stop_fail(droplet_id)
 
-        expect { snap_runner }
-          .to raise_error(DoSnapshot::DropletShutdownError)
+          expect { snap_runner }
+            .not_to raise_error
+          expect(DoSnapshot.logger.buffer)
+            .to include "Droplet id: #{droplet_id} is Failed to Power Off."
+        end
       end
-    end
 
-    context 'when no snapshot created' do
-      it 'raised with error' do
-        stub_droplet_snapshot_fail(droplet_id, snapshot_name)
+      context 'when no snapshot created' do
+        it 'raised with error' do
+          stub_droplet_inactive(droplet_id)
+          stub_droplet_snapshot_fail(droplet_id, snapshot_name)
 
-        expect { snap_runner }
-          .to raise_error(DoSnapshot::SnapshotCreateError)
+          expect { snap_runner }
+            .to raise_error(DoSnapshot::SnapshotCreateError)
+        end
       end
-    end
-  end
 
-  describe '.stop_droplet' do
-    it 'when raised with error' do
-      stub_droplet_stop_fail(droplet_id)
-      load_options
-      droplet = cmd.api.droplet droplet_id
-      expect { cmd.stop_droplet(droplet) }
-        .to raise_error(DoSnapshot::DropletShutdownError)
-    end
+      context 'when droplet not stopped' do
+        it 'skipped droplet' do
+          stub_droplet_stop_fail(droplet_id)
 
-    it 'when stopped' do
-      stub_droplet_stop(droplet_id)
-      load_options
-      droplet = cmd.api.droplet droplet_id
-      expect { cmd.stop_droplet(droplet) }
-        .not_to raise_error
+          expect { snap_runner }
+            .not_to raise_error
+          expect(DoSnapshot.logger.buffer)
+            .to include "Droplet id: #{droplet_id} is Failed to Power Off."
+        end
+      end
     end
-  end
 
-  describe '.create_snapshot' do
-    it 'when raised with error' do
-      stub_droplet_snapshot_fail(droplet_id, snapshot_name)
-      load_options
-      droplet = cmd.api.droplet droplet_id
-      expect { cmd.create_snapshot(droplet) }
-        .to raise_error(DoSnapshot::SnapshotCreateError)
-    end
+    describe '.stop_droplet' do
+      it 'when raised with error' do
+        stub_droplet_stop_fail(droplet_id)
+        load_options
+        droplet = cmd.api.droplet droplet_id
+        expect { cmd.stop_droplet(droplet) }
+          .not_to raise_error
+        expect(cmd.stop_droplet(droplet))
+          .to be_falsey
+      end
 
-    it 'when snapshot is created' do
-      stub_droplet_snapshot(droplet_id, snapshot_name)
-      load_options
-      droplet = cmd.api.droplet droplet_id
-      expect { cmd.create_snapshot(droplet) }
-        .not_to raise_error
+      it 'when stopped' do
+        stub_droplet_inactive(droplet_id)
+        stub_droplet_stop(droplet_id)
+        load_options
+        droplet = cmd.api.droplet droplet_id
+        expect { cmd.stop_droplet(droplet) }
+          .not_to raise_error
+        expect(cmd.stop_droplet(droplet))
+          .to be_truthy
+      end
     end
-  end
 
-  describe '.fail_power_off' do
-    it 'when success' do
-      stub_droplet_inactive(droplet_id)
+    describe '.create_snapshot' do
+      it 'when raised with error' do
+        stub_droplet_inactive(droplet_id)
+        stub_droplet_snapshot_fail(droplet_id, snapshot_name)
+        load_options
+        droplet = cmd.api.droplet droplet_id
+        expect { cmd.create_snapshot(droplet) }
+          .to raise_error(DoSnapshot::SnapshotCreateError)
+      end
 
-      expect { cmd.fail_power_off(DoSnapshot::DropletShutdownError.new(droplet_id)) }
-        .not_to raise_error
-      expect(DoSnapshot.logger.buffer)
-        .to include 'Power On has been requested.'
+      it 'when snapshot is created' do
+        stub_droplet_inactive(droplet_id)
+        stub_droplet_snapshot(droplet_id, snapshot_name)
+        load_options
+        droplet = cmd.api.droplet droplet_id
+        cmd.create_snapshot(droplet)
+        expect { cmd.create_snapshot(droplet) }
+          .not_to raise_error
+      end
+
+      it 'when droplet is running' do
+        stub_droplet(droplet_id)
+        load_options
+        droplet = cmd.api.droplet droplet_id
+        cmd.create_snapshot(droplet)
+        expect { cmd.create_snapshot(droplet) }
+          .not_to raise_error
+        expect(DoSnapshot.logger.buffer)
+          .to include "Droplet id: #{droplet_id} must be Powered Off!"
+      end
     end
 
-    it 'with request error' do
-      stub_droplet_fail(droplet_id)
+    describe '.fail_power_off' do
+      it 'when success' do
+        stub_droplet_inactive(droplet_id)
 
-      expect { cmd.fail_power_off(DoSnapshot::DropletShutdownError.new(droplet_id)) }
-        .to raise_error
-      expect(DoSnapshot.logger.buffer)
-        .to include 'Droplet id: 100823 is Failed to Power Off.'
-      expect(DoSnapshot.logger.buffer)
-        .to include 'Droplet Not Found'
-    end
+        expect { cmd.fail_power_off(DoSnapshot::DropletShutdownError.new(droplet_id)) }
+          .not_to raise_error
+        expect(DoSnapshot.logger.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(DoSnapshot::DropletFindError)
+        expect(DoSnapshot.logger.buffer)
+          .to include 'Droplet id: 100823 is Failed to Power Off.'
+        expect(DoSnapshot.logger.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)
+      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(DoSnapshot.logger.buffer)
-        .to include 'Power On failed to request.'
+        expect { cmd.fail_power_off(DoSnapshot::DropletShutdownError.new(droplet_id)) }
+          .not_to raise_error
+        expect(DoSnapshot.logger.buffer)
+          .to include 'Power On failed to request.'
+      end
     end
   end
 
@@ -143,7 +185,7 @@ describe DoSnapshot::Command do
   end
 
   def load_options(options = nil)
-    options ||= default_options
+    options ||= default_options.merge(protocol: 1)
     cmd.load_options(options, [:log, :mail, :smtp, :trace, :digital_ocean_client_id, :digital_ocean_api_key])
   end
 

data/spec/do_snapshot/configuration_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-describe DoSnapshot::Configuration do
+RSpec.describe DoSnapshot::Configuration do
   subject(:cli) { described_class }
 
   it { expect(cli.new).to respond_to(:logger) }

data/spec/do_snapshot/log_spec.rb

@@ -1,7 +1,7 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot::Log do
+RSpec.describe DoSnapshot::Log do
   include_context 'spec'
 
   subject(:log) { described_class }

data/spec/do_snapshot/mail_spec.rb

@@ -1,7 +1,7 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot::Mail do
+RSpec.describe DoSnapshot::Mail do
   include_context 'spec'
 
   subject(:mail) { described_class }

data/spec/do_snapshot/runner_spec.rb

@@ -134,7 +134,7 @@ RSpec.describe DoSnapshot::Runner, type: :aruba do
           hash_attribute_eq
 
           expect(last_command).to have_exit_status(0)
-          expect(all_stdout).to include(t_sending_email)
+          expect(all_stdout).not_to include(t_sending_email)
         end
 
         it 'with smtp' do

data/spec/do_snapshot_spec.rb

@@ -1,7 +1,7 @@
 # -*- encoding : utf-8 -*-
 require 'spec_helper'
 
-describe DoSnapshot do
+RSpec.describe DoSnapshot do
   include_context 'spec'
 
   describe DoSnapshot::DropletFindError do
@@ -14,6 +14,26 @@ describe DoSnapshot do
     end
   end
 
+  describe DoSnapshot::DropletPowerError do
+    subject(:error) { described_class }
+
+    it 'should work' do
+      error.new(droplet_id)
+      expect(DoSnapshot.logger.buffer)
+        .to include "Droplet id: #{droplet_id} must be Powered Off!"
+    end
+  end
+
+  describe DoSnapshot::EventError do
+    subject(:error) { described_class }
+
+    it 'should work' do
+      error.new(event_id)
+      expect(DoSnapshot.logger.buffer)
+        .to include "Event id: #{event_id} is failed!"
+    end
+  end
+
   describe DoSnapshot::DropletListError do
     subject(:error) { described_class }
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: do_snapshot
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.4
 platform: ruby
 authors:
 - Alexander Merkulov
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-19 00:00:00.000000000 Z
+date: 2015-07-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 4.0.0
+        version: '3.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 4.0.0
+        version: '3.2'
 - !ruby/object:Gem::Dependency
   name: digitalocean_c
   requirement: !ruby/object:Gem::Requirement