preserve env vars during rollbacks

Author: brianwarsing

app/models/shipit/deploy.rb

@@ -117,7 +117,7 @@ def trigger_rollback(user = AnonymousUser.new, env: nil, force: false, lock: tru
     end
 
     # Rolls the stack back to the most recent **previous** successful deploy
-    def trigger_revert(force: false, rollback_to: nil)
+    def trigger_revert(force: false, rollback_to: nil, env: nil)
       previous_successful_commit = rollback_to&.until_commit || commit_to_rollback_to
 
       rollback = Rollback.create!(
@@ -126,6 +126,7 @@ def trigger_revert(force: false, rollback_to: nil)
         parent_id: id,
         since_commit: until_commit,
         until_commit: previous_successful_commit,
+        env: env || self.env,
         allow_concurrency: force
       )
 
@@ -301,7 +302,11 @@ def trigger_revert_if_required
       return unless rollback_once_aborted?
       return unless supports_rollback?
 
-      trigger_revert(rollback_to: rollback_once_aborted_to)
+      if rollback_once_aborted_to
+        rollback_once_aborted_to.trigger_rollback(aborted_by, env: env, force: true)
+      else
+        trigger_revert(force: true, env: env)
+      end
     end
 
     def default_since_commit_id

test/models/deploys_test.rb

@@ -437,27 +437,35 @@ def setup
       @deploy = shipit_deploys(:shipit_running)
       @deploy.ping
       @deploy.pid = 42
+      @deploy.env = { 'CUSTOM_VAR' => 'value' }
       @deploy.abort!(rollback_once_aborted: true, aborted_by: @user)
 
       assert_difference -> { @stack.rollbacks.count }, 1 do
         assert_enqueued_with(job: PerformTaskJob) do
           @deploy.aborted!
         end
       end
+
+      rollback = @stack.rollbacks.last
+      assert_equal({ 'CUSTOM_VAR' => 'value' }, rollback.env)
     end
 
     test "transitioning to aborted schedule a rollback to the designated deploy if set" do
       @rollback_to = shipit_deploys(:shipit_pending)
       @deploy = shipit_deploys(:shipit_running)
       @deploy.ping
       @deploy.pid = 42
+      @deploy.env = { 'CUSTOM_VAR' => 'value' }
       @deploy.abort!(rollback_once_aborted: true, rollback_once_aborted_to: @rollback_to, aborted_by: @user)
 
       assert_difference -> { @stack.rollbacks.count }, 1 do
         assert_enqueued_with(job: PerformTaskJob) do
           @deploy.aborted!
         end
       end
+
+      rollback = @stack.rollbacks.last
+      assert_equal({ 'CUSTOM_VAR' => 'value' }, rollback.env)
     end
 
     test "transitioning to aborted locks the stack if a rollback is scheduled" do