shad-go/treeiter/treeiter_test.go

package treeiter_test

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"gitlab.com/slon/shad-go/treeiter"
)

type ValuesNode[T any] struct {
	value       T
	left, right *ValuesNode[T]
}

func (t ValuesNode[T]) Left() *ValuesNode[T] {
	return t.left
}

func (t ValuesNode[T]) Right() *ValuesNode[T] {
	return t.right
}

type Collector[T any] struct {
	nodes []*ValuesNode[T]
}

func (c *Collector[T]) Collect(tree *ValuesNode[T]) {
	c.nodes = append(c.nodes, tree)
}

func TestInOrderNil(t *testing.T) {
	var collector Collector[any]
	treeiter.DoInOrder(nil, collector.Collect)
}

func TestInOrderIntTree(t *testing.T) {
	root := &ValuesNode[int]{value: 1}
	collector := Collector[int]{}

	treeiter.DoInOrder(root, collector.Collect)

	assert.Equal(t, []*ValuesNode[int]{{value: 1}}, collector.nodes)
}

func TestInOrderStringTree(t *testing.T) {
	rightLeftRight := &ValuesNode[string]{
		value: "right left right",
	}
	rightLeft := &ValuesNode[string]{
		value: "right left",
		right: rightLeftRight,
	}
	right := &ValuesNode[string]{
		value: "right",
		left:  rightLeft,
	}
	left := &ValuesNode[string]{
		value: "left",
	}

	root := &ValuesNode[string]{
		value: "root",
		left:  left,
		right: right,
	}
	var collector Collector[string]

	treeiter.DoInOrder(root, collector.Collect)

	assert.Equal(t, []*ValuesNode[string]{left, root, rightLeft, rightLeftRight, right}, collector.nodes)
}