aptible-cli 0.7.2 → 0.7.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9eee8bccf78fd7a0814d5fb48e92c9c31aaef356
-  data.tar.gz: 80d7cf361bba1d14c1c50ab1f832de3c46533e72
+  metadata.gz: ed6d33fb4df228f78616e5a2a612584c1595fbce
+  data.tar.gz: 4795a89a24e7108ed2ee2c8eb590d331fbd4ecc6
 SHA512:
-  metadata.gz: 6c51adcf6c82f63b5129674a82070235d6ab079c7ed91fc8b02771ba81f8e243a459d39cf80135687797a6576320942a7ec767a5bec78bfc5155bfacdfca5caf
-  data.tar.gz: 2c5ccc49276560307c75d5dff6b6504d87b6f77a4bf9b820910c37900b65059e341ef2676e888870ffca0dbc9ae503ad82d328be7d93f16866b45605330cf78a
+  metadata.gz: 7d3e19769ee760b39e9d0fe388fdfa2a8c2dac97b78678758ecaacc467989982a34c993f9d606d09388053a8c7a3c1552b693896945ac596cc668896fc93325c
+  data.tar.gz: 77b04483d2f82409cbc2c03ba05e0fa66b7041bfaf72646335064a305ebaa2ed3f65c9128d7345f8e76c6a726a3313e1ad690721bef5f804ffffd6a77dbd7529

data/lib/aptible/cli/subcommands/restart.rb

@@ -8,10 +8,34 @@ module Aptible
             include Helpers::App
 
             desc 'restart', 'Restart all services associated with an app'
+            option :simulate_oom,
+                   type: :boolean,
+                   desc: 'Add this flag to simulate an OOM restart and test ' \
+                         "your app's response (not recommended on production " \
+                         'apps).'
+            option :force,
+                   type: :boolean,
+                   desc: 'Add this flag to use --simulate-oom in a ' \
+                         'production environment, which is not allowed by ' \
+                         'default.'
             app_options
             def restart
               app = ensure_app(options)
-              operation = app.create_operation!(type: 'restart')
+              type = 'restart'
+
+              if options[:simulate_oom]
+                type = 'captain_comeback_restart'
+
+                if app.account.type == 'production' && !options[:force]
+                  e = 'This operation is designed for test purposes only, ' \
+                      "but #{app.handle} is deployed in a production " \
+                      'environment. Are you sure you want to do this? If ' \
+                      'so, use the --force flag.'
+                  fail Thor::Error, e
+                end
+              end
+
+              operation = app.create_operation!(type: type)
               puts 'Restarting app...'
               attach_to_operation_logs(operation)
             end

data/lib/aptible/cli/version.rb

@@ -1,5 +1,5 @@
 module Aptible
   module CLI
-    VERSION = '0.7.2'.freeze
+    VERSION = '0.7.3'.freeze
   end
 end

data/spec/aptible/cli/subcommands/restart_spec.rb

@@ -0,0 +1,53 @@
+require 'spec_helper'
+
+describe Aptible::CLI::Agent do
+  let(:app) { Fabricate(:app) }
+  let(:operation) { Fabricate(:operation, resource: app) }
+  before { allow(subject).to receive(:ensure_app).and_return(app) }
+
+  describe '#restart' do
+    it 'restarts the app' do
+      expect(app).to receive(:create_operation!).with(type: 'restart')
+        .and_return(operation)
+      expect(subject).to receive(:attach_to_operation_logs).with(operation)
+
+      subject.send('restart')
+    end
+
+    it 'does not require the --force flag for a regular restart' do
+      app.account.type = 'production'
+      expect(app).to receive(:create_operation!)
+      expect(subject).to receive(:attach_to_operation_logs)
+
+      subject.send('restart')
+    end
+
+    it 'uses captain_comeback_restart if --simulate-oom is passed' do
+      subject.options = { simulate_oom: true }
+      expect(app).to receive(:create_operation!)
+        .with(type: 'captain_comeback_restart')
+        .and_return(operation)
+      expect(subject).to receive(:attach_to_operation_logs).with(operation)
+
+      subject.send('restart')
+    end
+
+    it 'fails a CC restart if the --force flag is not passed for a prod app' do
+      subject.options = { simulate_oom: true }
+      app.account.type = 'production'
+      expect(app).not_to receive(:create_operation!)
+      expect(subject).not_to receive(:attach_to_operation_logs)
+
+      expect { subject.send('restart') }.to raise_error(/are you sure/i)
+    end
+
+    it 'creates a CC restart if the --force flag is passed for a prod app' do
+      subject.options = { simulate_oom: true, force: true }
+      app.account.type = 'production'
+      expect(app).to receive(:create_operation!)
+      expect(subject).to receive(:attach_to_operation_logs)
+
+      subject.send('restart')
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aptible-cli
 version: !ruby/object:Gem::Version
-  version: 0.7.2
+  version: 0.7.3
 platform: ruby
 authors:
 - Frank Macreery
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-09-12 00:00:00.000000000 Z
+date: 2016-09-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aptible-api
@@ -256,6 +256,7 @@ files:
 - spec/aptible/cli/subcommands/domains_spec.rb
 - spec/aptible/cli/subcommands/logs_spec.rb
 - spec/aptible/cli/subcommands/ps_spec.rb
+- spec/aptible/cli/subcommands/restart_spec.rb
 - spec/fabricators/account_fabricator.rb
 - spec/fabricators/app_fabricator.rb
 - spec/fabricators/backup_fabricator.rb
@@ -304,6 +305,7 @@ test_files:
 - spec/aptible/cli/subcommands/domains_spec.rb
 - spec/aptible/cli/subcommands/logs_spec.rb
 - spec/aptible/cli/subcommands/ps_spec.rb
+- spec/aptible/cli/subcommands/restart_spec.rb
 - spec/fabricators/account_fabricator.rb
 - spec/fabricators/app_fabricator.rb
 - spec/fabricators/backup_fabricator.rb