microui-go/treenode_test.go at master · damiensgit/microui-go · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package microui

import (
	"testing"

	"github.com/user/microui-go/types"
)

func TestTreeNode_Basic(t *testing.T) {
	ui := New(Config{})
	ui.BeginFrame()

	ui.BeginWindow("Test", types.Rect{X: 0, Y: 0, W: 400, H: 300})
	ui.LayoutRow(1, []int{-1}, 0)

	if ui.BeginTreeNode("Node 1") {
		ui.Label("Child 1")
		ui.Label("Child 2")
		ui.EndTreeNode()
	}

	if ui.BeginTreeNode("Node 2") {
		ui.Label("Child A")
		ui.EndTreeNode()
	}

	ui.EndWindow()
	ui.EndFrame()
}

func TestTreeNode_Nested(t *testing.T) {
	ui := New(Config{})
	ui.BeginFrame()

	ui.BeginWindow("Test", types.Rect{X: 0, Y: 0, W: 400, H: 300})
	ui.LayoutRow(1, []int{-1}, 0)

	if ui.BeginTreeNode("Parent") {
		ui.Label("Direct child")

		if ui.BeginTreeNode("Nested") {
			ui.Label("Nested child")
			ui.EndTreeNode()
		}

		ui.EndTreeNode()
	}

	ui.EndWindow()
	ui.EndFrame()
}

func TestTreeNode_DefaultCollapsed(t *testing.T) {
	ui := New(Config{})

	// First frame - tree node should be collapsed by default
	ui.BeginFrame()
	ui.BeginWindow("Test", types.Rect{X: 0, Y: 0, W: 400, H: 300})
	ui.LayoutRow(1, []int{-1}, 30)

	expanded := ui.BeginTreeNode("Node")
	if expanded {
		ui.EndTreeNode()
	}

	ui.EndWindow()
	ui.EndFrame()

	// Default should be collapsed (unlike Header which is expanded by default)
	if expanded {
		t.Error("TreeNode should be collapsed by default")
	}
}

func TestTreeNode_OptExpanded(t *testing.T) {
	ui := New(Config{})

	// Use OptExpanded to start expanded
	ui.BeginFrame()
	ui.BeginWindow("Test", types.Rect{X: 0, Y: 0, W: 400, H: 300})
	ui.LayoutRow(1, []int{-1}, 30)

	expanded := ui.BeginTreeNodeEx("Node", OptExpanded)
	if expanded {
		ui.EndTreeNode()
	}

	ui.EndWindow()
	ui.EndFrame()

	// Should be expanded due to OptExpanded
	if !expanded {
		t.Error("TreeNode should be expanded when OptExpanded is set")
	}
}

func TestTreeNode_Toggle(t *testing.T) {
	ui := New(Config{})

	// First frame - tree node collapsed by default
	ui.BeginFrame()
	ui.BeginWindow("Test", types.Rect{X: 0, Y: 0, W: 400, H: 300})
	ui.LayoutRow(1, []int{-1}, 30)

	expanded1 := ui.BeginTreeNode("Node")
	if expanded1 {
		ui.EndTreeNode()
	}

	ui.EndWindow()
	ui.EndFrame()

	// Verify collapsed
	if expanded1 {
		t.Error("TreeNode should be collapsed initially")
	}

	// Click on tree node to toggle
	// TreeNode is at Y ~= 29 (after title 24 + padding 5), height ~= 20 (size.Y 10 + padding*2)
	// So click at Y=35 to be inside the control
	ui.MouseMove(50, 35)
	ui.MouseDown(50, 35, MouseLeft)
	ui.BeginFrame()

	ui.BeginWindow("Test", types.Rect{X: 0, Y: 0, W: 400, H: 300})
	ui.LayoutRow(1, []int{-1}, 30)

	expanded2 := ui.BeginTreeNode("Node")
	if expanded2 {
		ui.EndTreeNode()
	}

	ui.EndWindow()
	ui.EndFrame()

	// Should now be expanded after click
	if !expanded2 {
		t.Error("TreeNode should be expanded after click")
	}
}