git.giacomolli.uk - onechat/blob - tree_node_helper_test.go

   1 package main
   2
   3 import (
   4         "testing"
   5
   6         "github.com/stretchr/testify/require"
   7 )
   8
   9 func treeTestNode(level, index int, text string) Node {
  10         return Node{
  11                 Level: level,
  12                 Index: index,
  13                 Text:  text,
  14                 Size:  len(text),
  15         }
  16 }
  17
  18 // treeReadNode reads the node (l, i) of tree, which must be built.
  19 func treeReadNode(t *testing.T, tree *Tree, level, index int) Node {
  20         t.Helper()
  21
  22         node, built, err := tree.Node(level, index)
  23         require.NoError(t, err)
  24         require.True(t, built, "node (%d, %d)", level, index)
  25
  26         return node
  27 }
  28
  29 // treeIsBuilt tells if the node (l, i) of tree is built.
  30 func treeIsBuilt(t *testing.T, tree *Tree, level, index int) bool {
  31         t.Helper()
  32
  33         built, err := tree.Built(level, index)
  34         require.NoError(t, err)
  35
  36         return built
  37 }