108 lines
3.0 KiB
TypeScript
108 lines
3.0 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import { DEFAULT_HEARTBEAT_ACK_MAX_CHARS, stripHeartbeatToken } from "./heartbeat.js";
|
|
import { HEARTBEAT_TOKEN } from "./tokens.js";
|
|
|
|
describe("stripHeartbeatToken", () => {
|
|
it("skips empty or token-only replies", () => {
|
|
expect(stripHeartbeatToken(undefined, { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: false,
|
|
});
|
|
expect(stripHeartbeatToken(" ", { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: false,
|
|
});
|
|
expect(stripHeartbeatToken(HEARTBEAT_TOKEN, { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: true,
|
|
});
|
|
});
|
|
|
|
it("drops heartbeats with small junk in heartbeat mode", () => {
|
|
expect(stripHeartbeatToken("HEARTBEAT_OK 🦞", { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: true,
|
|
});
|
|
expect(stripHeartbeatToken(`🦞 ${HEARTBEAT_TOKEN}`, { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: true,
|
|
});
|
|
});
|
|
|
|
it("drops short remainder in heartbeat mode", () => {
|
|
expect(stripHeartbeatToken(`ALERT ${HEARTBEAT_TOKEN}`, { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: true,
|
|
});
|
|
});
|
|
|
|
it("keeps heartbeat replies when remaining content exceeds threshold", () => {
|
|
const long = "A".repeat(DEFAULT_HEARTBEAT_ACK_MAX_CHARS + 1);
|
|
expect(stripHeartbeatToken(`${long} ${HEARTBEAT_TOKEN}`, { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: false,
|
|
text: long,
|
|
didStrip: true,
|
|
});
|
|
});
|
|
|
|
it("strips token at edges for normal messages", () => {
|
|
expect(stripHeartbeatToken(`${HEARTBEAT_TOKEN} hello`, { mode: "message" })).toEqual({
|
|
shouldSkip: false,
|
|
text: "hello",
|
|
didStrip: true,
|
|
});
|
|
expect(stripHeartbeatToken(`hello ${HEARTBEAT_TOKEN}`, { mode: "message" })).toEqual({
|
|
shouldSkip: false,
|
|
text: "hello",
|
|
didStrip: true,
|
|
});
|
|
});
|
|
|
|
it("does not touch token in the middle", () => {
|
|
expect(
|
|
stripHeartbeatToken(`hello ${HEARTBEAT_TOKEN} there`, {
|
|
mode: "message",
|
|
}),
|
|
).toEqual({
|
|
shouldSkip: false,
|
|
text: `hello ${HEARTBEAT_TOKEN} there`,
|
|
didStrip: false,
|
|
});
|
|
});
|
|
|
|
it("strips HTML-wrapped heartbeat tokens", () => {
|
|
expect(stripHeartbeatToken(`<b>${HEARTBEAT_TOKEN}</b>`, { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: true,
|
|
});
|
|
});
|
|
|
|
it("strips markdown-wrapped heartbeat tokens", () => {
|
|
expect(stripHeartbeatToken(`**${HEARTBEAT_TOKEN}**`, { mode: "heartbeat" })).toEqual({
|
|
shouldSkip: true,
|
|
text: "",
|
|
didStrip: true,
|
|
});
|
|
});
|
|
|
|
it("removes markup-wrapped token and keeps trailing content", () => {
|
|
expect(
|
|
stripHeartbeatToken(`<code>${HEARTBEAT_TOKEN}</code> all good`, {
|
|
mode: "message",
|
|
}),
|
|
).toEqual({
|
|
shouldSkip: false,
|
|
text: "all good",
|
|
didStrip: true,
|
|
});
|
|
});
|
|
});
|