mirror of
https://github.com/discourse/discourse.git
synced 2024-12-19 23:44:04 +08:00
365 lines
9.7 KiB
Ruby
365 lines
9.7 KiB
Ruby
require 'rails_helper'
|
|
|
|
describe DiscoursePoll::PollsUpdater do
|
|
let(:post_with_two_polls) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll]
|
|
* 1
|
|
* 2
|
|
[/poll]
|
|
|
|
[poll name=test]
|
|
* 1
|
|
* 2
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
let(:post) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll]
|
|
* 1
|
|
* 2
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
let(:other_post) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll]
|
|
* 3
|
|
* 4
|
|
* 5
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
let(:polls) do
|
|
DiscoursePoll::PollsValidator.new(post).validate_polls
|
|
end
|
|
|
|
let(:polls_with_3_options) do
|
|
DiscoursePoll::PollsValidator.new(other_post).validate_polls
|
|
end
|
|
|
|
let(:two_polls) do
|
|
DiscoursePoll::PollsValidator.new(post_with_two_polls).validate_polls
|
|
end
|
|
|
|
describe '.update' do
|
|
describe 'when post does not contain any polls' do
|
|
it 'should update polls correctly' do
|
|
post = Fabricate(:post)
|
|
|
|
message = MessageBus.track_publish do
|
|
described_class.update(post, polls)
|
|
end.first
|
|
|
|
expect(post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]).to eq(polls)
|
|
expect(message.data[:post_id]).to eq(post.id)
|
|
expect(message.data[:polls]).to eq(polls)
|
|
end
|
|
end
|
|
|
|
describe 'when post contains existing polls' do
|
|
it "should be able to update polls correctly" do
|
|
message = MessageBus.track_publish do
|
|
described_class.update(post, polls_with_3_options)
|
|
end.first
|
|
|
|
expect(post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]).to eq(polls_with_3_options)
|
|
expect(message.data[:post_id]).to eq(post.id)
|
|
expect(message.data[:polls]).to eq(polls_with_3_options)
|
|
end
|
|
end
|
|
|
|
describe 'when there are no changes' do
|
|
it "should not do anything" do
|
|
messages = MessageBus.track_publish do
|
|
described_class.update(post, polls)
|
|
end
|
|
|
|
expect(messages).to eq([])
|
|
end
|
|
end
|
|
|
|
context "public polls" do
|
|
let(:post) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll public=true]
|
|
- A
|
|
- B
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
let(:private_poll_post) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll]
|
|
- A
|
|
- B
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
let(:private_poll) do
|
|
DiscoursePoll::PollsValidator.new(private_poll_post).validate_polls
|
|
end
|
|
|
|
let(:public_poll) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll public=true]
|
|
- A
|
|
- C
|
|
[/poll]
|
|
RAW
|
|
|
|
DiscoursePoll::PollsValidator.new(Fabricate(:post, raw: raw)).validate_polls
|
|
end
|
|
|
|
let(:user) { Fabricate(:user) }
|
|
|
|
before do
|
|
DiscoursePoll::Poll.vote(post.id, "poll", ["5c24fc1df56d764b550ceae1b9319125"], user)
|
|
post.reload
|
|
end
|
|
|
|
it "should not allow a private poll with votes to be made public" do
|
|
DiscoursePoll::Poll.vote(private_poll_post.id, "poll", ["5c24fc1df56d764b550ceae1b9319125"], user)
|
|
private_poll_post.reload
|
|
|
|
messages = MessageBus.track_publish do
|
|
described_class.update(private_poll_post, public_poll)
|
|
end
|
|
|
|
expect(messages).to eq([])
|
|
|
|
expect(private_poll_post.errors[:base]).to include(
|
|
I18n.t("poll.default_cannot_be_made_public")
|
|
)
|
|
end
|
|
|
|
it "should retain voter_ids when options have been edited" do
|
|
described_class.update(post, public_poll)
|
|
|
|
polls = post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]
|
|
|
|
expect(polls["poll"]["options"][0]["voter_ids"]).to eq([user.id])
|
|
expect(polls["poll"]["options"][1]["voter_ids"]).to eq([])
|
|
end
|
|
|
|
it "should delete voter_ids when poll is set to private" do
|
|
described_class.update(post, private_poll)
|
|
|
|
polls = post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]
|
|
|
|
expect(post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD])
|
|
.to eq(private_poll)
|
|
|
|
expect(polls["poll"]["options"][0]["voter_ids"]).to eq(nil)
|
|
expect(polls["poll"]["options"][1]["voter_ids"]).to eq(nil)
|
|
end
|
|
end
|
|
|
|
context "polls of type 'multiple'" do
|
|
let(:min_2_post) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll type=multiple min=2 max=3]
|
|
- Option 1
|
|
- Option 2
|
|
- Option 3
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
let(:min_2_poll) do
|
|
DiscoursePoll::PollsValidator.new(min_2_post).validate_polls
|
|
end
|
|
|
|
let(:min_1_post) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll type=multiple min=1 max=2]
|
|
- Option 1
|
|
- Option 2
|
|
- Option 3
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
let(:min_1_poll) do
|
|
DiscoursePoll::PollsValidator.new(min_1_post).validate_polls
|
|
end
|
|
|
|
it "should be able to update options" do
|
|
min_2_poll
|
|
|
|
message = MessageBus.track_publish do
|
|
described_class.update(min_2_post, min_1_poll)
|
|
end.first
|
|
|
|
expect(min_2_post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]).to eq(min_1_poll)
|
|
expect(message.data[:post_id]).to eq(min_2_post.id)
|
|
expect(message.data[:polls]).to eq(min_1_poll)
|
|
end
|
|
end
|
|
|
|
describe "when post has been created more than 5 minutes ago" do
|
|
let(:another_post) { Fabricate(:post, created_at: Time.zone.now - 5.minutes) }
|
|
|
|
before do
|
|
polls.each { |key, value| value["voters"] = 2 }
|
|
described_class.update(another_post, polls)
|
|
end
|
|
|
|
it "should not allow new polls to be added" do
|
|
messages = MessageBus.track_publish do
|
|
described_class.update(another_post, two_polls)
|
|
end
|
|
|
|
expect(another_post.errors[:base]).to include(I18n.t(
|
|
"poll.cannot_change_polls_after_5_minutes")
|
|
)
|
|
|
|
expect(messages).to eq([])
|
|
end
|
|
|
|
it "should not allow users to edit options of current poll" do
|
|
messages = MessageBus.track_publish do
|
|
described_class.update(another_post, polls_with_3_options)
|
|
end
|
|
|
|
expect(another_post.errors[:base]).to include(I18n.t(
|
|
"poll.op_cannot_edit_options_after_5_minutes"
|
|
))
|
|
|
|
expect(messages).to eq([])
|
|
end
|
|
|
|
context "staff" do
|
|
it "should not allow staff to add options if votes have been casted" do
|
|
another_post.update_attributes!(last_editor_id: User.staff.first.id)
|
|
|
|
messages = MessageBus.track_publish do
|
|
described_class.update(another_post, polls_with_3_options)
|
|
end
|
|
|
|
expect(another_post.errors[:base]).to include(I18n.t(
|
|
"poll.staff_cannot_add_or_remove_options_after_5_minutes"
|
|
))
|
|
|
|
expect(messages).to eq([])
|
|
end
|
|
|
|
it "should allow staff to add options if no votes have been casted" do
|
|
post.update_attributes!(
|
|
created_at: Time.zone.now - 5.minutes,
|
|
last_editor_id: User.staff.first.id
|
|
)
|
|
|
|
message = MessageBus.track_publish do
|
|
described_class.update(post, polls_with_3_options)
|
|
end.first
|
|
|
|
expect(post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]).to eq(polls_with_3_options)
|
|
expect(message.data[:post_id]).to eq(post.id)
|
|
expect(message.data[:polls]).to eq(polls_with_3_options)
|
|
end
|
|
|
|
it "should allow staff to edit options if votes have been casted" do
|
|
another_post.update_attributes!(last_editor_id: User.staff.first.id)
|
|
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll]
|
|
* 3
|
|
* 4
|
|
[/poll]
|
|
RAW
|
|
|
|
different_post = Fabricate(:post, raw: raw)
|
|
different_polls = DiscoursePoll::PollsValidator.new(different_post).validate_polls
|
|
|
|
message = MessageBus.track_publish do
|
|
described_class.update(another_post, different_polls)
|
|
end.first
|
|
|
|
different_polls.each { |key, value| value["voters"] = 2 }
|
|
|
|
expect(another_post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]).to eq(different_polls)
|
|
expect(message.data[:post_id]).to eq(another_post.id)
|
|
expect(message.data[:polls]).to eq(different_polls)
|
|
end
|
|
|
|
it "should allow staff to edit options if votes have not been casted" do
|
|
post.update_attributes!(last_editor_id: User.staff.first.id)
|
|
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll]
|
|
* 3
|
|
* 4
|
|
[/poll]
|
|
RAW
|
|
|
|
different_post = Fabricate(:post, raw: raw)
|
|
different_polls = DiscoursePoll::PollsValidator.new(different_post).validate_polls
|
|
|
|
message = MessageBus.track_publish do
|
|
described_class.update(post, different_polls)
|
|
end.first
|
|
|
|
expect(post.reload.custom_fields[DiscoursePoll::POLLS_CUSTOM_FIELD]).to eq(different_polls)
|
|
expect(message.data[:post_id]).to eq(post.id)
|
|
expect(message.data[:polls]).to eq(different_polls)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.extract_option_ids' do
|
|
it 'should return an array of the options id' do
|
|
expect(described_class.extract_option_ids(polls)).to eq(
|
|
["4d8a15e3cc35750f016ce15a43937620", "cd314db7dfbac2b10687b6f39abfdf41"]
|
|
)
|
|
end
|
|
end
|
|
|
|
describe '.total_votes' do
|
|
let!(:post) do
|
|
raw = <<-RAW.strip_heredoc
|
|
[poll]
|
|
* 1
|
|
* 2
|
|
[/poll]
|
|
|
|
[poll name=test]
|
|
* 1
|
|
* 2
|
|
[/poll]
|
|
RAW
|
|
|
|
Fabricate(:post, raw: raw)
|
|
end
|
|
|
|
it "should return the right number of votes" do
|
|
expect(described_class.total_votes(polls)).to eq(0)
|
|
|
|
polls.each { |key, value| value["voters"] = 2 }
|
|
|
|
expect(described_class.total_votes(polls)).to eq(4)
|
|
end
|
|
end
|
|
end
|