discourse/spec/services/badge_granter_spec.rb

252 lines
7.9 KiB
Ruby
Raw Normal View History

2014-03-05 07:52:20 -05:00
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)
2014-10-09 07:27:34 -04:00
BadgeGranter.grant(badge, user)
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
2014-08-12 22:25:56 -04:00
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))
2014-07-08 00:26:53 -04:00
b = UserBadge.find_by(user_id: post.user_id)
b.post_id.should == post.id
2014-07-08 00:26:53 -04:00
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)
2014-07-05 04:32:06 -04:00
2.times {
BadgeGranter.backfill(Badge.find(Badge::NiceTopic), post_ids: [post.id])
BadgeGranter.backfill(Badge.find(Badge::GoodTopic))
2014-07-05 04:32:06 -04:00
}
# TODO add welcome
post.user.user_badges.pluck(:badge_id).sort.should == [Badge::NiceTopic,Badge::GoodTopic]
2014-07-07 03:55:25 -04:00
post.user.notifications.count.should == 2
Badge.find(Badge::NiceTopic).grant_count.should == 1
Badge.find(Badge::GoodTopic).grant_count.should == 1
end
end
2014-03-05 07:52:20 -05:00
describe 'grant' do
it 'grants multiple badges' do
badge = Fabricate(:badge, multiple_grant: true)
user_badge = BadgeGranter.grant(badge, user)
2014-03-05 07:52:20 -05:00
user_badge = BadgeGranter.grant(badge, user)
user_badge.should be_present
UserBadge.where(user_id: user.id).count.should == 2
2014-03-05 07:52:20 -05:00
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
2014-03-05 07:52:20 -05:00
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
2014-03-05 07:52:20 -05:00
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
2014-03-05 07:52:20 -05:00
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
2014-03-05 07:52:20 -05:00
end
end
describe 'revoke' do
let(:admin) { Fabricate(:admin) }
2014-03-05 07:52:20 -05:00
let!(:user_badge) { BadgeGranter.grant(badge, user) }
it 'revokes the badge and does necessary cleanup' do
user.title = badge.name; user.save!
2014-03-05 07:52:20 -05:00
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
2014-03-05 07:52:20 -05:00
badge.reload.grant_count.should eq(0)
user.notifications.where(notification_type: Notification.types[:granted_badge]).should be_empty
user.reload.title.should == nil
2014-03-05 07:52:20 -05:00
end
end
2014-05-15 13:43:04 -04:00
context "update_badges" do
let(:user) { Fabricate(:user) }
2014-06-17 02:29:49 -04:00
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, { raw: "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
2014-05-15 13:43:04 -04:00
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
2014-06-17 02:29:49 -04:00
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])
2014-06-17 02:29:49 -04:00
# 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
2014-06-17 02:29:49 -04:00
# 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
2014-06-17 02:29:49 -04:00
# 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
2014-06-17 02:29:49 -04:00
# 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
2014-06-17 02:29:49 -04:00
end
end
2014-03-05 07:52:20 -05:00
end