# 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