osr-discourse-src/app/assets/javascripts/discourse/tests/unit/models/rest-model-test.js

import { module, test } from "qunit";
import RestAdapter from "discourse/adapters/rest";
import RestModel from "discourse/models/rest";
import sinon from "sinon";
import { getOwner } from "discourse-common/lib/get-owner";
import { setupTest } from "ember-qunit";

module("Unit | Model | rest-model", function (hooks) {
  setupTest(hooks);

  test("munging", function (assert) {
    const store = getOwner(this).lookup("service:store");
    class Grape extends RestModel {}
    Grape.reopenClass({
      munge: function (json) {
        json.inverse = 1 - json.percent;
        return json;
      },
    });

    getOwner(this).register("model:grape", Grape);
    const g = store.createRecord("grape", { store, percent: 0.4 });
    assert.strictEqual(g.inverse, 0.6, "it runs `munge` on `create`");
  });

  test("update", async function (assert) {
    const store = getOwner(this).lookup("service:store");
    const widget = await store.find("widget", 123);
    assert.strictEqual(widget.name, "Trout Lure");
    assert.ok(!widget.isSaving, "it is not saving");

    const spyBeforeUpdate = sinon.spy(widget, "beforeUpdate");
    const spyAfterUpdate = sinon.spy(widget, "afterUpdate");
    const promise = widget.update({ name: "new name" });
    assert.ok(widget.isSaving, "it is saving");
    assert.ok(spyBeforeUpdate.calledOn(widget));

    const result = await promise;
    assert.ok(spyAfterUpdate.calledOn(widget));
    assert.ok(!widget.isSaving, "it is no longer saving");
    assert.strictEqual(widget.name, "new name");

    assert.ok(result.target, "it has a reference to the record");
    assert.strictEqual(result.target.name, widget.name);
  });

  test("updating simultaneously", async function (assert) {
    assert.expect(2);

    const store = getOwner(this).lookup("service:store");
    const widget = await store.find("widget", 123);

    const firstPromise = widget.update({ name: "new name" });
    const secondPromise = widget.update({ name: "new name" });

    firstPromise.then(function () {
      assert.ok(true, "the first promise succeeds");
    });

    secondPromise.catch(function () {
      assert.ok(true, "the second promise fails");
    });
  });

  test("save new", async function (assert) {
    const store = getOwner(this).lookup("service:store");
    const widget = store.createRecord("widget");

    assert.ok(widget.isNew, "it is a new record");
    assert.ok(!widget.isCreated, "it is not created");
    assert.ok(!widget.isSaving, "it is not saving");

    const spyBeforeCreate = sinon.spy(widget, "beforeCreate");
    const spyAfterCreate = sinon.spy(widget, "afterCreate");
    const promise = widget.save({ name: "Evil Widget" });
    assert.ok(widget.isSaving, "it is not saving");
    assert.ok(spyBeforeCreate.calledOn(widget));

    const result = await promise;
    assert.ok(spyAfterCreate.calledOn(widget));
    assert.ok(!widget.isSaving, "it is no longer saving");
    assert.ok(widget.id, "it has an id");
    assert.ok(widget.name, "Evil Widget");
    assert.ok(widget.isCreated, "it is created");
    assert.ok(!widget.isNew, "it is no longer new");

    assert.ok(result.target, "it has a reference to the record");
    assert.strictEqual(result.target.name, widget.name);
  });

  test("creating simultaneously", function (assert) {
    assert.expect(2);

    const store = getOwner(this).lookup("service:store");
    const widget = store.createRecord("widget");

    const firstPromise = widget.save({ name: "Evil Widget" });
    const secondPromise = widget.save({ name: "Evil Widget" });
    firstPromise.then(function () {
      assert.ok(true, "the first promise succeeds");
    });

    secondPromise.catch(function () {
      assert.ok(true, "the second promise fails");
    });
  });

  test("destroyRecord", async function (assert) {
    const store = getOwner(this).lookup("service:store");
    const widget = await store.find("widget", 123);

    assert.ok(await widget.destroyRecord());
  });

  test("custom api name", async function (assert) {
    const store = getOwner(this).lookup("service:store");
    getOwner(this).register(
      "adapter:my-widget",
      class extends RestAdapter {
        // An adapter like this is used when the server-side key/url
        // do not match the name of the es6 class
        apiNameFor() {
          return "widget";
        }
      }
    );

    // The pretenders only respond to requests for `widget`
    // If these basic tests pass, the name override worked correctly

    // Create
    const widget = store.createRecord("my-widget");
    await widget.save({ name: "Evil Widget" });
    assert.strictEqual(widget.id, 100, "it saved a new record successfully");
    assert.strictEqual(widget.name, "Evil Widget");

    // Update
    await widget.update({ name: "new name" });
    assert.strictEqual(widget.name, "new name");

    // Destroy
    await widget.destroyRecord();

    // Lookup
    const foundWidget = await store.find("my-widget", 123);
    assert.strictEqual(foundWidget.name, "Trout Lure");
  });
});