discourse/spec/services/badge_granter_spec.rb

251 lines
7.8 KiB
Ruby

require 'spec_helper'
describe BadgeGranter do
let(:badge) { Fabricate(:badge) }
let(:user) { Fabricate(:user) }
describe 'revoke_titles' do
it 'can correctly revoke titles' do
badge = Fabricate(:badge, allow_title: true)
user = Fabricate(:user, title: badge.name)
user.reload
user.user_profile.update_column(:badge_granted_title, true)
BadgeGranter.revoke_ungranted_titles!
user.reload
user.title.should == badge.name
badge.update_column(:allow_title, false)
BadgeGranter.revoke_ungranted_titles!
user.reload
user.title.should == ''
user.title = "CEO"
user.save
BadgeGranter.revoke_ungranted_titles!
user.reload
user.title.should == "CEO"
end
end
describe 'preview' do
it 'can correctly preview' do
Fabricate(:user, email: 'sam@gmail.com')
result = BadgeGranter.preview('select id user_id, null post_id, created_at granted_at from users where email like \'%gmail.com\'')
result[:grant_count].should == 1
end
end
describe 'backfill' do
it 'has no broken badge queries' do
Badge.all.each do |b|
BadgeGranter.backfill(b)
end
end
it 'can backfill the welcome badge' do
post = Fabricate(:post)
user2 = Fabricate(:user)
PostAction.act(user2, post, PostActionType.types[:like])
UserBadge.destroy_all
BadgeGranter.backfill(Badge.find(Badge::Welcome))
BadgeGranter.backfill(Badge.find(Badge::FirstLike))
b = UserBadge.find_by(user_id: post.user_id)
b.post_id.should == post.id
b.badge_id = Badge::Welcome
b = UserBadge.find_by(user_id: user2.id)
b.post_id.should == post.id
b.badge_id = Badge::FirstLike
end
it 'should grant missing badges' do
post = Fabricate(:post, like_count: 30)
2.times {
BadgeGranter.backfill(Badge.find(Badge::NiceTopic), post_ids: [post.id])
BadgeGranter.backfill(Badge.find(Badge::GoodTopic))
}
# TODO add welcome
post.user.user_badges.pluck(:badge_id).sort.should == [Badge::NiceTopic,Badge::GoodTopic]
post.user.notifications.count.should == 2
Badge.find(Badge::NiceTopic).grant_count.should == 1
Badge.find(Badge::GoodTopic).grant_count.should == 1
end
end
describe 'grant' do
it 'grants multiple badges' do
badge = Fabricate(:badge, multiple_grant: true)
user_badge = BadgeGranter.grant(badge, user)
user_badge = BadgeGranter.grant(badge, user)
user_badge.should be_present
UserBadge.where(user_id: user.id).count.should == 2
end
it 'sets granted_at' do
time = Time.zone.now
Timecop.freeze time
user_badge = BadgeGranter.grant(badge, user)
user_badge.granted_at.should eq(time)
Timecop.return
end
it 'sets granted_by if the option is present' do
admin = Fabricate(:admin)
StaffActionLogger.any_instance.expects(:log_badge_grant).once
user_badge = BadgeGranter.grant(badge, user, granted_by: admin)
user_badge.granted_by.should eq(admin)
end
it 'defaults granted_by to the system user' do
StaffActionLogger.any_instance.expects(:log_badge_grant).never
user_badge = BadgeGranter.grant(badge, user)
user_badge.granted_by_id.should eq(Discourse.system_user.id)
end
it 'does not allow a regular user to grant badges' do
user_badge = BadgeGranter.grant(badge, user, granted_by: Fabricate(:user))
user_badge.should_not be_present
end
it 'increments grant_count on the badge and creates a notification' do
BadgeGranter.grant(badge, user)
badge.reload.grant_count.should eq(1)
user.notifications.find_by(notification_type: Notification.types[:granted_badge]).data_hash["badge_id"].should == badge.id
end
end
describe 'revoke' do
let(:admin) { Fabricate(:admin) }
let!(:user_badge) { BadgeGranter.grant(badge, user) }
it 'revokes the badge and does necessary cleanup' do
user.title = badge.name; user.save!
badge.reload.grant_count.should eq(1)
StaffActionLogger.any_instance.expects(:log_badge_revoke).with(user_badge)
BadgeGranter.revoke(user_badge, revoked_by: admin)
UserBadge.find_by(user: user, badge: badge).should_not be_present
badge.reload.grant_count.should eq(0)
user.notifications.where(notification_type: Notification.types[:granted_badge]).should be_empty
user.reload.title.should == nil
end
end
context "update_badges" do
let(:user) { Fabricate(:user) }
let(:liker) { Fabricate(:user) }
before do
BadgeGranter.clear_queue!
end
it "grants autobiographer" do
user.user_profile.bio_raw = "THIS IS MY bio it a long bio I like my bio"
user.uploaded_avatar_id = 10
user.user_profile.save
user.save
BadgeGranter.process_queue!
UserBadge.where(user_id: user.id, badge_id: Badge::Autobiographer).count.should eq(1)
end
it "grants read guidlines" do
user.user_stat.read_faq = Time.now
user.user_stat.save
BadgeGranter.process_queue!
UserBadge.where(user_id: user.id, badge_id: Badge::ReadGuidelines).count.should eq(1)
end
it "grants first link" do
post = create_post
post2 = create_post(raw: "#{Discourse.base_url}/t/slug/#{post.topic_id}")
BadgeGranter.process_queue!
UserBadge.where(user_id: post2.user.id, badge_id: Badge::FirstLink).count.should eq(1)
end
it "grants first edit" do
SiteSetting.ninja_edit_window = 0
post = create_post
user = post.user
UserBadge.where(user_id: user.id, badge_id: Badge::Editor).count.should eq(0)
PostRevisor.new(post).revise!(user, "This is my new test 1235 123")
BadgeGranter.process_queue!
UserBadge.where(user_id: user.id, badge_id: Badge::Editor).count.should eq(1)
end
it "grants and revokes trust level badges" do
user.change_trust_level!(TrustLevel[4])
BadgeGranter.process_queue!
UserBadge.where(user_id: user.id, badge_id: Badge.trust_level_badge_ids).count.should eq(4)
user.change_trust_level!(TrustLevel[1])
BadgeGranter.backfill(Badge.find(1))
BadgeGranter.backfill(Badge.find(2))
UserBadge.where(user_id: user.id, badge_id: 1).first.should_not == nil
UserBadge.where(user_id: user.id, badge_id: 2).first.should == nil
end
it "grants system like badges" do
post = create_post(user: user)
# Welcome badge
action = PostAction.act(liker, post, PostActionType.types[:like])
BadgeGranter.process_queue!
UserBadge.find_by(user_id: user.id, badge_id: 5).should_not == nil
post = create_post(topic: post.topic, user: user)
action = PostAction.act(liker, post, PostActionType.types[:like])
# Nice post badge
post.update_attributes like_count: 10
BadgeGranter.queue_badge_grant(Badge::Trigger::PostAction, post_action: action)
BadgeGranter.process_queue!
UserBadge.find_by(user_id: user.id, badge_id: Badge::NicePost).should_not == nil
UserBadge.where(user_id: user.id, badge_id: Badge::NicePost).count.should == 1
# Good post badge
post.update_attributes like_count: 25
BadgeGranter.queue_badge_grant(Badge::Trigger::PostAction, post_action: action)
BadgeGranter.process_queue!
UserBadge.find_by(user_id: user.id, badge_id: Badge::GoodPost).should_not == nil
# Great post badge
post.update_attributes like_count: 50
BadgeGranter.queue_badge_grant(Badge::Trigger::PostAction, post_action: action)
BadgeGranter.process_queue!
UserBadge.find_by(user_id: user.id, badge_id: Badge::GreatPost).should_not == nil
# Revoke badges on unlike
post.update_attributes like_count: 49
BadgeGranter.backfill(Badge.find(Badge::GreatPost))
UserBadge.find_by(user_id: user.id, badge_id: Badge::GreatPost).should == nil
end
end
end