File size: 747 Bytes
3e05655
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import { expect, test } from "bun:test"
import { shouldResetCodeTokens } from "./markdown-code-state"

const previous = {
  language: "ts",
  generation: 1,
  stableCount: 3,
  unstable: [],
  raw: "```ts\nconst x = 1\n```",
}

test("resets tokens for a non-prefix replacement with the same generation and token count", () => {
  expect(
    shouldResetCodeTokens(previous, {
      language: "ts",
      generation: 1,
      stableCount: 3,
      raw: "```ts\nlet y = 2\n```",
    }),
  ).toBe(true)
})

test("retains tokens for an append-only streaming update", () => {
  expect(
    shouldResetCodeTokens(previous, {
      language: "ts",
      generation: 1,
      stableCount: 4,
      raw: `${previous.raw}\nmore`,
    }),
  ).toBe(false)
})