git.oblomov.eu Git - wine/blob - 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 "msg.h"
  34
  35 #define NUM_MSG_SEQUENCES   1
  36 #define LISTVIEW_SEQ_INDEX  0
  37
  38 static struct msg_sequence *MsgSequences[NUM_MSG_SEQUENCES];
  39
  40 static const struct message FillRootSeq[] = {
  41     { TVM_INSERTITEM, sent },
  42     { TVM_GETITEM, sent },
  43     { TVM_INSERTITEM, sent },
  44     { 0 }
  45 };
  46
  47 static const struct message DoTest1Seq[] = {
  48     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  49     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  50     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  51     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  52     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  53     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  54     { 0 }
  55 };
  56
  57 static const struct message DoTest2Seq[] = {
  58     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  59     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  60     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  61     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  62     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  63     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  64     { 0 }
  65 };
  66
  67 static const struct message DoFocusTestSeq[] = {
  68     { TVM_INSERTITEM, sent },
  69     { TVM_INSERTITEM, sent },
  70     { WM_WINDOWPOSCHANGING, sent|defwinproc },
  71     { WM_NCCALCSIZE, sent|wparam|defwinproc, 0x00000001 },
  72     { WM_WINDOWPOSCHANGED, sent|defwinproc },
  73     { WM_SIZE, sent|defwinproc },
  74     { WM_WINDOWPOSCHANGING, sent },
  75     { WM_NCCALCSIZE, sent|wparam, 0x00000001 },
  76     { WM_WINDOWPOSCHANGED, sent },
  77     { WM_SIZE, sent|defwinproc },
  78     { WM_WINDOWPOSCHANGING, sent|defwinproc },
  79     { WM_NCCALCSIZE, sent|wparam|defwinproc, 0x00000001 },
  80     { WM_WINDOWPOSCHANGED, sent|defwinproc },
  81     { WM_SIZE, sent|defwinproc },
  82     { TVM_SELECTITEM, sent|wparam, 0x00000009 },
  83     /* The following end up out of order in wine */
  84     { WM_PAINT, sent|defwinproc },
  85     { WM_NCPAINT, sent|wparam|defwinproc, 0x00000001 },
  86     { WM_ERASEBKGND, sent|defwinproc },
  87     { TVM_EDITLABEL, sent },
  88     { WM_COMMAND, sent|wparam|defwinproc, 0x04000000 },
  89     { WM_COMMAND, sent|wparam|defwinproc, 0x03000000 },
  90     { WM_PARENTNOTIFY, sent|wparam|defwinproc, 0x00000001 },
  91     { WM_KILLFOCUS, sent|defwinproc },
  92     { WM_PAINT, sent|defwinproc },
  93     { WM_IME_SETCONTEXT, sent|defwinproc|optional },
  94     { WM_COMMAND, sent|wparam|defwinproc, 0x01000000},
  95     { WM_ERASEBKGND, sent|defwinproc },
  96     { WM_CTLCOLOREDIT, sent|defwinproc|optional },
  97     { WM_CTLCOLOREDIT, sent|defwinproc|optional },
  98     { 0 }
  99 };
 100
 101 static const struct message TestGetSetBkColorSeq[] = {
 102     { TVM_GETBKCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 103     { TVM_SETBKCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 104     { TVM_GETBKCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 105     { TVM_SETBKCOLOR, sent|wparam|lparam, 0x00000000, 0x00ffffff },
 106     { TVM_GETBKCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 107     { TVM_SETBKCOLOR, sent|wparam|lparam, 0x00000000, 0xffffffff },
 108     { 0 }
 109 };
 110
 111 static const struct message TestGetSetImageListSeq[] = {
 112     { TVM_SETIMAGELIST, sent|wparam|lparam, 0x00000000, 0x00000000 },
 113     { TVM_GETIMAGELIST, sent|wparam|lparam, 0x00000000, 0x00000000 },
 114     { 0 }
 115 };
 116
 117 static const struct message TestGetSetIndentSeq[] = {
 118     { TVM_SETINDENT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 119     { TVM_GETINDENT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 120     /* The actual amount to indent is dependent on the system for this message */
 121     { TVM_SETINDENT, sent },
 122     { TVM_GETINDENT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 123     { 0 }
 124 };
 125
 126 static const struct message TestGetSetInsertMarkColorSeq[] = {
 127     { TVM_SETINSERTMARKCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 128     { TVM_GETINSERTMARKCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 129     { 0 }
 130 };
 131
 132 static const struct message TestGetSetItemSeq[] = {
 133     { TVM_GETITEM, sent },
 134     { TVM_SETITEM, sent },
 135     { TVM_GETITEM, sent },
 136     { TVM_SETITEM, sent },
 137     { 0 }
 138 };
 139
 140 static const struct message TestGetSetItemHeightSeq[] = {
 141     { TVM_GETITEMHEIGHT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 142     { TVM_SETITEMHEIGHT, sent|wparam|lparam, 0xffffffff, 0x00000000 },
 143     { TVM_GETITEMHEIGHT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 144     { TVM_SETITEMHEIGHT, sent|wparam|lparam, 0x00000020, 0x00000000 },
 145     { TVM_GETITEMHEIGHT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 146     { TVM_SETITEMHEIGHT, sent|wparam|lparam, 0x00000009, 0x00000000 },
 147     { WM_WINDOWPOSCHANGING, sent|defwinproc },
 148     { WM_NCCALCSIZE, sent|wparam|defwinproc, 0x00000001 },
 149     { WM_WINDOWPOSCHANGED, sent|defwinproc },
 150     { WM_SIZE, sent|defwinproc },
 151     { TVM_GETITEMHEIGHT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 152     { 0 }
 153 };
 154
 155 static const struct message TestGetSetScrollTimeSeq[] = {
 156     { TVM_SETSCROLLTIME, sent|wparam|lparam, 0x00000014, 0x00000000 },
 157     { TVM_GETSCROLLTIME, sent|wparam|lparam, 0x00000000, 0x00000000 },
 158     { 0 }
 159 };
 160
 161 static const struct message TestGetSetTextColorSeq[] = {
 162     { TVM_GETTEXTCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 163     { TVM_SETTEXTCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 164     { TVM_GETTEXTCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 165     { TVM_SETTEXTCOLOR, sent|wparam|lparam, 0x00000000, 0x00ffffff },
 166     { TVM_GETTEXTCOLOR, sent|wparam|lparam, 0x00000000, 0x00000000 },
 167     { TVM_SETTEXTCOLOR, sent|wparam|lparam, 0x00000000, 0xffffffff },
 168     { 0 }
 169 };
 170
 171 static const struct message TestGetSetToolTipsSeq[] = {
 172     { WM_COMMAND,       sent|wparam,            0x02000000 },
 173     { WM_PARENTNOTIFY,  sent|wparam|defwinproc, 0x00020002 },
 174     { TVM_SETTOOLTIPS, sent|wparam|lparam, 0x00000000, 0x00000000 },
 175     { TVM_GETTOOLTIPS, sent|wparam|lparam, 0x00000000, 0x00000000 },
 176     { 0 }
 177 };
 178
 179 static const struct message TestGetSetUnicodeFormatSeq[] = {
 180     { TVM_SETUNICODEFORMAT, sent|wparam|lparam, 0x00000001, 0x00000000 },
 181     { TVM_GETUNICODEFORMAT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 182     { TVM_SETUNICODEFORMAT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 183     { TVM_GETUNICODEFORMAT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 184     { TVM_SETUNICODEFORMAT, sent|wparam|lparam, 0x00000000, 0x00000000 },
 185     { 0 }
 186 };
 187
 188 static HWND hMainWnd;
 189
 190 static HWND hTree, hEdit;
 191 static HTREEITEM hRoot, hChild;
 192
 193 static int pos = 0;
 194 static char sequence[256];
 195
 196 static void Clear(void)
 197 {
 198     pos = 0;
 199     sequence[0] = '\0';
 200 }
 201
 202 static void AddItem(char ch)
 203 {
 204     sequence[pos++] = ch;
 205     sequence[pos] = '\0';
 206 }
 207
 208 static void IdentifyItem(HTREEITEM hItem)
 209 {
 210     if (hItem == hRoot) {
 211         AddItem('R');
 212         return;
 213     }
 214     if (hItem == hChild) {
 215         AddItem('C');
 216         return;
 217     }
 218     if (hItem == NULL) {
 219         AddItem('n');
 220         return;
 221     }
 222     AddItem('?');
 223 }
 224
 225 static void FillRoot(void)
 226 {
 227     TVINSERTSTRUCTA ins;
 228     TVITEM tvi;
 229     static CHAR root[]  = "Root",
 230                 child[] = "Child";
 231
 232     Clear();
 233     AddItem('A');
 234     ins.hParent = TVI_ROOT;
 235     ins.hInsertAfter = TVI_ROOT;
 236     U(ins).item.mask = TVIF_TEXT;
 237     U(ins).item.pszText = root;
 238     hRoot = TreeView_InsertItem(hTree, &ins);
 239     assert(hRoot);
 240
 241     /* UMLPad 1.15 depends on this being not -1 (I_IMAGECALLBACK) */
 242     tvi.hItem = hRoot;
 243     tvi.mask = TVIF_IMAGE | TVIF_SELECTEDIMAGE;
 244     SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tvi );
 245     ok(tvi.iImage == 0, "tvi.iImage=%d\n", tvi.iImage);
 246     ok(tvi.iSelectedImage == 0, "tvi.iSelectedImage=%d\n", tvi.iSelectedImage);
 247
 248     AddItem('B');
 249     ins.hParent = hRoot;
 250     ins.hInsertAfter = TVI_FIRST;
 251     U(ins).item.mask = TVIF_TEXT;
 252     U(ins).item.pszText = child;
 253     hChild = TreeView_InsertItem(hTree, &ins);
 254     assert(hChild);
 255     AddItem('.');
 256
 257     ok(!strcmp(sequence, "AB."), "Item creation\n");
 258 }
 259
 260 static void DoTest1(void)
 261 {
 262     BOOL r;
 263     r = TreeView_SelectItem(hTree, NULL);
 264     Clear();
 265     AddItem('1');
 266     r = TreeView_SelectItem(hTree, hRoot);
 267     AddItem('2');
 268     r = TreeView_SelectItem(hTree, hRoot);
 269     AddItem('3');
 270     r = TreeView_SelectItem(hTree, NULL);
 271     AddItem('4');
 272     r = TreeView_SelectItem(hTree, NULL);
 273     AddItem('5');
 274     r = TreeView_SelectItem(hTree, hRoot);
 275     AddItem('.');
 276     ok(!strcmp(sequence, "1(nR)nR23(Rn)Rn45(nR)nR."), "root-none select test\n");
 277 }
 278
 279 static void DoTest2(void)
 280 {
 281     BOOL r;
 282     r = TreeView_SelectItem(hTree, NULL);
 283     Clear();
 284     AddItem('1');
 285     r = TreeView_SelectItem(hTree, hRoot);
 286     AddItem('2');
 287     r = TreeView_SelectItem(hTree, hRoot);
 288     AddItem('3');
 289     r = TreeView_SelectItem(hTree, hChild);
 290     AddItem('4');
 291     r = TreeView_SelectItem(hTree, hChild);
 292     AddItem('5');
 293     r = TreeView_SelectItem(hTree, hRoot);
 294     AddItem('.');
 295     ok(!strcmp(sequence, "1(nR)nR23(RC)RC45(CR)CR."), "root-child select test\n");
 296 }
 297
 298 static void DoFocusTest(void)
 299 {
 300     TVINSERTSTRUCTA ins;
 301     static CHAR child1[]  = "Edit",
 302                 child2[]  = "A really long string";
 303     HTREEITEM hChild1, hChild2;
 304
 305     /* This test verifies that when a label is being edited, scrolling
 306      * the treeview does not cause the label to lose focus. To test
 307      * this, first some additional entries are added to generate
 308      * scrollbars.
 309      */
 310     ins.hParent = hRoot;
 311     ins.hInsertAfter = hChild;
 312     U(ins).item.mask = TVIF_TEXT;
 313     U(ins).item.pszText = child1;
 314     hChild1 = TreeView_InsertItem(hTree, &ins);
 315     assert(hChild1);
 316     ins.hInsertAfter = hChild1;
 317     U(ins).item.mask = TVIF_TEXT;
 318     U(ins).item.pszText = child2;
 319     hChild2 = TreeView_InsertItem(hTree, &ins);
 320     assert(hChild2);
 321
 322     ShowWindow(hMainWnd,SW_SHOW);
 323     /* Using SendMessageA since Win98 doesn't have default unicode support */
 324     SendMessageA(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hChild);
 325     hEdit = TreeView_EditLabel(hTree, hChild);
 326     ScrollWindowEx(hTree, -10, 0, NULL, NULL, NULL, NULL, SW_SCROLLCHILDREN);
 327     ok(GetFocus() == hEdit, "Edit control should have focus\n");
 328 }
 329
 330 static void TestGetSetBkColor(void)
 331 {
 332     COLORREF crColor = RGB(0,0,0);
 333
 334     todo_wine{
 335         /* If the value is -1, the control is using the system color for the background color. */
 336         crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
 337         ok(crColor == -1, "Default background color reported as 0x%.8x\n", crColor);
 338     }
 339
 340     /* Test for black background */
 341     SendMessage( hTree, TVM_SETBKCOLOR, 0, (LPARAM)RGB(0,0,0) );
 342     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
 343     ok(crColor == RGB(0,0,0), "Black background color reported as 0x%.8x\n", crColor);
 344
 345     /* Test for white background */
 346     SendMessage( hTree, TVM_SETBKCOLOR, 0, (LPARAM)RGB(255,255,255) );
 347     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
 348     ok(crColor == RGB(255,255,255), "White background color reported as 0x%.8x\n", crColor);
 349
 350     /* Reset the default background */
 351     SendMessage( hTree, TVM_SETBKCOLOR, 0, -1 );
 352 }
 353
 354 static void TestGetSetImageList(void)
 355 {
 356     HIMAGELIST hImageList = NULL;
 357
 358     /* Test a NULL HIMAGELIST */
 359     SendMessage( hTree, TVM_SETIMAGELIST, TVSIL_NORMAL, (LPARAM)hImageList );
 360     hImageList = (HIMAGELIST)SendMessage( hTree, TVM_GETIMAGELIST, TVSIL_NORMAL, 0 );
 361     ok(hImageList == NULL, "NULL image list, reported as 0x%p, expected 0.\n", hImageList);
 362
 363     /* TODO: Test an actual image list */
 364 }
 365
 366 static void TestGetSetIndent(void)
 367 {
 368     int ulIndent = -1;
 369     int ulMinIndent = -1;
 370     int ulMoreThanTwiceMin = -1;
 371
 372     /* Finding the minimum indent */
 373     SendMessage( hTree, TVM_SETINDENT, 0, 0 );
 374     ulMinIndent = (int)SendMessage( hTree, TVM_GETINDENT, 0, 0 );
 375
 376     /* Checking an indent that is more than twice the default indent */
 377     ulMoreThanTwiceMin = 2*ulMinIndent+1;
 378     SendMessage( hTree, TVM_SETINDENT, ulMoreThanTwiceMin, 0 );
 379     ulIndent = (DWORD)SendMessage( hTree, TVM_GETINDENT, 0, 0 );
 380     ok(ulIndent == ulMoreThanTwiceMin, "Indent reported as %d, expected %d\n", ulIndent, ulMoreThanTwiceMin);
 381 }
 382
 383 static void TestGetSetInsertMarkColor(void)
 384 {
 385     COLORREF crColor = RGB(0,0,0);
 386     SendMessage( hTree, TVM_SETINSERTMARKCOLOR, 0, crColor );
 387     crColor = (COLORREF)SendMessage( hTree, TVM_GETINSERTMARKCOLOR, 0, 0 );
 388     ok(crColor == RGB(0,0,0), "Insert mark color reported as 0x%.8x, expected 0x00000000\n", crColor);
 389 }
 390
 391 static void TestGetSetItem(void)
 392 {
 393     TVITEM tviRoot = {0};
 394     int nBufferSize = 80;
 395     char szBuffer[80] = {0};
 396
 397     /* Test the root item */
 398     tviRoot.hItem = hRoot;
 399     tviRoot.mask = TVIF_TEXT;
 400     tviRoot.cchTextMax = nBufferSize;
 401     tviRoot.pszText = szBuffer;
 402     SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tviRoot );
 403     ok(!strcmp("Root", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Root\"\n", szBuffer);
 404
 405     /* Change the root text */
 406     strncpy(szBuffer, "Testing123", nBufferSize);
 407     SendMessage( hTree, TVM_SETITEM, 0, (LPARAM)&tviRoot );
 408     memset(szBuffer, 0, nBufferSize);
 409     SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tviRoot );
 410     ok(!strcmp("Testing123", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Testing123\"\n", szBuffer);
 411
 412     /* Reset the root text */
 413     memset(szBuffer, 0, nBufferSize);
 414     strncpy(szBuffer, "Root", nBufferSize);
 415     SendMessage( hTree, TVM_SETITEM, 0, (LPARAM)&tviRoot );
 416 }
 417
 418 static void TestGetSetItemHeight(void)
 419 {
 420     int ulOldHeight = 0;
 421     int ulNewHeight = 0;
 422
 423     /* Assuming default height to begin with */
 424     ulOldHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
 425
 426     /* Explicitly setting and getting the default height */
 427     SendMessage( hTree, TVM_SETITEMHEIGHT, -1, 0 );
 428     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
 429     ok(ulNewHeight == ulOldHeight, "Default height not set properly, reported %d, expected %d\n", ulNewHeight, ulOldHeight);
 430
 431     /* Explicitly setting and getting the height of twice the normal */
 432     SendMessage( hTree, TVM_SETITEMHEIGHT, 2*ulOldHeight, 0 );
 433     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
 434     ok(ulNewHeight == 2*ulOldHeight, "New height not set properly, reported %d, expected %d\n", ulNewHeight, 2*ulOldHeight);
 435
 436     /* Assuming tree doesn't have TVS_NONEVENHEIGHT set, so a set of 9 will round down to 8 */
 437     SendMessage( hTree, TVM_SETITEMHEIGHT, 9, 0 );
 438     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
 439     ok(ulNewHeight == 8, "Uneven height not set properly, reported %d, expected %d\n", ulNewHeight, 8);
 440 }
 441
 442 static void TestGetSetScrollTime(void)
 443 {
 444     int ulExpectedTime = 20;
 445     int ulTime = 0;
 446     SendMessage( hTree, TVM_SETSCROLLTIME, ulExpectedTime, 0 );
 447     ulTime = (int)SendMessage( hTree, TVM_GETSCROLLTIME, 0, 0 );
 448     ok(ulTime == ulExpectedTime, "Scroll time reported as %d, expected %d\n", ulTime, ulExpectedTime);
 449 }
 450
 451 static void TestGetSetTextColor(void)
 452 {
 453     /* If the value is -1, the control is using the system color for the text color. */
 454     COLORREF crColor = RGB(0,0,0);
 455     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
 456     ok(crColor == -1, "Default text color reported as 0x%.8x\n", crColor);
 457
 458     /* Test for black text */
 459     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, (LPARAM)RGB(0,0,0) );
 460     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
 461     ok(crColor == RGB(0,0,0), "Black text color reported as 0x%.8x\n", crColor);
 462
 463     /* Test for white text */
 464     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, (LPARAM)RGB(255,255,255) );
 465     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
 466     ok(crColor == RGB(255,255,255), "White text color reported as 0x%.8x\n", crColor);
 467
 468     /* Reset the default text color */
 469     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, -1 );
 470 }
 471
 472 static void TestGetSetToolTips(void)
 473 {
 474     HWND hwndLastToolTip = NULL;
 475     HWND hPopupTreeView;
 476
 477     /* show even WS_POPUP treeview don't send NM_TOOLTIPSCREATED */
 478     hPopupTreeView = CreateWindow(WC_TREEVIEW, NULL, WS_POPUP|WS_VISIBLE, 0, 0, 100, 100, hMainWnd, NULL, NULL, NULL);
 479     DestroyWindow(hPopupTreeView);
 480
 481     /* Testing setting a NULL ToolTip */
 482     SendMessage( hTree, TVM_SETTOOLTIPS, 0, 0 );
 483     hwndLastToolTip = (HWND)SendMessage( hTree, TVM_GETTOOLTIPS, 0, 0 );
 484     ok(hwndLastToolTip == NULL, "NULL tool tip, reported as 0x%p, expected 0.\n", hwndLastToolTip);
 485
 486     /* TODO: Add a test of an actual tooltip */
 487 }
 488
 489 static void TestGetSetUnicodeFormat(void)
 490 {
 491     BOOL bPreviousSetting = 0;
 492     BOOL bNewSetting = 0;
 493
 494     /* Set to Unicode */
 495     bPreviousSetting = (BOOL)SendMessage( hTree, TVM_SETUNICODEFORMAT, 1, 0 );
 496     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
 497     ok(bNewSetting == 1, "Unicode setting did not work.\n");
 498
 499     /* Set to ANSI */
 500     SendMessage( hTree, TVM_SETUNICODEFORMAT, 0, 0 );
 501     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
 502     ok(bNewSetting == 0, "ANSI setting did not work.\n");
 503
 504     /* Revert to original setting */
 505     SendMessage( hTree, TVM_SETUNICODEFORMAT, (LPARAM)bPreviousSetting, 0 );
 506 }
 507
 508 static void TestGetSet(void)
 509 {
 510     /* TVM_GETBKCOLOR and TVM_SETBKCOLOR */
 511     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 512     TestGetSetBkColor();
 513     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetBkColorSeq,
 514         "TestGetSetBkColor", FALSE);
 515
 516     /* TVM_GETIMAGELIST and TVM_SETIMAGELIST */
 517     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 518     TestGetSetImageList();
 519     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetImageListSeq,
 520         "TestGetImageList", FALSE);
 521
 522     /* TVM_SETINDENT and TVM_GETINDENT */
 523     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 524     TestGetSetIndent();
 525     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetIndentSeq,
 526         "TestGetSetIndent", FALSE);
 527
 528     /* TVM_GETINSERTMARKCOLOR and TVM_GETINSERTMARKCOLOR */
 529     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 530     TestGetSetInsertMarkColor();
 531     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetInsertMarkColorSeq,
 532         "TestGetSetInsertMarkColor", FALSE);
 533
 534     /* TVM_GETITEM and TVM_SETITEM */
 535     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 536     TestGetSetItem();
 537     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetItemSeq,
 538         "TestGetSetItem", FALSE);
 539
 540     /* TVM_GETITEMHEIGHT and TVM_SETITEMHEIGHT */
 541     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 542     TestGetSetItemHeight();
 543     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetItemHeightSeq,
 544         "TestGetSetItemHeight", FALSE);
 545
 546     /* TVM_GETSCROLLTIME and TVM_SETSCROLLTIME */
 547     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 548     TestGetSetScrollTime();
 549     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetScrollTimeSeq,
 550         "TestGetSetScrollTime", FALSE);
 551
 552     /* TVM_GETTEXTCOLOR and TVM_SETTEXTCOLOR */
 553     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 554     TestGetSetTextColor();
 555     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetTextColorSeq,
 556         "TestGetSetTextColor", FALSE);
 557
 558     /* TVM_GETTOOLTIPS and TVM_SETTOOLTIPS */
 559     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 560     TestGetSetToolTips();
 561     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetToolTipsSeq,
 562         "TestGetSetToolTips", TRUE);
 563
 564     /* TVM_GETUNICODEFORMAT and TVM_SETUNICODEFORMAT */
 565     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 566     TestGetSetUnicodeFormat();
 567     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, TestGetSetUnicodeFormatSeq,
 568         "TestGetSetUnicodeFormat", FALSE);
 569 }
 570
 571 /* This function hooks in and records all messages to the treeview control */
 572 static LRESULT WINAPI TreeviewWndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
 573 {
 574     static long defwndproc_counter = 0;
 575     LRESULT ret;
 576     struct message msg;
 577     WNDPROC *lpOldProc = (WNDPROC*)GetWindowLongA(hwnd, GWL_USERDATA);
 578
 579     msg.message = message;
 580     msg.flags = sent|wparam|lparam;
 581     if (defwndproc_counter) msg.flags |= defwinproc;
 582     msg.wParam = wParam;
 583     msg.lParam = lParam;
 584     add_message(MsgSequences, LISTVIEW_SEQ_INDEX, &msg);
 585
 586     defwndproc_counter++;
 587     ret = CallWindowProcA(*lpOldProc, hwnd, message, wParam, lParam);
 588     defwndproc_counter--;
 589
 590     return ret;
 591 }
 592
 593 static LRESULT CALLBACK MyWndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
 594 {
 595     WNDPROC *pOldWndProc;
 596
 597     switch(msg) {
 598
 599     case WM_CREATE:
 600     {
 601         hTree = CreateWindowExA(WS_EX_CLIENTEDGE, WC_TREEVIEWA, NULL, WS_CHILD|WS_VISIBLE|
 602             TVS_LINESATROOT|TVS_HASLINES|TVS_HASBUTTONS|TVS_EDITLABELS,
 603             0, 0, 120, 100, hWnd, (HMENU)100, GetModuleHandleA(0), 0);
 604
 605         SetFocus(hTree);
 606
 607         pOldWndProc = HeapAlloc(GetProcessHeap(), 0, sizeof(WNDPROC));
 608         if ( !ok(pOldWndProc != NULL, "Failed to allocate memory for subclass_info.\n") )
 609         {
 610             PostQuitMessage(1);
 611             break;
 612         }
 613
 614         /* Record the old WNDPROC so we can call it after recording the messages */
 615         *pOldWndProc = (WNDPROC)SetWindowLongA(hTree, GWL_WNDPROC, (LONG)TreeviewWndProc);
 616         SetWindowLongA(hTree, GWL_USERDATA, (LONG)pOldWndProc);
 617
 618         return 0;
 619     }
 620     case WM_NOTIFY:
 621     {
 622         NMHDR *pHdr = (NMHDR *)lParam;
 623
 624         ok(pHdr->code != NM_FIRST - 19, "Treeview should not send NM_TOOLTIPSCREATED\n");
 625         if (pHdr->idFrom == 100) {
 626             NMTREEVIEWA *pTreeView = (LPNMTREEVIEWA) lParam;
 627             switch(pHdr->code) {
 628             case TVN_SELCHANGINGA:
 629                 AddItem('(');
 630                 IdentifyItem(pTreeView->itemOld.hItem);
 631                 IdentifyItem(pTreeView->itemNew.hItem);
 632                 return 0;
 633             case TVN_SELCHANGEDA:
 634                 AddItem(')');
 635                 IdentifyItem(pTreeView->itemOld.hItem);
 636                 IdentifyItem(pTreeView->itemNew.hItem);
 637                 return 0;
 638             }
 639         }
 640         return 0;
 641     }
 642
 643     case WM_SIZE:
 644         MoveWindow(hTree, 0, 0, LOWORD(lParam), HIWORD(lParam), TRUE);
 645         break;
 646
 647     case WM_DESTROY:
 648         PostQuitMessage(0);
 649         break;
 650
 651     default:
 652         return DefWindowProcA(hWnd, msg, wParam, lParam);
 653     }
 654     return 0L;
 655 }
 656
 657 START_TEST(treeview)
 658 {
 659     WNDCLASSA wc;
 660     MSG msg;
 661     INITCOMMONCONTROLSEX icex;
 662
 663     icex.dwSize = sizeof(INITCOMMONCONTROLSEX);
 664     icex.dwICC   = ICC_TREEVIEW_CLASSES;
 665     InitCommonControlsEx(&icex);
 666     init_msg_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 667
 668     wc.style = CS_HREDRAW | CS_VREDRAW;
 669     wc.cbClsExtra = 0;
 670     wc.cbWndExtra = 0;
 671     wc.hInstance = GetModuleHandleA(NULL);
 672     wc.hIcon = NULL;
 673     wc.hCursor = LoadCursorA(NULL, MAKEINTRESOURCEA(IDC_IBEAM));
 674     wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
 675     wc.lpszMenuName = NULL;
 676     wc.lpszClassName = "MyTestWnd";
 677     wc.lpfnWndProc = MyWndProc;
 678     RegisterClassA(&wc);
 679
 680
 681     hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW,
 682       CW_USEDEFAULT, CW_USEDEFAULT, 130, 105, NULL, NULL, GetModuleHandleA(NULL), 0);
 683
 684     if ( !ok(hMainWnd != NULL, "Failed to create parent window. Tests aborted.\n") )
 685         return;
 686
 687     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 688     FillRoot();
 689     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, FillRootSeq, "FillRoot", FALSE);
 690
 691     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 692     DoTest1();
 693     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, DoTest1Seq, "DoTest1", FALSE);
 694
 695     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 696     DoTest2();
 697     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, DoTest2Seq, "DoTest2", FALSE);
 698
 699     flush_sequences(MsgSequences, NUM_MSG_SEQUENCES);
 700     DoFocusTest();
 701     ok_sequence(MsgSequences, LISTVIEW_SEQ_INDEX, DoFocusTestSeq, "DoFocusTest", TRUE);
 702
 703     /* Sequences tested inside due to number */
 704     TestGetSet();
 705
 706     PostMessageA(hMainWnd, WM_CLOSE, 0, 0);
 707     while(GetMessageA(&msg,0,0,0)) {
 708         TranslateMessage(&msg);
 709         DispatchMessageA(&msg);
 710     }
 711 }