discourse-ai/test/javascripts/acceptance/topic-summary-test.js

import { click, visit } from "@ember/test-helpers";
import { test } from "qunit";
import topicFixtures from "discourse/tests/fixtures/topic";
import {
  acceptance,
  publishToMessageBus,
  updateCurrentUser,
} from "discourse/tests/helpers/qunit-helpers";
import { cloneJSON } from "discourse-common/lib/object";

acceptance("Topic - Summary", function (needs) {
  const currentUserId = 5;

  needs.user();
  needs.pretender((server, helper) => {
    server.get("/t/1.json", () => {
      const json = cloneJSON(topicFixtures["/t/130.json"]);
      json.id = 1;
      json.summarizable = true;

      return helper.response(json);
    });

    server.get("/discourse-ai/summarization/t/1", () => {
      return helper.response({});
    });
  });

  needs.hooks.beforeEach(() => {
    updateCurrentUser({ id: currentUserId });
  });

  test("displays streamed summary", async function (assert) {
    await visit("/t/-/1");

    const partialSummary = "This a";
    await publishToMessageBus("/discourse-ai/summaries/topic/1", {
      done: false,
      ai_topic_summary: { summarized_text: partialSummary },
    });

    await click(".ai-topic-summarization");

    assert
      .dom(".ai-summary-box .generated-summary p")
      .hasText(partialSummary, "Updates the summary with a partial result");

    const finalSummary = "This is a completed summary";
    await publishToMessageBus("/discourse-ai/summaries/topic/1", {
      done: true,
      ai_topic_summary: {
        summarized_text: finalSummary,
        summarized_on: "2023-01-01T04:00:00.000Z",
        algorithm: "OpenAI GPT-4",
        outdated: false,
        new_posts_since_summary: false,
        can_regenerate: true,
      },
    });

    assert
      .dom(".ai-summary-box .generated-summary p")
      .hasText(finalSummary, "Updates the summary with a final result");

    assert
      .dom(".ai-summary-box .summarized-on")
      .exists("summary metadata exists");
  });
});

acceptance("Topic - Summary - Anon", function (needs) {
  const finalSummary = "This is a completed summary";

  needs.pretender((server, helper) => {
    server.get("/t/1.json", () => {
      const json = cloneJSON(topicFixtures["/t/280/1.json"]);
      json.id = 1;
      json.summarizable = true;

      return helper.response(json);
    });

    server.get("/discourse-ai/summarization/t/1", () => {
      return helper.response({
        ai_topic_summary: {
          summarized_text: finalSummary,
          summarized_on: "2023-01-01T04:00:00.000Z",
          algorithm: "OpenAI GPT-4",
          outdated: false,
          new_posts_since_summary: false,
          can_regenerate: false,
        },
      });
    });
  });

  test("displays cached summary immediately", async function (assert) {
    await visit("/t/-/1");

    await click(".ai-topic-summarization");

    assert
      .dom(".ai-summary-box .generated-summary p")
      .hasText(finalSummary, "Updates the summary with the result");

    assert
      .dom(".ai-summary-box .summarized-on")
      .exists("summary metadata exists");
  });
});