eBooru/test/unit/dmail_test.rb
Earlopain fc7d84affd
[RuboCop] Enable Style/FrozenStringLiteralComment
This reduces allocations on the posts page by about 5%, from basic testing
2024-02-25 18:15:55 +01:00

199 lines
6.3 KiB
Ruby

# frozen_string_literal: true
require "test_helper"
class DmailTest < ActiveSupport::TestCase
context "A dmail" do
setup do
@user = create(:user)
CurrentUser.user = @user
CurrentUser.ip_addr = "1.2.3.4"
ActionMailer::Base.delivery_method = :test
ActionMailer::Base.perform_deliveries = true
ActionMailer::Base.deliveries = []
end
context "filter" do
setup do
@recipient = create(:user)
@recipient.create_dmail_filter(words: "banned")
@dmail = build(:dmail, title: "xxx", owner: @recipient, body: "banned word here", to: @recipient, from: @user)
end
should "detect banned words" do
assert(@recipient.dmail_filter.filtered?(@dmail))
end
should "autoread if it has a banned word" do
@dmail.save
assert_equal(true, @dmail.is_read?)
end
should "not update the recipient's has_mail if filtered" do
@dmail.save
@recipient.reload
assert_equal(false, @recipient.has_mail?)
end
should "be ignored when sender is a moderator" do
as(create(:moderator_user)) do
@dmail = create(:dmail, owner: @recipient, body: "banned word here", to: @recipient)
end
assert_equal(false, !!@recipient.dmail_filter.filtered?(@dmail))
assert_equal(false, @dmail.is_read?)
assert_equal(true, @recipient.has_mail?)
end
context "that is empty" do
setup do
@recipient.dmail_filter.update(words: " ")
end
should "not filter everything" do
assert_not(@recipient.dmail_filter.filtered?(@dmail))
end
end
end
context "search" do
should "return results based on title contents" do
dmail = create(:dmail, title: "xxx", body: "bbb", owner: @user)
matches = Dmail.search(title_matches: "x*")
assert_equal([dmail.id], matches.map(&:id))
matches = Dmail.search(title_matches: "X*")
assert_equal([dmail.id], matches.map(&:id))
matches = Dmail.search(message_matches: "aaa")
assert(matches.empty?)
end
should "return results based on body contents" do
dmail = create(:dmail, body: "xxx", owner: @user)
matches = Dmail.search(message_matches: "xxx")
assert(matches.any?)
matches = Dmail.search(message_matches: "aaa")
assert(matches.empty?)
end
end
should "not create a senders copy when validations fail" do
Danbooru.config.stubs(:disable_throttles?).returns(false)
@user = create(:user, created_at: 2.weeks.ago)
@recipient = create(:user)
as(@user) do
(Danbooru.config.dmail_minute_limit + 1).times do
Dmail.create_split(attributes_for(:dmail, from_id: @user.id, to_id: @recipient.id))
end
end
assert_equal(Danbooru.config.dmail_minute_limit * 2, Dmail.count)
end
should "should parse user names" do
dmail = build(:dmail, owner: @user)
dmail.to_id = nil
dmail.to_name = @user.name
assert(dmail.to_id == @user.id)
end
should "construct a response" do
dmail = create(:dmail, owner: @user)
response = dmail.build_response
assert_equal("Re: #{dmail.title}", response.title)
assert_equal(dmail.from_id, response.to_id)
assert_equal(dmail.to_id, response.from_id)
end
should "create a copy for each user" do
@new_user = create(:user)
assert_difference(-> { Dmail.count }, 2) do
Dmail.create_split(to_id: @new_user.id, title: "foo", body: "foo")
end
end
should "record the creator's ip addr" do
dmail = create(:dmail, owner: @user)
assert_equal(CurrentUser.ip_addr, dmail.creator_ip_addr.to_s)
end
should "send an email if the user wants it" do
user = create(:user, receive_email_notifications: true)
assert_difference(-> { ActionMailer::Base.deliveries.size }, 1) do
create(:dmail, to: user, owner: user)
end
end
should "not send an email if no_email_notification is set" do
user = create(:user, receive_email_notifications: true)
assert_no_difference(-> { ActionMailer::Base.deliveries.size }) do
create(:dmail, to: user, owner: user, no_email_notification: true)
Dmail.create_automated(to: user, title: "test", body: "abc", no_email_notification: true)
end
assert_equal(2, Dmail.count)
end
should "create only one message for a split response" do
user = create(:user, receive_email_notifications: true)
assert_difference(-> { ActionMailer::Base.deliveries.size }, 1) do
Dmail.create_split(to_id: user.id, title: "foo", body: "foo")
end
end
should "be marked as read after the user reads it" do
dmail = create(:dmail, owner: @user)
assert_not(dmail.is_read?)
dmail.mark_as_read!
assert(dmail.is_read?)
end
should "notify the recipient he has mail" do
recipient = create(:user)
Dmail.create_split(title: "hello", body: "hello", to_id: recipient.id)
dmail = Dmail.where(owner_id: recipient.id).last
recipient.reload
assert(recipient.has_mail?)
assert_equal(1, recipient.unread_dmail_count)
as(recipient) do
dmail.mark_as_read!
end
recipient.reload
assert_not(recipient.has_mail?)
assert_equal(0, recipient.unread_dmail_count)
end
context "that is automated" do
setup do
@bot = create(:user)
User.stubs(:system).returns(@bot)
end
should "only create a copy for the recipient" do
Dmail.create_automated(to: @user, title: "test", body: "test")
assert @user.dmails.where(from: @bot, title: "test", body: "test").exists?
assert_not @bot.dmails.where(from: @bot, title: "test", body: "test").exists?
end
should "fail gracefully if recipient doesn't exist" do
assert_nothing_raised do
dmail = Dmail.create_automated(to_name: "this_name_does_not_exist", title: "test", body: "test")
assert_equal(["must exist"], dmail.errors[:to])
end
end
end
context "during validation" do
subject { build(:dmail) }
should_not allow_value(" ").for(:title)
should_not allow_value(" ").for(:body)
should_not allow_value(nil).for(:to)
should_not allow_value(nil).for(:owner)
end
end
end