1 files changed, 6 insertions, 0 deletions
diff --git a/spec/factories/broadcast_messages.rb b/spec/factories/broadcast_messages.rb
index ed6e267e7c4..fa8d255ae79 100644
--- a/spec/factories/broadcast_messages.rb
+++ b/spec/factories/broadcast_messages.rb
@@ -6,6 +6,8 @@ FactoryBot.define do
     starts_at { 1.day.ago }
     ends_at { 1.day.from_now }
 
+    broadcast_type { :banner }
+
     trait :expired do
       starts_at { 5.days.ago }
       ends_at { 3.days.ago }
@@ -15,5 +17,9 @@ FactoryBot.define do
       starts_at { 5.days.from_now }
       ends_at { 6.days.from_now }
     end
+
+    trait :notification do
+      broadcast_type { :notification }
+    end
   end
 end