mirror of
https://github.com/discourse/discourse.git
synced 2024-12-01 07:03:43 +08:00
86 lines
3.2 KiB
Ruby
86 lines
3.2 KiB
Ruby
# encoding: utf-8
|
|
|
|
require 'spec_helper'
|
|
require_dependency 'post_creator'
|
|
|
|
describe CategoryUser do
|
|
|
|
it 'allows batch set' do
|
|
user = Fabricate(:user)
|
|
category1 = Fabricate(:category)
|
|
category2 = Fabricate(:category)
|
|
|
|
watching = CategoryUser.where(user_id: user.id, notification_level: CategoryUser.notification_levels[:watching])
|
|
|
|
CategoryUser.batch_set(user, :watching, [category1.id, category2.id])
|
|
expect(watching.pluck(:category_id).sort).to eq [category1.id, category2.id]
|
|
|
|
CategoryUser.batch_set(user, :watching, [])
|
|
expect(watching.count).to eq 0
|
|
|
|
CategoryUser.batch_set(user, :watching, [category2.id])
|
|
expect(watching.count).to eq 1
|
|
end
|
|
|
|
|
|
context 'integration' do
|
|
before do
|
|
ActiveRecord::Base.observers.enable :all
|
|
end
|
|
|
|
it 'should operate correctly' do
|
|
watched_category = Fabricate(:category)
|
|
muted_category = Fabricate(:category)
|
|
tracked_category = Fabricate(:category)
|
|
|
|
user = Fabricate(:user)
|
|
|
|
CategoryUser.create!(user: user, category: watched_category, notification_level: CategoryUser.notification_levels[:watching])
|
|
CategoryUser.create!(user: user, category: muted_category, notification_level: CategoryUser.notification_levels[:muted])
|
|
CategoryUser.create!(user: user, category: tracked_category, notification_level: CategoryUser.notification_levels[:tracking])
|
|
|
|
watched_post = create_post(category: watched_category)
|
|
muted_post = create_post(category: muted_category)
|
|
tracked_post = create_post(category: tracked_category)
|
|
|
|
expect(Notification.where(user_id: user.id, topic_id: watched_post.topic_id).count).to eq 1
|
|
expect(Notification.where(user_id: user.id, topic_id: tracked_post.topic_id).count).to eq 0
|
|
|
|
tu = TopicUser.get(tracked_post.topic, user)
|
|
expect(tu.notification_level).to eq TopicUser.notification_levels[:tracking]
|
|
expect(tu.notifications_reason_id).to eq TopicUser.notification_reasons[:auto_track_category]
|
|
end
|
|
|
|
it "watches categories that have been changed" do
|
|
user = Fabricate(:user)
|
|
watched_category = Fabricate(:category)
|
|
CategoryUser.create!(user: user, category: watched_category, notification_level: CategoryUser.notification_levels[:watching])
|
|
|
|
post = create_post
|
|
expect(TopicUser.get(post.topic, user)).to be_blank
|
|
|
|
# Now, change the topic's category
|
|
post.topic.change_category_to_id(watched_category.id)
|
|
tu = TopicUser.get(post.topic, user)
|
|
expect(tu.notification_level).to eq TopicUser.notification_levels[:watching]
|
|
end
|
|
|
|
it "unwatches categories that have been changed" do
|
|
user = Fabricate(:user)
|
|
watched_category = Fabricate(:category)
|
|
CategoryUser.create!(user: user, category: watched_category, notification_level: CategoryUser.notification_levels[:watching])
|
|
|
|
post = create_post(category: watched_category)
|
|
tu = TopicUser.get(post.topic, user)
|
|
expect(tu.notification_level).to eq TopicUser.notification_levels[:watching]
|
|
|
|
# Now, change the topic's category
|
|
unwatched_category = Fabricate(:category)
|
|
post.topic.change_category_to_id(unwatched_category.id)
|
|
expect(TopicUser.get(post.topic, user)).to be_blank
|
|
end
|
|
|
|
end
|
|
end
|
|
|