X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/7e5cbe87ed37f9cba8224fa4049047d7f981f66a..bf4d8fc2d0e9fa8f828ff9a6b1dc7c6e0a41170b:/test/factories/user_blocks.rb?ds=sidebyside

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