1 files changed, 8 insertions, 0 deletions
diff --git a/spec/factories/ci/builds.rb b/spec/factories/ci/builds.rb
index 5bba1dec7db..f8922275860 100644
--- a/spec/factories/ci/builds.rb
+++ b/spec/factories/ci/builds.rb
@@ -226,5 +226,13 @@ FactoryGirl.define do
       status 'created'
       self.when 'manual'
     end
+
+    trait(:protected) do
+      protected true
+    end
+
+    trait(:unprotected) do
+      protected false
+    end
   end
 end