[Port semi-linear history from EE] Move migration to CE and `merge_method` method

1 files changed, 13 insertions, 1 deletions

diff --git a/app/models/project.rb b/app/models/project.rb
index 1c3a78ff05a..fe442755619 100644
--- a/app/models/project.rb
+++ b/app/models/project.rb
@@ -1567,13 +1567,25 @@ class Project < ActiveRecord::Base
   def merge_method
     if self.merge_requests_ff_only_enabled
       :ff
+    elsif self.merge_requests_rebase_enabled
+      :rebase_merge
     else
       :merge
     end
   end
 
   def merge_method=(method)
-    self.merge_requests_ff_only_enabled = method.to_s == "ff"
+    case method.to_s
+    when "ff"
+      self.merge_requests_ff_only_enabled = true
+      self.merge_requests_rebase_enabled = true
+    when "rebase_merge"
+      self.merge_requests_ff_only_enabled = false
+      self.merge_requests_rebase_enabled = true
+    when "merge"
+      self.merge_requests_ff_only_enabled = false
+      self.merge_requests_rebase_enabled = false
+    end
   end
 
   def ff_merge_must_be_possible?