discourse/spec/jobs/mass_award_badge_spec.rb

# frozen_string_literal: true

require 'rails_helper'

describe Jobs::MassAwardBadge do
  describe '#execute' do
    fab!(:badge) { Fabricate(:badge) }
    fab!(:user) { Fabricate(:user) }
    let(:email_mode) { 'email' }

    it 'creates the badge for an existing user' do
      execute_job(user)

      expect(UserBadge.where(user: user, badge: badge).exists?).to eq(true)
    end

    it 'also creates a notification for the user' do
      execute_job(user)

      expect(Notification.exists?(user: user)).to eq(true)
      expect(UserBadge.where.not(notification_id: nil).exists?(user: user, badge: badge)).to eq(true)
    end

    it 'updates badge ranks correctly' do
      user_2 = Fabricate(:user)

      UserBadge.create!(badge_id: Badge::Member, user: user, granted_by: Discourse.system_user, granted_at: Time.now)

      execute_job(user)
      execute_job(user_2)

      expect(UserBadge.find_by(user: user, badge: badge).featured_rank).to eq(2)
      expect(UserBadge.find_by(user: user_2, badge: badge).featured_rank).to eq(1)
    end

    it 'grants a badge multiple times to a user' do
      badge.update!(multiple_grant: true)
      Notification.destroy_all
      execute_job(user, count: 4, grant_existing_holders: true)
      instances = UserBadge.where(user: user, badge: badge)
      expect(instances.count).to eq(4)
      expect(instances.pluck(:seq).sort).to eq((0...4).to_a)
      notifications = Notification.where(user: user)
      expect(notifications.count).to eq(1)
      expect(instances.map(&:notification_id).uniq).to contain_exactly(notifications.first.id)
    end

    def execute_job(user, count: 1, grant_existing_holders: false)
      subject.execute(user: user.id, badge:  badge.id, count: count, grant_existing_holders: grant_existing_holders)
    end
  end
end