discourse/plugins/chat/test/javascripts/components/chat-composer-uploads-test.js

import { setupRenderingTest } from "discourse/tests/helpers/component-test";
import pretender from "discourse/tests/helpers/create-pretender";
import {
  count,
  createFile,
  exists,
} from "discourse/tests/helpers/qunit-helpers";
import hbs from "htmlbars-inline-precompile";
import { click, render, settled, waitFor } from "@ember/test-helpers";
import { module, test } from "qunit";
import { run } from "@ember/runloop";

const fakeUpload = {
  type: ".png",
  extension: "png",
  name: "myfile.png",
  short_path: "/images/avatar.png",
};

const mockUploadResponse = {
  extension: "jpeg",
  filesize: 126177,
  height: 800,
  human_filesize: "123 KB",
  id: 202,
  original_filename: "avatar.PNG.jpg",
  retain_hours: null,
  short_path: "/images/avatar.png",
  short_url: "upload://yoj8pf9DdIeHRRULyw7i57GAYdz.jpeg",
  thumbnail_height: 320,
  thumbnail_width: 690,
  url: "/images/avatar.png",
  width: 1920,
};

function setupUploadPretender() {
  pretender.post(
    "/uploads.json",
    () => {
      return [200, { "Content-Type": "application/json" }, mockUploadResponse];
    },
    500 // this delay is important to slow down the uploads a bit so we can click elements in the UI like the cancel button
  );
}

module("Discourse Chat | Component | chat-composer-uploads", function (hooks) {
  setupRenderingTest(hooks);

  test("loading uploads from an outside source (e.g. draft or editing message)", async function (assert) {
    await render(hbs`
      <ChatComposerUploads @fileUploadElementId="chat-widget-uploader" />
    `);

    this.appEvents = this.container.lookup("service:appEvents");
    this.appEvents.trigger("chat-composer:load-uploads", [fakeUpload]);
    await settled();

    assert.strictEqual(count(".chat-composer-upload"), 1);
    assert.strictEqual(exists(".chat-composer-upload"), true);
  });

  test("upload starts and completes", async function (assert) {
    setupUploadPretender();
    this.set("changedUploads", null);
    this.set("onUploadChanged", (uploads) => {
      this.set("changedUploads", uploads);
    });

    await render(hbs`
      <ChatComposerUploads @fileUploadElementId="chat-widget-uploader" @onUploadChanged={{this.onUploadChanged}} />
    `);

    const done = assert.async();
    this.appEvents = this.container.lookup("service:appEvents");
    this.appEvents.on(
      "upload-mixin:chat-composer-uploader:upload-success",
      (fileName, upload) => {
        assert.strictEqual(fileName, "avatar.png");
        assert.deepEqual(upload, mockUploadResponse);
        done();
      }
    );

    this.appEvents.trigger(
      "upload-mixin:chat-composer-uploader:add-files",
      createFile("avatar.png")
    );

    await waitFor(".chat-composer-upload");
    assert.strictEqual(count(".chat-composer-upload"), 1);
  });

  test("removing a completed upload", async function (assert) {
    this.set("changedUploads", null);
    this.set("onUploadChanged", (uploads) => {
      this.set("changedUploads", uploads);
    });

    await render(hbs`
      <ChatComposerUploads @fileUploadElementId="chat-widget-uploader" @onUploadChanged={{this.onUploadChanged}} />
    `);

    this.appEvents = this.container.lookup("service:appEvents");
    run(() =>
      this.appEvents.trigger("chat-composer:load-uploads", [fakeUpload])
    );
    assert.strictEqual(count(".chat-composer-upload"), 1);

    await click(".remove-upload");
    assert.strictEqual(count(".chat-composer-upload"), 0);
  });

  test("cancelling in progress upload", async function (assert) {
    setupUploadPretender();

    this.set("changedUploads", null);
    this.set("onUploadChanged", (uploads) => {
      this.set("changedUploads", uploads);
    });

    await render(hbs`
      <ChatComposerUploads @fileUploadElementId="chat-widget-uploader" @onUploadChanged={{this.onUploadChanged}} />
    `);

    const image = createFile("avatar.png");
    const done = assert.async();
    this.appEvents = this.container.lookup("service:appEvents");

    this.appEvents.on(
      `upload-mixin:chat-composer-uploader:upload-cancelled`,
      (fileId) => {
        assert.strictEqual(
          fileId.includes("uppy-avatar/"),
          true,
          "upload was cancelled"
        );
        done();
      }
    );

    this.appEvents.trigger(
      "upload-mixin:chat-composer-uploader:add-files",
      image
    );

    await waitFor(".chat-composer-upload");
    assert.strictEqual(count(".chat-composer-upload"), 1);

    await click(".remove-upload");
    assert.strictEqual(count(".chat-composer-upload"), 0);
  });
});