discourse/plugins/poll/test/javascripts/acceptance/poll-builder-enabled-test.js.es6

import { acceptance, replaceCurrentUser } from "helpers/qunit-helpers";
import { displayPollBuilderButton } from "discourse/plugins/poll/helpers/display-poll-builder-button";
import { clearPopupMenuOptionsCallback } from "discourse/controllers/composer";

acceptance("Poll Builder - polls are enabled", {
  loggedIn: true,
  settings: {
    poll_enabled: true,
    poll_minimum_trust_level_to_create: 1
  },
  beforeEach: function() {
    clearPopupMenuOptionsCallback();
  }
});

test("regular user - sufficient trust level", assert => {
  replaceCurrentUser({ staff: false, trust_level: 1 });

  displayPollBuilderButton();

  andThen(() => {
    assert.ok(
      exists(".select-kit-row[title='Build Poll']"),
      "it shows the builder button"
    );
  });
});

test("regular user - insufficient trust level", assert => {
  replaceCurrentUser({ staff: false, trust_level: 0 });

  displayPollBuilderButton();

  andThen(() => {
    assert.ok(
      !exists(".select-kit-row[title='Build Poll']"),
      "it hides the builder button"
    );
  });
});

test("staff - with insufficient trust level", assert => {
  replaceCurrentUser({ staff: true, trust_level: 0 });

  displayPollBuilderButton();

  andThen(() => {
    assert.ok(
      exists(".select-kit-row[title='Build Poll']"),
      "it shows the builder button"
    );
  });
});