X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/000ae97acb21a1c0c90e9278177a74411ba6eab2..c781f79f3c8cb9a7577ee1574d196988da2719b3:/tree_test.go?ds=sidebyside diff --git a/tree_test.go b/tree_test.go index f2cec17..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)) } @@ -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)) }