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     flush_sequences(sequences, NUM_MSG_SEQUENCES);
 960
 961     /* Set to Unicode */
 962     bPreviousSetting = (BOOL)SendMessage( hTree, TVM_SETUNICODEFORMAT, 1, 0 );
 963     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
 964     ok(bNewSetting == 1, "Unicode setting did not work.\n");
 965
 966     /* Set to ANSI */
 967     SendMessage( hTree, TVM_SETUNICODEFORMAT, 0, 0 );
 968     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
 969     ok(bNewSetting == 0, "ANSI setting did not work.\n");
 970
 971     /* Revert to original setting */
 972     SendMessage( hTree, TVM_SETUNICODEFORMAT, bPreviousSetting, 0 );
 973
 974     ok_sequence(sequences, TREEVIEW_SEQ_INDEX, test_get_set_unicodeformat_seq,
 975         "test get set unicode format", FALSE);
 976
 977     DestroyWindow(hTree);
 978 }
 979
 980 static LRESULT CALLBACK parent_wnd_proc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
 981 {
 982     static LONG defwndproc_counter = 0;
 983     struct message msg;
 984     LRESULT ret;
 985     RECT rect;
 986     HTREEITEM visibleItem;
 987
 988     msg.message = message;
 989     msg.flags = sent|wparam|lparam;
 990     if (defwndproc_counter) msg.flags |= defwinproc;
 991     msg.wParam = wParam;
 992     msg.lParam = lParam;
 993     if (message == WM_NOTIFY && lParam) msg.id = ((NMHDR*)lParam)->code;
 994
 995     /* log system messages, except for painting */
 996     if (message < WM_USER &&
 997         message != WM_PAINT &&
 998         message != WM_ERASEBKGND &&
 999         message != WM_NCPAINT &&
1000         message != WM_NCHITTEST &&
1001         message != WM_GETTEXT &&
1002         message != WM_GETICON &&
1003         message != WM_DEVICECHANGE)
1004     {
1005         trace("parent: %p, %04x, %08lx, %08lx\n", hWnd, message, wParam, lParam);
1006         add_message(sequences, PARENT_SEQ_INDEX, &msg);
1007     }
1008
1009     switch(message) {
1010     case WM_NOTIFY:
1011     {
1012         NMHDR *pHdr = (NMHDR *)lParam;
1013
1014         ok(pHdr->code != NM_TOOLTIPSCREATED, "Treeview should not send NM_TOOLTIPSCREATED\n");
1015         if (pHdr->idFrom == 100)
1016         {
1017             NMTREEVIEWA *pTreeView = (LPNMTREEVIEWA) lParam;
1018             switch(pHdr->code)
1019             {
1020             case TVN_SELCHANGINGA:
1021                 AddItem('(');
1022                 IdentifyItem(pTreeView->itemOld.hItem);
1023                 IdentifyItem(pTreeView->itemNew.hItem);
1024                 break;
1025             case TVN_SELCHANGEDA:
1026                 AddItem(')');
1027                 IdentifyItem(pTreeView->itemOld.hItem);
1028                 IdentifyItem(pTreeView->itemNew.hItem);
1029                 break;
1030             case TVN_GETDISPINFOA: {
1031                 NMTVDISPINFOA *disp = (NMTVDISPINFOA *)lParam;
1032                 if (disp->item.mask & TVIF_TEXT) {
1033                     lstrcpyn(disp->item.pszText, TEST_CALLBACK_TEXT, disp->item.cchTextMax);
1034                 }
1035
1036                 if (g_disp_A_to_W && (disp->item.mask & TVIF_TEXT)) {
1037                     static const WCHAR testW[] = {'T','E','S','T','2',0};
1038
1039                     disp->hdr.code = TVN_GETDISPINFOW;
1040                     memcpy(disp->item.pszText, testW, sizeof(testW));
1041                 }
1042
1043                 if (g_disp_set_stateimage)
1044                 {
1045                     ok(disp->item.mask == TVIF_IMAGE, "got %x\n", disp->item.mask);
1046                     /* both masks set here are necessary to change state bits */
1047                     disp->item.mask |= TVIF_STATE;
1048                     disp->item.state = TVIS_SELECTED | INDEXTOSTATEIMAGEMASK(2) | INDEXTOOVERLAYMASK(3);
1049                     disp->item.stateMask = TVIS_SELECTED | TVIS_OVERLAYMASK | TVIS_STATEIMAGEMASK;
1050                 }
1051
1052                 break;
1053               }
1054             case TVN_BEGINLABELEDIT:
1055               {
1056                 if (g_beginedit_alter_text)
1057                 {
1058                     static const char* textA = "<edittextaltered>";
1059                     HWND edit;
1060
1061                     edit = (HWND)SendMessageA(pHdr->hwndFrom, TVM_GETEDITCONTROL, 0, 0);
1062                     ok(IsWindow(edit), "failed to get edit handle\n");
1063                     SetWindowTextA(edit, textA);
1064                 }
1065
1066                 break;
1067               }
1068
1069             case TVN_ENDLABELEDIT: return TRUE;
1070             case TVN_ITEMEXPANDINGA:
1071                 ok(pTreeView->itemNew.mask ==
1072                    (TVIF_HANDLE | TVIF_SELECTEDIMAGE | TVIF_IMAGE | TVIF_PARAM | TVIF_STATE),
1073                    "got wrong mask %x\n", pTreeView->itemNew.mask);
1074                 ok((pTreeView->itemNew.state & TVIS_EXPANDED) == 0,
1075                    "got wrong state %x\n", pTreeView->itemNew.state);
1076                 ok(pTreeView->itemOld.mask == 0,
1077                    "got wrong mask %x\n", pTreeView->itemOld.mask);
1078
1079                 if (g_get_from_expand)
1080                 {
1081                   g_item_expanding.mask = TVIF_STATE;
1082                   g_item_expanding.hItem = hRoot;
1083                   ret = SendMessageA(pHdr->hwndFrom, TVM_GETITEMA, 0, (LPARAM)&g_item_expanding);
1084                   ok(ret == TRUE, "got %lu\n", ret);
1085                 }
1086                 break;
1087             case TVN_ITEMEXPANDEDA:
1088                 ok(pTreeView->itemNew.mask & TVIF_STATE, "got wrong mask %x\n", pTreeView->itemNew.mask);
1089                 ok(pTreeView->itemNew.state & (TVIS_EXPANDED|TVIS_EXPANDEDONCE),
1090                    "got wrong mask %x\n", pTreeView->itemNew.mask);
1091                 ok(pTreeView->itemOld.mask == 0,
1092                    "got wrong mask %x\n", pTreeView->itemOld.mask);
1093
1094                 if (g_get_from_expand)
1095                 {
1096                   g_item_expanded.mask = TVIF_STATE;
1097                   g_item_expanded.hItem = hRoot;
1098                   ret = SendMessageA(pHdr->hwndFrom, TVM_GETITEMA, 0, (LPARAM)&g_item_expanded);
1099                   ok(ret == TRUE, "got %lu\n", ret);
1100                 }
1101                 if (g_get_rect_in_expand)
1102                 {
1103                   visibleItem = TreeView_GetNextItem(pHdr->hwndFrom, NULL, TVGN_FIRSTVISIBLE);
1104                   ok(pTreeView->itemNew.hItem == visibleItem, "expanded item == first visible item\n");
1105                   *(HTREEITEM*)&rect = visibleItem;
1106                   ok(SendMessage(pHdr->hwndFrom, TVM_GETITEMRECT, TRUE, (LPARAM)&rect), "Failed to get rect for first visible item.\n");
1107                   visibleItem = TreeView_GetNextItem(pHdr->hwndFrom, visibleItem, TVGN_NEXTVISIBLE);
1108                   *(HTREEITEM*)&rect = visibleItem;
1109                   ok(visibleItem != NULL, "There must be a visible item after the first visisble item.\n");
1110                   ok(SendMessage(pHdr->hwndFrom, TVM_GETITEMRECT, TRUE, (LPARAM)&rect), "Failed to get rect for second visible item.\n");
1111                 }
1112                 break;
1113             case TVN_DELETEITEMA:
1114             {
1115                 struct message item;
1116
1117                 ok(pTreeView->itemNew.mask == 0, "got wrong mask 0x%x\n", pTreeView->itemNew.mask);
1118
1119                 ok(pTreeView->itemOld.mask == (TVIF_HANDLE | TVIF_PARAM), "got wrong mask 0x%x\n", pTreeView->itemOld.mask);
1120                 ok(pTreeView->itemOld.hItem != NULL, "got %p\n", pTreeView->itemOld.hItem);
1121
1122                 memset(&item, 0, sizeof(item));
1123                 item.lParam = (LPARAM)pTreeView->itemOld.hItem;
1124                 add_message(item_sequence, 0, &item);
1125
1126                 break;
1127             }
1128             }
1129         }
1130         break;
1131     }
1132
1133     case WM_DESTROY:
1134         PostQuitMessage(0);
1135         break;
1136     }
1137
1138     defwndproc_counter++;
1139     ret = DefWindowProcA(hWnd, message, wParam, lParam);
1140     defwndproc_counter--;
1141
1142     return ret;
1143 }
1144
1145 static void test_expandinvisible(void)
1146 {
1147     static CHAR nodeText[][5] = {"0", "1", "2", "3", "4"};
1148     TVINSERTSTRUCTA ins;
1149     HTREEITEM node[5];
1150     RECT dummyRect;
1151     BOOL nodeVisible;
1152     LRESULT ret;
1153     HWND hTree;
1154
1155     hTree = create_treeview_control(0);
1156
1157     /* The test builds the following tree and expands node 1, while node 0 is collapsed.
1158      *
1159      * 0
1160      * |- 1
1161      * |  |- 2
1162      * |  |- 3
1163      * |- 4
1164      *
1165      */
1166
1167     ret = TreeView_DeleteAllItems(hTree);
1168     ok(ret == TRUE, "ret\n");
1169     ins.hParent = TVI_ROOT;
1170     ins.hInsertAfter = TVI_ROOT;
1171     U(ins).item.mask = TVIF_TEXT;
1172     U(ins).item.pszText = nodeText[0];
1173     node[0] = TreeView_InsertItem(hTree, &ins);
1174     assert(node[0]);
1175
1176     ins.hInsertAfter = TVI_LAST;
1177     U(ins).item.mask = TVIF_TEXT;
1178     ins.hParent = node[0];
1179
1180     U(ins).item.pszText = nodeText[1];
1181     node[1] = TreeView_InsertItem(hTree, &ins);
1182     assert(node[1]);
1183     U(ins).item.pszText = nodeText[4];
1184     node[4] = TreeView_InsertItem(hTree, &ins);
1185     assert(node[4]);
1186
1187     ins.hParent = node[1];
1188
1189     U(ins).item.pszText = nodeText[2];
1190     node[2] = TreeView_InsertItem(hTree, &ins);
1191     assert(node[2]);
1192     U(ins).item.pszText = nodeText[3];
1193     node[3] = TreeView_InsertItem(hTree, &ins);
1194     assert(node[3]);
1195
1196
1197     nodeVisible = TreeView_GetItemRect(hTree, node[1], &dummyRect, FALSE);
1198     ok(!nodeVisible, "Node 1 should not be visible.\n");
1199     nodeVisible = TreeView_GetItemRect(hTree, node[2], &dummyRect, FALSE);
1200     ok(!nodeVisible, "Node 2 should not be visible.\n");
1201     nodeVisible = TreeView_GetItemRect(hTree, node[3], &dummyRect, FALSE);
1202     ok(!nodeVisible, "Node 3 should not be visible.\n");
1203     nodeVisible = TreeView_GetItemRect(hTree, node[4], &dummyRect, FALSE);
1204     ok(!nodeVisible, "Node 4 should not be visible.\n");
1205
1206     ok(TreeView_Expand(hTree, node[1], TVE_EXPAND), "Expand of node 1 failed.\n");
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     DestroyWindow(hTree);
1218 }
1219
1220 static void test_itemedit(void)
1221 {
1222     DWORD r;
1223     HWND edit;
1224     TVITEMA item;
1225     CHAR buffA[20];
1226     HWND hTree;
1227
1228     hTree = create_treeview_control(0);
1229     fill_tree(hTree);
1230
1231     /* try with null item */
1232     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, 0);
1233     ok(!IsWindow(edit), "Expected valid handle\n");
1234
1235     /* trigger edit */
1236     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1237     ok(IsWindow(edit), "Expected valid handle\n");
1238     /* item shouldn't be selected automatically after TVM_EDITLABEL */
1239     r = SendMessage(hTree, TVM_GETITEMSTATE, (WPARAM)hRoot, TVIS_SELECTED);
1240     expect(0, r);
1241     /* try to cancel with wrong edit handle */
1242     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), 0);
1243     expect(0, r);
1244     ok(IsWindow(edit), "Expected edit control to be valid\n");
1245     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1246     expect(0, r);
1247     ok(!IsWindow(edit), "Expected edit control to be destroyed\n");
1248     /* try to cancel without creating edit */
1249     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), 0);
1250     expect(0, r);
1251
1252     /* try to cancel with wrong (not null) handle */
1253     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1254     ok(IsWindow(edit), "Expected valid handle\n");
1255     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)hTree);
1256     expect(0, r);
1257     ok(IsWindow(edit), "Expected edit control to be valid\n");
1258     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1259     expect(0, r);
1260
1261     /* remove selection after starting edit */
1262     r = TreeView_SelectItem(hTree, hRoot);
1263     expect(TRUE, r);
1264     edit = (HWND)SendMessage(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1265     ok(IsWindow(edit), "Expected valid handle\n");
1266     r = TreeView_SelectItem(hTree, NULL);
1267     expect(TRUE, r);
1268     /* alter text */
1269     strcpy(buffA, "x");
1270     r = SendMessage(edit, WM_SETTEXT, 0, (LPARAM)buffA);
1271     expect(TRUE, r);
1272     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1273     expect(0, r);
1274     ok(!IsWindow(edit), "Expected edit control to be destroyed\n");
1275     /* check that text is saved */
1276     item.mask = TVIF_TEXT;
1277     item.hItem = hRoot;
1278     item.pszText = buffA;
1279     item.cchTextMax = sizeof(buffA)/sizeof(CHAR);
1280     r = SendMessage(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1281     expect(TRUE, r);
1282     ok(!strcmp("x", buffA), "Expected item text to change\n");
1283
1284     /* try A/W messages */
1285     edit = (HWND)SendMessageA(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1286     ok(IsWindow(edit), "Expected valid handle\n");
1287     ok(IsWindowUnicode(edit), "got ansi window\n");
1288     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1289     expect(0, r);
1290     ok(!IsWindow(edit), "expected invalid handle\n");
1291
1292     edit = (HWND)SendMessageA(hTree, TVM_EDITLABELW, 0, (LPARAM)hRoot);
1293     ok(IsWindow(edit), "Expected valid handle\n");
1294     ok(IsWindowUnicode(edit), "got ansi window\n");
1295     r = SendMessage(hTree, WM_COMMAND, MAKEWPARAM(0, EN_KILLFOCUS), (LPARAM)edit);
1296     expect(0, r);
1297
1298     /* alter text during TVM_BEGINLABELEDIT, check that it's preserved */
1299     strcpy(buffA, "<root>");
1300
1301     item.mask = TVIF_TEXT;
1302     item.hItem = hRoot;
1303     item.pszText = buffA;
1304     item.cchTextMax = 0;
1305     r = SendMessage(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1306     expect(TRUE, r);
1307
1308     g_beginedit_alter_text = TRUE;
1309     edit = (HWND)SendMessageA(hTree, TVM_EDITLABELA, 0, (LPARAM)hRoot);
1310     ok(IsWindow(edit), "Expected valid handle\n");
1311     g_beginedit_alter_text = FALSE;
1312
1313     GetWindowTextA(edit, buffA, sizeof(buffA)/sizeof(CHAR));
1314     ok(!strcmp(buffA, "<edittextaltered>"), "got string %s\n", buffA);
1315
1316     DestroyWindow(hTree);
1317 }
1318
1319 static void test_treeview_classinfo(void)
1320 {
1321     WNDCLASSA cls;
1322
1323     memset(&cls, 0, sizeof(cls));
1324     GetClassInfo(GetModuleHandleA("comctl32.dll"), WC_TREEVIEWA, &cls);
1325     ok(cls.hbrBackground == NULL, "Expected NULL background brush, got %p\n", cls.hbrBackground);
1326     ok(cls.style == (CS_GLOBALCLASS | CS_DBLCLKS), "Expected got %x\n", cls.style);
1327     expect(0, cls.cbClsExtra);
1328 }
1329
1330 static void test_get_linecolor(void)
1331 {
1332     COLORREF clr;
1333     HWND hTree;
1334
1335     hTree = create_treeview_control(0);
1336
1337     /* newly created control has default color */
1338     clr = (COLORREF)SendMessage(hTree, TVM_GETLINECOLOR, 0, 0);
1339     if (clr == 0)
1340         win_skip("TVM_GETLINECOLOR is not supported on comctl32 < 5.80\n");
1341     else
1342         expect(CLR_DEFAULT, clr);
1343
1344     DestroyWindow(hTree);
1345 }
1346
1347 static void test_get_insertmarkcolor(void)
1348 {
1349     COLORREF clr;
1350     HWND hTree;
1351
1352     hTree = create_treeview_control(0);
1353
1354     /* newly created control has default color */
1355     clr = (COLORREF)SendMessage(hTree, TVM_GETINSERTMARKCOLOR, 0, 0);
1356     if (clr == 0)
1357         win_skip("TVM_GETINSERTMARKCOLOR is not supported on comctl32 < 5.80\n");
1358     else
1359         expect(CLR_DEFAULT, clr);
1360
1361     DestroyWindow(hTree);
1362 }
1363
1364 static void test_expandnotify(void)
1365 {
1366     HWND hTree;
1367     BOOL ret;
1368     TVITEMA item;
1369
1370     hTree = create_treeview_control(0);
1371     fill_tree(hTree);
1372
1373     item.hItem = hRoot;
1374     item.mask = TVIF_STATE;
1375
1376     item.state = TVIS_EXPANDED;
1377     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1378     expect(TRUE, ret);
1379     ok((item.state & TVIS_EXPANDED) == 0, "expected collapsed\n");
1380
1381     /* preselect root node here */
1382     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hRoot);
1383     expect(TRUE, ret);
1384
1385     g_get_from_expand = TRUE;
1386     /* expand */
1387     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1388     g_item_expanding.state = 0xdeadbeef;
1389     g_item_expanded.state = 0xdeadbeef;
1390     ret = SendMessageA(hTree, TVM_EXPAND, TVE_EXPAND, (LPARAM)hRoot);
1391     expect(TRUE, ret);
1392     ok(g_item_expanding.state == TVIS_SELECTED, "got state on TVN_ITEMEXPANDING 0x%08x\n",
1393        g_item_expanding.state);
1394     ok(g_item_expanded.state == (TVIS_SELECTED|TVIS_EXPANDED), "got state on TVN_ITEMEXPANDED 0x%08x\n",
1395        g_item_expanded.state);
1396     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_seq, "expand notifications", FALSE);
1397     g_get_from_expand = FALSE;
1398
1399     /* check that it's expanded */
1400     item.state = TVIS_EXPANDED;
1401     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1402     expect(TRUE, ret);
1403     ok((item.state & TVIS_EXPANDED) == TVIS_EXPANDED, "expected expanded\n");
1404
1405     /* collapse */
1406     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1407     ret = SendMessageA(hTree, TVM_EXPAND, TVE_COLLAPSE, (LPARAM)hRoot);
1408     expect(TRUE, ret);
1409     item.state = TVIS_EXPANDED;
1410     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1411     expect(TRUE, ret);
1412     ok((item.state & TVIS_EXPANDED) == 0, "expected collapsed\n");
1413     /* all further collapse/expand attempts won't produce any notifications,
1414        the only way is to reset with all children removed */
1415     ok_sequence(sequences, PARENT_SEQ_INDEX, empty_seq, "collapse after expand notifications", FALSE);
1416
1417     /* try to toggle child that doesn't have children itself */
1418     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1419     ret = SendMessageA(hTree, TVM_EXPAND, TVE_TOGGLE, (LPARAM)hChild);
1420     expect(FALSE, ret);
1421     ok_sequence(sequences, PARENT_SEQ_INDEX, empty_seq, "toggle node without children", FALSE);
1422
1423     DestroyWindow(hTree);
1424
1425     /* test TVM_GETITEMRECT inside TVN_ITEMEXPANDED notification */
1426     hTree = create_treeview_control(0);
1427     fill_tree(hTree);
1428     g_get_rect_in_expand = TRUE;
1429     ret = TreeView_Select(hTree, hChild, TVGN_CARET);
1430     expect(TRUE, ret);
1431     g_get_rect_in_expand = FALSE;
1432
1433     DestroyWindow(hTree);
1434
1435     /* TVE_TOGGLE acts as any other TVM_EXPAND */
1436     hTree = create_treeview_control(0);
1437     fill_tree(hTree);
1438
1439     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1440     ret = SendMessageA(hTree, TVM_EXPAND, TVE_TOGGLE, (LPARAM)hRoot);
1441     expect(TRUE, ret);
1442     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_seq, "toggle node (expand)", FALSE);
1443
1444     /* toggle again - no notifications */
1445     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1446     ret = SendMessageA(hTree, TVM_EXPAND, TVE_TOGGLE, (LPARAM)hRoot);
1447     expect(TRUE, ret);
1448     ok_sequence(sequences, PARENT_SEQ_INDEX, empty_seq, "toggle node (collapse)", FALSE);
1449
1450     DestroyWindow(hTree);
1451
1452     /* some keyboard events are also translated to expand */
1453     hTree = create_treeview_control(0);
1454     fill_tree(hTree);
1455
1456     /* preselect root node here */
1457     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hRoot);
1458     expect(TRUE, ret);
1459
1460     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1461     ret = SendMessageA(hTree, WM_KEYDOWN, VK_ADD, 0);
1462     expect(FALSE, ret);
1463     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_kb_seq, "expand node", FALSE);
1464
1465     /* go to child */
1466     ret = SendMessageA(hTree, WM_KEYDOWN, VK_RIGHT, 0);
1467     expect(FALSE, ret);
1468
1469     /* try to expand child that doesn't have children itself */
1470     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1471     ret = SendMessageA(hTree, WM_KEYDOWN, VK_ADD, 0);
1472     expect(FALSE, ret);
1473     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_expand_empty_kb_seq, "expand node with no children", TRUE);
1474
1475     DestroyWindow(hTree);
1476 }
1477
1478 static void test_expandedimage(void)
1479 {
1480     TVITEMEXA item;
1481     HWND hTree;
1482     BOOL ret;
1483
1484     hTree = create_treeview_control(0);
1485     fill_tree(hTree);
1486
1487     item.mask = TVIF_EXPANDEDIMAGE;
1488     item.iExpandedImage = 1;
1489     item.hItem = hRoot;
1490     ret = SendMessageA(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1491     ok(ret, "got %d\n", ret);
1492
1493     item.mask = TVIF_EXPANDEDIMAGE;
1494     item.iExpandedImage = -1;
1495     item.hItem = hRoot;
1496     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1497     ok(ret, "got %d\n", ret);
1498
1499     if (item.iExpandedImage != 1)
1500     {
1501         win_skip("TVIF_EXPANDEDIMAGE not supported\n");
1502         DestroyWindow(hTree);
1503         return;
1504     }
1505
1506     /* test for default iExpandedImage value */
1507     item.mask = TVIF_EXPANDEDIMAGE;
1508     item.iExpandedImage = -1;
1509     item.hItem = hChild;
1510     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1511     ok(ret, "got %d\n", ret);
1512     ok(item.iExpandedImage == (WORD)I_IMAGENONE, "got %d\n", item.iExpandedImage);
1513
1514     DestroyWindow(hTree);
1515 }
1516
1517 static void test_TVS_SINGLEEXPAND(void)
1518 {
1519     HWND hTree;
1520     BOOL ret;
1521
1522     hTree = create_treeview_control(0);
1523     SetWindowLongA(hTree, GWL_STYLE, GetWindowLong(hTree, GWL_STYLE) | TVS_SINGLEEXPAND);
1524     /* to avoid painting related notifications */
1525     ShowWindow(hTree, SW_HIDE);
1526     fill_tree(hTree);
1527
1528     flush_sequences(sequences, NUM_MSG_SEQUENCES);
1529     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hRoot);
1530     ok(ret, "got %d\n", ret);
1531     ok_sequence(sequences, PARENT_SEQ_INDEX, parent_singleexpand_seq, "singleexpand notifications", FALSE);
1532
1533     /* a workaround for NT4 that sends expand notifications when nothing is about to expand */
1534     ret = SendMessageA(hTree, TVM_DELETEITEM, 0, (LPARAM)hRoot);
1535     ok(ret, "got %d\n", ret);
1536     fill_tree(hTree);
1537     ret = SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, 0);
1538     ok(ret, "got %d\n", ret);
1539
1540     DestroyWindow(hTree);
1541 }
1542
1543 static void test_WM_PAINT(void)
1544 {
1545     HWND hTree;
1546     COLORREF clr;
1547     LONG ret;
1548     RECT rc;
1549     HDC hdc;
1550
1551     hTree = create_treeview_control(0);
1552
1553     clr = SendMessageA(hTree, TVM_SETBKCOLOR, 0, RGB(255, 0, 0));
1554     ok(clr == ~0u, "got %d, expected -1\n", clr);
1555
1556     hdc = GetDC(hMainWnd);
1557
1558     GetClientRect(hMainWnd, &rc);
1559     FillRect(hdc, &rc, GetStockObject(BLACK_BRUSH));
1560
1561     clr = GetPixel(hdc, 1, 1);
1562     ok(clr == RGB(0, 0, 0), "got 0x%x\n", clr);
1563
1564     ret = SendMessageA(hTree, WM_PAINT, (WPARAM)hdc, 0);
1565     ok(ret == 0, "got %d\n", ret);
1566
1567     clr = GetPixel(hdc, 1, 1);
1568     ok(clr == RGB(255, 0, 0) || broken(clr == RGB(0, 0, 0)) /* win98 */,
1569         "got 0x%x\n", clr);
1570
1571     ReleaseDC(hMainWnd, hdc);
1572
1573     DestroyWindow(hTree);
1574 }
1575
1576 static void test_delete_items(void)
1577 {
1578     const struct message *msg;
1579     HWND hTree;
1580     INT ret;
1581
1582     hTree = create_treeview_control(0);
1583     fill_tree(hTree);
1584
1585     /* check delete order */
1586     flush_sequences(item_sequence, 1);
1587     ret = SendMessage(hTree, TVM_DELETEITEM, 0, 0);
1588     ok(ret == TRUE, "got %d\n", ret);
1589
1590     msg = item_sequence[0]->sequence;
1591     ok(item_sequence[0]->count == 2, "expected 2 items, got %d\n", item_sequence[0]->count);
1592
1593     if (item_sequence[0]->count == 2)
1594     {
1595       ok(msg[0].lParam == (LPARAM)hChild, "expected %p, got 0x%lx\n", hChild, msg[0].lParam);
1596       ok(msg[1].lParam == (LPARAM)hRoot, "expected %p, got 0x%lx\n", hRoot, msg[1].lParam);
1597     }
1598
1599     ret = SendMessageA(hTree, TVM_GETCOUNT, 0, 0);
1600     ok(ret == 0, "got %d\n", ret);
1601
1602     DestroyWindow(hTree);
1603 }
1604
1605 static void test_cchildren(void)
1606 {
1607     HWND hTree;
1608     INT ret;
1609     TVITEMA item;
1610
1611     hTree = create_treeview_control(0);
1612     fill_tree(hTree);
1613
1614     ret = SendMessage(hTree, TVM_DELETEITEM, 0, (LPARAM)hChild);
1615     expect(TRUE, ret);
1616
1617     /* check cChildren - automatic mode */
1618     item.hItem = hRoot;
1619     item.mask = TVIF_CHILDREN;
1620
1621     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1622     expect(TRUE, ret);
1623     expect(0, item.cChildren);
1624
1625     DestroyWindow(hTree);
1626
1627     /* start over */
1628     hTree = create_treeview_control(0);
1629     fill_tree(hTree);
1630
1631     /* turn off automatic mode by setting cChildren explicitly */
1632     item.hItem = hRoot;
1633     item.mask = TVIF_CHILDREN;
1634
1635     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1636     expect(TRUE, ret);
1637     expect(1, item.cChildren);
1638
1639     ret = SendMessageA(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1640     expect(TRUE, ret);
1641
1642     ret = SendMessage(hTree, TVM_DELETEITEM, 0, (LPARAM)hChild);
1643     expect(TRUE, ret);
1644
1645     /* check cChildren */
1646     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1647     expect(TRUE, ret);
1648 todo_wine
1649     expect(1, item.cChildren);
1650
1651     DestroyWindow(hTree);
1652 }
1653
1654 struct _ITEM_DATA
1655 {
1656     HTREEITEM  parent; /* for root value of parent field is unidetified */
1657     HTREEITEM  nextsibling;
1658     HTREEITEM  firstchild;
1659 };
1660
1661 static void _check_item(HTREEITEM item, HTREEITEM parent, HTREEITEM nextsibling, HTREEITEM firstchild, int line)
1662 {
1663     struct _ITEM_DATA *data = (struct _ITEM_DATA*)item;
1664
1665     ok_(__FILE__, line)(data->parent == parent, "parent %p, got %p\n", parent, data->parent);
1666     ok_(__FILE__, line)(data->nextsibling == nextsibling, "sibling %p, got %p\n", nextsibling, data->nextsibling);
1667     ok_(__FILE__, line)(data->firstchild == firstchild, "firstchild %p, got %p\n", firstchild, data->firstchild);
1668 }
1669
1670 #define check_item(a, b, c, d) _check_item(a, b, c, d, __LINE__)
1671
1672 static void test_htreeitem_layout(void)
1673 {
1674     TVINSERTSTRUCTA ins;
1675     HTREEITEM item1, item2;
1676     HWND hTree;
1677
1678     hTree = create_treeview_control(0);
1679     fill_tree(hTree);
1680
1681     /* root has some special pointer in parent field */
1682     check_item(hRoot, ((struct _ITEM_DATA*)hRoot)->parent, 0, hChild);
1683     check_item(hChild, hRoot, 0, 0);
1684
1685     ins.hParent = hChild;
1686     ins.hInsertAfter = TVI_FIRST;
1687     U(ins).item.mask = 0;
1688     item1 = TreeView_InsertItem(hTree, &ins);
1689
1690     check_item(item1, hChild, 0, 0);
1691
1692     ins.hParent = hRoot;
1693     ins.hInsertAfter = TVI_FIRST;
1694     U(ins).item.mask = 0;
1695     item2 = TreeView_InsertItem(hTree, &ins);
1696
1697     check_item(item2, hRoot, hChild, 0);
1698
1699     SendMessage(hTree, TVM_DELETEITEM, 0, (LPARAM)hChild);
1700
1701     /* without children now */
1702     check_item(hRoot, ((struct _ITEM_DATA*)hRoot)->parent, 0, item2);
1703
1704     DestroyWindow(hTree);
1705 }
1706
1707 static void test_TVS_CHECKBOXES(void)
1708 {
1709     HIMAGELIST himl, himl2;
1710     HWND hTree, hTree2;
1711     TVITEMA item;
1712     DWORD ret;
1713
1714     hTree = create_treeview_control(0);
1715     fill_tree(hTree);
1716
1717     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1718     ok(himl == NULL, "got %p\n", himl);
1719
1720     item.hItem = hRoot;
1721     item.mask = TVIF_STATE;
1722     item.state = INDEXTOSTATEIMAGEMASK(1);
1723     item.stateMask = TVIS_STATEIMAGEMASK;
1724     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1725     expect(TRUE, ret);
1726     ok(item.state == 0, "got 0x%x\n", item.state);
1727
1728     /* set some index for a child */
1729     item.hItem = hChild;
1730     item.mask = TVIF_STATE;
1731     item.state = INDEXTOSTATEIMAGEMASK(4);
1732     item.stateMask = TVIS_STATEIMAGEMASK;
1733     ret = SendMessageA(hTree, TVM_SETITEMA, 0, (LPARAM)&item);
1734     expect(TRUE, ret);
1735
1736     /* enabling check boxes set all items to 1 state image index */
1737     SetWindowLongA(hTree, GWL_STYLE, GetWindowLongA(hTree, GWL_STYLE) | TVS_CHECKBOXES);
1738     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1739     ok(himl != NULL, "got %p\n", himl);
1740
1741     himl2 = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1742     ok(himl2 != NULL, "got %p\n", himl2);
1743     ok(himl2 == himl, "got %p, expected %p\n", himl2, himl);
1744
1745     item.hItem = hRoot;
1746     item.mask = TVIF_STATE;
1747     item.state = 0;
1748     item.stateMask = TVIS_STATEIMAGEMASK;
1749     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1750     expect(TRUE, ret);
1751     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1752
1753     item.hItem = hChild;
1754     item.mask = TVIF_STATE;
1755     item.state = 0;
1756     item.stateMask = TVIS_STATEIMAGEMASK;
1757     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1758     expect(TRUE, ret);
1759     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1760
1761     /* create another control and check its checkbox list */
1762     hTree2 = create_treeview_control(0);
1763     fill_tree(hTree2);
1764
1765     /* set some index for a child */
1766     item.hItem = hChild;
1767     item.mask = TVIF_STATE;
1768     item.state = INDEXTOSTATEIMAGEMASK(4);
1769     item.stateMask = TVIS_STATEIMAGEMASK;
1770     ret = SendMessageA(hTree2, TVM_SETITEMA, 0, (LPARAM)&item);
1771     expect(TRUE, ret);
1772
1773     /* enabling check boxes set all items to 1 state image index */
1774     SetWindowLongA(hTree2, GWL_STYLE, GetWindowLongA(hTree, GWL_STYLE) | TVS_CHECKBOXES);
1775     himl2 = (HIMAGELIST)SendMessageA(hTree2, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1776     ok(himl2 != NULL, "got %p\n", himl2);
1777     ok(himl != himl2, "got %p, expected %p\n", himl2, himl);
1778
1779     DestroyWindow(hTree2);
1780     DestroyWindow(hTree);
1781
1782     /* the same, but initially created with TVS_CHECKBOXES */
1783     hTree = create_treeview_control(0);
1784     fill_tree(hTree);
1785     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1786     ok(himl == NULL, "got %p\n", himl);
1787     DestroyWindow(hTree);
1788
1789     hTree = create_treeview_control(TVS_CHECKBOXES);
1790     fill_tree(hTree);
1791     himl = (HIMAGELIST)SendMessageA(hTree, TVM_GETIMAGELIST, TVSIL_STATE, 0);
1792     todo_wine ok(himl == NULL, "got %p\n", himl);
1793
1794     item.hItem = hRoot;
1795     item.mask = TVIF_STATE;
1796     item.state = 0;
1797     item.stateMask = TVIS_STATEIMAGEMASK;
1798     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1799     expect(TRUE, ret);
1800     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1801
1802     item.hItem = hChild;
1803     item.mask = TVIF_STATE;
1804     item.state = 0;
1805     item.stateMask = TVIS_STATEIMAGEMASK;
1806     ret = SendMessageA(hTree, TVM_GETITEMA, 0, (LPARAM)&item);
1807     expect(TRUE, ret);
1808     ok(item.state == INDEXTOSTATEIMAGEMASK(1), "got 0x%x\n", item.state);
1809
1810     DestroyWindow(hTree);
1811 }
1812
1813 static void test_TVM_GETNEXTITEM(void)
1814 {
1815     HTREEITEM item;
1816     HWND hTree;
1817
1818     hTree = create_treeview_control(0);
1819     fill_tree(hTree);
1820
1821     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, 0);
1822     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1823
1824     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, (LPARAM)TVI_ROOT);
1825     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1826
1827     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, (LPARAM)hRoot);
1828     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1829
1830     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_ROOT, (LPARAM)hChild);
1831     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1832
1833     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_CHILD, 0);
1834     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1835
1836     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_CHILD, (LPARAM)hRoot);
1837     ok(item == hChild, "got %p, expected %p\n", item, hChild);
1838
1839     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_CHILD, (LPARAM)TVI_ROOT);
1840     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1841
1842     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_PARENT, 0);
1843     ok(item == NULL, "got %p\n", item);
1844
1845     item = (HTREEITEM)SendMessageA(hTree, TVM_GETNEXTITEM, TVGN_PARENT, (LPARAM)hChild);
1846     ok(item == hRoot, "got %p, expected %p\n", item, hRoot);
1847
1848     DestroyWindow(hTree);
1849 }
1850
1851 static void test_TVM_HITTEST(void)
1852 {
1853     HWND hTree;
1854     LRESULT ret;
1855     RECT rc;
1856     TVHITTESTINFO ht;
1857
1858     hTree = create_treeview_control(0);
1859     fill_tree(hTree);
1860
1861     *(HTREEITEM*)&rc = hRoot;
1862     ret = SendMessage(hTree, TVM_GETITEMRECT, TRUE, (LPARAM)&rc);
1863     expect(TRUE, (BOOL)ret);
1864
1865     ht.pt.x = rc.left-1;
1866     ht.pt.y = rc.top;
1867
1868     ret = SendMessage(hTree, TVM_HITTEST, 0, (LPARAM)&ht);
1869     ok((HTREEITEM)ret == hRoot, "got %p, expected %p\n", (HTREEITEM)ret, hRoot);
1870     ok(ht.hItem == hRoot, "got %p, expected %p\n", ht.hItem, hRoot);
1871     ok(ht.flags == TVHT_ONITEMBUTTON, "got %d, expected %d\n", ht.flags, TVHT_ONITEMBUTTON);
1872
1873     ret = SendMessageA(hTree, TVM_EXPAND, TVE_EXPAND, (LPARAM)hRoot);
1874     expect(TRUE, (BOOL)ret);
1875
1876     *(HTREEITEM*)&rc = hChild;
1877     ret = SendMessage(hTree, TVM_GETITEMRECT, TRUE, (LPARAM)&rc);
1878     expect(TRUE, (BOOL)ret);
1879
1880     ht.pt.x = rc.left-1;
1881     ht.pt.y = rc.top;
1882
1883     ret = SendMessage(hTree, TVM_HITTEST, 0, (LPARAM)&ht);
1884     ok((HTREEITEM)ret == hChild, "got %p, expected %p\n", (HTREEITEM)ret, hChild);
1885     ok(ht.hItem == hChild, "got %p, expected %p\n", ht.hItem, hChild);
1886     /* Wine returns item button here, but this item has no button */
1887     todo_wine ok(ht.flags == TVHT_ONITEMINDENT, "got %d, expected %d\n", ht.flags, TVHT_ONITEMINDENT);
1888
1889     DestroyWindow(hTree);
1890 }
1891
1892 START_TEST(treeview)
1893 {
1894     HMODULE hComctl32;
1895     BOOL (WINAPI *pInitCommonControlsEx)(const INITCOMMONCONTROLSEX*);
1896     WNDCLASSA wc;
1897     MSG msg;
1898
1899     ULONG_PTR ctx_cookie;
1900     HANDLE hCtx;
1901     HWND hwnd;
1902
1903     hComctl32 = GetModuleHandleA("comctl32.dll");
1904     pInitCommonControlsEx = (void*)GetProcAddress(hComctl32, "InitCommonControlsEx");
1905     if (pInitCommonControlsEx)
1906     {
1907         INITCOMMONCONTROLSEX iccex;
1908         iccex.dwSize = sizeof(iccex);
1909         iccex.dwICC  = ICC_TREEVIEW_CLASSES;
1910         pInitCommonControlsEx(&iccex);
1911     }
1912     else
1913         InitCommonControls();
1914
1915     init_msg_sequences(sequences, NUM_MSG_SEQUENCES);
1916     init_msg_sequences(item_sequence, 1);
1917
1918     wc.style = CS_HREDRAW | CS_VREDRAW;
1919     wc.cbClsExtra = 0;
1920     wc.cbWndExtra = 0;
1921     wc.hInstance = GetModuleHandleA(NULL);
1922     wc.hIcon = NULL;
1923     wc.hCursor = LoadCursorA(NULL, IDC_IBEAM);
1924     wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
1925     wc.lpszMenuName = NULL;
1926     wc.lpszClassName = "MyTestWnd";
1927     wc.lpfnWndProc = parent_wnd_proc;
1928     RegisterClassA(&wc);
1929
1930     hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW,
1931       CW_USEDEFAULT, CW_USEDEFAULT, 130, 105, NULL, NULL, GetModuleHandleA(NULL), 0);
1932
1933     ok(hMainWnd != NULL, "Failed to create parent window. Tests aborted.\n");
1934     if (!hMainWnd) return;
1935
1936     test_fillroot();
1937     test_select();
1938     test_getitemtext();
1939     test_focus();
1940     test_get_set_bkcolor();
1941     test_get_set_imagelist();
1942     test_get_set_indent();
1943     test_get_set_insertmark();
1944     test_get_set_item();
1945     test_get_set_itemheight();
1946     test_get_set_scrolltime();
1947     test_get_set_textcolor();
1948     test_get_linecolor();
1949     test_get_insertmarkcolor();
1950     test_get_set_tooltips();
1951     test_get_set_unicodeformat();
1952     test_callback();
1953     test_expandinvisible();
1954     test_itemedit();
1955     test_treeview_classinfo();
1956     test_expandnotify();
1957     test_TVS_SINGLEEXPAND();
1958     test_WM_PAINT();
1959     test_delete_items();
1960     test_cchildren();
1961     test_htreeitem_layout();
1962     test_TVS_CHECKBOXES();
1963     test_TVM_GETNEXTITEM();
1964     test_TVM_HITTEST();
1965
1966     if (!load_v6_module(&ctx_cookie, &hCtx))
1967     {
1968         DestroyWindow(hMainWnd);
1969         return;
1970     }
1971
1972     /* this is a XP SP3 failure workaround */
1973     hwnd = CreateWindowExA(0, WC_TREEVIEW, "foo",
1974                            WS_CHILD | WS_BORDER | WS_VISIBLE,
1975                            0, 0, 100, 100,
1976                            hMainWnd, NULL, GetModuleHandleA(NULL), NULL);
1977     if (!IsWindow(hwnd))
1978     {
1979         win_skip("FIXME: failed to create TreeView window.\n");
1980         unload_v6_module(ctx_cookie, hCtx);
1981         DestroyWindow(hMainWnd);
1982         return;
1983     }
1984     else
1985         DestroyWindow(hwnd);
1986
1987     /* comctl32 version 6 tests start here */
1988     test_expandedimage();
1989     test_htreeitem_layout();
1990
1991     unload_v6_module(ctx_cookie, hCtx);
1992
1993     PostMessageA(hMainWnd, WM_CLOSE, 0, 0);
1994     while(GetMessageA(&msg, 0, 0, 0))
1995     {
1996         TranslateMessage(&msg);
1997         DispatchMessageA(&msg);
1998     }
1999 }