maintenance: consolidate maintenance tasks in logical/maintenance.rb.

test/unit/maintenance_test.rb

@@ -0,0 +1,46 @@
+require 'test_helper'
+
+class MaintenanceTest < ActiveSupport::TestCase
+  context "daily maintenance" do
+    setup do
+      # have ApiCacheGenerator save files to a temp dir.
+      @temp_shared_dir_path = "/tmp/#{SecureRandom.uuid}"
+      Danbooru.config.stubs(:shared_dir_path).returns(@temp_shared_dir_path)
+
+      FactoryBot.create(:tag, post_count: 1) # for ApiCacheGenerator
+      FactoryBot.create(:admin_user) # for SuperVoter.init!
+    end
+
+    teardown do
+      FileUtils.rm_rf(@temp_shared_dir_path)
+      Danbooru.config.unstub(:shared_dir_path)
+    end
+
+    should "work" do
+      assert_nothing_raised { Maintenance.daily }
+    end
+
+    should "prune expired posts" do
+      @pending = FactoryBot.create(:post, is_pending: true, created_at: 4.days.ago)
+      @flagged = FactoryBot.create(:post, is_flagged: true, created_at: 4.days.ago)
+
+      Maintenance.daily
+
+      assert(true, @pending.reload.is_deleted)
+      assert(true, @flagged.reload.is_deleted)
+    end
+
+    context "when pruning bans" do
+      should "clear the is_banned flag for users who are no longer banned" do
+        banner = FactoryBot.create(:admin_user)
+        user = FactoryBot.create(:user)
+
+        CurrentUser.as(banner) { FactoryBot.create(:ban, user: user, banner: banner, duration: 1) }
+
+        assert_equal(true, user.reload.is_banned)
+        travel_to(2.days.from_now) { Maintenance.daily }
+        assert_equal(false, user.reload.is_banned)
+      end
+    end
+  end
+end