Improved test coverage

src/build/expression/Expression_test.go

@@ -1,6 +1,7 @@
 package expression_test
 
 import (
+	"fmt"
 	"testing"
 
 	"git.akyoto.dev/cli/q/src/build/expression"
@@ -95,12 +96,47 @@ func TestExpressionParse(t *testing.T) {
 	}
 }
 
-func BenchmarkExpression(b *testing.B) {
-	src := []byte("(1+2-3*4)+(5*6-7+8)\n")
+func TestEachLeaf(t *testing.T) {
+	src := []byte("(1+2-3*4)+(5*6-7+8)")
 	tokens := token.Tokenize(src)
+	expr := expression.Parse(tokens)
+	leaves := []string{}
 
-	for i := 0; i < b.N; i++ {
-		expr := expression.Parse(tokens)
-		expr.Close()
-	}
+	expr.EachLeaf(func(leaf *expression.Expression) error {
+		leaves = append(leaves, leaf.Token.Text())
+		return nil
+	})
+
+	assert.DeepEqual(t, leaves, []string{"1", "2", "3", "4", "5", "6", "7", "8"})
+
+	err := expr.EachLeaf(func(leaf *expression.Expression) error {
+		return fmt.Errorf("error")
+	})
+
+	assert.NotNil(t, err)
+	assert.Equal(t, err.Error(), "error")
+}
+
+func TestRemoveChild(t *testing.T) {
+	src := []byte("(1+2-3*4)+(5*6-7+8)")
+	tokens := token.Tokenize(src)
+	expr := expression.Parse(tokens)
+	left := expr.Children[0]
+	right := expr.Children[1]
+	expr.RemoveChild(left)
+	assert.Equal(t, expr.Children[0], right)
+}
+
+func TestNilExpression(t *testing.T) {
+	src := []byte("")
+	tokens := token.Tokenize(src)
+	expr := expression.Parse(tokens)
+	assert.Nil(t, expr)
+}
+
+func TestNilGroup(t *testing.T) {
+	src := []byte("()")
+	tokens := token.Tokenize(src)
+	expr := expression.Parse(tokens)
+	assert.Nil(t, expr)
 }