discourse/plugins/poll/spec/poll_plugin/poll_spec.rb

require 'spec_helper'

describe PollPlugin::Poll do
  let(:topic) { create_topic(title: "Poll: Chitoge vs Onodera") }
  let(:post) { create_post(topic: topic, raw: "Pick one.\n\n[poll]\n* Chitoge\n* Onodera\n[/poll]") }
  let(:poll) { PollPlugin::Poll.new(post) }
  let(:user) { Fabricate(:user) }

  it "should detect poll post correctly" do
    expect(poll.is_poll?).to be_true
    post2 = create_post(topic: topic, raw: "This is a generic reply.")
    expect(PollPlugin::Poll.new(post2).is_poll?).to be_false
    post.topic.title = "Not a poll"
    expect(poll.is_poll?).to be_false
  end

  it "strips whitespace from the prefix translation" do
    topic.title = "Polll: This might be a poll"
    topic.save
    expect(PollPlugin::Poll.new(post).is_poll?).to be_false
    I18n.expects(:t).with('poll.prefix').returns("Polll ")
    I18n.expects(:t).with('poll.closed_prefix').returns("Closed Poll ")
    expect(PollPlugin::Poll.new(post).is_poll?).to be_true
  end

  it "should get options correctly" do
    expect(poll.options).to eq(["Chitoge", "Onodera"])
  end

  it "should fall back to using the first list if [poll] markup is not present" do
    topic = create_topic(title: "This is not a poll topic")
    post = create_post(topic: topic, raw: "Pick one.\n\n* Chitoge\n* Onodera")
    poll = PollPlugin::Poll.new(post)
    expect(poll.options).to eq(["Chitoge", "Onodera"])
  end

  it "should get details correctly" do
    expect(poll.details).to eq({"Chitoge" => 0, "Onodera" => 0})
  end

  it "should set details correctly" do
    poll.set_details!({})
    poll.details.should eq({})
    PollPlugin::Poll.new(post).details.should eq({})
  end

  it "should get and set votes correctly" do
    poll.get_vote(user).should eq(nil)
    poll.set_vote!(user, "Onodera")
    poll.get_vote(user).should eq("Onodera")
    poll.details["Onodera"].should eq(1)
  end

  it "should not set votes on closed polls" do
    poll.set_vote!(user, "Onodera")
    post.topic.closed = true
    post.topic.save!
    poll.set_vote!(user, "Chitoge")
    poll.get_vote(user).should eq("Onodera")
  end

  it "should serialize correctly" do
    poll.serialize(user).should eq({options: poll.details, selected: nil, closed: false})
    poll.set_vote!(user, "Onodera")
    poll.serialize(user).should eq({options: poll.details, selected: "Onodera", closed: false})
    poll.serialize(nil).should eq({options: poll.details, selected: nil, closed: false})

    topic.title = "Closed Poll: my poll"
    topic.save

    post.topic.reload
    poll = PollPlugin::Poll.new(post)
    poll.serialize(nil).should eq({options: poll.details, selected: nil, closed: true})
  end

  it "should serialize to nil if there are no poll options" do
    topic = create_topic(title: "This is not a poll topic")
    post = create_post(topic: topic, raw: "no options in the content")
    poll = PollPlugin::Poll.new(post)
    poll.serialize(user).should eq(nil)
  end

  it "stores poll options to plugin store" do
    poll.set_vote!(user, "Onodera")
    poll.stubs(:options).returns(["Chitoge", "Onodera", "Inferno Cop"])
    poll.update_options!
    poll.details.keys.sort.should eq(["Chitoge", "Inferno Cop", "Onodera"])
    poll.details["Inferno Cop"].should eq(0)
    poll.details["Onodera"].should eq(1)

    poll.stubs(:options).returns(["Chitoge", "Onodera v2", "Inferno Cop"])
    poll.update_options!
    poll.details.keys.sort.should eq(["Chitoge", "Inferno Cop", "Onodera v2"])
    poll.details["Onodera v2"].should eq(1)
    poll.get_vote(user).should eq("Onodera v2")
  end
end