Improve external architecture

CE mirror of 4f17c7b2c30188302e6a73421acbf5a09fb2c823

1 files changed, 4 insertions, 4 deletions

diff --git a/app/models/blob_viewer/gitlab_ci_yml.rb b/app/models/blob_viewer/gitlab_ci_yml.rb
index f5e4367fc09..54462d01b66 100644
--- a/app/models/blob_viewer/gitlab_ci_yml.rb
+++ b/app/models/blob_viewer/gitlab_ci_yml.rb
@@ -10,16 +10,16 @@ module BlobViewer
     self.file_types = %i(gitlab_ci)
     self.binary = false
 
-    def validation_message
+    def validation_message(project, ref)
       return @validation_message if defined?(@validation_message)
 
       prepare!
 
-      @validation_message = Gitlab::Ci::YamlProcessor.validation_message(blob.data, project)
+      @validation_message = Gitlab::Ci::YamlProcessor.validation_message(blob.data, { project: project, branch_name: ref })
     end
 
-    def valid?
-      validation_message.blank?
+    def valid?(project, ref)
+      validation_message(project, ref).blank?
     end
   end
 end