X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/45487e1ff8e7d334792440a2b5b3b1067269bfaf..b87983c2baf55d437c09e668188454d01b5f20b0:/test/factories/user_blocks.rb

diff --git a/test/factories/user_blocks.rb b/test/factories/user_blocks.rb
index fefbba45a..8bdd6875e 100644
--- a/test/factories/user_blocks.rb
+++ b/test/factories/user_blocks.rb
@@ -2,20 +2,30 @@ FactoryBot.define do
   factory :user_block do
     sequence(:reason) { |n| "User Block #{n}" }
     ends_at { Time.now.utc + 1.day }
+    deactivates_at { ends_at }
 
     user
-    association :creator, :factory => :moderator_user
+    creator :factory => :moderator_user
+
+    trait :zero_hour do
+      now = Time.now.utc
+      created_at { now }
+      ends_at { now }
+    end
 
     trait :needs_view do
       needs_view { true }
+      deactivates_at { nil }
     end
 
     trait :expired do
+      created_at { Time.now.utc - 2.days }
       ends_at { Time.now.utc - 1.day }
     end
 
     trait :revoked do
-      association :revoker, :factory => :moderator_user
+      expired
+      revoker :factory => :moderator_user
     end
   end
 end