discourse/test/javascripts/helpers/component-test.js

import EmberObject from "@ember/object";
import createStore from "helpers/create-store";
import { autoLoadModules } from "discourse/initializers/auto-load-modules";
import TopicTrackingState from "discourse/models/topic-tracking-state";
import User from "discourse/models/user";
import Site from "discourse/models/site";

export default function(name, opts) {
  opts = opts || {};

  if (opts.skip) {
    return;
  }

  test(name, function(assert) {
    this.site = Site.current();

    this.registry.register("site-settings:main", Discourse.SiteSettings, {
      instantiate: false
    });
    this.registry.register("capabilities:main", EmberObject);
    this.registry.register("site:main", this.site, { instantiate: false });
    this.registry.injection("component", "siteSettings", "site-settings:main");
    this.registry.injection("component", "appEvents", "service:app-events");
    this.registry.injection("component", "capabilities", "capabilities:main");
    this.registry.injection("component", "site", "site:main");

    this.siteSettings = Discourse.SiteSettings;

    autoLoadModules(this.registry, this.registry);

    const store = createStore();
    if (!opts.anonymous) {
      const currentUser = User.create({ username: "eviltrout" });
      this.currentUser = currentUser;
      this.registry.register("current-user:main", this.currentUser, {
        instantiate: false
      });
      this.registry.injection("component", "currentUser", "current-user:main");
      this.registry.register(
        "topic-tracking-state:main",
        TopicTrackingState.create({ currentUser }),
        { instantiate: false }
      );
    }

    this.registry.register("service:store", store, { instantiate: false });

    if (opts.beforeEach) {
      opts.beforeEach.call(this, store);
    }

    andThen(() => {
      return this.render(opts.template);
    });

    andThen(() => {
      return opts.test.call(this, assert);
    }).finally(() => {
      if (opts.afterEach) {
        andThen(() => {
          return opts.afterEach.call(opts);
        });
      }
    });
  });
}