discourse/plugins/poll/test/javascripts/unit/controllers/poll-ui-builder-test.js

import { module, test } from "qunit";
import { setupTest } from "ember-qunit";
import {
  MULTIPLE_POLL_TYPE,
  NUMBER_POLL_TYPE,
  REGULAR_POLL_TYPE,
} from "discourse/plugins/poll/controllers/poll-ui-builder";
import { settled } from "@ember/test-helpers";

function setupController(ctx) {
  const controller = ctx.owner.lookup("controller:poll-ui-builder");
  controller.set("toolbarEvent", { getText: () => "" });
  controller.onShow();
  return controller;
}

module("Unit | Controller | poll-ui-builder", function (hooks) {
  setupTest(hooks);

  test("isMultiple", function (assert) {
    const controller = setupController(this);

    controller.setProperties({
      pollType: MULTIPLE_POLL_TYPE,
      pollOptions: [{ value: "a" }],
    });
    assert.strictEqual(controller.isMultiple, true, "it should be true");

    controller.setProperties({
      pollType: "random",
      pollOptions: [{ value: "b" }],
    });
    assert.strictEqual(controller.isMultiple, false, "it should be false");
  });

  test("isNumber", function (assert) {
    const controller = setupController(this);

    controller.set("pollType", REGULAR_POLL_TYPE);
    assert.strictEqual(controller.isNumber, false, "it should be false");

    controller.set("pollType", NUMBER_POLL_TYPE);
    assert.strictEqual(controller.isNumber, true, "it should be true");
  });

  test("pollOptionsCount", function (assert) {
    const controller = setupController(this);

    controller.set("pollOptions", [{ value: "1" }, { value: "2" }]);
    assert.strictEqual(controller.pollOptionsCount, 2, "it should equal 2");

    controller.set("pollOptions", []);
    assert.strictEqual(controller.pollOptionsCount, 0, "it should equal 0");
  });

  test("disableInsert", function (assert) {
    const controller = setupController(this);

    controller.siteSettings.poll_maximum_options = 20;
    assert.strictEqual(controller.disableInsert, true, "it should be true");

    controller.set("pollOptions", [{ value: "a" }, { value: "b" }]);
    assert.strictEqual(controller.disableInsert, false, "it should be false");

    controller.set("pollType", NUMBER_POLL_TYPE);
    assert.strictEqual(controller.disableInsert, false, "it should be false");

    controller.setProperties({
      pollType: REGULAR_POLL_TYPE,
      pollOptions: [{ value: "a" }, { value: "b" }, { value: "c" }],
    });
    assert.strictEqual(controller.disableInsert, false, "it should be false");

    controller.setProperties({
      pollType: REGULAR_POLL_TYPE,
      pollOptions: [],
    });
    assert.strictEqual(controller.disableInsert, true, "it should be true");

    controller.setProperties({
      pollType: REGULAR_POLL_TYPE,
      pollOptions: [{ value: "w" }],
    });
    assert.strictEqual(controller.disableInsert, false, "it should be false");
  });

  test("number pollOutput", async function (assert) {
    const controller = setupController(this);
    controller.siteSettings.poll_maximum_options = 20;

    controller.setProperties({
      pollType: NUMBER_POLL_TYPE,
      pollMin: 1,
    });
    await settled();
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=number results=always min=1 max=20 step=1]\n[/poll]\n",
      "it should return the right output"
    );

    controller.set("pollStep", 2);
    await settled();
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=number results=always min=1 max=20 step=2]\n[/poll]\n",
      "it should return the right output"
    );

    controller.set("publicPoll", true);
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=number results=always min=1 max=20 step=2 public=true]\n[/poll]\n",
      "it should return the right output"
    );

    controller.set("pollStep", 0);
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=number results=always min=1 max=20 step=1 public=true]\n[/poll]\n",
      "it should return the right output"
    );
  });

  test("regular pollOutput", function (assert) {
    const controller = setupController(this);
    controller.siteSettings.poll_maximum_options = 20;

    controller.setProperties({
      pollOptions: [{ value: "1" }, { value: "2" }],
      pollType: REGULAR_POLL_TYPE,
    });
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=regular results=always chartType=bar]\n* 1\n* 2\n[/poll]\n",
      "it should return the right output"
    );

    controller.set("publicPoll", "true");
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=regular results=always public=true chartType=bar]\n* 1\n* 2\n[/poll]\n",
      "it should return the right output"
    );

    controller.set("pollGroups", "test");
    assert.strictEqual(
      controller.get("pollOutput"),
      "[poll type=regular results=always public=true chartType=bar groups=test]\n* 1\n* 2\n[/poll]\n",
      "it should return the right output"
    );
  });

  test("multiple pollOutput", function (assert) {
    const controller = setupController(this);
    controller.siteSettings.poll_maximum_options = 20;

    controller.setProperties({
      pollType: MULTIPLE_POLL_TYPE,
      pollMin: 1,
      pollOptions: [{ value: "1" }, { value: "2" }],
    });
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=multiple results=always min=1 max=2 chartType=bar]\n* 1\n* 2\n[/poll]\n",
      "it should return the right output"
    );

    controller.set("publicPoll", "true");
    assert.strictEqual(
      controller.pollOutput,
      "[poll type=multiple results=always min=1 max=2 public=true chartType=bar]\n* 1\n* 2\n[/poll]\n",
      "it should return the right output"
    );
  });

  test("staff_only option is not present for non-staff", async function (assert) {
    const controller = setupController(this);
    controller.currentUser = { staff: false };
    controller.notifyPropertyChange("pollResults");

    assert.strictEqual(
      controller.pollResults.filterBy("value", "staff_only").length,
      0,
      "staff_only is not present"
    );
  });

  test("poll result is always by default", function (assert) {
    const controller = setupController(this);
    assert.strictEqual(controller.pollResult, "always");
  });

  test("staff_only option is present for staff", async function (assert) {
    const controller = setupController(this);
    controller.currentUser = { staff: true };
    controller.notifyPropertyChange("pollResults");

    assert.strictEqual(
      controller.pollResults.filterBy("value", "staff_only").length,
      1,
      "staff_only is present"
    );
  });
});