1 /* Unit tests for treeview.
3 * Copyright 2005 Krzysztof Foltman
4 * Copyright 2007 Christopher James Peterson
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
32 #include "wine/test.h"
35 const char *TEST_CALLBACK_TEXT
= "callback_text";
37 #define NUM_MSG_SEQUENCES 1
38 #define TREEVIEW_SEQ_INDEX 0
40 #define expect(expected, got) ok(got == expected, "Expected %d, got %d\n", expected, got)
42 static struct msg_sequence
*MsgSequences
[NUM_MSG_SEQUENCES
];
44 static const struct message FillRootSeq
[] = {
45 { TVM_INSERTITEM
, sent
},
46 { TVM_INSERTITEM
, sent
},
50 static const struct message rootnone_select_seq
[] = {
51 { TVM_SELECTITEM
, sent
|wparam
, 9 },
52 { TVM_SELECTITEM
, sent
|wparam
, 9 },
53 { TVM_SELECTITEM
, sent
|wparam
, 9 },
54 { TVM_SELECTITEM
, sent
|wparam
, 9 },
55 { TVM_SELECTITEM
, sent
|wparam
, 9 },
56 { TVM_SELECTITEM
, sent
|wparam
, 9 },
60 static const struct message rootchild_select_seq
[] = {
61 { TVM_SELECTITEM
, sent
|wparam
, 9 },
62 { TVM_SELECTITEM
, sent
|wparam
, 9 },
63 { TVM_SELECTITEM
, sent
|wparam
, 9 },
64 { TVM_SELECTITEM
, sent
|wparam
, 9 },
65 { TVM_SELECTITEM
, sent
|wparam
, 9 },
66 { TVM_SELECTITEM
, sent
|wparam
, 9 },
70 static const struct message getitemtext_seq
[] = {
71 { TVM_INSERTITEM
, sent
},
72 { TVM_GETITEM
, sent
},
73 { TVM_DELETEITEM
, sent
},
77 static const struct message focus_seq
[] = {
78 { TVM_INSERTITEM
, sent
},
79 { TVM_INSERTITEM
, sent
},
80 { TVM_SELECTITEM
, sent
|wparam
, 9 },
81 /* The following end up out of order in wine */
82 { WM_WINDOWPOSCHANGING
, sent
|defwinproc
},
83 { WM_NCCALCSIZE
, sent
|wparam
|defwinproc
, TRUE
},
84 { WM_WINDOWPOSCHANGED
, sent
|defwinproc
},
85 { WM_SIZE
, sent
|defwinproc
},
86 { WM_PAINT
, sent
|defwinproc
},
87 { WM_NCPAINT
, sent
|wparam
|defwinproc
, 1 },
88 { WM_ERASEBKGND
, sent
|defwinproc
},
89 { TVM_EDITLABEL
, sent
},
90 { WM_COMMAND
, sent
|wparam
|defwinproc
, MAKEWPARAM(0, EN_UPDATE
) },
91 { WM_COMMAND
, sent
|wparam
|defwinproc
, MAKEWPARAM(0, EN_CHANGE
) },
92 { WM_PARENTNOTIFY
, sent
|wparam
|defwinproc
, MAKEWPARAM(WM_CREATE
, 0) },
93 { WM_KILLFOCUS
, sent
|defwinproc
},
94 { WM_PAINT
, sent
|defwinproc
},
95 { WM_IME_SETCONTEXT
, sent
|defwinproc
|optional
},
96 { WM_COMMAND
, sent
|wparam
|defwinproc
, MAKEWPARAM(0, EN_SETFOCUS
) },
97 { WM_ERASEBKGND
, sent
|defwinproc
|optional
},
98 { WM_CTLCOLOREDIT
, sent
|defwinproc
|optional
},
99 { WM_CTLCOLOREDIT
, sent
|defwinproc
|optional
},
103 static const struct message test_get_set_bkcolor_seq
[] = {
104 { TVM_GETBKCOLOR
, sent
|wparam
|lparam
, 0, 0 },
105 { TVM_SETBKCOLOR
, sent
|wparam
|lparam
, 0, 0 },
106 { TVM_GETBKCOLOR
, sent
|wparam
|lparam
, 0, 0 },
107 { TVM_SETBKCOLOR
, sent
|wparam
|lparam
, 0, 0x00ffffff },
108 { TVM_GETBKCOLOR
, sent
|wparam
|lparam
, 0, 0 },
109 { TVM_SETBKCOLOR
, sent
|wparam
|lparam
, 0, -1 },
113 static const struct message test_get_set_imagelist_seq
[] = {
114 { TVM_SETIMAGELIST
, sent
|wparam
|lparam
, 0, 0 },
115 { TVM_GETIMAGELIST
, sent
|wparam
|lparam
, 0, 0 },
119 static const struct message test_get_set_indent_seq
[] = {
120 { TVM_SETINDENT
, sent
|wparam
|lparam
, 0, 0 },
121 { TVM_GETINDENT
, sent
|wparam
|lparam
, 0, 0 },
122 /* The actual amount to indent is dependent on the system for this message */
123 { TVM_SETINDENT
, sent
},
124 { TVM_GETINDENT
, sent
|wparam
|lparam
, 0, 0 },
128 static const struct message test_get_set_insertmarkcolor_seq
[] = {
129 { TVM_SETINSERTMARKCOLOR
, sent
|wparam
|lparam
, 0, 0 },
130 { TVM_GETINSERTMARKCOLOR
, sent
|wparam
|lparam
, 0, 0 },
134 static const struct message test_get_set_item_seq
[] = {
135 { TVM_GETITEM
, sent
},
136 { TVM_SETITEM
, sent
},
137 { TVM_GETITEM
, sent
},
138 { TVM_SETITEM
, sent
},
142 static const struct message test_get_set_itemheight_seq
[] = {
143 { TVM_GETITEMHEIGHT
, sent
|wparam
|lparam
, 0, 0 },
144 { TVM_SETITEMHEIGHT
, sent
|wparam
|lparam
, -1, 0 },
145 { TVM_GETITEMHEIGHT
, sent
|wparam
|lparam
, 0, 0 },
146 { TVM_SETITEMHEIGHT
, sent
|lparam
, 0xcccccccc, 0 },
147 { TVM_GETITEMHEIGHT
, sent
|wparam
|lparam
|optional
, 0, 0 },
148 { TVM_SETITEMHEIGHT
, sent
|wparam
|lparam
|optional
, 9, 0 },
149 { TVM_GETITEMHEIGHT
, sent
|wparam
|lparam
, 0, 0 },
153 static const struct message test_get_set_scrolltime_seq
[] = {
154 { TVM_SETSCROLLTIME
, sent
|wparam
|lparam
, 20, 0 },
155 { TVM_GETSCROLLTIME
, sent
|wparam
|lparam
, 0, 0 },
159 static const struct message test_get_set_textcolor_seq
[] = {
160 { TVM_GETTEXTCOLOR
, sent
|wparam
|lparam
, 0, 0 },
161 { TVM_SETTEXTCOLOR
, sent
|wparam
|lparam
, 0, 0 },
162 { TVM_GETTEXTCOLOR
, sent
|wparam
|lparam
, 0, 0 },
163 { TVM_SETTEXTCOLOR
, sent
|wparam
|lparam
, 0, RGB(255, 255, 255) },
164 { TVM_GETTEXTCOLOR
, sent
|wparam
|lparam
, 0, 0 },
165 { TVM_SETTEXTCOLOR
, sent
|wparam
|lparam
, 0, CLR_NONE
},
169 static const struct message test_get_set_tooltips_seq
[] = {
170 { WM_KILLFOCUS
, sent
},
171 { WM_IME_SETCONTEXT
, sent
|optional
},
172 { WM_IME_NOTIFY
, sent
|optional
},
173 { TVM_SETTOOLTIPS
, sent
|wparam
|lparam
, 0, 0 },
174 { TVM_GETTOOLTIPS
, sent
|wparam
|lparam
, 0, 0 },
178 static const struct message test_get_set_unicodeformat_seq
[] = {
179 { TVM_SETUNICODEFORMAT
, sent
|wparam
|lparam
, TRUE
, 0 },
180 { TVM_GETUNICODEFORMAT
, sent
|wparam
|lparam
, 0, 0 },
181 { TVM_SETUNICODEFORMAT
, sent
|wparam
|lparam
, 0, 0 },
182 { TVM_GETUNICODEFORMAT
, sent
|wparam
|lparam
, 0, 0 },
183 { TVM_SETUNICODEFORMAT
, sent
|wparam
|lparam
, 0, 0 },
187 static HWND hMainWnd
;
189 static HTREEITEM hRoot
, hChild
;
192 static char sequence
[256];
194 static void Clear(void)
200 static void AddItem(char ch
)
202 sequence
[pos
++] = ch
;
203 sequence
[pos
] = '\0';
206 static void IdentifyItem(HTREEITEM hItem
)
208 if (hItem
== hRoot
) {
212 if (hItem
== hChild
) {
223 /* This function hooks in and records all messages to the treeview control */
224 static LRESULT WINAPI
TreeviewWndProc(HWND hwnd
, UINT message
, WPARAM wParam
, LPARAM lParam
)
226 static LONG defwndproc_counter
= 0;
229 WNDPROC lpOldProc
= (WNDPROC
)GetWindowLongPtrA(hwnd
, GWLP_USERDATA
);
231 msg
.message
= message
;
232 msg
.flags
= sent
|wparam
|lparam
;
233 if (defwndproc_counter
) msg
.flags
|= defwinproc
;
236 add_message(MsgSequences
, TREEVIEW_SEQ_INDEX
, &msg
);
238 defwndproc_counter
++;
239 ret
= CallWindowProcA(lpOldProc
, hwnd
, message
, wParam
, lParam
);
240 defwndproc_counter
--;
245 static HWND
create_treeview_control(void)
250 hTree
= CreateWindowExA(WS_EX_CLIENTEDGE
, WC_TREEVIEWA
, NULL
, WS_CHILD
|WS_VISIBLE
|
251 TVS_LINESATROOT
|TVS_HASLINES
|TVS_HASBUTTONS
|TVS_EDITLABELS
,
252 0, 0, 120, 100, hMainWnd
, (HMENU
)100, GetModuleHandleA(0), 0);
256 /* Record the old WNDPROC so we can call it after recording the messages */
257 pOldWndProc
= (WNDPROC
)SetWindowLongPtrA(hTree
, GWLP_WNDPROC
, (LONG_PTR
)TreeviewWndProc
);
258 SetWindowLongPtrA(hTree
, GWLP_USERDATA
, (LONG_PTR
)pOldWndProc
);
263 static void fill_tree(HWND hTree
)
266 static CHAR root
[] = "Root",
269 ins
.hParent
= TVI_ROOT
;
270 ins
.hInsertAfter
= TVI_ROOT
;
271 U(ins
).item
.mask
= TVIF_TEXT
;
272 U(ins
).item
.pszText
= root
;
273 hRoot
= TreeView_InsertItem(hTree
, &ins
);
276 ins
.hInsertAfter
= TVI_FIRST
;
277 U(ins
).item
.mask
= TVIF_TEXT
;
278 U(ins
).item
.pszText
= child
;
279 hChild
= TreeView_InsertItem(hTree
, &ins
);
282 static void test_fillroot(void)
287 hTree
= create_treeview_control();
289 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
299 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, FillRootSeq
, "FillRoot", FALSE
);
300 ok(!strcmp(sequence
, "AB."), "Item creation\n");
302 /* UMLPad 1.15 depends on this being not -1 (I_IMAGECALLBACK) */
304 tvi
.mask
= TVIF_IMAGE
| TVIF_SELECTEDIMAGE
;
305 SendMessage( hTree
, TVM_GETITEM
, 0, (LPARAM
)&tvi
);
306 ok(tvi
.iImage
== 0, "tvi.iImage=%d\n", tvi
.iImage
);
307 ok(tvi
.iSelectedImage
== 0, "tvi.iSelectedImage=%d\n", tvi
.iSelectedImage
);
309 DestroyWindow(hTree
);
312 static void test_callback(void)
315 HTREEITEM hItem1
, hItem2
;
318 CHAR test_string
[] = "Test_string";
323 hTree
= create_treeview_control();
326 ret
= TreeView_DeleteAllItems(hTree
);
327 ok(ret
== TRUE
, "ret\n");
328 ins
.hParent
= TVI_ROOT
;
329 ins
.hInsertAfter
= TVI_ROOT
;
330 U(ins
).item
.mask
= TVIF_TEXT
;
331 U(ins
).item
.pszText
= LPSTR_TEXTCALLBACK
;
332 hRoot
= TreeView_InsertItem(hTree
, &ins
);
336 tvi
.mask
= TVIF_TEXT
;
338 tvi
.cchTextMax
= sizeof(buf
)/sizeof(buf
[0]);
339 ret
= TreeView_GetItem(hTree
, &tvi
);
340 ok(ret
== 1, "ret\n");
341 ok(strcmp(tvi
.pszText
, TEST_CALLBACK_TEXT
) == 0, "Callback item text mismatch %s vs %s\n",
342 tvi
.pszText
, TEST_CALLBACK_TEXT
);
345 ins
.hInsertAfter
= TVI_FIRST
;
346 U(ins
).item
.mask
= TVIF_TEXT
;
347 U(ins
).item
.pszText
= test_string
;
348 hItem1
= TreeView_InsertItem(hTree
, &ins
);
352 ret
= TreeView_GetItem(hTree
, &tvi
);
353 ok(ret
== TRUE
, "ret\n");
354 ok(strcmp(tvi
.pszText
, test_string
) == 0, "Item text mismatch %s vs %s\n",
355 tvi
.pszText
, test_string
);
357 /* undocumented: pszText of NULL also means LPSTR_CALLBACK: */
359 ret
= TreeView_SetItem(hTree
, &tvi
);
360 ok(ret
== 1, "Expected SetItem return 1, got %ld\n", ret
);
362 ret
= TreeView_GetItem(hTree
, &tvi
);
363 ok(ret
== TRUE
, "Expected GetItem return TRUE, got %ld\n", ret
);
364 ok(strcmp(tvi
.pszText
, TEST_CALLBACK_TEXT
) == 0, "Item text mismatch %s vs %s\n",
365 tvi
.pszText
, TEST_CALLBACK_TEXT
);
367 U(ins
).item
.pszText
= NULL
;
368 hItem2
= TreeView_InsertItem(hTree
, &ins
);
371 memset(buf
, 0, sizeof(buf
));
372 ret
= TreeView_GetItem(hTree
, &tvi
);
373 ok(ret
== TRUE
, "Expected GetItem return TRUE, got %ld\n", ret
);
374 ok(strcmp(tvi
.pszText
, TEST_CALLBACK_TEXT
) == 0, "Item text mismatch %s vs %s\n",
375 tvi
.pszText
, TEST_CALLBACK_TEXT
);
377 DestroyWindow(hTree
);
380 static void test_select(void)
385 hTree
= create_treeview_control();
388 /* root-none select tests */
389 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
390 r
= TreeView_SelectItem(hTree
, NULL
);
394 r
= TreeView_SelectItem(hTree
, hRoot
);
397 r
= TreeView_SelectItem(hTree
, hRoot
);
400 r
= TreeView_SelectItem(hTree
, NULL
);
403 r
= TreeView_SelectItem(hTree
, NULL
);
406 r
= TreeView_SelectItem(hTree
, hRoot
);
409 ok(!strcmp(sequence
, "1(nR)nR23(Rn)Rn45(nR)nR."), "root-none select test\n");
410 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, rootnone_select_seq
,
411 "root-none select seq", FALSE
);
413 /* root-child select tests */
414 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
415 r
= TreeView_SelectItem(hTree
, NULL
);
420 r
= TreeView_SelectItem(hTree
, hRoot
);
423 r
= TreeView_SelectItem(hTree
, hRoot
);
426 r
= TreeView_SelectItem(hTree
, hChild
);
429 r
= TreeView_SelectItem(hTree
, hChild
);
432 r
= TreeView_SelectItem(hTree
, hRoot
);
435 ok(!strcmp(sequence
, "1(nR)nR23(RC)RC45(CR)CR."), "root-child select test\n");
436 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, rootchild_select_seq
,
437 "root-child select seq", FALSE
);
439 DestroyWindow(hTree
);
442 static void test_getitemtext(void)
449 CHAR szBuffer
[80] = "Blah";
450 int nBufferSize
= sizeof(szBuffer
)/sizeof(CHAR
);
452 hTree
= create_treeview_control();
455 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
457 /* add an item without TVIF_TEXT mask and pszText == NULL */
459 ins
.hInsertAfter
= TVI_ROOT
;
460 U(ins
).item
.mask
= 0;
461 U(ins
).item
.pszText
= NULL
;
462 U(ins
).item
.cchTextMax
= 0;
463 hChild
= TreeView_InsertItem(hTree
, &ins
);
466 /* retrieve it with TVIF_TEXT mask */
468 tvi
.mask
= TVIF_TEXT
;
469 tvi
.cchTextMax
= nBufferSize
;
470 tvi
.pszText
= szBuffer
;
472 SendMessageA( hTree
, TVM_GETITEM
, 0, (LPARAM
)&tvi
);
473 ok(!strcmp(szBuffer
, ""), "szBuffer=\"%s\", expected \"\"\n", szBuffer
);
474 ok(SendMessageA(hTree
, TVM_DELETEITEM
, 0, (LPARAM
)hChild
), "DeleteItem failed\n");
475 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, getitemtext_seq
, "get item text seq", FALSE
);
477 DestroyWindow(hTree
);
480 static void test_focus(void)
483 static CHAR child1
[] = "Edit",
484 child2
[] = "A really long string";
485 HTREEITEM hChild1
, hChild2
;
489 hTree
= create_treeview_control();
492 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
494 /* This test verifies that when a label is being edited, scrolling
495 * the treeview does not cause the label to lose focus. To test
496 * this, first some additional entries are added to generate
500 ins
.hInsertAfter
= hChild
;
501 U(ins
).item
.mask
= TVIF_TEXT
;
502 U(ins
).item
.pszText
= child1
;
503 hChild1
= TreeView_InsertItem(hTree
, &ins
);
505 ins
.hInsertAfter
= hChild1
;
506 U(ins
).item
.mask
= TVIF_TEXT
;
507 U(ins
).item
.pszText
= child2
;
508 hChild2
= TreeView_InsertItem(hTree
, &ins
);
511 ShowWindow(hMainWnd
,SW_SHOW
);
512 SendMessageA(hTree
, TVM_SELECTITEM
, TVGN_CARET
, (LPARAM
)hChild
);
513 hEdit
= TreeView_EditLabel(hTree
, hChild
);
514 ScrollWindowEx(hTree
, -10, 0, NULL
, NULL
, NULL
, NULL
, SW_SCROLLCHILDREN
);
515 ok(GetFocus() == hEdit
, "Edit control should have focus\n");
516 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, focus_seq
, "focus test", TRUE
);
518 DestroyWindow(hTree
);
521 static void test_get_set_bkcolor(void)
523 COLORREF crColor
= RGB(0,0,0);
526 hTree
= create_treeview_control();
529 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
531 /* If the value is -1, the control is using the system color for the background color. */
532 crColor
= (COLORREF
)SendMessage( hTree
, TVM_GETBKCOLOR
, 0, 0 );
533 ok(crColor
== -1, "Default background color reported as 0x%.8x\n", crColor
);
535 /* Test for black background */
536 SendMessage( hTree
, TVM_SETBKCOLOR
, 0, RGB(0,0,0) );
537 crColor
= (COLORREF
)SendMessage( hTree
, TVM_GETBKCOLOR
, 0, 0 );
538 ok(crColor
== RGB(0,0,0), "Black background color reported as 0x%.8x\n", crColor
);
540 /* Test for white background */
541 SendMessage( hTree
, TVM_SETBKCOLOR
, 0, RGB(255,255,255) );
542 crColor
= (COLORREF
)SendMessage( hTree
, TVM_GETBKCOLOR
, 0, 0 );
543 ok(crColor
== RGB(255,255,255), "White background color reported as 0x%.8x\n", crColor
);
545 /* Reset the default background */
546 SendMessage( hTree
, TVM_SETBKCOLOR
, 0, -1 );
548 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_bkcolor_seq
,
549 "test get set bkcolor", FALSE
);
551 DestroyWindow(hTree
);
554 static void test_get_set_imagelist(void)
556 HIMAGELIST hImageList
= NULL
;
559 hTree
= create_treeview_control();
562 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
564 /* Test a NULL HIMAGELIST */
565 SendMessage( hTree
, TVM_SETIMAGELIST
, TVSIL_NORMAL
, (LPARAM
)hImageList
);
566 hImageList
= (HIMAGELIST
)SendMessage( hTree
, TVM_GETIMAGELIST
, TVSIL_NORMAL
, 0 );
567 ok(hImageList
== NULL
, "NULL image list, reported as 0x%p, expected 0.\n", hImageList
);
569 /* TODO: Test an actual image list */
571 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_imagelist_seq
,
572 "test get imagelist", FALSE
);
574 DestroyWindow(hTree
);
577 static void test_get_set_indent(void)
580 int ulMinIndent
= -1;
581 int ulMoreThanTwiceMin
= -1;
584 hTree
= create_treeview_control();
587 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
589 /* Finding the minimum indent */
590 SendMessage( hTree
, TVM_SETINDENT
, 0, 0 );
591 ulMinIndent
= (int)SendMessage( hTree
, TVM_GETINDENT
, 0, 0 );
593 /* Checking an indent that is more than twice the default indent */
594 ulMoreThanTwiceMin
= 2*ulMinIndent
+1;
595 SendMessage( hTree
, TVM_SETINDENT
, ulMoreThanTwiceMin
, 0 );
596 ulIndent
= (DWORD
)SendMessage( hTree
, TVM_GETINDENT
, 0, 0 );
597 ok(ulIndent
== ulMoreThanTwiceMin
, "Indent reported as %d, expected %d\n", ulIndent
, ulMoreThanTwiceMin
);
599 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_indent_seq
,
600 "test get set indent", FALSE
);
602 DestroyWindow(hTree
);
605 static void test_get_set_insertmark(void)
607 COLORREF crColor
= RGB(0,0,0);
610 hTree
= create_treeview_control();
613 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
615 SendMessage( hTree
, TVM_SETINSERTMARKCOLOR
, 0, crColor
);
616 crColor
= (COLORREF
)SendMessage( hTree
, TVM_GETINSERTMARKCOLOR
, 0, 0 );
617 ok(crColor
== RGB(0,0,0), "Insert mark color reported as 0x%.8x, expected 0x00000000\n", crColor
);
619 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_insertmarkcolor_seq
,
620 "test get set insertmark color", FALSE
);
622 DestroyWindow(hTree
);
625 static void test_get_set_item(void)
627 TVITEM tviRoot
= {0};
628 int nBufferSize
= 80;
629 char szBuffer
[80] = {0};
632 hTree
= create_treeview_control();
635 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
637 /* Test the root item */
638 tviRoot
.hItem
= hRoot
;
639 tviRoot
.mask
= TVIF_TEXT
;
640 tviRoot
.cchTextMax
= nBufferSize
;
641 tviRoot
.pszText
= szBuffer
;
642 SendMessage( hTree
, TVM_GETITEM
, 0, (LPARAM
)&tviRoot
);
643 ok(!strcmp("Root", szBuffer
), "GetItem: szBuffer=\"%s\", expected \"Root\"\n", szBuffer
);
645 /* Change the root text */
646 strncpy(szBuffer
, "Testing123", nBufferSize
);
647 SendMessage( hTree
, TVM_SETITEM
, 0, (LPARAM
)&tviRoot
);
648 memset(szBuffer
, 0, nBufferSize
);
649 SendMessage( hTree
, TVM_GETITEM
, 0, (LPARAM
)&tviRoot
);
650 ok(!strcmp("Testing123", szBuffer
), "GetItem: szBuffer=\"%s\", expected \"Testing123\"\n", szBuffer
);
652 /* Reset the root text */
653 memset(szBuffer
, 0, nBufferSize
);
654 strncpy(szBuffer
, "Root", nBufferSize
);
655 SendMessage( hTree
, TVM_SETITEM
, 0, (LPARAM
)&tviRoot
);
657 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_item_seq
,
658 "test get set item", FALSE
);
660 DestroyWindow(hTree
);
663 static void test_get_set_itemheight(void)
669 hTree
= create_treeview_control();
672 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
674 /* Assuming default height to begin with */
675 ulOldHeight
= (int) SendMessage( hTree
, TVM_GETITEMHEIGHT
, 0, 0 );
677 /* Explicitly setting and getting the default height */
678 SendMessage( hTree
, TVM_SETITEMHEIGHT
, -1, 0 );
679 ulNewHeight
= (int) SendMessage( hTree
, TVM_GETITEMHEIGHT
, 0, 0 );
680 ok(ulNewHeight
== ulOldHeight
, "Default height not set properly, reported %d, expected %d\n", ulNewHeight
, ulOldHeight
);
682 /* Explicitly setting and getting the height of twice the normal */
683 SendMessage( hTree
, TVM_SETITEMHEIGHT
, 2*ulOldHeight
, 0 );
684 ulNewHeight
= (int) SendMessage( hTree
, TVM_GETITEMHEIGHT
, 0, 0 );
685 ok(ulNewHeight
== 2*ulOldHeight
, "New height not set properly, reported %d, expected %d\n", ulNewHeight
, 2*ulOldHeight
);
687 /* Assuming tree doesn't have TVS_NONEVENHEIGHT set, so a set of 9 will round down to 8 */
688 SendMessage( hTree
, TVM_SETITEMHEIGHT
, 9, 0 );
689 ulNewHeight
= (int) SendMessage( hTree
, TVM_GETITEMHEIGHT
, 0, 0 );
690 ok(ulNewHeight
== 8, "Uneven height not set properly, reported %d, expected %d\n", ulNewHeight
, 8);
692 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_itemheight_seq
,
693 "test get set item height", FALSE
);
695 DestroyWindow(hTree
);
698 static void test_get_set_scrolltime(void)
700 int ulExpectedTime
= 20;
704 hTree
= create_treeview_control();
707 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
709 SendMessage( hTree
, TVM_SETSCROLLTIME
, ulExpectedTime
, 0 );
710 ulTime
= (int)SendMessage( hTree
, TVM_GETSCROLLTIME
, 0, 0 );
711 ok(ulTime
== ulExpectedTime
, "Scroll time reported as %d, expected %d\n", ulTime
, ulExpectedTime
);
713 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_scrolltime_seq
,
714 "test get set scroll time", FALSE
);
716 DestroyWindow(hTree
);
719 static void test_get_set_textcolor(void)
721 /* If the value is -1, the control is using the system color for the text color. */
722 COLORREF crColor
= RGB(0,0,0);
725 hTree
= create_treeview_control();
728 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
730 crColor
= (COLORREF
)SendMessage( hTree
, TVM_GETTEXTCOLOR
, 0, 0 );
731 ok(crColor
== -1, "Default text color reported as 0x%.8x\n", crColor
);
733 /* Test for black text */
734 SendMessage( hTree
, TVM_SETTEXTCOLOR
, 0, RGB(0,0,0) );
735 crColor
= (COLORREF
)SendMessage( hTree
, TVM_GETTEXTCOLOR
, 0, 0 );
736 ok(crColor
== RGB(0,0,0), "Black text color reported as 0x%.8x\n", crColor
);
738 /* Test for white text */
739 SendMessage( hTree
, TVM_SETTEXTCOLOR
, 0, RGB(255,255,255) );
740 crColor
= (COLORREF
)SendMessage( hTree
, TVM_GETTEXTCOLOR
, 0, 0 );
741 ok(crColor
== RGB(255,255,255), "White text color reported as 0x%.8x\n", crColor
);
743 /* Reset the default text color */
744 SendMessage( hTree
, TVM_SETTEXTCOLOR
, 0, CLR_NONE
);
746 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_textcolor_seq
,
747 "test get set text color", FALSE
);
749 DestroyWindow(hTree
);
752 static void test_get_set_tooltips(void)
754 HWND hwndLastToolTip
= NULL
;
758 hTree
= create_treeview_control();
761 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
763 /* show even WS_POPUP treeview don't send NM_TOOLTIPSCREATED */
764 hPopupTreeView
= CreateWindow(WC_TREEVIEW
, NULL
, WS_POPUP
|WS_VISIBLE
, 0, 0, 100, 100, hMainWnd
, NULL
, NULL
, NULL
);
765 DestroyWindow(hPopupTreeView
);
767 /* Testing setting a NULL ToolTip */
768 SendMessage( hTree
, TVM_SETTOOLTIPS
, 0, 0 );
769 hwndLastToolTip
= (HWND
)SendMessage( hTree
, TVM_GETTOOLTIPS
, 0, 0 );
770 ok(hwndLastToolTip
== NULL
, "NULL tool tip, reported as 0x%p, expected 0.\n", hwndLastToolTip
);
772 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_tooltips_seq
,
773 "test get set tooltips", TRUE
);
775 /* TODO: Add a test of an actual tooltip */
776 DestroyWindow(hTree
);
779 static void test_get_set_unicodeformat(void)
781 BOOL bPreviousSetting
= 0;
782 BOOL bNewSetting
= 0;
785 hTree
= create_treeview_control();
788 flush_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
791 bPreviousSetting
= (BOOL
)SendMessage( hTree
, TVM_SETUNICODEFORMAT
, 1, 0 );
792 bNewSetting
= (BOOL
)SendMessage( hTree
, TVM_GETUNICODEFORMAT
, 0, 0 );
793 ok(bNewSetting
== 1, "Unicode setting did not work.\n");
796 SendMessage( hTree
, TVM_SETUNICODEFORMAT
, 0, 0 );
797 bNewSetting
= (BOOL
)SendMessage( hTree
, TVM_GETUNICODEFORMAT
, 0, 0 );
798 ok(bNewSetting
== 0, "ANSI setting did not work.\n");
800 /* Revert to original setting */
801 SendMessage( hTree
, TVM_SETUNICODEFORMAT
, bPreviousSetting
, 0 );
803 ok_sequence(MsgSequences
, TREEVIEW_SEQ_INDEX
, test_get_set_unicodeformat_seq
,
804 "test get set unicode format", FALSE
);
806 DestroyWindow(hTree
);
809 static LRESULT CALLBACK
MyWndProc(HWND hWnd
, UINT msg
, WPARAM wParam
, LPARAM lParam
)
814 NMHDR
*pHdr
= (NMHDR
*)lParam
;
816 ok(pHdr
->code
!= NM_FIRST
- 19, "Treeview should not send NM_TOOLTIPSCREATED\n");
817 if (pHdr
->idFrom
== 100) {
818 NMTREEVIEWA
*pTreeView
= (LPNMTREEVIEWA
) lParam
;
820 case TVN_SELCHANGINGA
:
822 IdentifyItem(pTreeView
->itemOld
.hItem
);
823 IdentifyItem(pTreeView
->itemNew
.hItem
);
825 case TVN_SELCHANGEDA
:
827 IdentifyItem(pTreeView
->itemOld
.hItem
);
828 IdentifyItem(pTreeView
->itemNew
.hItem
);
830 case TVN_GETDISPINFOA
: {
831 NMTVDISPINFOA
*disp
= (NMTVDISPINFOA
*)lParam
;
832 if (disp
->item
.mask
& TVIF_TEXT
) {
833 lstrcpyn(disp
->item
.pszText
, TEST_CALLBACK_TEXT
, disp
->item
.cchTextMax
);
837 case TVN_ENDLABELEDIT
: return TRUE
;
848 return DefWindowProcA(hWnd
, msg
, wParam
, lParam
);
853 static void test_expandinvisible(void)
855 static CHAR nodeText
[][5] = {"0", "1", "2", "3", "4"};
863 hTree
= create_treeview_control();
865 /* The test builds the following tree and expands then node 1, while node 0 is collapsed.
875 ret
= TreeView_DeleteAllItems(hTree
);
876 ok(ret
== TRUE
, "ret\n");
877 ins
.hParent
= TVI_ROOT
;
878 ins
.hInsertAfter
= TVI_ROOT
;
879 U(ins
).item
.mask
= TVIF_TEXT
;
880 U(ins
).item
.pszText
= nodeText
[0];
881 node
[0] = TreeView_InsertItem(hTree
, &ins
);
884 ins
.hInsertAfter
= TVI_LAST
;
885 U(ins
).item
.mask
= TVIF_TEXT
;
886 ins
.hParent
= node
[0];
888 U(ins
).item
.pszText
= nodeText
[1];
889 node
[1] = TreeView_InsertItem(hTree
, &ins
);
891 U(ins
).item
.pszText
= nodeText
[4];
892 node
[4] = TreeView_InsertItem(hTree
, &ins
);
895 ins
.hParent
= node
[1];
897 U(ins
).item
.pszText
= nodeText
[2];
898 node
[2] = TreeView_InsertItem(hTree
, &ins
);
900 U(ins
).item
.pszText
= nodeText
[3];
901 node
[3] = TreeView_InsertItem(hTree
, &ins
);
905 nodeVisible
= TreeView_GetItemRect(hTree
, node
[1], &dummyRect
, FALSE
);
906 ok(!nodeVisible
, "Node 1 should not be visible.\n");
907 nodeVisible
= TreeView_GetItemRect(hTree
, node
[2], &dummyRect
, FALSE
);
908 ok(!nodeVisible
, "Node 2 should not be visible.\n");
909 nodeVisible
= TreeView_GetItemRect(hTree
, node
[3], &dummyRect
, FALSE
);
910 ok(!nodeVisible
, "Node 3 should not be visible.\n");
911 nodeVisible
= TreeView_GetItemRect(hTree
, node
[4], &dummyRect
, FALSE
);
912 ok(!nodeVisible
, "Node 4 should not be visible.\n");
914 ok(TreeView_Expand(hTree
, node
[1], TVE_EXPAND
), "Expand of node 1 failed.\n");
916 nodeVisible
= TreeView_GetItemRect(hTree
, node
[1], &dummyRect
, FALSE
);
917 ok(!nodeVisible
, "Node 1 should not be visible.\n");
918 nodeVisible
= TreeView_GetItemRect(hTree
, node
[2], &dummyRect
, FALSE
);
919 ok(!nodeVisible
, "Node 2 should not be visible.\n");
920 nodeVisible
= TreeView_GetItemRect(hTree
, node
[3], &dummyRect
, FALSE
);
921 ok(!nodeVisible
, "Node 3 should not be visible.\n");
922 nodeVisible
= TreeView_GetItemRect(hTree
, node
[4], &dummyRect
, FALSE
);
923 ok(!nodeVisible
, "Node 4 should not be visible.\n");
925 DestroyWindow(hTree
);
928 static void test_itemedit(void)
936 hTree
= create_treeview_control();
939 /* try with null item */
940 edit
= (HWND
)SendMessage(hTree
, TVM_EDITLABEL
, 0, 0);
941 ok(!IsWindow(edit
), "Expected valid handle\n");
944 edit
= (HWND
)SendMessage(hTree
, TVM_EDITLABEL
, 0, (LPARAM
)hRoot
);
945 ok(IsWindow(edit
), "Expected valid handle\n");
946 /* item shouldn't be selected automatically after TVM_EDITLABEL */
947 r
= SendMessage(hTree
, TVM_GETITEMSTATE
, (WPARAM
)hRoot
, TVIS_SELECTED
);
949 /* try to cancel with wrong edit handle */
950 r
= SendMessage(hTree
, WM_COMMAND
, MAKEWPARAM(0, EN_KILLFOCUS
), 0);
952 ok(IsWindow(edit
), "Expected edit control to be valid\n");
953 r
= SendMessage(hTree
, WM_COMMAND
, MAKEWPARAM(0, EN_KILLFOCUS
), (LPARAM
)edit
);
955 ok(!IsWindow(edit
), "Expected edit control to be destroyed\n");
956 /* try to cancel without creating edit */
957 r
= SendMessage(hTree
, WM_COMMAND
, MAKEWPARAM(0, EN_KILLFOCUS
), 0);
960 /* try to cancel with wrong (not null) handle */
961 edit
= (HWND
)SendMessage(hTree
, TVM_EDITLABEL
, 0, (LPARAM
)hRoot
);
962 ok(IsWindow(edit
), "Expected valid handle\n");
963 r
= SendMessage(hTree
, WM_COMMAND
, MAKEWPARAM(0, EN_KILLFOCUS
), (LPARAM
)hTree
);
965 ok(IsWindow(edit
), "Expected edit control to be valid\n");
966 r
= SendMessage(hTree
, WM_COMMAND
, MAKEWPARAM(0, EN_KILLFOCUS
), (LPARAM
)edit
);
969 /* remove selection after starting edit */
970 r
= TreeView_SelectItem(hTree
, hRoot
);
972 edit
= (HWND
)SendMessage(hTree
, TVM_EDITLABEL
, 0, (LPARAM
)hRoot
);
973 ok(IsWindow(edit
), "Expected valid handle\n");
974 r
= TreeView_SelectItem(hTree
, NULL
);
977 strncpy(buff
, "x", sizeof(buff
)/sizeof(CHAR
));
978 r
= SendMessage(edit
, WM_SETTEXT
, 0, (LPARAM
)buff
);
980 r
= SendMessage(hTree
, WM_COMMAND
, MAKEWPARAM(0, EN_KILLFOCUS
), (LPARAM
)edit
);
982 ok(!IsWindow(edit
), "Expected edit control to be destroyed\n");
983 /* check that text is saved */
984 item
.mask
= TVIF_TEXT
;
987 item
.cchTextMax
= sizeof(buff
)/sizeof(CHAR
);
988 r
= SendMessage(hTree
, TVM_GETITEM
, 0, (LPARAM
)&item
);
990 ok(!strcmp("x", buff
), "Expected item text to change\n");
992 DestroyWindow(hTree
);
995 static void test_treeview_classinfo(void)
999 memset(&cls
, 0, sizeof(cls
));
1000 GetClassInfo(GetModuleHandleA("comctl32.dll"), WC_TREEVIEWA
, &cls
);
1001 ok(cls
.hbrBackground
== NULL
, "Expected NULL background brush, got %p\n", cls
.hbrBackground
);
1002 ok(cls
.style
== (CS_GLOBALCLASS
| CS_DBLCLKS
), "Expected got %x\n", cls
.style
);
1003 expect(0, cls
.cbClsExtra
);
1006 static void test_get_linecolor(void)
1011 hTree
= create_treeview_control();
1013 /* newly created control has default color */
1014 clr
= (COLORREF
)SendMessage(hTree
, TVM_GETLINECOLOR
, 0, 0);
1016 win_skip("TVM_GETLINECOLOR is not supported on comctl32 < 5.80\n");
1018 expect(CLR_DEFAULT
, clr
);
1020 DestroyWindow(hTree
);
1023 static void test_get_insertmarkcolor(void)
1028 hTree
= create_treeview_control();
1030 /* newly created control has default color */
1031 clr
= (COLORREF
)SendMessage(hTree
, TVM_GETINSERTMARKCOLOR
, 0, 0);
1033 win_skip("TVM_GETINSERTMARKCOLOR is not supported on comctl32 < 5.80\n");
1035 expect(CLR_DEFAULT
, clr
);
1037 DestroyWindow(hTree
);
1040 START_TEST(treeview
)
1043 BOOL (WINAPI
*pInitCommonControlsEx
)(const INITCOMMONCONTROLSEX
*);
1047 hComctl32
= GetModuleHandleA("comctl32.dll");
1048 pInitCommonControlsEx
= (void*)GetProcAddress(hComctl32
, "InitCommonControlsEx");
1049 if (pInitCommonControlsEx
)
1051 INITCOMMONCONTROLSEX iccex
;
1052 iccex
.dwSize
= sizeof(iccex
);
1053 iccex
.dwICC
= ICC_TREEVIEW_CLASSES
;
1054 pInitCommonControlsEx(&iccex
);
1057 InitCommonControls();
1059 init_msg_sequences(MsgSequences
, NUM_MSG_SEQUENCES
);
1061 wc
.style
= CS_HREDRAW
| CS_VREDRAW
;
1064 wc
.hInstance
= GetModuleHandleA(NULL
);
1066 wc
.hCursor
= LoadCursorA(NULL
, IDC_IBEAM
);
1067 wc
.hbrBackground
= GetSysColorBrush(COLOR_WINDOW
);
1068 wc
.lpszMenuName
= NULL
;
1069 wc
.lpszClassName
= "MyTestWnd";
1070 wc
.lpfnWndProc
= MyWndProc
;
1071 RegisterClassA(&wc
);
1073 hMainWnd
= CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW
,
1074 CW_USEDEFAULT
, CW_USEDEFAULT
, 130, 105, NULL
, NULL
, GetModuleHandleA(NULL
), 0);
1076 ok(hMainWnd
!= NULL
, "Failed to create parent window. Tests aborted.\n");
1077 if (!hMainWnd
) return;
1083 test_get_set_bkcolor();
1084 test_get_set_imagelist();
1085 test_get_set_indent();
1086 test_get_set_insertmark();
1087 test_get_set_item();
1088 test_get_set_itemheight();
1089 test_get_set_scrolltime();
1090 test_get_set_textcolor();
1091 test_get_linecolor();
1092 test_get_insertmarkcolor();
1093 test_get_set_tooltips();
1094 test_get_set_unicodeformat();
1096 test_expandinvisible();
1098 test_treeview_classinfo();
1100 PostMessageA(hMainWnd
, WM_CLOSE
, 0, 0);
1101 while(GetMessageA(&msg
,0,0,0)) {
1102 TranslateMessage(&msg
);
1103 DispatchMessageA(&msg
);