Use before_save :set_protected

1 files changed, 5 insertions, 1 deletions

diff --git a/app/models/ci/build.rb b/app/models/ci/build.rb
index ba3156154ac..11390d3aa0d 100644
--- a/app/models/ci/build.rb
+++ b/app/models/ci/build.rb
@@ -27,7 +27,6 @@ module Ci
 
     validates :coverage, numericality: true, allow_blank: true
     validates :ref, presence: true
-    validates :protected, inclusion: { in: [true, false], unless: :importing? }, on: :create
 
     scope :unstarted, ->() { where(runner_id: nil) }
     scope :ignore_failures, ->() { where(allow_failure: false) }
@@ -47,6 +46,7 @@ module Ci
 
     before_save :update_artifacts_size, if: :artifacts_file_changed?
     before_save :ensure_token
+    before_save :set_protected
     before_destroy { unscoped_project }
 
     after_create do |build|
@@ -461,6 +461,10 @@ module Ci
                             end
     end
 
+    def set_protected
+      self.protected = pipeline.protected
+    end
+
     def erase_trace!
       trace.erase!
     end