import { getOwner } from "@ember/application"; import { clearRender, render } from "@ember/test-helpers"; import hbs from "htmlbars-inline-precompile"; import { module, test } from "qunit"; import { setupRenderingTest } from "discourse/tests/helpers/component-test"; import { exists } from "discourse/tests/helpers/qunit-helpers"; import ChatFabricators from "discourse/plugins/chat/discourse/lib/fabricators"; module("Discourse Chat | Component | chat-message", function (hooks) { setupRenderingTest(hooks); const template = hbs` `; test("Message with edits", async function (assert) { this.message = new ChatFabricators(getOwner(this)).message({ edited: true, }); await render(template); assert.true(exists(".chat-message-edited"), "has the correct css class"); }); test("Deleted message", async function (assert) { this.message = new ChatFabricators(getOwner(this)).message({ user: this.currentUser, deleted_at: moment(), }); await render(template); assert.true( exists(".chat-message-text.-deleted .chat-message-expand"), "has the correct css class and expand button within" ); }); test("Hidden message", async function (assert) { this.message = new ChatFabricators(getOwner(this)).message({ hidden: true, }); await render(template); assert.true( exists(".chat-message-text.-hidden .chat-message-expand"), "has the correct css class and expand button within" ); }); test("Message with reply", async function (assert) { this.message = new ChatFabricators(getOwner(this)).message({ inReplyTo: new ChatFabricators(getOwner(this)).message(), }); await render(template); assert.true( exists(".chat-message-container.has-reply"), "has the correct css class" ); }); test("Message with streaming", async function (assert) { // admin this.currentUser.admin = true; this.message = new ChatFabricators(getOwner(this)).message({ inReplyTo: new ChatFabricators(getOwner(this)).message(), streaming: true, }); await this.message.cook(); await render(template); assert .dom(".stop-streaming-btn") .exists("when admin, it has the stop streaming button"); await clearRender(); // not admin - not replying to current user this.currentUser.admin = false; this.message = new ChatFabricators(getOwner(this)).message({ inReplyTo: new ChatFabricators(getOwner(this)).message(), streaming: true, }); await this.message.cook(); await render(template); assert .dom(".stop-streaming-btn") .doesNotExist("when admin, it doesn't have the stop streaming button"); await clearRender(); // not admin - replying to current user this.currentUser.admin = false; this.message = new ChatFabricators(getOwner(this)).message({ inReplyTo: new ChatFabricators(getOwner(this)).message({ user: this.currentUser, }), streaming: true, }); await this.message.cook(); await render(template); assert .dom(".stop-streaming-btn") .exists( "when replying to current user, it has the stop streaming button" ); }); });