discourse/plugins/discourse-presence/spec/presence_manager_spec.rb

107 lines
3.4 KiB
Ruby

require 'rails_helper'
describe ::Presence::PresenceManager do
let(:user1) { Fabricate(:user) }
let(:user2) { Fabricate(:user) }
let(:user3) { Fabricate(:user) }
let(:manager) { ::Presence::PresenceManager }
let(:post1) { Fabricate(:post) }
let(:post2) { Fabricate(:post) }
after(:each) do
$redis.del("presence:topic:#{post1.topic.id}")
$redis.del("presence:topic:#{post2.topic.id}")
$redis.del("presence:post:#{post1.id}")
$redis.del("presence:post:#{post2.id}")
end
it 'adds, removes and lists users correctly' do
expect(manager.get_users('post', post1.id).count).to eq(0)
expect(manager.add('post', post1.id, user1.id)).to be true
expect(manager.add('post', post1.id, user2.id)).to be true
expect(manager.add('post', post2.id, user3.id)).to be true
expect(manager.get_users('post', post1.id).count).to eq(2)
expect(manager.get_users('post', post2.id).count).to eq(1)
expect(manager.get_users('post', post1.id)).to contain_exactly(user1, user2)
expect(manager.get_users('post', post2.id)).to contain_exactly(user3)
expect(manager.remove('post', post1.id, user1.id)).to be true
expect(manager.get_users('post', post1.id).count).to eq(1)
expect(manager.get_users('post', post1.id)).to contain_exactly(user2)
end
it 'publishes correctly' do
expect(manager.get_users('post', post1.id).count).to eq(0)
manager.add('post', post1.id, user1.id)
manager.add('post', post1.id, user2.id)
messages = MessageBus.track_publish do
manager.publish('post', post1.id)
end
expect(messages.count).to eq (1)
message = messages.first
expect(message.channel).to eq("/presence/post/#{post1.id}")
expect(message.data["users"].map { |u| u[:id] }).to contain_exactly(user1.id, user2.id)
end
it 'publishes private message securely' do
private_post = Fabricate(:private_message_post)
manager.add('post', private_post.id, user2.id)
messages = MessageBus.track_publish do
manager.publish('post', private_post.id)
end
expect(messages.count).to eq (1)
message = messages.first
expect(message.channel).to eq("/presence/post/#{private_post.id}")
user_ids = User.where('admin or moderator').pluck(:id)
user_ids += private_post.topic.allowed_users.pluck(:id)
expect(message.user_ids).to contain_exactly(*user_ids)
end
it 'publishes private category securely' do
group = Fabricate(:group)
category = Fabricate(:private_category, group: group)
private_topic = Fabricate(:topic, category: category)
manager.add('topic', private_topic.id, user2.id)
messages = MessageBus.track_publish do
manager.publish('topic', private_topic.id)
end
expect(messages.count).to eq (1)
message = messages.first
expect(message.channel).to eq("/presence/topic/#{private_topic.id}")
expect(message.group_ids).to contain_exactly(*private_topic.secure_group_ids)
end
it 'cleans up correctly' do
freeze_time Time.zone.now do
expect(manager.add('post', post1.id, user1.id)).to be true
expect(manager.cleanup('post', post1.id)).to be false # Nothing to cleanup
expect(manager.get_users('post', post1.id).count).to eq(1)
end
# Anything older than 20 seconds should be cleaned up
freeze_time 30.seconds.from_now do
expect(manager.cleanup('post', post1.id)).to be true
expect(manager.get_users('post', post1.id).count).to eq(0)
end
end
end