supermemory/apps/web/lib/plugin-document.test.ts
2026-07-13 21:22:07 +05:30

118 lines
3.4 KiB
TypeScript

import { describe, expect, it } from "bun:test"
import { parsePluginDocument } from "./plugin-document"
type PluginDocumentInput = Parameters<typeof parsePluginDocument>[0]
function makeCodexSessionDocument(content: string): PluginDocumentInput {
return {
id: "doc_1",
title: "Codex session",
content,
metadata: { sm_source: "codex" },
memoryEntries: [],
} as unknown as PluginDocumentInput
}
describe("parsePluginDocument — session transcripts", () => {
it("keeps the Codex source badge inside a shared Agents container", () => {
const parsed = parsePluginDocument({
...makeCodexSessionDocument(
["[Session abc-123]", "1. [user] Shared memory"].join("\n"),
),
source: "codex",
containerTags: ["user_project_0123456789abcdef"],
} as unknown as PluginDocumentInput)
expect(parsed?.pluginLabel).toBe("Codex")
expect(parsed?.pluginIconSrc).toBe("/images/plugins/codex.png")
})
it("keeps the Claude Code source badge inside a shared Agents container", () => {
const parsed = parsePluginDocument({
id: "doc_claude",
title: "Claude memory",
content: "Remember this project convention",
source: "claude-code",
metadata: {
sm_source: "claude-code",
type: "manual",
project: "supermemory",
},
containerTags: ["repo_supermemory"],
memoryEntries: [],
} as unknown as PluginDocumentInput)
expect(parsed?.pluginLabel).toBe("Claude Code")
expect(parsed?.pluginIconSrc).toBe("/images/plugins/claude-code.svg")
})
it("keeps multi-line message bodies intact", () => {
const parsed = parsePluginDocument(
makeCodexSessionDocument(
[
"[Session abc-123]",
"1. [user] Hello there",
"Here is more context on line two",
"2. [assistant] Sure!",
"Second line of the reply",
].join("\n"),
),
)
expect(parsed).not.toBeNull()
expect(parsed?.kind).toBe("codex-session")
expect(parsed?.messages).toHaveLength(2)
expect(parsed?.messages[0]?.text).toBe(
"Hello there\nHere is more context on line two",
)
expect(parsed?.messages[1]?.text).toBe("Sure!\nSecond line of the reply")
})
it("surfaces memory id artifacts from continuation lines", () => {
const parsed = parsePluginDocument(
makeCodexSessionDocument(
[
"[Session abc-123]",
"1. [user] Remember my editor is Neovim",
"memory id: mem_456",
"2. [assistant] Saved it.",
].join("\n"),
),
)
expect(parsed?.messages[0]?.text).toBe("Remember my editor is Neovim")
expect(parsed?.artifacts).toContainEqual({
label: "Memory ID",
value: "mem_456",
})
})
it("normalizes literal \\n escapes before splitting messages", () => {
const parsed = parsePluginDocument(
makeCodexSessionDocument(
"[Session abc-123]\\n1. [user] First line\\nSecond line\\n2. [assistant] Reply",
),
)
expect(parsed?.messages).toHaveLength(2)
expect(parsed?.messages[0]?.text).toBe("First line\nSecond line")
expect(parsed?.messages[1]?.text).toBe("Reply")
})
it("parses single-line messages as before", () => {
const parsed = parsePluginDocument(
makeCodexSessionDocument(
["[Session abc-123]", "1. [user] Hi", "2. [assistant] Hello!"].join(
"\n",
),
),
)
expect(parsed?.messages).toHaveLength(2)
expect(parsed?.messages[0]?.text).toBe("Hi")
expect(parsed?.messages[1]?.text).toBe("Hello!")
expect(parsed?.summary).toBe(
"1 user message and 1 assistant message captured from Codex.",
)
})
})