Refactor MergeRequestDiff initialize method

app/models/merge_request_diff.rb

@@ -25,22 +25,18 @@ class MergeRequestDiff < ActiveRecord::Base
   validates :start_commit_sha, presence: true, unless: :importing?
   validates :head_commit_sha,  presence: true, unless: :importing?
 
-  after_initialize :set_diff_range, unless: :importing?
+  after_initialize :initialize_commits_sha, unless: :importing?
   after_create :save_git_content, unless: :importing?
   after_save :keep_around_commits, unless: :importing?
 
-  def set_diff_range
-    if persisted?
-      # Workaround for old MergeRequestDiff object
-      # that does not have head_commit_sha in the database
+  # Those variables are used for collecting commits and diff from git repository.
+  # After object is created those sha are stored in the database.
+  # However some old MergeRequestDiff records don't
+  # have those variables in the database so we try to initialize it
+  def initialize_commits_sha
     self.start_commit_sha ||= merge_request.target_branch_sha
-      self.head_commit_sha  ||= last_commit.sha
+    self.head_commit_sha  ||= last_commit.try(:sha) || merge_request.source_branch_sha
     self.base_commit_sha  ||= find_base_sha
-    else
-      self.start_commit_sha ||= merge_request.target_branch_sha
-      self.head_commit_sha  ||= merge_request.source_branch_sha
-      self.base_commit_sha  ||= find_base_sha
-    end
   end
 
   # Collect information about commits and diff from repository

spec/features/merge_requests/merge_request_versions_spec.rb

@@ -22,7 +22,6 @@ feature 'Merge Request versions', js: true, feature: true do
     before do
       page.within '.mr-version-switch' do
         find('.btn-link').click
-        #find('a', text: '6f6d7e7e').click
         click_link '6f6d7e7e'
       end
     end