80f43f9a7c
* Rename comments.ip_addr to comments.creator_ip_addr. * Fix belongs_to_creator to not clobber ip_addr field.
37 lines
1.2 KiB
Ruby
37 lines
1.2 KiB
Ruby
require 'test_helper'
|
|
|
|
class IpBanTest < ActiveSupport::TestCase
|
|
setup do
|
|
@user = FactoryBot.create(:user)
|
|
CurrentUser.user = FactoryBot.create(:mod_user)
|
|
CurrentUser.ip_addr = "127.0.0.1"
|
|
Danbooru.config.stubs(:member_comment_time_threshold).returns(1.week.from_now)
|
|
end
|
|
|
|
teardown do
|
|
CurrentUser.user = nil
|
|
CurrentUser.ip_addr = nil
|
|
end
|
|
|
|
should "be able to count the number of comments an IP address is associated with" do
|
|
comment = FactoryBot.create(:comment)
|
|
counts = IpBan.count_by_ip_addr("comments", [comment.creator_id], "creator_id", "creator_ip_addr")
|
|
assert_equal([{"creator_ip_addr" => "127.0.0.1", "count" => 1}], counts)
|
|
end
|
|
|
|
should "be able to count any updates from a user, groupiny by IP address" do
|
|
CurrentUser.scoped(@user, "1.2.3.4") do
|
|
comment = FactoryBot.create(:comment, :body => "aaa")
|
|
counts = IpBan.query([comment.creator_id])
|
|
assert_equal([{"creator_ip_addr" => "1.2.3.4", "count" => 1}], counts["comments"])
|
|
end
|
|
end
|
|
|
|
should "be able to ban a user" do
|
|
ip_ban = IpBan.create(ip_addr: "1.2.3.4", reason: "test")
|
|
|
|
assert_equal("1.2.3.4", ip_ban.ip_addr.to_s)
|
|
assert(IpBan.is_banned?("1.2.3.4"))
|
|
end
|
|
end
|