jscript: Get rid of BSTR in date.c.
[wine] / dlls / comctl32 / tests / treeview.c
1 /* Unit tests for treeview.
2  *
3  * Copyright 2005 Krzysztof Foltman
4  * Copyright 2007 Christopher James Peterson
5  *
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.
10  *
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.
15  *
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
19  */
20
21 #include <assert.h>
22 #include <stdarg.h>
23
24 #include "windef.h"
25 #include "winbase.h"
26 #include "wingdi.h"
27 #include "winuser.h"
28 #include "winnls.h"
29 #include "winreg.h"
30 #include "commctrl.h" 
31
32 #include "wine/test.h"
33 #include "v6util.h"
34 #include "msg.h"
35
36 static const char *TEST_CALLBACK_TEXT = "callback_text";
37
38 static TVITEMA g_item_expanding, g_item_expanded;
39 static BOOL g_get_from_expand;
40 static BOOL g_get_rect_in_expand;
41 static BOOL g_disp_A_to_W;
42 static BOOL g_disp_set_stateimage;
43 static BOOL g_beginedit_alter_text;
44
45 #define NUM_MSG_SEQUENCES   2
46 #define TREEVIEW_SEQ_INDEX  0
47 #define PARENT_SEQ_INDEX    1
48
49 #define expect(expected, got) ok(got == expected, "Expected %d, got %d\n", expected, got)
50
51 static struct msg_sequence *sequences[NUM_MSG_SEQUENCES];
52 static struct msg_sequence *item_sequence[1];
53
54 static const struct message FillRootSeq[] = {
55     { TVM_INSERTITEM, sent },
56     { TVM_INSERTITEM, sent },
57     { 0 }
58 };
59
60 static const struct message rootnone_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 },
67     { 0 }
68 };
69
70 static const struct message rootchild_select_seq[] = {
71     { TVM_SELECTITEM, sent|wparam, 9 },
72     { TVM_SELECTITEM, sent|wparam, 9 },
73     { TVM_SELECTITEM, sent|wparam, 9 },
74     { TVM_SELECTITEM, sent|wparam, 9 },
75     { TVM_SELECTITEM, sent|wparam, 9 },
76     { TVM_SELECTITEM, sent|wparam, 9 },
77     { 0 }
78 };
79
80 static const struct message getitemtext_seq[] = {
81     { TVM_INSERTITEMA, sent },
82     { TVM_GETITEMA, sent },
83     { TVM_DELETEITEM, sent },
84     { 0 }
85 };
86
87 static const struct message focus_seq[] = {
88     { TVM_INSERTITEM, sent },
89     { TVM_INSERTITEM, sent },
90     { TVM_SELECTITEM, sent|wparam, 9 },
91     /* The following end up out of order in wine */
92     { WM_WINDOWPOSCHANGING, sent|defwinproc },
93     { WM_NCCALCSIZE, sent|wparam|defwinproc, TRUE },
94     { WM_WINDOWPOSCHANGED, sent|defwinproc },
95     { WM_SIZE, sent|defwinproc },
96     { WM_WINDOWPOSCHANGING, sent|defwinproc|optional },
97     { WM_NCCALCSIZE, sent|wparam|defwinproc|optional, TRUE },
98     { WM_WINDOWPOSCHANGED, sent|defwinproc|optional },
99     { WM_SIZE, sent|defwinproc|optional },
100     { WM_PAINT, sent|defwinproc },
101     { WM_NCPAINT, sent|wparam|defwinproc, 1 },
102     { WM_ERASEBKGND, sent|defwinproc },
103     { TVM_EDITLABELA, sent },
104     { WM_COMMAND, sent|wparam|defwinproc, MAKEWPARAM(0, EN_UPDATE) },
105     { WM_COMMAND, sent|wparam|defwinproc, MAKEWPARAM(0, EN_CHANGE) },
106     { WM_PARENTNOTIFY, sent|wparam|defwinproc, MAKEWPARAM(WM_CREATE, 0) },
107     { WM_KILLFOCUS, sent|defwinproc },
108     { WM_PAINT, sent|defwinproc },
109     { WM_IME_SETCONTEXT, sent|defwinproc|optional },
110     { WM_COMMAND, sent|wparam|defwinproc, MAKEWPARAM(0, EN_SETFOCUS) },
111     { WM_ERASEBKGND, sent|defwinproc|optional },
112     { WM_CTLCOLOREDIT, sent|defwinproc|optional },
113     { WM_CTLCOLOREDIT, sent|defwinproc|optional },
114     { 0 }
115 };
116
117 static const struct message test_get_set_bkcolor_seq[] = {
118     { TVM_GETBKCOLOR, sent|wparam|lparam, 0, 0 },
119     { TVM_SETBKCOLOR, sent|wparam|lparam, 0, 0 },
120     { TVM_GETBKCOLOR, sent|wparam|lparam, 0, 0 },
121     { TVM_SETBKCOLOR, sent|wparam|lparam, 0, 0x00ffffff },
122     { TVM_GETBKCOLOR, sent|wparam|lparam, 0, 0 },
123     { TVM_SETBKCOLOR, sent|wparam|lparam, 0, -1 },
124     { 0 }
125 };
126
127 static const struct message test_get_set_imagelist_seq[] = {
128     { TVM_SETIMAGELIST, sent|wparam|lparam, 0, 0 },
129     { TVM_GETIMAGELIST, sent|wparam|lparam, 0, 0 },
130     { 0 }
131 };
132
133 static const struct message test_get_set_indent_seq[] = {
134     { TVM_SETINDENT, sent|wparam|lparam, 0, 0 },
135     { TVM_GETINDENT, sent|wparam|lparam, 0, 0 },
136     /* The actual amount to indent is dependent on the system for this message */
137     { TVM_SETINDENT, sent },
138     { TVM_GETINDENT, sent|wparam|lparam, 0, 0 },
139     { 0 }
140 };
141
142 static const struct message test_get_set_insertmarkcolor_seq[] = {
143     { TVM_SETINSERTMARKCOLOR, sent|wparam|lparam, 0, 0 },
144     { TVM_GETINSERTMARKCOLOR, sent|wparam|lparam, 0, 0 },
145     { 0 }
146 };
147
148 static const struct message test_get_set_item_seq[] = {
149     { TVM_GETITEMA, sent },
150     { TVM_SETITEMA, sent },
151     { TVM_GETITEMA, sent },
152     { TVM_SETITEMA, sent },
153     { 0 }
154 };
155
156 static const struct message test_get_set_itemheight_seq[] = {
157     { TVM_GETITEMHEIGHT, sent|wparam|lparam, 0, 0 },
158     { TVM_SETITEMHEIGHT, sent|wparam|lparam, -1, 0 },
159     { TVM_GETITEMHEIGHT, sent|wparam|lparam, 0, 0 },
160     { TVM_SETITEMHEIGHT, sent|lparam, 0xcccccccc, 0 },
161     { TVM_GETITEMHEIGHT, sent|wparam|lparam|optional, 0, 0 },
162     { TVM_SETITEMHEIGHT, sent|wparam|lparam|optional, 9, 0 },
163     { TVM_GETITEMHEIGHT, sent|wparam|lparam, 0, 0 },
164     { 0 }
165 };
166
167 static const struct message test_get_set_scrolltime_seq[] = {
168     { TVM_SETSCROLLTIME, sent|wparam|lparam, 20, 0 },
169     { TVM_GETSCROLLTIME, sent|wparam|lparam, 0, 0 },
170     { 0 }
171 };
172
173 static const struct message test_get_set_textcolor_seq[] = {
174     { TVM_GETTEXTCOLOR, sent|wparam|lparam, 0, 0 },
175     { TVM_SETTEXTCOLOR, sent|wparam|lparam, 0, 0 },
176     { TVM_GETTEXTCOLOR, sent|wparam|lparam, 0, 0 },
177     { TVM_SETTEXTCOLOR, sent|wparam|lparam, 0, RGB(255, 255, 255) },
178     { TVM_GETTEXTCOLOR, sent|wparam|lparam, 0, 0 },
179     { TVM_SETTEXTCOLOR, sent|wparam|lparam, 0, CLR_NONE },
180     { 0 }
181 };
182
183 static const struct message test_get_set_tooltips_seq[] = {
184     { WM_KILLFOCUS,    sent },
185     { WM_IME_SETCONTEXT, sent|optional },
186     { WM_IME_NOTIFY, sent|optional },
187     { TVM_SETTOOLTIPS, sent|wparam|lparam, 0, 0 },
188     { TVM_GETTOOLTIPS, sent|wparam|lparam, 0, 0 },
189     { 0 }
190 };
191
192 static const struct message test_get_set_unicodeformat_seq[] = {
193     { TVM_SETUNICODEFORMAT, sent|wparam|lparam, TRUE, 0 },
194     { TVM_GETUNICODEFORMAT, sent|wparam|lparam, 0, 0 },
195     { TVM_SETUNICODEFORMAT, sent|wparam|lparam, 0, 0 },
196     { TVM_GETUNICODEFORMAT, sent|wparam|lparam, 0, 0 },
197     { TVM_SETUNICODEFORMAT, sent|wparam|lparam, 0, 0 },
198     { 0 }
199 };
200
201 static const struct message parent_expand_seq[] = {
202     { WM_NOTIFY, sent|id, 0, 0, TVN_ITEMEXPANDINGA },
203     { WM_NOTIFY, sent|id, 0, 0, TVN_ITEMEXPANDEDA },
204     { 0 }
205 };
206
207 static const struct message parent_expand_kb_seq[] = {
208     { WM_NOTIFY, sent|id|optional, 0, 0, TVN_KEYDOWN },
209     { WM_NOTIFY, sent|id, 0, 0, TVN_ITEMEXPANDINGA },
210     { WM_NOTIFY, sent|id, 0, 0, TVN_ITEMEXPANDEDA },
211     { WM_CHANGEUISTATE, sent|optional },
212     { 0 }
213 };
214
215 static const struct message parent_expand_empty_kb_seq[] = {
216     { WM_NOTIFY, sent|id|optional, 0, 0, TVN_KEYDOWN },
217     { WM_CHANGEUISTATE, sent|optional },
218     { 0 }
219 };
220
221 static const struct message parent_singleexpand_seq[] = {
222     { WM_NOTIFY, sent|id, 0, 0, TVN_SELCHANGINGA },
223     { WM_NOTIFY, sent|id, 0, 0, TVN_SELCHANGEDA },
224     { WM_NOTIFY, sent|id, 0, 0, TVN_SINGLEEXPAND },
225     { WM_NOTIFY, sent|id, 0, 0, TVN_ITEMEXPANDINGA },
226     { WM_NOTIFY, sent|id, 0, 0, TVN_ITEMEXPANDEDA },
227     { 0 }
228 };
229
230 static const struct message parent_get_dispinfo_seq[] = {
231     { WM_NOTIFY, sent|id, 0, 0, TVN_GETDISPINFOA },
232     { 0 }
233 };
234
235 static const struct message empty_seq[] = {
236     { 0 }
237 };
238
239 static HWND hMainWnd;
240
241 static HTREEITEM hRoot, hChild;
242
243 static int pos = 0;
244 static char sequence[256];
245
246 static void Clear(void)
247 {
248     pos = 0;
249     sequence[0] = '\0';
250 }
251
252 static void AddItem(char ch)
253 {
254     sequence[pos++] = ch;
255     sequence[pos] = '\0';
256 }
257
258 static void IdentifyItem(HTREEITEM hItem)
259 {
260     if (hItem == hRoot) {
261         AddItem('R');
262         return;
263     }
264     if (hItem == hChild) {
265         AddItem('C');
266         return;
267     }
268     if (hItem == NULL) {
269         AddItem('n');
270         return;
271     }
272     AddItem('?');
273 }
274
275 /* This function hooks in and records all messages to the treeview control */
276 static LRESULT WINAPI TreeviewWndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
277 {
278     static LONG defwndproc_counter = 0;
279     LRESULT ret;
280     struct message msg;
281     WNDPROC lpOldProc = (WNDPROC)GetWindowLongPtrA(hwnd, GWLP_USERDATA);
282
283     msg.message = message;
284     msg.flags = sent|wparam|lparam;
285     if (defwndproc_counter) msg.flags |= defwinproc;
286     msg.wParam = wParam;
287     msg.lParam = lParam;
288     add_message(sequences, TREEVIEW_SEQ_INDEX, &msg);
289
290     defwndproc_counter++;
291     ret = CallWindowProcA(lpOldProc, hwnd, message, wParam, lParam);
292     defwndproc_counter--;
293
294     return ret;
295 }
296
297 static HWND create_treeview_control(DWORD style)
298 {
299     WNDPROC pOldWndProc;
300     HWND hTree;
301
302     hTree = CreateWindowExA(WS_EX_CLIENTEDGE, WC_TREEVIEWA, NULL, WS_CHILD|WS_VISIBLE|
303             TVS_LINESATROOT|TVS_HASLINES|TVS_HASBUTTONS|TVS_EDITLABELS|style,
304             0, 0, 120, 100, hMainWnd, (HMENU)100, GetModuleHandleA(0), 0);
305
306     SetFocus(hTree);
307
308     /* Record the old WNDPROC so we can call it after recording the messages */
309     pOldWndProc = (WNDPROC)SetWindowLongPtrA(hTree, GWLP_WNDPROC, (LONG_PTR)TreeviewWndProc);
310     SetWindowLongPtrA(hTree, GWLP_USERDATA, (LONG_PTR)pOldWndProc);
311
312     return hTree;
313 }
314
315 static void fill_tree(HWND hTree)
316 {
317     TVINSERTSTRUCTA ins;
318     static CHAR root[]  = "Root",
319                 child[] = "Child";
320
321     ins.hParent = TVI_ROOT;
322     ins.hInsertAfter = TVI_ROOT;
323     U(ins).item.mask = TVIF_TEXT;
324     U(ins).item.pszText = root;
325     hRoot = TreeView_InsertItem(hTree, &ins);
326
327     ins.hParent = hRoot;
328     ins.hInsertAfter = TVI_FIRST;
329     U(ins).item.mask = TVIF_TEXT;
330     U(ins).item.pszText = child;
331     hChild = TreeView_InsertItem(hTree, &ins);
332 }
333
334 static void test_fillroot(void)
335 {
336     TVITEMA tvi;
337     HWND hTree;
338
339     hTree = create_treeview_control(0);
340
341     flush_sequences(sequences, NUM_MSG_SEQUENCES);
342
343     fill_tree(hTree);
344
345     Clear();
346     AddItem('A');
347     assert(hRoot);
348     AddItem('B');
349     assert(hChild);
350     AddItem('.');
351     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, FillRootSeq, "FillRoot", FALSE);
352     ok(!strcmp(sequence, "AB."), "Item creation\n");
353
354     /* UMLPad 1.15 depends on this being not -1 (I_IMAGECALLBACK) */
355     tvi.hItem = hRoot;
356     tvi.mask = TVIF_IMAGE | TVIF_SELECTEDIMAGE;
357     SendMessage( hTree, TVM_GETITEMA, 0, (LPARAM)&tvi );
358     ok(tvi.iImage == 0, "tvi.iImage=%d\n", tvi.iImage);
359     ok(tvi.iSelectedImage == 0, "tvi.iSelectedImage=%d\n", tvi.iSelectedImage);
360
361     DestroyWindow(hTree);
362 }
363
364 static void test_callback(void)
365 {
366     HTREEITEM hRoot;
367     HTREEITEM hItem1, hItem2;
368     TVINSERTSTRUCTA ins;
369     TVITEMA tvi;
370     CHAR test_string[] = "Test_string";
371     static const CHAR test2A[] = "TEST2";
372     CHAR buf[128];
373     HWND hTree;
374     DWORD ret;
375
376     hTree = create_treeview_control(0);
377
378     ret = TreeView_DeleteAllItems(hTree);
379     expect(TRUE, ret);
380     ins.hParent = TVI_ROOT;
381     ins.hInsertAfter = TVI_ROOT;
382     U(ins).item.mask = TVIF_TEXT;
383     U(ins).item.pszText = LPSTR_TEXTCALLBACK;
384     hRoot = TreeView_InsertItem(hTree, &ins);
385     assert(hRoot);
386
387     tvi.hItem = hRoot;
388     tvi.mask = TVIF_TEXT;
389     tvi.pszText = buf;
390     tvi.cchTextMax = sizeof(buf)/sizeof(buf[0]);
391     ret = TreeView_GetItem(hTree, &tvi);
392     expect(TRUE, ret);
393     ok(strcmp(tvi.pszText, TEST_CALLBACK_TEXT) == 0, "Callback item text mismatch %s vs %s\n",
394         tvi.pszText, TEST_CALLBACK_TEXT);
395
396     ins.hParent = hRoot;
397     ins.hInsertAfter = TVI_FIRST;
398     U(ins).item.mask = TVIF_TEXT;
399     U(ins).item.pszText = test_string;
400     hItem1 = TreeView_InsertItem(hTree, &ins);
401     assert(hItem1);
402
403     tvi.hItem = hItem1;
404     ret = TreeView_GetItem(hTree, &tvi);
405     expect(TRUE, ret);
406     ok(strcmp(tvi.pszText, test_string) == 0, "Item text mismatch %s vs %s\n",
407         tvi.pszText, test_string);
408
409     /* undocumented: pszText of NULL also means LPSTR_CALLBACK: */
410     tvi.pszText = NULL;
411     ret = TreeView_SetItem(hTree, &tvi);
412     expect(TRUE, ret);
413     tvi.pszText = buf;
414     ret = TreeView_GetItem(hTree, &tvi);
415     expect(TRUE, ret);
416     ok(strcmp(tvi.pszText, TEST_CALLBACK_TEXT) == 0, "Item text mismatch %s vs %s\n",
417         tvi.pszText, TEST_CALLBACK_TEXT);
418
419     U(ins).item.pszText = NULL;
420     hItem2 = TreeView_InsertItem(hTree, &ins);
421     assert(hItem2);
422     tvi.hItem = hItem2;
423     memset(buf, 0, sizeof(buf));
424     ret = TreeView_GetItem(hTree, &tvi);
425     expect(TRUE, ret);
426     ok(strcmp(tvi.pszText, TEST_CALLBACK_TEXT) == 0, "Item text mismatch %s vs %s\n",
427         tvi.pszText, TEST_CALLBACK_TEXT);
428
429     /* notification handler changed A->W */
430     g_disp_A_to_W = TRUE;
431     tvi.hItem = hItem2;
432     memset(buf, 0, sizeof(buf));
433     ret = TreeView_GetItem(hTree, &tvi);
434     expect(TRUE, ret);
435     ok(strcmp(tvi.pszText, test2A) == 0, "got %s, expected %s\n",
436         tvi.pszText, test2A);
437     g_disp_A_to_W = FALSE;
438
439     /* handler changes state image index */
440     SetWindowLongA(hTree, GWL_STYLE, GetWindowLongA(hTree, GWL_STYLE) | TVS_CHECKBOXES);
441
442     /* clear selection, handler will set selected state */
443     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, 0);
444     expect(TRUE, ret);
445
446     flush_sequences(sequences, NUM_MSG_SEQUENCES);
447
448     tvi.hItem = hRoot;
449     tvi.mask = TVIF_STATE;
450     tvi.state = TVIS_SELECTED;
451     ret = TreeView_GetItem(hTree, &tvi);
452     expect(TRUE, ret);
453     ok(tvi.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", tvi.state);
454
455     ok_sequence(sequences, PARENT_SEQ_INDEX, empty_seq,
456                 "no TVN_GETDISPINFO for a state seq", FALSE);
457
458     tvi.hItem     = hRoot;
459     tvi.mask      = TVIF_IMAGE | TVIF_STATE;
460     tvi.state     = TVIS_FOCUSED;
461     tvi.stateMask = TVIS_FOCUSED;
462     tvi.iImage    = I_IMAGECALLBACK;
463     ret = TreeView_SetItem(hTree, &tvi);
464     expect(TRUE, ret);
465
466     /* ask for item image index through callback - state is also set with state image index */
467     flush_sequences(sequences, NUM_MSG_SEQUENCES);
468
469     tvi.hItem = hRoot;
470     tvi.mask = TVIF_IMAGE;
471     tvi.state = 0;
472     ret = TreeView_GetItem(hTree, &tvi);
473     expect(TRUE, ret);
474     ok(tvi.state == (INDEXTOSTATEIMAGEMASK(1) | TVIS_FOCUSED), "got 0x%x\n", tvi.state);
475
476     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_get_dispinfo_seq,
477                 "callback for state/overlay image index, noop seq", FALSE);
478
479     /* ask for image again and overwrite state to some value in handler */
480     flush_sequences(sequences, NUM_MSG_SEQUENCES);
481
482     g_disp_set_stateimage = TRUE;
483     tvi.hItem = hRoot;
484     tvi.mask = TVIF_IMAGE;
485     tvi.state = INDEXTOSTATEIMAGEMASK(1);
486     tvi.stateMask = 0;
487     ret = TreeView_GetItem(hTree, &tvi);
488     expect(TRUE, ret);
489     /* handler sets TVIS_SELECTED as well */
490     ok(tvi.state == (TVIS_FOCUSED | TVIS_SELECTED | INDEXTOSTATEIMAGEMASK(2) | INDEXTOOVERLAYMASK(3)), "got 0x%x\n", tvi.state);
491     g_disp_set_stateimage = FALSE;
492
493     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_get_dispinfo_seq,
494                 "callback for state/overlay image index seq", FALSE);
495
496     DestroyWindow(hTree);
497 }
498
499 static void test_select(void)
500 {
501     BOOL r;
502     HWND hTree;
503
504     hTree = create_treeview_control(0);
505     fill_tree(hTree);
506
507     /* root-none select tests */
508     flush_sequences(sequences, NUM_MSG_SEQUENCES);
509     r = TreeView_SelectItem(hTree, NULL);
510     expect(TRUE, r);
511     Clear();
512     AddItem('1');
513     r = TreeView_SelectItem(hTree, hRoot);
514     expect(TRUE, r);
515     AddItem('2');
516     r = TreeView_SelectItem(hTree, hRoot);
517     expect(TRUE, r);
518     AddItem('3');
519     r = TreeView_SelectItem(hTree, NULL);
520     expect(TRUE, r);
521     AddItem('4');
522     r = TreeView_SelectItem(hTree, NULL);
523     expect(TRUE, r);
524     AddItem('5');
525     r = TreeView_SelectItem(hTree, hRoot);
526     expect(TRUE, r);
527     AddItem('.');
528     ok(!strcmp(sequence, "1(nR)nR23(Rn)Rn45(nR)nR."), "root-none select test\n");
529     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, rootnone_select_seq,
530                 "root-none select seq", FALSE);
531
532     /* root-child select tests */
533     flush_sequences(sequences, NUM_MSG_SEQUENCES);
534     r = TreeView_SelectItem(hTree, NULL);
535     expect(TRUE, r);
536
537     Clear();
538     AddItem('1');
539     r = TreeView_SelectItem(hTree, hRoot);
540     expect(TRUE, r);
541     AddItem('2');
542     r = TreeView_SelectItem(hTree, hRoot);
543     expect(TRUE, r);
544     AddItem('3');
545     r = TreeView_SelectItem(hTree, hChild);
546     expect(TRUE, r);
547     AddItem('4');
548     r = TreeView_SelectItem(hTree, hChild);
549     expect(TRUE, r);
550     AddItem('5');
551     r = TreeView_SelectItem(hTree, hRoot);
552     expect(TRUE, r);
553     AddItem('.');
554     ok(!strcmp(sequence, "1(nR)nR23(RC)RC45(CR)CR."), "root-child select test\n");
555     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, rootchild_select_seq,
556                 "root-child select seq", FALSE);
557
558     DestroyWindow(hTree);
559 }
560
561 static void test_getitemtext(void)
562 {
563     TVINSERTSTRUCTA ins;
564     HTREEITEM hChild;
565     TVITEMA tvi;
566     HWND hTree;
567
568     CHAR szBuffer[80] = "Blah";
569     int nBufferSize = sizeof(szBuffer)/sizeof(CHAR);
570
571     hTree = create_treeview_control(0);
572     fill_tree(hTree);
573
574     flush_sequences(sequences, NUM_MSG_SEQUENCES);
575
576     /* add an item without TVIF_TEXT mask and pszText == NULL */
577     ins.hParent = hRoot;
578     ins.hInsertAfter = TVI_ROOT;
579     U(ins).item.mask = 0;
580     U(ins).item.pszText = NULL;
581     U(ins).item.cchTextMax = 0;
582     hChild = TreeView_InsertItem(hTree, &ins);
583     assert(hChild);
584
585     /* retrieve it with TVIF_TEXT mask */
586     tvi.hItem = hChild;
587     tvi.mask = TVIF_TEXT;
588     tvi.cchTextMax = nBufferSize;
589     tvi.pszText = szBuffer;
590
591     SendMessageA( hTree, TVM_GETITEMA, 0, (LPARAM)&tvi );
592     ok(!strcmp(szBuffer, ""), "szBuffer=\"%s\", expected \"\"\n", szBuffer);
593     ok(SendMessageA(hTree, TVM_DELETEITEM, 0, (LPARAM)hChild), "DeleteItem failed\n");
594     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, getitemtext_seq, "get item text seq", FALSE);
595
596     DestroyWindow(hTree);
597 }
598
599 static void test_focus(void)
600 {
601     TVINSERTSTRUCTA ins;
602     static CHAR child1[]  = "Edit",
603                 child2[]  = "A really long string";
604     HTREEITEM hChild1, hChild2;
605     HWND hTree;
606     HWND hEdit;
607
608     hTree = create_treeview_control(0);
609     fill_tree(hTree);
610
611     flush_sequences(sequences, NUM_MSG_SEQUENCES);
612
613     /* This test verifies that when a label is being edited, scrolling
614      * the treeview does not cause the label to lose focus. To test
615      * this, first some additional entries are added to generate
616      * scrollbars.
617      */
618     ins.hParent = hRoot;
619     ins.hInsertAfter = hChild;
620     U(ins).item.mask = TVIF_TEXT;
621     U(ins).item.pszText = child1;
622     hChild1 = TreeView_InsertItem(hTree, &ins);
623     assert(hChild1);
624     ins.hInsertAfter = hChild1;
625     U(ins).item.mask = TVIF_TEXT;
626     U(ins).item.pszText = child2;
627     hChild2 = TreeView_InsertItem(hTree, &ins);
628     assert(hChild2);
629
630     ShowWindow(hMainWnd,SW_SHOW);
631     SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hChild);
632     hEdit = TreeView_EditLabel(hTree, hChild);
633     ScrollWindowEx(hTree, -10, 0, NULL, NULL, NULL, NULL, SW_SCROLLCHILDREN);
634     ok(GetFocus() == hEdit, "Edit control should have focus\n");
635     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, focus_seq, "focus test", TRUE);
636
637     DestroyWindow(hTree);
638 }
639
640 static void test_get_set_bkcolor(void)
641 {
642     COLORREF crColor = RGB(0,0,0);
643     HWND hTree;
644
645     hTree = create_treeview_control(0);
646     fill_tree(hTree);
647
648     flush_sequences(sequences, NUM_MSG_SEQUENCES);
649
650     /* If the value is -1, the control is using the system color for the background color. */
651     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
652     ok(crColor == ~0u, "Default background color reported as 0x%.8x\n", crColor);
653
654     /* Test for black background */
655     SendMessage( hTree, TVM_SETBKCOLOR, 0, RGB(0,0,0) );
656     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
657     ok(crColor == RGB(0,0,0), "Black background color reported as 0x%.8x\n", crColor);
658
659     /* Test for white background */
660     SendMessage( hTree, TVM_SETBKCOLOR, 0, RGB(255,255,255) );
661     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
662     ok(crColor == RGB(255,255,255), "White background color reported as 0x%.8x\n", crColor);
663
664     /* Reset the default background */
665     SendMessage( hTree, TVM_SETBKCOLOR, 0, -1 );
666
667     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_bkcolor_seq,
668         "test get set bkcolor", FALSE);
669
670     DestroyWindow(hTree);
671 }
672
673 static void test_get_set_imagelist(void)
674 {
675     HIMAGELIST himl;
676     HWND hTree;
677
678     hTree = create_treeview_control(0);
679     fill_tree(hTree);
680
681     flush_sequences(sequences, NUM_MSG_SEQUENCES);
682
683     /* Test a NULL HIMAGELIST */
684     SendMessage( hTree, TVM_SETIMAGELIST, TVSIL_NORMAL, 0 );
685     himl = (HIMAGELIST)SendMessage( hTree, TVM_GETIMAGELIST, TVSIL_NORMAL, 0 );
686     ok(himl == NULL, "NULL image list, reported as %p, expected 0.\n", himl);
687
688     /* TODO: Test an actual image list */
689
690     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_imagelist_seq,
691         "test get imagelist", FALSE);
692
693     DestroyWindow(hTree);
694 }
695
696 static void test_get_set_indent(void)
697 {
698     int ulIndent = -1;
699     int ulMinIndent = -1;
700     int ulMoreThanTwiceMin = -1;
701     HWND hTree;
702
703     hTree = create_treeview_control(0);
704     fill_tree(hTree);
705
706     flush_sequences(sequences, NUM_MSG_SEQUENCES);
707
708     /* Finding the minimum indent */
709     SendMessage( hTree, TVM_SETINDENT, 0, 0 );
710     ulMinIndent = (int)SendMessage( hTree, TVM_GETINDENT, 0, 0 );
711
712     /* Checking an indent that is more than twice the default indent */
713     ulMoreThanTwiceMin = 2*ulMinIndent+1;
714     SendMessage( hTree, TVM_SETINDENT, ulMoreThanTwiceMin, 0 );
715     ulIndent = (DWORD)SendMessage( hTree, TVM_GETINDENT, 0, 0 );
716     ok(ulIndent == ulMoreThanTwiceMin, "Indent reported as %d, expected %d\n", ulIndent, ulMoreThanTwiceMin);
717
718     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_indent_seq,
719         "test get set indent", FALSE);
720
721     DestroyWindow(hTree);
722 }
723
724 static void test_get_set_insertmark(void)
725 {
726     COLORREF crColor = RGB(0,0,0);
727     HWND hTree;
728
729     hTree = create_treeview_control(0);
730     fill_tree(hTree);
731
732     flush_sequences(sequences, NUM_MSG_SEQUENCES);
733
734     SendMessage( hTree, TVM_SETINSERTMARKCOLOR, 0, crColor );
735     crColor = (COLORREF)SendMessage( hTree, TVM_GETINSERTMARKCOLOR, 0, 0 );
736     ok(crColor == RGB(0,0,0), "Insert mark color reported as 0x%.8x, expected 0x00000000\n", crColor);
737
738     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_insertmarkcolor_seq,
739         "test get set insertmark color", FALSE);
740
741     DestroyWindow(hTree);
742 }
743
744 static void test_get_set_item(void)
745 {
746     TVITEMA tviRoot = {0};
747     int nBufferSize = 80;
748     char szBuffer[80] = {0};
749     HWND hTree, hTree2;
750     DWORD ret;
751
752     hTree = create_treeview_control(0);
753     fill_tree(hTree);
754
755     tviRoot.hItem = hRoot;
756     tviRoot.mask  = TVIF_STATE;
757     tviRoot.state = TVIS_FOCUSED;
758     tviRoot.stateMask = TVIS_FOCUSED;
759     ret = SendMessage( hTree, TVM_SETITEMA, 0, (LPARAM)&tviRoot );
760     expect(TRUE, ret);
761
762     flush_sequences(sequences, NUM_MSG_SEQUENCES);
763
764     /* Test the root item, state is set even when not requested */
765     tviRoot.hItem = hRoot;
766     tviRoot.mask = TVIF_TEXT;
767     tviRoot.state = 0;
768     tviRoot.stateMask = 0;
769     tviRoot.cchTextMax = nBufferSize;
770     tviRoot.pszText = szBuffer;
771     ret = SendMessage( hTree, TVM_GETITEMA, 0, (LPARAM)&tviRoot );
772     expect(TRUE, ret);
773     ok(!strcmp("Root", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Root\"\n", szBuffer);
774     ok(tviRoot.state == TVIS_FOCUSED, "got 0x%0x\n", tviRoot.state);
775
776     /* Change the root text */
777     strncpy(szBuffer, "Testing123", nBufferSize);
778     ret = SendMessage( hTree, TVM_SETITEMA, 0, (LPARAM)&tviRoot );
779     expect(TRUE, ret);
780     memset(szBuffer, 0, nBufferSize);
781     ret = SendMessage( hTree, TVM_GETITEMA, 0, (LPARAM)&tviRoot );
782     expect(TRUE, ret);
783     ok(!strcmp("Testing123", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Testing123\"\n", szBuffer);
784
785     /* Reset the root text */
786     memset(szBuffer, 0, nBufferSize);
787     strncpy(szBuffer, "Root", nBufferSize);
788     ret = SendMessage( hTree, TVM_SETITEMA, 0, (LPARAM)&tviRoot );
789     expect(TRUE, ret);
790
791     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_item_seq,
792         "test get set item", FALSE);
793
794     /* get item from a different tree */
795     hTree2 = create_treeview_control(0);
796
797     tviRoot.hItem = hRoot;
798     tviRoot.mask = TVIF_STATE;
799     tviRoot.state = 0;
800     ret = SendMessage( hTree2, TVM_GETITEMA, 0, (LPARAM)&tviRoot );
801     expect(TRUE, ret);
802     ok(tviRoot.state == TVIS_FOCUSED, "got state 0x%0x\n", tviRoot.state);
803
804     DestroyWindow(hTree);
805     DestroyWindow(hTree2);
806 }
807
808 static void test_get_set_itemheight(void)
809 {
810     int ulOldHeight = 0;
811     int ulNewHeight = 0;
812     HWND hTree;
813
814     hTree = create_treeview_control(0);
815     fill_tree(hTree);
816
817     flush_sequences(sequences, NUM_MSG_SEQUENCES);
818
819     /* Assuming default height to begin with */
820     ulOldHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
821
822     /* Explicitly setting and getting the default height */
823     SendMessage( hTree, TVM_SETITEMHEIGHT, -1, 0 );
824     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
825     ok(ulNewHeight == ulOldHeight, "Default height not set properly, reported %d, expected %d\n", ulNewHeight, ulOldHeight);
826
827     /* Explicitly setting and getting the height of twice the normal */
828     SendMessage( hTree, TVM_SETITEMHEIGHT, 2*ulOldHeight, 0 );
829     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
830     ok(ulNewHeight == 2*ulOldHeight, "New height not set properly, reported %d, expected %d\n", ulNewHeight, 2*ulOldHeight);
831
832     /* Assuming tree doesn't have TVS_NONEVENHEIGHT set, so a set of 9 will round down to 8 */
833     SendMessage( hTree, TVM_SETITEMHEIGHT, 9, 0 );
834     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
835     ok(ulNewHeight == 8, "Uneven height not set properly, reported %d, expected %d\n", ulNewHeight, 8);
836
837     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_itemheight_seq,
838         "test get set item height", FALSE);
839
840     /* without TVS_NONEVENHEIGHT */
841     SetWindowLong(hTree, GWL_STYLE, GetWindowLong(hTree, GWL_STYLE) & ~TVS_NONEVENHEIGHT);
842     /* odd value */
843     ulOldHeight = SendMessage( hTree, TVM_SETITEMHEIGHT, 3, 0);
844     ok(ulOldHeight == 8, "got %d, expected %d\n", ulOldHeight, 8);
845     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
846     ok(ulNewHeight == 2, "got %d, expected %d\n", ulNewHeight, 2);
847
848     ulOldHeight = SendMessage( hTree, TVM_SETITEMHEIGHT, 4, 0);
849     ok(ulOldHeight == 2, "got %d, expected %d\n", ulOldHeight, 2);
850     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
851     ok(ulNewHeight == 4, "got %d, expected %d\n", ulNewHeight, 4);
852
853     /* with TVS_NONEVENHEIGHT */
854     SetWindowLong(hTree, GWL_STYLE, GetWindowLong(hTree, GWL_STYLE) | TVS_NONEVENHEIGHT);
855     /* odd value */
856     ulOldHeight = SendMessage( hTree, TVM_SETITEMHEIGHT, 3, 0);
857     ok(ulOldHeight == 4, "got %d, expected %d\n", ulOldHeight, 4);
858     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
859     ok(ulNewHeight == 3, "got %d, expected %d\n", ulNewHeight, 3);
860     /* even value */
861     ulOldHeight = SendMessage( hTree, TVM_SETITEMHEIGHT, 10, 0);
862     ok(ulOldHeight == 3, "got %d, expected %d\n", ulOldHeight, 3);
863     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
864     ok(ulNewHeight == 10, "got %d, expected %d\n", ulNewHeight, 10);
865
866     DestroyWindow(hTree);
867 }
868
869 static void test_get_set_scrolltime(void)
870 {
871     int ulExpectedTime = 20;
872     int ulTime = 0;
873     HWND hTree;
874
875     hTree = create_treeview_control(0);
876     fill_tree(hTree);
877
878     flush_sequences(sequences, NUM_MSG_SEQUENCES);
879
880     SendMessage( hTree, TVM_SETSCROLLTIME, ulExpectedTime, 0 );
881     ulTime = (int)SendMessage( hTree, TVM_GETSCROLLTIME, 0, 0 );
882     ok(ulTime == ulExpectedTime, "Scroll time reported as %d, expected %d\n", ulTime, ulExpectedTime);
883
884     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_scrolltime_seq,
885         "test get set scroll time", FALSE);
886
887     DestroyWindow(hTree);
888 }
889
890 static void test_get_set_textcolor(void)
891 {
892     /* If the value is -1, the control is using the system color for the text color. */
893     COLORREF crColor = RGB(0,0,0);
894     HWND hTree;
895
896     hTree = create_treeview_control(0);
897     fill_tree(hTree);
898
899     flush_sequences(sequences, NUM_MSG_SEQUENCES);
900
901     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
902     ok(crColor == ~0u, "Default text color reported as 0x%.8x\n", crColor);
903
904     /* Test for black text */
905     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, RGB(0,0,0) );
906     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
907     ok(crColor == RGB(0,0,0), "Black text color reported as 0x%.8x\n", crColor);
908
909     /* Test for white text */
910     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, RGB(255,255,255) );
911     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
912     ok(crColor == RGB(255,255,255), "White text color reported as 0x%.8x\n", crColor);
913
914     /* Reset the default text color */
915     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, CLR_NONE );
916
917     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_textcolor_seq,
918         "test get set text color", FALSE);
919
920     DestroyWindow(hTree);
921 }
922
923 static void test_get_set_tooltips(void)
924 {
925     HWND hwndLastToolTip = NULL;
926     HWND hPopupTreeView;
927     HWND hTree;
928
929     hTree = create_treeview_control(0);
930     fill_tree(hTree);
931
932     flush_sequences(sequences, NUM_MSG_SEQUENCES);
933
934     /* show even WS_POPUP treeview don't send NM_TOOLTIPSCREATED */
935     hPopupTreeView = CreateWindow(WC_TREEVIEW, NULL, WS_POPUP|WS_VISIBLE, 0, 0, 100, 100, hMainWnd, NULL, NULL, NULL);
936     DestroyWindow(hPopupTreeView);
937
938     /* Testing setting a NULL ToolTip */
939     SendMessage( hTree, TVM_SETTOOLTIPS, 0, 0 );
940     hwndLastToolTip = (HWND)SendMessage( hTree, TVM_GETTOOLTIPS, 0, 0 );
941     ok(hwndLastToolTip == NULL, "NULL tool tip, reported as 0x%p, expected 0.\n", hwndLastToolTip);
942
943     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_tooltips_seq,
944         "test get set tooltips", TRUE);
945
946     /* TODO: Add a test of an actual tooltip */
947     DestroyWindow(hTree);
948 }
949
950 static void test_get_set_unicodeformat(void)
951 {
952     BOOL bPreviousSetting = 0;
953     BOOL bNewSetting = 0;
954     HWND hTree;
955
956     hTree = create_treeview_control(0);
957     fill_tree(hTree);
958
959     /* Check that an invalid format returned by NF_QUERY defaults to ANSI */
960     bPreviousSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
961     ok(bPreviousSetting == 0, "Format should be ANSI.\n");
962
963     flush_sequences(sequences, NUM_MSG_SEQUENCES);
964
965     /* Set to Unicode */
966     bPreviousSetting = (BOOL)SendMessage( hTree, TVM_SETUNICODEFORMAT, 1, 0 );
967     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
968     ok(bNewSetting == 1, "Unicode setting did not work.\n");
969
970     /* Set to ANSI */
971     SendMessage( hTree, TVM_SETUNICODEFORMAT, 0, 0 );
972     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
973     ok(bNewSetting == 0, "ANSI setting did not work.\n");
974
975     /* Revert to original setting */
976     SendMessage( hTree, TVM_SETUNICODEFORMAT, bPreviousSetting, 0 );
977
978     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_unicodeformat_seq,
979         "test get set unicode format", FALSE);
980
981     DestroyWindow(hTree);
982 }
983
984 static LRESULT CALLBACK parent_wnd_proc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
985 {
986     static LONG defwndproc_counter = 0;
987     struct message msg;
988     LRESULT ret;
989     RECT rect;
990     HTREEITEM visibleItem;
991
992     msg.message = message;
993     msg.flags = sent|wparam|lparam;
994     if (defwndproc_counter) msg.flags |= defwinproc;
995     msg.wParam = wParam;
996     msg.lParam = lParam;
997     if (message == WM_NOTIFY && lParam) msg.id = ((NMHDR*)lParam)->code;
998
999     /* log system messages, except for painting */
1000     if (message < WM_USER &&
1001         message != WM_PAINT &&
1002         message != WM_ERASEBKGND &&
1003         message != WM_NCPAINT &&
1004         message != WM_NCHITTEST &&
1005         message != WM_GETTEXT &&
1006         message != WM_GETICON &&
1007         message != WM_DEVICECHANGE)
1008     {
1009         trace("parent: %p, %04x, %08lx, %08lx\n", hWnd, message, wParam, lParam);
1010         add_message(sequences, PARENT_SEQ_INDEX, &msg);
1011     }
1012
1013     switch(message) {
1014     case WM_NOTIFYFORMAT:
1015     {
1016         /* Make NF_QUERY return an invalid format to show that it defaults to ANSI */
1017         if (lParam == NF_QUERY) return 0;
1018         break;
1019     }
1020
1021     case WM_NOTIFY:
1022     {
1023         NMHDR *pHdr = (NMHDR *)lParam;
1024
1025         ok(pHdr->code != NM_TOOLTIPSCREATED, "Treeview should not send NM_TOOLTIPSCREATED\n");
1026         if (pHdr->idFrom == 100)
1027         {
1028             NMTREEVIEWA *pTreeView = (LPNMTREEVIEWA) lParam;
1029             switch(pHdr->code)
1030             {
1031             case TVN_SELCHANGINGA:
1032                 AddItem('(');
1033                 IdentifyItem(pTreeView->itemOld.hItem);
1034                 IdentifyItem(pTreeView->itemNew.hItem);
1035                 break;
1036             case TVN_SELCHANGEDA:
1037                 AddItem(')');
1038                 IdentifyItem(pTreeView->itemOld.hItem);
1039                 IdentifyItem(pTreeView->itemNew.hItem);
1040                 break;
1041             case TVN_GETDISPINFOA: {
1042                 NMTVDISPINFOA *disp = (NMTVDISPINFOA *)lParam;
1043                 if (disp->item.mask & TVIF_TEXT) {
1044                     lstrcpyn(disp->item.pszText, TEST_CALLBACK_TEXT, disp->item.cchTextMax);
1045                 }
1046
1047                 if (g_disp_A_to_W && (disp->item.mask & TVIF_TEXT)) {
1048                     static const WCHAR testW[] = {'T','E','S','T','2',0};
1049
1050                     disp->hdr.code = TVN_GETDISPINFOW;
1051                     memcpy(disp->item.pszText, testW, sizeof(testW));
1052                 }
1053
1054                 if (g_disp_set_stateimage)
1055                 {
1056                     ok(disp->item.mask == TVIF_IMAGE, "got %x\n", disp->item.mask);
1057                     /* both masks set here are necessary to change state bits */
1058                     disp->item.mask |= TVIF_STATE;
1059                     disp->item.state = TVIS_SELECTED | INDEXTOSTATEIMAGEMASK(2) | INDEXTOOVERLAYMASK(3);
1060                     disp->item.stateMask = TVIS_SELECTED | TVIS_OVERLAYMASK | TVIS_STATEIMAGEMASK;
1061                 }
1062
1063                 break;
1064               }
1065             case TVN_BEGINLABELEDIT:
1066               {
1067                 if (g_beginedit_alter_text)
1068                 {
1069                     static const char* textA = "<edittextaltered>";
1070                     HWND edit;
1071
1072                     edit = (HWND)SendMessageA(pHdr->hwndFrom, TVM_GETEDITCONTROL, 0, 0);
1073                     ok(IsWindow(edit), "failed to get edit handle\n");
1074                     SetWindowTextA(edit, textA);
1075                 }
1076
1077                 break;
1078               }
1079
1080             case TVN_ENDLABELEDIT: return TRUE;
1081             case TVN_ITEMEXPANDINGA:
1082                 ok(pTreeView->itemNew.mask ==
1083                    (TVIF_HANDLE | TVIF_SELECTEDIMAGE | TVIF_IMAGE | TVIF_PARAM | TVIF_STATE),
1084                    "got wrong mask %x\n", pTreeView->itemNew.mask);
1085                 ok((pTreeView->itemNew.state & TVIS_EXPANDED) == 0,
1086                    "got wrong state %x\n", pTreeView->itemNew.state);
1087                 ok(pTreeView->itemOld.mask == 0,
1088                    "got wrong mask %x\n", pTreeView->itemOld.mask);
1089
1090                 if (g_get_from_expand)
1091                 {
1092                   g_item_expanding.mask = TVIF_STATE;
1093                   g_item_expanding.hItem = hRoot;
1094                   ret = SendMessageA(pHdr->hwndFrom, TVM_GETITEMA, 0, (LPARAM)&g_item_expanding);
1095                   ok(ret == TRUE, "got %lu\n", ret);
1096                 }
1097                 break;
1098             case TVN_ITEMEXPANDEDA:
1099                 ok(pTreeView->itemNew.mask & TVIF_STATE, "got wrong mask %x\n", pTreeView->itemNew.mask);
1100                 ok(pTreeView->itemNew.state & (TVIS_EXPANDED|TVIS_EXPANDEDONCE),
1101                    "got wrong mask %x\n", pTreeView->itemNew.mask);
1102                 ok(pTreeView->itemOld.mask == 0,
1103                    "got wrong mask %x\n", pTreeView->itemOld.mask);
1104
1105                 if (g_get_from_expand)
1106                 {
1107                   g_item_expanded.mask = TVIF_STATE;
1108                   g_item_expanded.hItem = hRoot;
1109                   ret = SendMessageA(pHdr->hwndFrom, TVM_GETITEMA, 0, (LPARAM)&g_item_expanded);
1110                   ok(ret == TRUE, "got %lu\n", ret);
1111                 }
1112                 if (g_get_rect_in_expand)
1113                 {
1114                   visibleItem = TreeView_GetNextItem(pHdr->hwndFrom, NULL, TVGN_FIRSTVISIBLE);
1115                   ok(pTreeView->itemNew.hItem == visibleItem, "expanded item == first visible item\n");
1116                   *(HTREEITEM*)&rect = visibleItem;
1117                   ok(SendMessage(pHdr->hwndFrom, TVM_GETITEMRECT, TRUE, (LPARAM)&rect), "Failed to get rect for first visible item.\n");
1118                   visibleItem = TreeView_GetNextItem(pHdr->hwndFrom, visibleItem, TVGN_NEXTVISIBLE);
1119                   *(HTREEITEM*)&rect = visibleItem;
1120                   ok(visibleItem != NULL, "There must be a visible item after the first visisble item.\n");
1121                   ok(SendMessage(pHdr->hwndFrom, TVM_GETITEMRECT, TRUE, (LPARAM)&rect), "Failed to get rect for second visible item.\n");
1122                 }
1123                 break;
1124             case TVN_DELETEITEMA:
1125             {
1126                 struct message item;
1127
1128                 ok(pTreeView->itemNew.mask == 0, "got wrong mask 0x%x\n", pTreeView->itemNew.mask);
1129
1130                 ok(pTreeView->itemOld.mask == (TVIF_HANDLE | TVIF_PARAM), "got wrong mask 0x%x\n", pTreeView->itemOld.mask);
1131                 ok(pTreeView->itemOld.hItem != NULL, "got %p\n", pTreeView->itemOld.hItem);
1132
1133                 memset(&item, 0, sizeof(item));
1134                 item.lParam = (LPARAM)pTreeView->itemOld.hItem;
1135                 add_message(item_sequence, 0, &item);
1136
1137                 break;
1138             }
1139             }
1140         }
1141         break;
1142     }
1143
1144     case WM_DESTROY:
1145         PostQuitMessage(0);
1146         break;
1147     }
1148
1149     defwndproc_counter++;
1150     ret = DefWindowProcA(hWnd, message, wParam, lParam);
1151     defwndproc_counter--;
1152
1153     return ret;
1154 }
1155
1156 static void test_expandinvisible(void)
1157 {
1158     static CHAR nodeText[][5] = {"0", "1", "2", "3", "4"};
1159     TVINSERTSTRUCTA ins;
1160     HTREEITEM node[5];
1161     RECT dummyRect;
1162     BOOL nodeVisible;
1163     LRESULT ret;
1164     HWND hTree;
1165
1166     hTree = create_treeview_control(0);
1167
1168     /* The test builds the following tree and expands node 1, while node 0 is collapsed.
1169      *
1170      * 0
1171      * |- 1
1172      * |  |- 2
1173      * |  |- 3
1174      * |- 4
1175      *
1176      */
1177
1178     ret = TreeView_DeleteAllItems(hTree);
1179     ok(ret == TRUE, "ret\n");
1180     ins.hParent = TVI_ROOT;
1181     ins.hInsertAfter = TVI_ROOT;
1182     U(ins).item.mask = TVIF_TEXT;
1183     U(ins).item.pszText = nodeText[0];
1184     node[0] = TreeView_InsertItem(hTree, &ins);
1185     assert(node[0]);
1186
1187     ins.hInsertAfter = TVI_LAST;
1188     U(ins).item.mask = TVIF_TEXT;
1189     ins.hParent = node[0];
1190
1191     U(ins).item.pszText = nodeText[1];
1192     node[1] = TreeView_InsertItem(hTree, &ins);
1193     assert(node[1]);
1194     U(ins).item.pszText = nodeText[4];
1195     node[4] = TreeView_InsertItem(hTree, &ins);
1196     assert(node[4]);
1197
1198     ins.hParent = node[1];
1199
1200     U(ins).item.pszText = nodeText[2];
1201     node[2] = TreeView_InsertItem(hTree, &ins);
1202     assert(node[2]);
1203     U(ins).item.pszText = nodeText[3];
1204     node[3] = TreeView_InsertItem(hTree, &ins);
1205     assert(node[3]);
1206
1207
1208     nodeVisible = TreeView_GetItemRect(hTree, node[1], &dummyRect, FALSE);
1209     ok(!nodeVisible, "Node 1 should not be visible.\n");
1210     nodeVisible = TreeView_GetItemRect(hTree, node[2], &dummyRect, FALSE);
1211     ok(!nodeVisible, "Node 2 should not be visible.\n");
1212     nodeVisible = TreeView_GetItemRect(hTree, node[3], &dummyRect, FALSE);
1213     ok(!nodeVisible, "Node 3 should not be visible.\n");
1214     nodeVisible = TreeView_GetItemRect(hTree, node[4], &dummyRect, FALSE);
1215     ok(!nodeVisible, "Node 4 should not be visible.\n");
1216
1217     ok(TreeView_Expand(hTree, node[1], TVE_EXPAND), "Expand of node 1 failed.\n");
1218
1219     nodeVisible = TreeView_GetItemRect(hTree, node[1], &dummyRect, FALSE);
1220     ok(!nodeVisible, "Node 1 should not be visible.\n");
1221     nodeVisible = TreeView_GetItemRect(hTree, node[2], &dummyRect, FALSE);
1222     ok(!nodeVisible, "Node 2 should not be visible.\n");
1223     nodeVisible = TreeView_GetItemRect(hTree, node[3], &dummyRect, FALSE);
1224     ok(!nodeVisible, "Node 3 should not be visible.\n");
1225     nodeVisible = TreeView_GetItemRect(hTree, node[4], &dummyRect, FALSE);
1226     ok(!nodeVisible, "Node 4 should not be visible.\n");
1227
1228     DestroyWindow(hTree);
1229 }
1230
1231 static void test_itemedit(void)
1232 {
1233     DWORD r;
1234     HWND edit;
1235     TVITEMA item;
1236     CHAR buffA[20];
1237     HWND hTree;
1238
1239     hTree = create_treeview_control(0);
1240     fill_tree(hTree);
1241
1242     /* try with null item */
1243     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, 0);
1244     ok(!IsWindow(edit), "Expected valid handle\n");
1245
1246     /* trigger edit */
1247     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1248     ok(IsWindow(edit), "Expected valid handle\n");
1249     /* item shouldn't be selected automatically after TVM_EDITLABEL */
1250     r = SendMessage(hTree, TVM_GETITEMSTATE, (WPARAM)hRoot, TVIS_SELECTED);
1251     expect(0, r);
1252     /* try to cancel with wrong edit handle */
1253     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), 0);
1254     expect(0, r);
1255     ok(IsWindow(edit), "Expected edit control to be valid\n");
1256     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1257     expect(0, r);
1258     ok(!IsWindow(edit), "Expected edit control to be destroyed\n");
1259     /* try to cancel without creating edit */
1260     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), 0);
1261     expect(0, r);
1262
1263     /* try to cancel with wrong (not null) handle */
1264     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1265     ok(IsWindow(edit), "Expected valid handle\n");
1266     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)hTree);
1267     expect(0, r);
1268     ok(IsWindow(edit), "Expected edit control to be valid\n");
1269     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1270     expect(0, r);
1271
1272     /* remove selection after starting edit */
1273     r = TreeView_SelectItem(hTree, hRoot);
1274     expect(TRUE, r);
1275     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1276     ok(IsWindow(edit), "Expected valid handle\n");
1277     r = TreeView_SelectItem(hTree, NULL);
1278     expect(TRUE, r);
1279     /* alter text */
1280     strcpy(buffA, "x");
1281     r = SendMessage(edit, WM_SETTEXT, 0, (LPARAM)buffA);
1282     expect(TRUE, r);
1283     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1284     expect(0, r);
1285     ok(!IsWindow(edit), "Expected edit control to be destroyed\n");
1286     /* check that text is saved */
1287     item.mask = TVIF_TEXT;
1288     item.hItem = hRoot;
1289     item.pszText = buffA;
1290     item.cchTextMax = sizeof(buffA)/sizeof(CHAR);
1291     r = SendMessage(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1292     expect(TRUE, r);
1293     ok(!strcmp("x", buffA), "Expected item text to change\n");
1294
1295     /* try A/W messages */
1296     edit = (HWND)SendMessageA(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1297     ok(IsWindow(edit), "Expected valid handle\n");
1298     ok(IsWindowUnicode(edit), "got ansi window\n");
1299     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1300     expect(0, r);
1301     ok(!IsWindow(edit), "expected invalid handle\n");
1302
1303     edit = (HWND)SendMessageA(hTree, TVM_EDITLABELW, 0, (LPARAM)hRoot);
1304     ok(IsWindow(edit), "Expected valid handle\n");
1305     ok(IsWindowUnicode(edit), "got ansi window\n");
1306     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1307     expect(0, r);
1308
1309     /* alter text during TVM_BEGINLABELEDIT, check that it's preserved */
1310     strcpy(buffA, "<root>");
1311
1312     item.mask = TVIF_TEXT;
1313     item.hItem = hRoot;
1314     item.pszText = buffA;
1315     item.cchTextMax = 0;
1316     r = SendMessage(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1317     expect(TRUE, r);
1318
1319     g_beginedit_alter_text = TRUE;
1320     edit = (HWND)SendMessageA(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1321     ok(IsWindow(edit), "Expected valid handle\n");
1322     g_beginedit_alter_text = FALSE;
1323
1324     GetWindowTextA(edit, buffA, sizeof(buffA)/sizeof(CHAR));
1325     ok(!strcmp(buffA, "<edittextaltered>"), "got string %s\n", buffA);
1326
1327     DestroyWindow(hTree);
1328 }
1329
1330 static void test_treeview_classinfo(void)
1331 {
1332     WNDCLASSA cls;
1333
1334     memset(&cls, 0, sizeof(cls));
1335     GetClassInfo(GetModuleHandleA("comctl32.dll"), WC_TREEVIEWA, &cls);
1336     ok(cls.hbrBackground == NULL, "Expected NULL background brush, got %p\n", cls.hbrBackground);
1337     ok(cls.style == (CS_GLOBALCLASS | CS_DBLCLKS), "Expected got %x\n", cls.style);
1338     expect(0, cls.cbClsExtra);
1339 }
1340
1341 static void test_get_linecolor(void)
1342 {
1343     COLORREF clr;
1344     HWND hTree;
1345
1346     hTree = create_treeview_control(0);
1347
1348     /* newly created control has default color */
1349     clr = (COLORREF)SendMessage(hTree, TVM_GETLINECOLOR, 0, 0);
1350     if (clr == 0)
1351         win_skip("TVM_GETLINECOLOR is not supported on comctl32 < 5.80\n");
1352     else
1353         expect(CLR_DEFAULT, clr);
1354
1355     DestroyWindow(hTree);
1356 }
1357
1358 static void test_get_insertmarkcolor(void)
1359 {
1360     COLORREF clr;
1361     HWND hTree;
1362
1363     hTree = create_treeview_control(0);
1364
1365     /* newly created control has default color */
1366     clr = (COLORREF)SendMessage(hTree, TVM_GETINSERTMARKCOLOR, 0, 0);
1367     if (clr == 0)
1368         win_skip("TVM_GETINSERTMARKCOLOR is not supported on comctl32 < 5.80\n");
1369     else
1370         expect(CLR_DEFAULT, clr);
1371
1372     DestroyWindow(hTree);
1373 }
1374
1375 static void test_expandnotify(void)
1376 {
1377     HWND hTree;
1378     BOOL ret;
1379     TVITEMA item;
1380
1381     hTree = create_treeview_control(0);
1382     fill_tree(hTree);
1383
1384     item.hItem = hRoot;
1385     item.mask = TVIF_STATE;
1386
1387     item.state = TVIS_EXPANDED;
1388     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1389     expect(TRUE, ret);
1390     ok((item.state & TVIS_EXPANDED) == 0, "expected collapsed\n");
1391
1392     /* preselect root node here */
1393     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hRoot);
1394     expect(TRUE, ret);
1395
1396     g_get_from_expand = TRUE;
1397     /* expand */
1398     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1399     g_item_expanding.state = 0xdeadbeef;
1400     g_item_expanded.state = 0xdeadbeef;
1401     ret = SendMessageA(hTree, TVM_EXPAND, TVE_EXPAND, (LPARAM)hRoot);
1402     expect(TRUE, ret);
1403     ok(g_item_expanding.state == TVIS_SELECTED, "got state on TVN_ITEMEXPANDING 0x%08x\n",
1404        g_item_expanding.state);
1405     ok(g_item_expanded.state == (TVIS_SELECTED|TVIS_EXPANDED), "got state on TVN_ITEMEXPANDED 0x%08x\n",
1406        g_item_expanded.state);
1407     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_seq, "expand notifications", FALSE);
1408     g_get_from_expand = FALSE;
1409
1410     /* check that it's expanded */
1411     item.state = TVIS_EXPANDED;
1412     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1413     expect(TRUE, ret);
1414     ok((item.state & TVIS_EXPANDED) == TVIS_EXPANDED, "expected expanded\n");
1415
1416     /* collapse */
1417     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1418     ret = SendMessageA(hTree, TVM_EXPAND, TVE_COLLAPSE, (LPARAM)hRoot);
1419     expect(TRUE, ret);
1420     item.state = TVIS_EXPANDED;
1421     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1422     expect(TRUE, ret);
1423     ok((item.state & TVIS_EXPANDED) == 0, "expected collapsed\n");
1424     /* all further collapse/expand attempts won't produce any notifications,
1425        the only way is to reset with all children removed */
1426     ok_sequence(sequences, PARENT_SEQ_INDEX, empty_seq, "collapse after expand notifications", FALSE);
1427
1428     /* try to toggle child that doesn't have children itself */
1429     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1430     ret = SendMessageA(hTree, TVM_EXPAND, TVE_TOGGLE, (LPARAM)hChild);
1431     expect(FALSE, ret);
1432     ok_sequence(sequences, PARENT_SEQ_INDEX, empty_seq, "toggle node without children", FALSE);
1433
1434     DestroyWindow(hTree);
1435
1436     /* test TVM_GETITEMRECT inside TVN_ITEMEXPANDED notification */
1437     hTree = create_treeview_control(0);
1438     fill_tree(hTree);
1439     g_get_rect_in_expand = TRUE;
1440     ret = TreeView_Select(hTree, hChild, TVGN_CARET);
1441     expect(TRUE, ret);
1442     g_get_rect_in_expand = FALSE;
1443
1444     DestroyWindow(hTree);
1445
1446     /* TVE_TOGGLE acts as any other TVM_EXPAND */
1447     hTree = create_treeview_control(0);
1448     fill_tree(hTree);
1449
1450     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1451     ret = SendMessageA(hTree, TVM_EXPAND, TVE_TOGGLE, (LPARAM)hRoot);
1452     expect(TRUE, ret);
1453     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_seq, "toggle node (expand)", FALSE);
1454
1455     /* toggle again - no notifications */
1456     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1457     ret = SendMessageA(hTree, TVM_EXPAND, TVE_TOGGLE, (LPARAM)hRoot);
1458     expect(TRUE, ret);
1459     ok_sequence(sequences, PARENT_SEQ_INDEX, empty_seq, "toggle node (collapse)", FALSE);
1460
1461     DestroyWindow(hTree);
1462
1463     /* some keyboard events are also translated to expand */
1464     hTree = create_treeview_control(0);
1465     fill_tree(hTree);
1466
1467     /* preselect root node here */
1468     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hRoot);
1469     expect(TRUE, ret);
1470
1471     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1472     ret = SendMessageA(hTree, WM_KEYDOWN, VK_ADD, 0);
1473     expect(FALSE, ret);
1474     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_kb_seq, "expand node", FALSE);
1475
1476     /* go to child */
1477     ret = SendMessageA(hTree, WM_KEYDOWN, VK_RIGHT, 0);
1478     expect(FALSE, ret);
1479
1480     /* try to expand child that doesn't have children itself */
1481     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1482     ret = SendMessageA(hTree, WM_KEYDOWN, VK_ADD, 0);
1483     expect(FALSE, ret);
1484     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_empty_kb_seq, "expand node with no children", TRUE);
1485
1486     DestroyWindow(hTree);
1487 }
1488
1489 static void test_expandedimage(void)
1490 {
1491     TVITEMEXA item;
1492     HWND hTree;
1493     BOOL ret;
1494
1495     hTree = create_treeview_control(0);
1496     fill_tree(hTree);
1497
1498     item.mask = TVIF_EXPANDEDIMAGE;
1499     item.iExpandedImage = 1;
1500     item.hItem = hRoot;
1501     ret = SendMessageA(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1502     ok(ret, "got %d\n", ret);
1503
1504     item.mask = TVIF_EXPANDEDIMAGE;
1505     item.iExpandedImage = -1;
1506     item.hItem = hRoot;
1507     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1508     ok(ret, "got %d\n", ret);
1509
1510     if (item.iExpandedImage != 1)
1511     {
1512         win_skip("TVIF_EXPANDEDIMAGE not supported\n");
1513         DestroyWindow(hTree);
1514         return;
1515     }
1516
1517     /* test for default iExpandedImage value */
1518     item.mask = TVIF_EXPANDEDIMAGE;
1519     item.iExpandedImage = -1;
1520     item.hItem = hChild;
1521     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1522     ok(ret, "got %d\n", ret);
1523     ok(item.iExpandedImage == (WORD)I_IMAGENONE, "got %d\n", item.iExpandedImage);
1524
1525     DestroyWindow(hTree);
1526 }
1527
1528 static void test_TVS_SINGLEEXPAND(void)
1529 {
1530     HWND hTree;
1531     BOOL ret;
1532
1533     hTree = create_treeview_control(0);
1534     SetWindowLongA(hTree, GWL_STYLE, GetWindowLong(hTree, GWL_STYLE) | TVS_SINGLEEXPAND);
1535     /* to avoid painting related notifications */
1536     ShowWindow(hTree, SW_HIDE);
1537     fill_tree(hTree);
1538
1539     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1540     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hRoot);
1541     ok(ret, "got %d\n", ret);
1542     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_singleexpand_seq, "singleexpand notifications", FALSE);
1543
1544     /* a workaround for NT4 that sends expand notifications when nothing is about to expand */
1545     ret = SendMessageA(hTree, TVM_DELETEITEM, 0, (LPARAM)hRoot);
1546     ok(ret, "got %d\n", ret);
1547     fill_tree(hTree);
1548     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, 0);
1549     ok(ret, "got %d\n", ret);
1550
1551     DestroyWindow(hTree);
1552 }
1553
1554 static void test_WM_PAINT(void)
1555 {
1556     HWND hTree;
1557     COLORREF clr;
1558     LONG ret;
1559     RECT rc;
1560     HDC hdc;
1561
1562     hTree = create_treeview_control(0);
1563
1564     clr = SendMessageA(hTree, TVM_SETBKCOLOR, 0, RGB(255, 0, 0));
1565     ok(clr == ~0u, "got %d, expected -1\n", clr);
1566
1567     hdc = GetDC(hMainWnd);
1568
1569     GetClientRect(hMainWnd, &rc);
1570     FillRect(hdc, &rc, GetStockObject(BLACK_BRUSH));
1571
1572     clr = GetPixel(hdc, 1, 1);
1573     ok(clr == RGB(0, 0, 0), "got 0x%x\n", clr);
1574
1575     ret = SendMessageA(hTree, WM_PAINT, (WPARAM)hdc, 0);
1576     ok(ret == 0, "got %d\n", ret);
1577
1578     clr = GetPixel(hdc, 1, 1);
1579     ok(clr == RGB(255, 0, 0) || broken(clr == RGB(0, 0, 0)) /* win98 */,
1580         "got 0x%x\n", clr);
1581
1582     ReleaseDC(hMainWnd, hdc);
1583
1584     DestroyWindow(hTree);
1585 }
1586
1587 static void test_delete_items(void)
1588 {
1589     const struct message *msg;
1590     HWND hTree;
1591     INT ret;
1592
1593     hTree = create_treeview_control(0);
1594     fill_tree(hTree);
1595
1596     /* check delete order */
1597     flush_sequences(item_sequence, 1);
1598     ret = SendMessage(hTree, TVM_DELETEITEM, 0, 0);
1599     ok(ret == TRUE, "got %d\n", ret);
1600
1601     msg = item_sequence[0]->sequence;
1602     ok(item_sequence[0]->count == 2, "expected 2 items, got %d\n", item_sequence[0]->count);
1603
1604     if (item_sequence[0]->count == 2)
1605     {
1606       ok(msg[0].lParam == (LPARAM)hChild, "expected %p, got 0x%lx\n", hChild, msg[0].lParam);
1607       ok(msg[1].lParam == (LPARAM)hRoot, "expected %p, got 0x%lx\n", hRoot, msg[1].lParam);
1608     }
1609
1610     ret = SendMessageA(hTree, TVM_GETCOUNT, 0, 0);
1611     ok(ret == 0, "got %d\n", ret);
1612
1613     DestroyWindow(hTree);
1614 }
1615
1616 static void test_cchildren(void)
1617 {
1618     HWND hTree;
1619     INT ret;
1620     TVITEMA item;
1621
1622     hTree = create_treeview_control(0);
1623     fill_tree(hTree);
1624
1625     ret = SendMessage(hTree, TVM_DELETEITEM, 0, (LPARAM)hChild);
1626     expect(TRUE, ret);
1627
1628     /* check cChildren - automatic mode */
1629     item.hItem = hRoot;
1630     item.mask = TVIF_CHILDREN;
1631
1632     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1633     expect(TRUE, ret);
1634     expect(0, item.cChildren);
1635
1636     DestroyWindow(hTree);
1637
1638     /* start over */
1639     hTree = create_treeview_control(0);
1640     fill_tree(hTree);
1641
1642     /* turn off automatic mode by setting cChildren explicitly */
1643     item.hItem = hRoot;
1644     item.mask = TVIF_CHILDREN;
1645
1646     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1647     expect(TRUE, ret);
1648     expect(1, item.cChildren);
1649
1650     ret = SendMessageA(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1651     expect(TRUE, ret);
1652
1653     ret = SendMessage(hTree, TVM_DELETEITEM, 0, (LPARAM)hChild);
1654     expect(TRUE, ret);
1655
1656     /* check cChildren */
1657     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1658     expect(TRUE, ret);
1659 todo_wine
1660     expect(1, item.cChildren);
1661
1662     DestroyWindow(hTree);
1663 }
1664
1665 struct _ITEM_DATA
1666 {
1667     HTREEITEM  parent; /* for root value of parent field is unidetified */
1668     HTREEITEM  nextsibling;
1669     HTREEITEM  firstchild;
1670 };
1671
1672 static void _check_item(HTREEITEM item, HTREEITEM parent, HTREEITEM nextsibling, HTREEITEM firstchild, int line)
1673 {
1674     struct _ITEM_DATA *data = (struct _ITEM_DATA*)item;
1675
1676     ok_(__FILE__, line)(data->parent == parent, "parent %p, got %p\n", parent, data->parent);
1677     ok_(__FILE__, line)(data->nextsibling == nextsibling, "sibling %p, got %p\n", nextsibling, data->nextsibling);
1678     ok_(__FILE__, line)(data->firstchild == firstchild, "firstchild %p, got %p\n", firstchild, data->firstchild);
1679 }
1680
1681 #define check_item(a, b, c, d) _check_item(a, b, c, d, __LINE__)
1682
1683 static void test_htreeitem_layout(void)
1684 {
1685     TVINSERTSTRUCTA ins;
1686     HTREEITEM item1, item2;
1687     HWND hTree;
1688
1689     hTree = create_treeview_control(0);
1690     fill_tree(hTree);
1691
1692     /* root has some special pointer in parent field */
1693     check_item(hRoot, ((struct _ITEM_DATA*)hRoot)->parent, 0, hChild);
1694     check_item(hChild, hRoot, 0, 0);
1695
1696     ins.hParent = hChild;
1697     ins.hInsertAfter = TVI_FIRST;
1698     U(ins).item.mask = 0;
1699     item1 = TreeView_InsertItem(hTree, &ins);
1700
1701     check_item(item1, hChild, 0, 0);
1702
1703     ins.hParent = hRoot;
1704     ins.hInsertAfter = TVI_FIRST;
1705     U(ins).item.mask = 0;
1706     item2 = TreeView_InsertItem(hTree, &ins);
1707
1708     check_item(item2, hRoot, hChild, 0);
1709
1710     SendMessage(hTree, TVM_DELETEITEM, 0, (LPARAM)hChild);
1711
1712     /* without children now */
1713     check_item(hRoot, ((struct _ITEM_DATA*)hRoot)->parent, 0, item2);
1714
1715     DestroyWindow(hTree);
1716 }
1717
1718 static void test_TVS_CHECKBOXES(void)
1719 {
1720     HIMAGELIST himl, himl2;
1721     HWND hTree, hTree2;
1722     TVITEMA item;
1723     DWORD ret;
1724
1725     hTree = create_treeview_control(0);
1726     fill_tree(hTree);
1727
1728     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1729     ok(himl == NULL, "got %p\n", himl);
1730
1731     item.hItem = hRoot;
1732     item.mask = TVIF_STATE;
1733     item.state = INDEXTOSTATEIMAGEMASK(1);
1734     item.stateMask = TVIS_STATEIMAGEMASK;
1735     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1736     expect(TRUE, ret);
1737     ok(item.state == 0, "got 0x%x\n", item.state);
1738
1739     /* set some index for a child */
1740     item.hItem = hChild;
1741     item.mask = TVIF_STATE;
1742     item.state = INDEXTOSTATEIMAGEMASK(4);
1743     item.stateMask = TVIS_STATEIMAGEMASK;
1744     ret = SendMessageA(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1745     expect(TRUE, ret);
1746
1747     /* enabling check boxes set all items to 1 state image index */
1748     SetWindowLongA(hTree, GWL_STYLE, GetWindowLongA(hTree, GWL_STYLE) | TVS_CHECKBOXES);
1749     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1750     ok(himl != NULL, "got %p\n", himl);
1751
1752     himl2 = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1753     ok(himl2 != NULL, "got %p\n", himl2);
1754     ok(himl2 == himl, "got %p, expected %p\n", himl2, himl);
1755
1756     item.hItem = hRoot;
1757     item.mask = TVIF_STATE;
1758     item.state = 0;
1759     item.stateMask = TVIS_STATEIMAGEMASK;
1760     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1761     expect(TRUE, ret);
1762     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1763
1764     item.hItem = hChild;
1765     item.mask = TVIF_STATE;
1766     item.state = 0;
1767     item.stateMask = TVIS_STATEIMAGEMASK;
1768     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1769     expect(TRUE, ret);
1770     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1771
1772     /* create another control and check its checkbox list */
1773     hTree2 = create_treeview_control(0);
1774     fill_tree(hTree2);
1775
1776     /* set some index for a child */
1777     item.hItem = hChild;
1778     item.mask = TVIF_STATE;
1779     item.state = INDEXTOSTATEIMAGEMASK(4);
1780     item.stateMask = TVIS_STATEIMAGEMASK;
1781     ret = SendMessageA(hTree2, TVM_SETITEMA, 0, (LPARAM)&item);
1782     expect(TRUE, ret);
1783
1784     /* enabling check boxes set all items to 1 state image index */
1785     SetWindowLongA(hTree2, GWL_STYLE, GetWindowLongA(hTree, GWL_STYLE) | TVS_CHECKBOXES);
1786     himl2 = (HIMAGELIST)SendMessageA(hTree2, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1787     ok(himl2 != NULL, "got %p\n", himl2);
1788     ok(himl != himl2, "got %p, expected %p\n", himl2, himl);
1789
1790     DestroyWindow(hTree2);
1791     DestroyWindow(hTree);
1792
1793     /* the same, but initially created with TVS_CHECKBOXES */
1794     hTree = create_treeview_control(0);
1795     fill_tree(hTree);
1796     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1797     ok(himl == NULL, "got %p\n", himl);
1798     DestroyWindow(hTree);
1799
1800     hTree = create_treeview_control(TVS_CHECKBOXES);
1801     fill_tree(hTree);
1802     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1803     todo_wine ok(himl == NULL, "got %p\n", himl);
1804
1805     item.hItem = hRoot;
1806     item.mask = TVIF_STATE;
1807     item.state = 0;
1808     item.stateMask = TVIS_STATEIMAGEMASK;
1809     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1810     expect(TRUE, ret);
1811     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1812
1813     item.hItem = hChild;
1814     item.mask = TVIF_STATE;
1815     item.state = 0;
1816     item.stateMask = TVIS_STATEIMAGEMASK;
1817     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1818     expect(TRUE, ret);
1819     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1820
1821     DestroyWindow(hTree);
1822 }
1823
1824 static void test_TVM_GETNEXTITEM(void)
1825 {
1826     HTREEITEM item;
1827     HWND hTree;
1828
1829     hTree = create_treeview_control(0);
1830     fill_tree(hTree);
1831
1832     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, 0);
1833     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1834
1835     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, (LPARAM)TVI_ROOT);
1836     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1837
1838     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, (LPARAM)hRoot);
1839     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1840
1841     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, (LPARAM)hChild);
1842     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1843
1844     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_CHILD, 0);
1845     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1846
1847     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_CHILD, (LPARAM)hRoot);
1848     ok(item == hChild, "got %p, expected %p\n", item, hChild);
1849
1850     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_CHILD, (LPARAM)TVI_ROOT);
1851     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1852
1853     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_PARENT, 0);
1854     ok(item == NULL, "got %p\n", item);
1855
1856     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_PARENT, (LPARAM)hChild);
1857     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1858
1859     DestroyWindow(hTree);
1860 }
1861
1862 static void test_TVM_HITTEST(void)
1863 {
1864     HWND hTree;
1865     LRESULT ret;
1866     RECT rc;
1867     TVHITTESTINFO ht;
1868
1869     hTree = create_treeview_control(0);
1870     fill_tree(hTree);
1871
1872     *(HTREEITEM*)&rc = hRoot;
1873     ret = SendMessage(hTree, TVM_GETITEMRECT, TRUE, (LPARAM)&rc);
1874     expect(TRUE, (BOOL)ret);
1875
1876     ht.pt.x = rc.left-1;
1877     ht.pt.y = rc.top;
1878
1879     ret = SendMessage(hTree, TVM_HITTEST, 0, (LPARAM)&ht);
1880     ok((HTREEITEM)ret == hRoot, "got %p, expected %p\n", (HTREEITEM)ret, hRoot);
1881     ok(ht.hItem == hRoot, "got %p, expected %p\n", ht.hItem, hRoot);
1882     ok(ht.flags == TVHT_ONITEMBUTTON, "got %d, expected %d\n", ht.flags, TVHT_ONITEMBUTTON);
1883
1884     ret = SendMessageA(hTree, TVM_EXPAND, TVE_EXPAND, (LPARAM)hRoot);
1885     expect(TRUE, (BOOL)ret);
1886
1887     *(HTREEITEM*)&rc = hChild;
1888     ret = SendMessage(hTree, TVM_GETITEMRECT, TRUE, (LPARAM)&rc);
1889     expect(TRUE, (BOOL)ret);
1890
1891     ht.pt.x = rc.left-1;
1892     ht.pt.y = rc.top;
1893
1894     ret = SendMessage(hTree, TVM_HITTEST, 0, (LPARAM)&ht);
1895     ok((HTREEITEM)ret == hChild, "got %p, expected %p\n", (HTREEITEM)ret, hChild);
1896     ok(ht.hItem == hChild, "got %p, expected %p\n", ht.hItem, hChild);
1897     /* Wine returns item button here, but this item has no button */
1898     todo_wine ok(ht.flags == TVHT_ONITEMINDENT, "got %d, expected %d\n", ht.flags, TVHT_ONITEMINDENT);
1899
1900     DestroyWindow(hTree);
1901 }
1902
1903 START_TEST(treeview)
1904 {
1905     HMODULE hComctl32;
1906     BOOL (WINAPI *pInitCommonControlsEx)(const INITCOMMONCONTROLSEX*);
1907     WNDCLASSA wc;
1908     MSG msg;
1909
1910     ULONG_PTR ctx_cookie;
1911     HANDLE hCtx;
1912     HWND hwnd;
1913   
1914     hComctl32 = GetModuleHandleA("comctl32.dll");
1915     pInitCommonControlsEx = (void*)GetProcAddress(hComctl32, "InitCommonControlsEx");
1916     if (pInitCommonControlsEx)
1917     {
1918         INITCOMMONCONTROLSEX iccex;
1919         iccex.dwSize = sizeof(iccex);
1920         iccex.dwICC  = ICC_TREEVIEW_CLASSES;
1921         pInitCommonControlsEx(&iccex);
1922     }
1923     else
1924         InitCommonControls();
1925
1926     init_msg_sequences(sequences, NUM_MSG_SEQUENCES);
1927     init_msg_sequences(item_sequence, 1);
1928   
1929     wc.style = CS_HREDRAW | CS_VREDRAW;
1930     wc.cbClsExtra = 0;
1931     wc.cbWndExtra = 0;
1932     wc.hInstance = GetModuleHandleA(NULL);
1933     wc.hIcon = NULL;
1934     wc.hCursor = LoadCursorA(NULL, IDC_IBEAM);
1935     wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
1936     wc.lpszMenuName = NULL;
1937     wc.lpszClassName = "MyTestWnd";
1938     wc.lpfnWndProc = parent_wnd_proc;
1939     RegisterClassA(&wc);
1940
1941     hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW,
1942       CW_USEDEFAULT, CW_USEDEFAULT, 130, 105, NULL, NULL, GetModuleHandleA(NULL), 0);
1943
1944     ok(hMainWnd != NULL, "Failed to create parent window. Tests aborted.\n");
1945     if (!hMainWnd) return;
1946
1947     test_fillroot();
1948     test_select();
1949     test_getitemtext();
1950     test_focus();
1951     test_get_set_bkcolor();
1952     test_get_set_imagelist();
1953     test_get_set_indent();
1954     test_get_set_insertmark();
1955     test_get_set_item();
1956     test_get_set_itemheight();
1957     test_get_set_scrolltime();
1958     test_get_set_textcolor();
1959     test_get_linecolor();
1960     test_get_insertmarkcolor();
1961     test_get_set_tooltips();
1962     test_get_set_unicodeformat();
1963     test_callback();
1964     test_expandinvisible();
1965     test_itemedit();
1966     test_treeview_classinfo();
1967     test_expandnotify();
1968     test_TVS_SINGLEEXPAND();
1969     test_WM_PAINT();
1970     test_delete_items();
1971     test_cchildren();
1972     test_htreeitem_layout();
1973     test_TVS_CHECKBOXES();
1974     test_TVM_GETNEXTITEM();
1975     test_TVM_HITTEST();
1976
1977     if (!load_v6_module(&ctx_cookie, &hCtx))
1978     {
1979         DestroyWindow(hMainWnd);
1980         return;
1981     }
1982
1983     /* this is a XP SP3 failure workaround */
1984     hwnd = CreateWindowExA(0, WC_TREEVIEW, "foo",
1985                            WS_CHILD | WS_BORDER | WS_VISIBLE,
1986                            0, 0, 100, 100,
1987                            hMainWnd, NULL, GetModuleHandleA(NULL), NULL);
1988     if (!IsWindow(hwnd))
1989     {
1990         win_skip("FIXME: failed to create TreeView window.\n");
1991         unload_v6_module(ctx_cookie, hCtx);
1992         DestroyWindow(hMainWnd);
1993         return;
1994     }
1995     else
1996         DestroyWindow(hwnd);
1997
1998     /* comctl32 version 6 tests start here */
1999     test_expandedimage();
2000     test_htreeitem_layout();
2001
2002     unload_v6_module(ctx_cookie, hCtx);
2003
2004     PostMessageA(hMainWnd, WM_CLOSE, 0, 0);
2005     while(GetMessageA(&msg, 0, 0, 0))
2006     {
2007         TranslateMessage(&msg);
2008         DispatchMessageA(&msg);
2009     }
2010 }