From 159b0e8a22c7e7d2211396525b1e3903039c84cc Mon Sep 17 00:00:00 2001
From: evazion <noizave@gmail.com>
Date: Wed, 11 Apr 2018 20:50:27 -0500
Subject: [PATCH] tests: add daily & weekly maintenance tests.

---
 test/unit/daily_maintenance_test.rb  | 33 ++++++++++++++++++++++++++++
 test/unit/weekly_maintenance_test.rb | 13 +++++++++++
 2 files changed, 46 insertions(+)
 create mode 100644 test/unit/daily_maintenance_test.rb
 create mode 100644 test/unit/weekly_maintenance_test.rb

diff --git a/test/unit/daily_maintenance_test.rb b/test/unit/daily_maintenance_test.rb
new file mode 100644
index 000000000..13198deff
--- /dev/null
+++ b/test/unit/daily_maintenance_test.rb
@@ -0,0 +1,33 @@
+require 'test_helper'
+
+class DailyMaintenanceTest < 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 { DailyMaintenance.new.run }
+    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)
+
+      DailyMaintenance.new.run
+
+      assert(true, @pending.reload.is_deleted)
+      assert(true, @flagged.reload.is_deleted)
+    end
+  end
+end
diff --git a/test/unit/weekly_maintenance_test.rb b/test/unit/weekly_maintenance_test.rb
new file mode 100644
index 000000000..278c7184e
--- /dev/null
+++ b/test/unit/weekly_maintenance_test.rb
@@ -0,0 +1,13 @@
+require 'test_helper'
+
+class WeeklyMaintenanceTest < ActiveSupport::TestCase
+  context "weekly maintenance" do
+    should "prune password resets" do
+      @user = FactoryBot.create(:user, email: "test@example.com")
+      @nonce = FactoryBot.create(:user_password_reset_nonce, email: "test@example.com", created_at: 1.month.ago)
+
+      WeeklyMaintenance.new.run
+      assert_equal(0, UserPasswordResetNonce.count)
+    end
+  end
+end