Can a merge request transition to a given state? Added a method which returns whether this is allowed and another method that lets you wrap a block to be executed if allowed.

Added another merge request fixture that allows the state change.

1 files changed, 21 insertions, 1 deletions

diff --git a/app/models/merge_request.rb b/app/models/merge_request.rb
index 6c6d304..e5e75d4 100644
--- a/app/models/merge_request.rb
+++ b/app/models/merge_request.rb
@@ -54,7 +54,11 @@ class MergeRequest < ActiveRecord::Base
   end
   
   def status_string
-    self.class.statuses.invert[status].downcase
+    self.class.status_string(status)
+  end
+  
+  def self.status_string(status_code)
+    statuses.invert[status_code.to_i].downcase
   end
   
   def open?
@@ -72,6 +76,22 @@ class MergeRequest < ActiveRecord::Base
   def pending_acceptance_of_terms?
     status == STATUS_PENDING_ACCEPTANCE_OF_TERMS
   end
+
+  def can_transition_to?(new_state)
+    if status == STATUS_OPEN
+      return [STATUS_MERGED, STATUS_REJECTED].include?(new_state)
+    else
+      return status == STATUS_PENDING_ACCEPTANCE_OF_TERMS && new_state == STATUS_OPEN
+    end
+  end
+  
+  def transition_to(status)
+    if can_transition_to?(status)
+      self.status = status
+      yield
+      return true
+    end
+  end
   
   def source_branch
     super || "master"