do_snapshot 0.0.6 → 0.0.7

data/lib/do_snapshot/core_ext/hash.rb

@@ -1,3 +1,4 @@
+# -*- encoding : utf-8 -*-
 # Rails symbolize keys methods for Hash
 #
 class Hash
@@ -25,12 +26,12 @@ class Hash
   end
 
   def symbolize_keys
-    transform_keys { |key| key.to_sym rescue key }
+    transform_keys { |key| key.to_sym rescue key } # rubocop:disable Style/RescueModifier
   end
 
   # Destructively convert all keys to symbols, as long as they respond
   # to +to_sym+. Same as +symbolize_keys+, but modifies +self+.
   def symbolize_keys!
-    transform_keys! { |key| key.to_sym rescue key }
+    transform_keys! { |key| key.to_sym rescue key } # rubocop:disable Style/RescueModifier
   end
 end

data/lib/do_snapshot/log.rb

@@ -0,0 +1,59 @@
+# -*- encoding : utf-8 -*-
+require 'logger'
+
+module DoSnapshot
+  # Shared logger
+  #
+  class Log
+    class << self
+      attr_accessor :logger, :shell, :quiet, :verbose
+      attr_writer :buffer
+
+      def buffer
+        @buffer ||= %w()
+      end
+
+      def info(message)
+        log :info, message
+      end
+
+      def warning(message)
+        log :warn, message
+      end
+
+      def error(message)
+        log :error, message
+      end
+
+      def debug(message)
+        log :debug, message
+      end
+
+      protected
+
+      def log(type, message)
+        buffer << message
+        logger.send(type, message) if logger
+
+        say message, color(type) unless type == :debug && !verbose
+      end
+
+      def say(message, color)
+        shell.say message, color if shell
+      end
+
+      def color(type)
+        case type
+        when :debug
+          :white
+        when :error
+          :red
+        when :warn
+          :yellow
+        else
+          :green
+        end
+      end
+    end
+  end
+end

data/lib/do_snapshot/mail.rb

@@ -0,0 +1,70 @@
+# -*- encoding : utf-8 -*-
+require 'date'
+require 'pony'
+require 'do_snapshot/core_ext/hash'
+
+module DoSnapshot
+  # Shared mailer.
+  #
+  class Mail
+    class << self
+      attr_accessor :opts
+      attr_writer :smtp, :opts_default, :smtp_default
+
+      def smtp
+        @smtp ||= {}
+      end
+
+      # Sending message via Hash params.
+      #
+      # Options:: --mail to:mail@somehost.com from:from@host.com --smtp address:smtp.gmail.com user_name:someuser password:somepassword
+      #
+      def notify
+        return unless opts
+
+        opts.symbolize_keys!
+        smtp.symbolize_keys!
+
+        opts_setup
+        smtp_setup
+
+        Log.debug 'Sending e-mail notification.'
+        # Look into your inbox :)
+        Pony.mail(opts)
+      end
+
+      protected
+
+      def opts_default
+        @opts_default ||= {
+          subject: 'Digital Ocean: maximum snapshots is reached.',
+          body: "Please cleanup your Digital Ocean account.\nSnapshot maximum is reached.",
+          from: 'noreply@someonelse.com',
+          to: 'to@someonelse.com',
+          via: :smtp
+        }
+      end
+
+      def smtp_default
+        @smtp_default ||= {
+          domain: 'localhost.localdomain',
+          port: '25'
+        }
+      end
+
+      def opts_setup
+        opts_default.each_pair do |key, value|
+          opts[key] = value unless opts.include? key
+        end
+        opts[:body]    = "#{opts[:body]}\n\nTrace: #{DateTime.now}\n#{Log.buffer.join("\n")}"
+      end
+
+      def smtp_setup
+        smtp_default.each_pair do |key, value|
+          smtp[key] = value unless smtp.include? key
+        end
+        opts[:via_options] = smtp
+      end
+    end
+  end
+end

data/lib/do_snapshot/version.rb

@@ -1,5 +1,6 @@
+# -*- encoding : utf-8 -*-
 # Current version
 #
 module DoSnapshot
-  VERSION = '0.0.6'
+  VERSION = '0.0.7'
 end

data/spec/do_snapshot/api_spec.rb

@@ -0,0 +1,218 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+describe DoSnapshot::API do
+  include_context 'spec'
+
+  subject(:api) { described_class }
+  subject(:log) { DoSnapshot::Log }
+
+  describe '.initialize' do
+    describe '#delay' do
+      let(:delay) { 5 }
+      let(:instance) { api.new(delay: delay) }
+      it('with custom delay') { expect(instance.delay).to eq delay  }
+    end
+
+    describe '#timeout' do
+      let(:timeout) { 5 }
+      let(:instance) { api.new(timeout: timeout) }
+      it('with custom timeout') { expect(instance.timeout).to eq timeout  }
+    end
+  end
+
+  describe 'droplets' do
+    let(:instance) { api.new(delay: delay, timeout: timeout) }
+    include_context 'uri_helpers'
+
+    describe '.droplet' do
+      it 'with droplet' do
+        stub_droplet(droplet_id)
+
+        instance.droplet(droplet_id)
+
+        expect(a_request(:get, droplet_url))
+          .to have_been_made
+      end
+
+      it 'with error' do
+        stub_droplet_fail(droplet_id)
+
+        expect { instance.droplet(droplet_id) }
+          .to raise_error
+        expect(log.buffer)
+          .to include 'Droplet Not Found'
+
+        expect(a_request(:get, droplet_url))
+          .to have_been_made
+      end
+    end
+
+    describe '.droplets' do
+      it 'with droplets' do
+        stub_droplets
+
+        instance.droplets
+
+        expect(a_request(:get, droplets_uri))
+          .to have_been_made
+      end
+
+      it 'with error' do
+        stub_droplets_fail
+
+        expect { instance.droplets }.to raise_error
+        expect(log.buffer)
+          .to include 'Droplet Listing is failed to retrieve'
+
+        expect(a_request(:get, droplets_uri))
+          .to have_been_made
+      end
+    end
+
+    describe '.start_droplet' do
+      it 'with event' do
+        stub_droplet_inactive(droplet_id)
+        stub_droplet_start(droplet_id)
+
+        instance.start_droplet(droplet_id)
+        expect(log.buffer).to include 'Power On has been requested.'
+
+        expect(a_request(:get, droplet_start_url))
+          .to have_been_made
+        expect(a_request(:get, droplet_url))
+          .to have_been_made
+      end
+
+      it 'with warning message' do
+        stub_droplet(droplet_id)
+
+        expect { instance.start_droplet(droplet_id) }
+          .not_to raise_error
+        expect(log.buffer)
+          .to include 'Droplet is still running.'
+
+        expect(a_request(:get, droplet_url))
+          .to have_been_made
+      end
+
+      it 'with error' do
+        stub_droplet_fail(droplet_id)
+
+        expect { instance.start_droplet(droplet_id) }
+          .to raise_error
+
+        expect(a_request(:get, droplet_url))
+          .to have_been_made
+      end
+    end
+
+    describe '.stop_droplet' do
+      it 'with event' do
+        stub_event_done(event_id)
+        stub_droplet_stop(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))
+          .to have_been_made
+      end
+
+      it 'with error' do
+        stub_droplet_stop_fail(droplet_id)
+
+        expect { instance.stop_droplet(droplet_id) }
+          .to raise_error
+        expect(log.buffer)
+          .to include 'Droplet id: 100823 is Failed to Power Off.'
+
+        expect(a_request(:get, droplet_stop_url))
+          .to have_been_made
+      end
+    end
+
+    describe '.create_snapshot' do
+      it 'with success' do
+        stub_event_done(event_id)
+        stub_droplet_snapshot(droplet_id, snapshot_name)
+
+        expect { instance.create_snapshot(droplet_id, snapshot_name) }
+          .not_to raise_error
+
+        expect(a_request(:get, droplet_snapshot_url))
+          .to have_been_made
+        expect(a_request(:get, event_find_url))
+          .to have_been_made
+      end
+
+      it 'with error' do
+        stub_droplet_snapshot_fail(droplet_id, snapshot_name)
+
+        expect { instance.create_snapshot(droplet_id, snapshot_name) }
+          .to raise_error
+
+        expect(a_request(:get, droplet_snapshot_url))
+          .to have_been_made
+      end
+
+      it 'with event error' do
+        stub_droplet_snapshot(droplet_id, snapshot_name)
+        stub_event_fail(event_id)
+
+        expect { instance.create_snapshot(droplet_id, snapshot_name) }
+          .to raise_error
+
+        expect(a_request(:get, droplet_snapshot_url))
+          .to have_been_made
+        expect(a_request(:get, event_find_url))
+          .to have_been_made
+      end
+    end
+
+    describe '.cleanup_snapshots' do
+      it 'with success' do
+        stub_droplet(droplet_id)
+        stub_image_destroy(image_id)
+        stub_image_destroy(image_id2)
+
+        droplet = instance.droplet(droplet_id)
+        expect { instance.cleanup_snapshots(droplet.droplet, 1) }
+          .not_to raise_error
+        expect(log.buffer)
+          .to include 'Snapshot name: mrcr.ru_2014_07_19 delete requested.'
+
+        expect(a_request(:get, droplet_url))
+          .to have_been_made
+        expect(a_request(:get, image_destroy_url))
+          .to have_been_made
+        expect(a_request(:get, image_destroy2_url))
+          .to have_been_made
+      end
+
+      it 'with warning message' do
+        stub_droplet(droplet_id)
+        stub_image_destroy_fail(image_id)
+        stub_image_destroy_fail(image_id2)
+
+        droplet = instance.droplet(droplet_id)
+        expect { instance.cleanup_snapshots(droplet.droplet, 1) }
+          .not_to raise_error
+        expect(log.buffer)
+          .to include 'Some Message'
+
+        expect(a_request(:get, droplet_url))
+          .to have_been_made
+        expect(a_request(:get, image_destroy_url))
+          .to have_been_made
+        expect(a_request(:get, image_destroy2_url))
+        .to have_been_made
+      end
+    end
+  end
+
+  before(:each) do
+    log.buffer = %w()
+  end
+end

data/spec/do_snapshot/cli_spec.rb

@@ -0,0 +1,173 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+describe DoSnapshot::CLI do
+  include_context 'spec'
+
+  subject(:cli)     { described_class }
+  subject(:command) { DoSnapshot::Command }
+  subject(:api)     { DoSnapshot::API }
+
+  describe '.snap' do
+
+    it 'with exclude' do
+      excluded_droplets = %w( 100824 )
+      stub_all_api(%w(100825 100823))
+      hash_attribute_eq_no_stub(exclude: excluded_droplets, only: %w())
+
+      expect(command.send('exclude'))
+        .to eq excluded_droplets
+    end
+
+    it 'with only' do
+      selected_droplets = %w( 100823 )
+      stub_all_api(selected_droplets)
+      hash_attribute_eq_no_stub(only: selected_droplets)
+
+      expect(command.send('only'))
+        .to eq selected_droplets
+    end
+
+    it 'with 1 delay' do
+      set_api_attribute(delay: 1, timeout: timeout)
+      attribute_eq 'delay', 1
+    end
+
+    it 'with 0 delay' do
+      set_api_attribute(delay: 0, timeout: timeout)
+      attribute_eq 'delay', 0
+    end
+
+    it 'with custom timeout' do
+      set_api_attribute(timeout: 1, delay: delay)
+      attribute_eq 'timeout', 1
+    end
+
+    it 'with keep' do
+      attribute_eq 'keep', 7
+    end
+
+    it 'with quiet' do
+      attribute_eq 'quiet', true
+    end
+
+    it 'with no quiet' do
+      attribute_eq 'quiet', false
+    end
+
+    it 'with stop' do
+      attribute_eq 'stop', true
+    end
+
+    it 'with no stop' do
+      attribute_eq 'stop', false
+    end
+
+    it 'with clean' do
+      attribute_eq 'clean', true
+    end
+
+    it 'with no clean' do
+      attribute_eq 'clean', false
+    end
+
+    it 'with digital ocean credentials' do
+      with_hash_attribute_eq(cli_keys)
+    end
+
+    it 'with no digital ocean credentials' do
+      without_hash_attribute_eq(cli_keys)
+    end
+
+    it 'with mail' do
+      hash_attribute_eq(mail_options)
+    end
+
+    it 'with no mail' do
+      without_hash_attribute_eq(mail_options)
+    end
+
+    it 'with smtp' do
+      hash_attribute_eq(smtp_options)
+    end
+
+    it 'with no smtp' do
+      without_hash_attribute_eq(smtp_options)
+    end
+
+    it 'with log' do
+      hash_attribute_eq(log)
+    end
+
+    it 'with no log' do
+      without_hash_attribute_eq(log)
+    end
+  end
+
+  describe '.version' do
+    it 'shows the correct version' do
+      @cli.options = @cli.options.merge(version: true)
+      @cli.version
+
+      expect($stdout.string.chomp)
+        .to eq("#{DoSnapshot::VERSION}")
+    end
+  end
+
+  describe '.help' do
+    it 'shows a help message' do
+      @cli.help
+      expect($stdout.string)
+        .to match('Commands:')
+    end
+
+    it 'shows a help message for specific commands' do
+      @cli.help 'snap'
+      expect($stdout.string)
+        .to match('Usage:')
+    end
+  end
+
+  def attribute_eq(name, value)
+    stub_all_api
+    options = default_options.merge!(:"#{name}" => value)
+    @cli.options = @cli.options.merge(options)
+    @cli.snap
+
+    expect(command.send(name))
+      .to eq value
+  end
+
+  def hash_attribute_eq(hash)
+    stub_all_api
+    options = default_options.merge!(hash)
+    @cli.options = @cli.options.merge(options)
+    @cli.snap
+  end
+
+  def with_hash_attribute_eq(hash)
+    hash_attribute_eq hash
+    expect(@cli.options)
+      .to include(hash)
+  end
+
+  def without_hash_attribute_eq(hash)
+    hash_attribute_eq({})
+    expect(@cli.options)
+      .not_to include(hash)
+  end
+
+  def hash_attribute_eq_no_stub(hash)
+    options = default_options.merge!(hash)
+    @cli.options = @cli.options.merge(options)
+    @cli.snap
+  end
+
+  def set_api_attribute(options = { delay: delay, timeout: timeout }) # rubocop:disable Style/AccessorMethodName
+    command.send('api=', api.new(options))
+  end
+
+  after(:each) do
+    # stub_cleanup
+  end
+end