discourse/plugins/poll/spec/serializers/poll_option_serializer_spec.rb

# frozen_string_literal: true

def serialize_option(option, user)
  PollOptionSerializer.new(
    option,
    root: false,
    scope: { can_see_results: poll.can_see_results?(user) }
  )
end

describe PollOptionSerializer do
  let(:voter) { Fabricate(:user) }
  let(:poll) { post.polls.first }

  before do
    poll.poll_votes.create!(poll_option_id: poll.poll_options.first.id, user_id: voter.id)
  end

  context 'poll results are public' do
    let(:post) { Fabricate(:post, raw: "[poll]\n- A\n- B\n[/poll]") }

    context 'user is not staff' do
      let(:user) { Fabricate(:user) }

      it 'include votes' do
        serializer = serialize_option(poll.poll_options.first, user)

        expect(serializer.include_votes?).to eq(true)
      end
    end
  end

  context 'poll results are staff only' do
    let(:post) { Fabricate(:post, raw: "[poll results=staff_only]\n- A\n- B\n[/poll]") }

    context 'user is not staff' do
      let(:user) { Fabricate(:user) }

      it 'doesn’t include votes' do
        serializer = serialize_option(poll.poll_options.first, user)

        expect(serializer.include_votes?).to eq(false)
      end
    end

    context 'user staff' do
      let(:admin) { Fabricate(:admin) }

      it 'includes votes' do
        serializer = serialize_option(poll.poll_options.first, admin)

        expect(serializer.include_votes?).to eq(true)
      end
    end
  end
end