data, err := os.ReadFile(filepath.Join(root, "tree", "2026-01-01.jsonl"))
require.NoError(t, err)
- require.Equal(t, `{"l":0,"i":1,"text":"reply: b","size":8}`+"\n", strings.SplitAfter(string(data), "\n")[0])
+ require.JSONEq(t, `{"l":0,"i":1,"text":"reply: b","size":8}`+"\n", strings.SplitAfter(string(data), "\n")[0])
tree, err = openTree(t.Context(), root, 2, clock.Now)
require.NoError(t, err)
require.Equal(t, "reply: b", treeReadNode(t, tree, 0, 1).Text)
require.NoError(t, tree.Add(treeTestNode(1, 0, "user: a\nreply: b")))
}
-
-func treeTestNode(level, index int, text string) Node {
- return Node{
- Level: level,
- Index: index,
- Text: text,
- Size: len(text),
- }
-}
-
-// treeReadNode reads the node (l, i) of tree, which must be built.
-func treeReadNode(t *testing.T, tree *Tree, level, index int) Node {
- t.Helper()
-
- node, built, err := tree.Node(level, index)
- require.NoError(t, err)
- require.True(t, built, "node (%d, %d)", level, index)
-
- return node
-}
-
-// treeIsBuilt tells if the node (l, i) of tree is built.
-func treeIsBuilt(t *testing.T, tree *Tree, level, index int) bool {
- t.Helper()
-
- built, err := tree.Built(level, index)
- require.NoError(t, err)
-
- return built
-}