X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/6b21fcf78372995c6d8130c63f2f8aa2609bf7fe..c781f79f3c8cb9a7577ee1574d196988da2719b3:/tree_test.go?ds=inline diff --git a/tree_test.go b/tree_test.go index 46c17c7..2f23d4d 100644 --- a/tree_test.go +++ b/tree_test.go @@ -72,6 +72,7 @@ func TestTreeReopenLoadsTheSameNodes(t *testing.T) { treeTestNode(0, 0, "user: a"), treeTestNode(1, 0, "user: a\nreply: b"), } + for _, node := range nodes { require.NoError(t, tree.Add(node)) } @@ -80,7 +81,7 @@ func TestTreeReopenLoadsTheSameNodes(t *testing.T) { 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) @@ -90,6 +91,7 @@ func TestTreeReopenLoadsTheSameNodes(t *testing.T) { }() require.Equal(t, len(nodes), tree.Count()) + for _, node := range nodes { require.Equal(t, node, treeReadNode(t, tree, node.Level, node.Index)) } @@ -106,6 +108,7 @@ func TestTreeBuildsAMissingIndexAgain(t *testing.T) { treeTestNode(0, 0, "user: a"), treeTestNode(1, 0, "user: a\nreply: b"), } + for _, node := range nodes { require.NoError(t, tree.Add(node)) } @@ -125,6 +128,7 @@ func TestTreeBuildsAMissingIndexAgain(t *testing.T) { }() require.Equal(t, len(nodes), tree.Count()) + for _, node := range nodes { require.Equal(t, node, treeReadNode(t, tree, node.Level, node.Index)) }