discourse/plugins/poll/test/javascripts/widgets/discourse-poll-test.js.es6

import {
  count,
  discourseModule,
  exists,
  queryAll,
} from "discourse/tests/helpers/qunit-helpers";
import componentTest, {
  setupRenderingTest,
} from "discourse/tests/helpers/component-test";
import EmberObject from "@ember/object";
import I18n from "I18n";
import pretender from "discourse/tests/helpers/create-pretender";
import hbs from "htmlbars-inline-precompile";

let requests = 0;

discourseModule(
  "Integration | Component | Widget | discourse-poll",
  function (hooks) {
    setupRenderingTest(hooks);

    pretender.put("/polls/vote", () => {
      ++requests;
      return [
        200,
        { "Content-Type": "application/json" },
        {
          poll: {
            name: "poll",
            type: "regular",
            status: "open",
            results: "always",
            options: [
              {
                id: "1f972d1df351de3ce35a787c89faad29",
                html: "yes",
                votes: 1,
              },
              {
                id: "d7ebc3a9beea2e680815a1e4f57d6db6",
                html: "no",
                votes: 0,
              },
            ],
            voters: 1,
            chart_type: "bar",
          },
          vote: ["1f972d1df351de3ce35a787c89faad29"],
        },
      ];
    });

    pretender.put("/polls/vote", () => {
      ++requests;
      return [
        200,
        { "Content-Type": "application/json" },
        {
          poll: {
            name: "poll",
            type: "regular",
            status: "open",
            results: "always",
            options: [
              {
                id: "1f972d1df351de3ce35a787c89faad29",
                html: "yes",
                votes: 1,
              },
              {
                id: "d7ebc3a9beea2e680815a1e4f57d6db6",
                html: "no",
                votes: 0,
              },
            ],
            voters: 1,
            chart_type: "bar",
          },
          vote: ["1f972d1df351de3ce35a787c89faad29"],
        },
      ];
    });

    const template = hbs`{{mount-widget
                    widget="discourse-poll"
                    args=(hash id=id
                               post=post
                               poll=poll
                               vote=vote
                               groupableUserFields=groupableUserFields)}}`;

    componentTest("can vote", {
      template,

      beforeEach() {
        this.setProperties({
          post: EmberObject.create({
            id: 42,
            topic: {
              archived: false,
            },
          }),
          poll: EmberObject.create({
            name: "poll",
            type: "regular",
            status: "open",
            results: "always",
            options: [
              { id: "1f972d1df351de3ce35a787c89faad29", html: "yes", votes: 0 },
              { id: "d7ebc3a9beea2e680815a1e4f57d6db6", html: "no", votes: 0 },
            ],
            voters: 0,
            chart_type: "bar",
          }),
          vote: [],
          groupableUserFields: [],
        });
      },

      async test(assert) {
        requests = 0;

        await click(
          "li[data-poll-option-id='1f972d1df351de3ce35a787c89faad29']"
        );
        assert.equal(requests, 1);
        assert.equal(count(".chosen"), 1);
        assert.equal(queryAll(".chosen").text(), "100%yes");

        await click(".toggle-results");
        assert.equal(
          queryAll("li[data-poll-option-id='1f972d1df351de3ce35a787c89faad29']")
            .length,
          1
        );
      },
    });

    componentTest("cannot vote if not member of the right group", {
      template,

      beforeEach() {
        this.setProperties({
          post: EmberObject.create({
            id: 42,
            topic: {
              archived: false,
            },
          }),
          poll: EmberObject.create({
            name: "poll",
            type: "regular",
            status: "open",
            results: "always",
            options: [
              { id: "1f972d1df351de3ce35a787c89faad29", html: "yes", votes: 0 },
              { id: "d7ebc3a9beea2e680815a1e4f57d6db6", html: "no", votes: 0 },
            ],
            voters: 0,
            chart_type: "bar",
            groups: "foo",
          }),
          vote: [],
          groupableUserFields: [],
        });
      },

      async test(assert) {
        requests = 0;

        await click(
          "li[data-poll-option-id='1f972d1df351de3ce35a787c89faad29']"
        );
        assert.equal(
          queryAll(".poll-container .alert").text(),
          I18n.t("poll.results.groups.title", { groups: "foo" })
        );
        assert.equal(requests, 0);
        assert.ok(!exists(".chosen"));
      },
    });
  }
);