suspend certain autoscaling processes before cycling servers

spec/aws-missing-tools/aws-ha-release/aws-ha-release_spec.rb

@@ -19,7 +19,7 @@ describe 'aws-ha-release' do
 
   describe '#initialize' do
     it 'initializes the AWS connection' do
-      as.groups.create 'test_group'
+      as.groups.create opts[:as_group_name]
 
       AWS.should_receive(:config).with(access_key_id: 'testaccesskey', secret_access_key: 'testsecretkey', region: 'test-region')
       AwsHaRelease.new(opts)
@@ -31,4 +31,17 @@ describe 'aws-ha-release' do
       }.should raise_error
     end
   end
+
+  describe '#execute!' do
+    before do
+      as.groups.create opts[:as_group_name]
+      @aws_ha_release = AwsHaRelease.new(opts)
+    end
+
+    it 'suspends certain autoscaling processes' do
+      AWS::FakeAutoScaling::Group.any_instance.should_receive(:suspend_processes)
+          .with('ReplaceUnhealthy', 'AlarmNotification', 'ScheduledActions', 'AZRebalance')
+      @aws_ha_release.execute!
+    end
+  end
 end

spec/support/fake_auto_scaling.rb

@@ -26,6 +26,13 @@ module AWS
 
       def initialize(name)
         @name = name
+        @suspended_processes = {}
+      end
+
+      def suspend_processes(*processes)
+        processes.each do |process|
+          @suspended_processes[process] = 'test'
+        end
       end
     end
   end