osr-discourse-src/app/assets/javascripts/discourse/tests/helpers/component-test.js

import { render } from "@ember/test-helpers";
import Session from "discourse/models/session";
import Site from "discourse/models/site";
import TopicTrackingState from "discourse/models/topic-tracking-state";
import User from "discourse/models/user";
import { autoLoadModules } from "discourse/initializers/auto-load-modules";
import QUnit, { test } from "qunit";
import { setupRenderingTest as emberSetupRenderingTest } from "ember-qunit";
import { currentSettings } from "discourse/tests/helpers/site-settings";
import { testCleanup } from "discourse/tests/helpers/qunit-helpers";
import { injectServiceIntoService } from "discourse/pre-initializers/inject-discourse-objects";

export function setupRenderingTest(hooks) {
  emberSetupRenderingTest(hooks);

  hooks.beforeEach(function () {
    if (!hooks.usingDiscourseModule) {
      this.siteSettings = currentSettings();

      if (!this.registry) {
        this.registry = this.owner.__registry__;
      }

      this.container = this.owner;
    }

    this.site = Site.current();
    this.session = Session.current();

    const currentUser = User.create({
      username: "eviltrout",
      timezone: "Australia/Brisbane",
    });
    this.currentUser = currentUser;
    this.owner.unregister("service:current-user");
    this.owner.register("service:current-user", currentUser, {
      instantiate: false,
    });
    this.owner.inject("component", "currentUser", "service:current-user");
    injectServiceIntoService({
      app: this.owner.application,
      property: "currentUser",
      specifier: "service:current-user",
    });

    this.owner.unregister("topic-tracking-state:main");
    this.owner.register(
      "topic-tracking-state:main",
      TopicTrackingState.create({ currentUser }),
      { instantiate: false }
    );
    this.owner.inject(
      "service",
      "topicTrackingState",
      "topic-tracking-state:main"
    );

    autoLoadModules(this.owner, this.registry);
    this.owner.lookup("service:store");

    $.fn.autocomplete = function () {};
  });

  if (!hooks.usingDiscourseModule) {
    hooks.afterEach(function () {
      testCleanup(this.container);
    });
  }
}

export default function (name, hooks, opts) {
  if (opts === undefined) {
    opts = hooks;
  }

  opts = opts || {};

  if (opts.skip) {
    return;
  }

  if (typeof opts.template === "string") {
    let testName = QUnit.config.currentModule.name + " " + name;
    // eslint-disable-next-line
    console.warn(
      `${testName} skipped; template must be compiled and not a string`
    );
    return;
  }

  test(name, async function (assert) {
    if (opts.anonymous) {
      this.owner.unregister("service:current-user");
    }

    if (opts.beforeEach) {
      const store = this.owner.lookup("service:store");
      await opts.beforeEach.call(this, store);
    }

    try {
      await render(opts.template);
      await opts.test.call(this, assert);
    } finally {
      if (opts.afterEach) {
        await opts.afterEach.call(opts);
      }
    }
  });
}