dlls/comctl32/tests/treeview.c - wine - Git at Google

 /* Unit tests for treeview.
  *
  * Copyright 2005 Krzysztof Foltman
  * Copyright 2007 Christopher James Peterson
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with this library; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  */

 #include <assert.h>
 #include <stdarg.h>

 #include "windef.h"
 #include "winbase.h"
 #include "wingdi.h"
 #include "winuser.h"
 #include "winnls.h"
 #include "winreg.h"
 #include "commctrl.h"

 #include "wine/test.h"

 static HWND hMainWnd;

 static HWND hTree, hEdit;
 static HTREEITEM hRoot, hChild;

 static int pos = 0;
 static char sequence[256];

 static void Clear(void)
 {
     pos = 0;
     sequence[0] = '\0';
 }

 static void AddItem(char ch)
 {
     sequence[pos++] = ch;
     sequence[pos] = '\0';
 }

 static void IdentifyItem(HTREEITEM hItem)
 {
     if (hItem == hRoot) {
         AddItem('R');
         return;
     }
     if (hItem == hChild) {
         AddItem('C');
         return;
     }
     if (hItem == NULL) {
         AddItem('n');
         return;
     }
     AddItem('?');
 }

 static void FillRoot(void)
 {
     TVINSERTSTRUCTA ins;
     TVITEM tvi;
     static CHAR root[]  = "Root",
                 child[] = "Child";

     Clear();
     AddItem('A');
     ins.hParent = TVI_ROOT;
     ins.hInsertAfter = TVI_ROOT;
     U(ins).item.mask = TVIF_TEXT;
     U(ins).item.pszText = root;
     hRoot = TreeView_InsertItem(hTree, &ins);
     assert(hRoot);

     /* UMLPad 1.15 depends on this being not -1 (I_IMAGECALLBACK) */
     tvi.hItem = hRoot;
     tvi.mask = TVIF_IMAGE | TVIF_SELECTEDIMAGE;
     SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tvi );
     ok(tvi.iImage == 0, "tvi.iImage=%d\n", tvi.iImage);
     ok(tvi.iSelectedImage == 0, "tvi.iSelectedImage=%d\n", tvi.iSelectedImage);

     AddItem('B');
     ins.hParent = hRoot;
     ins.hInsertAfter = TVI_FIRST;
     U(ins).item.mask = TVIF_TEXT;
     U(ins).item.pszText = child;
     hChild = TreeView_InsertItem(hTree, &ins);
     assert(hChild);
     AddItem('.');

     ok(!strcmp(sequence, "AB."), "Item creation\n");
 }

 static void DoTest1(void)
 {
     BOOL r;
     r = TreeView_SelectItem(hTree, NULL);
     Clear();
     AddItem('1');
     r = TreeView_SelectItem(hTree, hRoot);
     AddItem('2');
     r = TreeView_SelectItem(hTree, hRoot);
     AddItem('3');
     r = TreeView_SelectItem(hTree, NULL);
     AddItem('4');
     r = TreeView_SelectItem(hTree, NULL);
     AddItem('5');
     r = TreeView_SelectItem(hTree, hRoot);
     AddItem('.');
     ok(!strcmp(sequence, "1(nR)nR23(Rn)Rn45(nR)nR."), "root-none select test\n");
 }

 static void DoTest2(void)
 {
     BOOL r;
     r = TreeView_SelectItem(hTree, NULL);
     Clear();
     AddItem('1');
     r = TreeView_SelectItem(hTree, hRoot);
     AddItem('2');
     r = TreeView_SelectItem(hTree, hRoot);
     AddItem('3');
     r = TreeView_SelectItem(hTree, hChild);
     AddItem('4');
     r = TreeView_SelectItem(hTree, hChild);
     AddItem('5');
     r = TreeView_SelectItem(hTree, hRoot);
     AddItem('.');
     ok(!strcmp(sequence, "1(nR)nR23(RC)RC45(CR)CR."), "root-child select test\n");
 }

 static void DoFocusTest(void)
 {
     TVINSERTSTRUCTA ins;
     static CHAR child1[]  = "Edit",
                 child2[]  = "A really long string";
     HTREEITEM hChild1, hChild2;

     /* This test verifies that when a label is being edited, scrolling
      * the treeview does not cause the label to lose focus. To test
      * this, first some additional entries are added to generate
      * scrollbars.
      */
     ins.hParent = hRoot;
     ins.hInsertAfter = hChild;
     U(ins).item.mask = TVIF_TEXT;
     U(ins).item.pszText = child1;
     hChild1 = TreeView_InsertItem(hTree, &ins);
     assert(hChild1);
     ins.hInsertAfter = hChild1;
     U(ins).item.mask = TVIF_TEXT;
     U(ins).item.pszText = child2;
     hChild2 = TreeView_InsertItem(hTree, &ins);
     assert(hChild2);

     ShowWindow(hMainWnd,SW_SHOW);
     SendMessageW(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hChild);
     hEdit = TreeView_EditLabel(hTree, hChild);
     ScrollWindowEx(hTree, -10, 0, NULL, NULL, NULL, NULL, SW_SCROLLCHILDREN);
     ok(GetFocus() == hEdit, "Edit control should have focus\n");
 }

 static void TestGetSetBkColor(void)
 {
     COLORREF crColor = RGB(0,0,0);

     todo_wine{
         /* If the value is -1, the control is using the system color for the background color. */
         crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
         ok(crColor == -1, "Default background color reported as 0x%.8x\n", crColor);
     }

     /* Test for black background */
     SendMessage( hTree, TVM_SETBKCOLOR, 0, (LPARAM)RGB(0,0,0) );
     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
     ok(crColor == RGB(0,0,0), "Black background color reported as 0x%.8x\n", crColor);

     /* Test for white background */
     SendMessage( hTree, TVM_SETBKCOLOR, 0, (LPARAM)RGB(255,255,255) );
     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
     ok(crColor == RGB(255,255,255), "White background color reported as 0x%.8x\n", crColor);

     /* Reset the default background */
     SendMessage( hTree, TVM_SETBKCOLOR, 0, -1 );
 }

 static void TestGetSetImageList(void)
 {
     HIMAGELIST hImageList = NULL;

     /* Test a NULL HIMAGELIST */
     SendMessage( hTree, TVM_SETIMAGELIST, TVSIL_NORMAL, (LPARAM)hImageList );
     hImageList = (HIMAGELIST)SendMessage( hTree, TVM_GETIMAGELIST, TVSIL_NORMAL, 0 );
     ok(hImageList == NULL, "NULL image list, reported as 0x%p, expected 0.\n", hImageList);

     /* TODO: Test an actual image list */
 }

 static void TestGetSetIndent(void)
 {
     int ulIndent = -1;
     int ulMinIndent = -1;
     int ulMoreThanTwiceMin = -1;

     /* Finding the minimum indent */
     SendMessage( hTree, TVM_SETINDENT, 0, 0 );
     ulMinIndent = (int)SendMessage( hTree, TVM_GETINDENT, 0, 0 );

     /* Checking an indent that is more than twice the default indent */
     ulMoreThanTwiceMin = 2*ulMinIndent+1;
     SendMessage( hTree, TVM_SETINDENT, ulMoreThanTwiceMin, 0 );
     ulIndent = (DWORD)SendMessage( hTree, TVM_GETINDENT, 0, 0 );
     ok(ulIndent == ulMoreThanTwiceMin, "Indent reported as %d, expected %d\n", ulIndent, ulMoreThanTwiceMin);
 }

 static void TestGetSetInsertMarkColor(void)
 {
     COLORREF crColor = RGB(0,0,0);
     SendMessage( hTree, TVM_SETBKCOLOR, 0, crColor );
     crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
     ok(crColor == RGB(0,0,0), "Insert mark color reported as 0x%.8x, expected 0x00000000\n", crColor);
 }

 static void TestGetSetItem(void)
 {
     TVITEM tviRoot = {0};
     int nBufferSize = 80;
     char szBuffer[80] = {0};

     /* Test the root item */
     tviRoot.hItem = hRoot;
     tviRoot.mask = TVIF_TEXT;
     tviRoot.cchTextMax = nBufferSize;
     tviRoot.pszText = szBuffer;
     SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tviRoot );
     ok(!strcmp("Root", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Root\"\n", szBuffer);

     /* Change the root text */
     strncpy(szBuffer, "Testing123", nBufferSize);
     SendMessage( hTree, TVM_SETITEM, 0, (LPARAM)&tviRoot );
     memset(szBuffer, 0, nBufferSize);
     SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tviRoot );
     ok(!strcmp("Testing123", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Testing123\"\n", szBuffer);

     /* Reset the root text */
     memset(szBuffer, 0, nBufferSize);
     strncpy(szBuffer, "Root", nBufferSize);
     SendMessage( hTree, TVM_SETITEM, 0, (LPARAM)&tviRoot );
 }

 static void TestGetSetItemHeight(void)
 {
     int ulOldHeight = 0;
     int ulNewHeight = 0;

     /* Assuming default height to begin with */
     ulOldHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );

     /* Explicitly setting and getting the default height */
     SendMessage( hTree, TVM_SETITEMHEIGHT, -1, 0 );
     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
     ok(ulNewHeight == ulOldHeight, "Default height not set properly, reported %d, expected %d\n", ulNewHeight, ulOldHeight);

     /* Explicitly setting and getting the height of twice the normal */
     SendMessage( hTree, TVM_SETITEMHEIGHT, 2*ulOldHeight, 0 );
     ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
     ok(ulNewHeight == 2*ulOldHeight, "New height not set properly, reported %d, expected %d\n", ulNewHeight, 2*ulOldHeight);

     todo_wine {
         /* Assuming tree doesn't have TVS_NONEVENHEIGHT set, so a set of 9 will round down to 8 */
         SendMessage( hTree, TVM_SETITEMHEIGHT, 9, 0 );
         ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
         ok(ulNewHeight == 8, "Uneven height not set properly, reported %d, expected %d\n", ulNewHeight, 8);
     }
 }

 static void TestGetSetScrollTime(void)
 {
     int ulExpectedTime = 20;
     int ulTime = 0;
     SendMessage( hTree, TVM_SETSCROLLTIME, ulExpectedTime, 0 );
     ulTime = (int)SendMessage( hTree, TVM_GETSCROLLTIME, 0, 0 );
     ok(ulTime == ulExpectedTime, "Scroll time reported as %d, expected %d\n", ulTime, ulExpectedTime);
 }

 static void TestGetSetTextColor(void)
 {
     /* If the value is -1, the control is using the system color for the text color. */
     COLORREF crColor = RGB(0,0,0);
     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
     ok(crColor == -1, "Default text color reported as 0x%.8x\n", crColor);

     /* Test for black text */
     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, (LPARAM)RGB(0,0,0) );
     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
     ok(crColor == RGB(0,0,0), "Black text color reported as 0x%.8x\n", crColor);

     /* Test for white text */
     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, (LPARAM)RGB(255,255,255) );
     crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
     ok(crColor == RGB(255,255,255), "White text color reported as 0x%.8x\n", crColor);

     /* Reset the default text color */
     SendMessage( hTree, TVM_SETTEXTCOLOR, 0, -1 );
 }

 static void TestGetSetToolTips(void)
 {
     HWND hwndLastToolTip = NULL;

     /* Testing setting a NULL ToolTip */
     SendMessage( hTree, TVM_SETTOOLTIPS, 0, 0 );
     hwndLastToolTip = (HWND)SendMessage( hTree, TVM_GETTOOLTIPS, 0, 0 );
     ok(hwndLastToolTip == NULL, "NULL tool tip, reported as 0x%p, expected 0.\n", hwndLastToolTip);

     /* TODO: Add a test of an actual tooltip */
 }

 static void TestGetSetUnicodeFormat(void)
 {
     BOOL bPreviousSetting = 0;
     BOOL bNewSetting = 0;

     /* Set to Unicode */
     bPreviousSetting = (BOOL)SendMessage( hTree, TVM_SETUNICODEFORMAT, 1, 0 );
     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
     ok(bNewSetting == 1, "Unicode setting did not work.\n");

     /* Set to ANSI */
     SendMessage( hTree, TVM_SETUNICODEFORMAT, 0, 0 );
     bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
     ok(bNewSetting == 0, "ANSI setting did not work.\n");

     /* Revert to original setting */
     SendMessage( hTree, TVM_SETUNICODEFORMAT, (LPARAM)bPreviousSetting, 0 );
 }

 static void TestGetSet(void)
 {
     TestGetSetBkColor();            /* TVM_GETBKCOLOR and TVM_SETBKCOLOR */
     TestGetSetImageList();          /* TVM_GETIMAGELIST and TVM_SETIMAGELIST */
     TestGetSetIndent();             /* TVM_SETINDENT and TVM_GETINDENT */
     TestGetSetInsertMarkColor();    /* TVM_GETINSERTMARKCOLOR and TVM_GETINSERTMARKCOLOR */
     TestGetSetItem();               /* TVM_GETITEM and TVM_SETITEM */
     TestGetSetItemHeight();         /* TVM_GETITEMHEIGHT and TVM_SETITEMHEIGHT*/
     TestGetSetScrollTime();         /* TVM_GETSCROLLTIME and TVM_SETSCROLLTIME */
     TestGetSetTextColor();          /* TVM_GETTEXTCOLOR and TVM_SETTEXTCOLOR */
     TestGetSetToolTips();           /* TVM_GETTOOLTIPS and TVM_SETTOOLTIPS */
     TestGetSetUnicodeFormat();      /* TVM_GETUNICODEFORMAT and TVM_SETUNICODEFORMAT */
 }

 static LRESULT CALLBACK MyWndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
 {
     switch(msg) {

     case WM_CREATE:
     {
         hTree = CreateWindowExA(WS_EX_CLIENTEDGE, WC_TREEVIEWA, NULL, WS_CHILD|WS_VISIBLE|
             TVS_LINESATROOT|TVS_HASLINES|TVS_HASBUTTONS|TVS_EDITLABELS,
             0, 0, 120, 100, hWnd, (HMENU)100, GetModuleHandleA(0), 0);

         SetFocus(hTree);
         return 0;
     }
     case WM_NOTIFY:
     {
         NMHDR *pHdr = (NMHDR *)lParam;

         if (pHdr->idFrom == 100) {
             NMTREEVIEWA *pTreeView = (LPNMTREEVIEWA) lParam;
             switch(pHdr->code) {
             case TVN_SELCHANGINGA:
                 AddItem('(');
                 IdentifyItem(pTreeView->itemOld.hItem);
                 IdentifyItem(pTreeView->itemNew.hItem);
                 return 0;
             case TVN_SELCHANGEDA:
                 AddItem(')');
                 IdentifyItem(pTreeView->itemOld.hItem);
                 IdentifyItem(pTreeView->itemNew.hItem);
                 return 0;
             }
         }
         return 0;
     }

     case WM_SIZE:
         MoveWindow(hTree, 0, 0, LOWORD(lParam), HIWORD(lParam), TRUE);
         break;

     case WM_DESTROY:
         PostQuitMessage(0);
         break;

     default:
         return DefWindowProcA(hWnd, msg, wParam, lParam);
     }
     return 0L;
 }

 START_TEST(treeview)
 {
     WNDCLASSA wc;
     MSG msg;
     INITCOMMONCONTROLSEX icex;
     RECT rc;

     icex.dwSize = sizeof(INITCOMMONCONTROLSEX);
     icex.dwICC   = ICC_TREEVIEW_CLASSES;
     InitCommonControlsEx(&icex);

     wc.style = CS_HREDRAW | CS_VREDRAW;
     wc.cbClsExtra = 0;
     wc.cbWndExtra = 0;
     wc.hInstance = GetModuleHandleA(NULL);
     wc.hIcon = NULL;
     wc.hCursor = LoadCursorA(NULL, MAKEINTRESOURCEA(IDC_IBEAM));
     wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
     wc.lpszMenuName = NULL;
     wc.lpszClassName = "MyTestWnd";
     wc.lpfnWndProc = MyWndProc;
     RegisterClassA(&wc);


     hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW,
       CW_USEDEFAULT, CW_USEDEFAULT, 130, 105, NULL, NULL, GetModuleHandleA(NULL), 0);
     GetClientRect(hMainWnd, &rc);

     FillRoot();
     DoTest1();
     DoTest2();
     DoFocusTest();
     TestGetSet();

     PostMessageA(hMainWnd, WM_CLOSE, 0, 0);
     while(GetMessageA(&msg,0,0,0)) {
         TranslateMessage(&msg);
         DispatchMessageA(&msg);
     }
 }
	/* Unit tests for treeview.
	*
	* Copyright 2005 Krzysztof Foltman
	* Copyright 2007 Christopher James Peterson
	*
	* This library is free software; you can redistribute it and/or
	* modify it under the terms of the GNU Lesser General Public
	* License as published by the Free Software Foundation; either
	* version 2.1 of the License, or (at your option) any later version.
	*
	* This library is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General Public
	* License along with this library; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
	*/

	#include <assert.h>
	#include <stdarg.h>

	#include "windef.h"
	#include "winbase.h"
	#include "wingdi.h"
	#include "winuser.h"
	#include "winnls.h"
	#include "winreg.h"
	#include "commctrl.h"

	#include "wine/test.h"

	static HWND hMainWnd;

	static HWND hTree, hEdit;
	static HTREEITEM hRoot, hChild;

	static int pos = 0;
	static char sequence[256];

	static void Clear(void)
	{
	pos = 0;
	sequence[0] = '\0';
	}

	static void AddItem(char ch)
	{
	sequence[pos++] = ch;
	sequence[pos] = '\0';
	}

	static void IdentifyItem(HTREEITEM hItem)
	{
	if (hItem == hRoot) {
	AddItem('R');
	return;
	}
	if (hItem == hChild) {
	AddItem('C');
	return;
	}
	if (hItem == NULL) {
	AddItem('n');
	return;
	}
	AddItem('?');
	}

	static void FillRoot(void)
	{
	TVINSERTSTRUCTA ins;
	TVITEM tvi;
	static CHAR root[] = "Root",
	child[] = "Child";

	Clear();
	AddItem('A');
	ins.hParent = TVI_ROOT;
	ins.hInsertAfter = TVI_ROOT;
	U(ins).item.mask = TVIF_TEXT;
	U(ins).item.pszText = root;
	hRoot = TreeView_InsertItem(hTree, &ins);
	assert(hRoot);

	/* UMLPad 1.15 depends on this being not -1 (I_IMAGECALLBACK) */
	tvi.hItem = hRoot;
	tvi.mask = TVIF_IMAGE \| TVIF_SELECTEDIMAGE;
	SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tvi );
	ok(tvi.iImage == 0, "tvi.iImage=%d\n", tvi.iImage);
	ok(tvi.iSelectedImage == 0, "tvi.iSelectedImage=%d\n", tvi.iSelectedImage);

	AddItem('B');
	ins.hParent = hRoot;
	ins.hInsertAfter = TVI_FIRST;
	U(ins).item.mask = TVIF_TEXT;
	U(ins).item.pszText = child;
	hChild = TreeView_InsertItem(hTree, &ins);
	assert(hChild);
	AddItem('.');

	ok(!strcmp(sequence, "AB."), "Item creation\n");
	}

	static void DoTest1(void)
	{
	BOOL r;
	r = TreeView_SelectItem(hTree, NULL);
	Clear();
	AddItem('1');
	r = TreeView_SelectItem(hTree, hRoot);
	AddItem('2');
	r = TreeView_SelectItem(hTree, hRoot);
	AddItem('3');
	r = TreeView_SelectItem(hTree, NULL);
	AddItem('4');
	r = TreeView_SelectItem(hTree, NULL);
	AddItem('5');
	r = TreeView_SelectItem(hTree, hRoot);
	AddItem('.');
	ok(!strcmp(sequence, "1(nR)nR23(Rn)Rn45(nR)nR."), "root-none select test\n");
	}

	static void DoTest2(void)
	{
	BOOL r;
	r = TreeView_SelectItem(hTree, NULL);
	Clear();
	AddItem('1');
	r = TreeView_SelectItem(hTree, hRoot);
	AddItem('2');
	r = TreeView_SelectItem(hTree, hRoot);
	AddItem('3');
	r = TreeView_SelectItem(hTree, hChild);
	AddItem('4');
	r = TreeView_SelectItem(hTree, hChild);
	AddItem('5');
	r = TreeView_SelectItem(hTree, hRoot);
	AddItem('.');
	ok(!strcmp(sequence, "1(nR)nR23(RC)RC45(CR)CR."), "root-child select test\n");
	}

	static void DoFocusTest(void)
	{
	TVINSERTSTRUCTA ins;
	static CHAR child1[] = "Edit",
	child2[] = "A really long string";
	HTREEITEM hChild1, hChild2;

	/* This test verifies that when a label is being edited, scrolling
	* the treeview does not cause the label to lose focus. To test
	* this, first some additional entries are added to generate
	* scrollbars.
	*/
	ins.hParent = hRoot;
	ins.hInsertAfter = hChild;
	U(ins).item.mask = TVIF_TEXT;
	U(ins).item.pszText = child1;
	hChild1 = TreeView_InsertItem(hTree, &ins);
	assert(hChild1);
	ins.hInsertAfter = hChild1;
	U(ins).item.mask = TVIF_TEXT;
	U(ins).item.pszText = child2;
	hChild2 = TreeView_InsertItem(hTree, &ins);
	assert(hChild2);

	ShowWindow(hMainWnd,SW_SHOW);
	SendMessageW(hTree, TVM_SELECTITEM, TVGN_CARET, (LPARAM)hChild);
	hEdit = TreeView_EditLabel(hTree, hChild);
	ScrollWindowEx(hTree, -10, 0, NULL, NULL, NULL, NULL, SW_SCROLLCHILDREN);
	ok(GetFocus() == hEdit, "Edit control should have focus\n");
	}

	static void TestGetSetBkColor(void)
	{
	COLORREF crColor = RGB(0,0,0);

	todo_wine{
	/* If the value is -1, the control is using the system color for the background color. */
	crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
	ok(crColor == -1, "Default background color reported as 0x%.8x\n", crColor);
	}

	/* Test for black background */
	SendMessage( hTree, TVM_SETBKCOLOR, 0, (LPARAM)RGB(0,0,0) );
	crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
	ok(crColor == RGB(0,0,0), "Black background color reported as 0x%.8x\n", crColor);

	/* Test for white background */
	SendMessage( hTree, TVM_SETBKCOLOR, 0, (LPARAM)RGB(255,255,255) );
	crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
	ok(crColor == RGB(255,255,255), "White background color reported as 0x%.8x\n", crColor);

	/* Reset the default background */
	SendMessage( hTree, TVM_SETBKCOLOR, 0, -1 );
	}

	static void TestGetSetImageList(void)
	{
	HIMAGELIST hImageList = NULL;

	/* Test a NULL HIMAGELIST */
	SendMessage( hTree, TVM_SETIMAGELIST, TVSIL_NORMAL, (LPARAM)hImageList );
	hImageList = (HIMAGELIST)SendMessage( hTree, TVM_GETIMAGELIST, TVSIL_NORMAL, 0 );
	ok(hImageList == NULL, "NULL image list, reported as 0x%p, expected 0.\n", hImageList);

	/* TODO: Test an actual image list */
	}

	static void TestGetSetIndent(void)
	{
	int ulIndent = -1;
	int ulMinIndent = -1;
	int ulMoreThanTwiceMin = -1;

	/* Finding the minimum indent */
	SendMessage( hTree, TVM_SETINDENT, 0, 0 );
	ulMinIndent = (int)SendMessage( hTree, TVM_GETINDENT, 0, 0 );

	/* Checking an indent that is more than twice the default indent */
	ulMoreThanTwiceMin = 2*ulMinIndent+1;
	SendMessage( hTree, TVM_SETINDENT, ulMoreThanTwiceMin, 0 );
	ulIndent = (DWORD)SendMessage( hTree, TVM_GETINDENT, 0, 0 );
	ok(ulIndent == ulMoreThanTwiceMin, "Indent reported as %d, expected %d\n", ulIndent, ulMoreThanTwiceMin);
	}

	static void TestGetSetInsertMarkColor(void)
	{
	COLORREF crColor = RGB(0,0,0);
	SendMessage( hTree, TVM_SETBKCOLOR, 0, crColor );
	crColor = (COLORREF)SendMessage( hTree, TVM_GETBKCOLOR, 0, 0 );
	ok(crColor == RGB(0,0,0), "Insert mark color reported as 0x%.8x, expected 0x00000000\n", crColor);
	}

	static void TestGetSetItem(void)
	{
	TVITEM tviRoot = {0};
	int nBufferSize = 80;
	char szBuffer[80] = {0};

	/* Test the root item */
	tviRoot.hItem = hRoot;
	tviRoot.mask = TVIF_TEXT;
	tviRoot.cchTextMax = nBufferSize;
	tviRoot.pszText = szBuffer;
	SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tviRoot );
	ok(!strcmp("Root", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Root\"\n", szBuffer);

	/* Change the root text */
	strncpy(szBuffer, "Testing123", nBufferSize);
	SendMessage( hTree, TVM_SETITEM, 0, (LPARAM)&tviRoot );
	memset(szBuffer, 0, nBufferSize);
	SendMessage( hTree, TVM_GETITEM, 0, (LPARAM)&tviRoot );
	ok(!strcmp("Testing123", szBuffer), "GetItem: szBuffer=\"%s\", expected \"Testing123\"\n", szBuffer);

	/* Reset the root text */
	memset(szBuffer, 0, nBufferSize);
	strncpy(szBuffer, "Root", nBufferSize);
	SendMessage( hTree, TVM_SETITEM, 0, (LPARAM)&tviRoot );
	}

	static void TestGetSetItemHeight(void)
	{
	int ulOldHeight = 0;
	int ulNewHeight = 0;

	/* Assuming default height to begin with */
	ulOldHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );

	/* Explicitly setting and getting the default height */
	SendMessage( hTree, TVM_SETITEMHEIGHT, -1, 0 );
	ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
	ok(ulNewHeight == ulOldHeight, "Default height not set properly, reported %d, expected %d\n", ulNewHeight, ulOldHeight);

	/* Explicitly setting and getting the height of twice the normal */
	SendMessage( hTree, TVM_SETITEMHEIGHT, 2*ulOldHeight, 0 );
	ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
	ok(ulNewHeight == 2ulOldHeight, "New height not set properly, reported %d, expected %d\n", ulNewHeight, 2ulOldHeight);

	todo_wine {
	/* Assuming tree doesn't have TVS_NONEVENHEIGHT set, so a set of 9 will round down to 8 */
	SendMessage( hTree, TVM_SETITEMHEIGHT, 9, 0 );
	ulNewHeight = (int) SendMessage( hTree, TVM_GETITEMHEIGHT, 0, 0 );
	ok(ulNewHeight == 8, "Uneven height not set properly, reported %d, expected %d\n", ulNewHeight, 8);
	}
	}

	static void TestGetSetScrollTime(void)
	{
	int ulExpectedTime = 20;
	int ulTime = 0;
	SendMessage( hTree, TVM_SETSCROLLTIME, ulExpectedTime, 0 );
	ulTime = (int)SendMessage( hTree, TVM_GETSCROLLTIME, 0, 0 );
	ok(ulTime == ulExpectedTime, "Scroll time reported as %d, expected %d\n", ulTime, ulExpectedTime);
	}

	static void TestGetSetTextColor(void)
	{
	/* If the value is -1, the control is using the system color for the text color. */
	COLORREF crColor = RGB(0,0,0);
	crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
	ok(crColor == -1, "Default text color reported as 0x%.8x\n", crColor);

	/* Test for black text */
	SendMessage( hTree, TVM_SETTEXTCOLOR, 0, (LPARAM)RGB(0,0,0) );
	crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
	ok(crColor == RGB(0,0,0), "Black text color reported as 0x%.8x\n", crColor);

	/* Test for white text */
	SendMessage( hTree, TVM_SETTEXTCOLOR, 0, (LPARAM)RGB(255,255,255) );
	crColor = (COLORREF)SendMessage( hTree, TVM_GETTEXTCOLOR, 0, 0 );
	ok(crColor == RGB(255,255,255), "White text color reported as 0x%.8x\n", crColor);

	/* Reset the default text color */
	SendMessage( hTree, TVM_SETTEXTCOLOR, 0, -1 );
	}

	static void TestGetSetToolTips(void)
	{
	HWND hwndLastToolTip = NULL;

	/* Testing setting a NULL ToolTip */
	SendMessage( hTree, TVM_SETTOOLTIPS, 0, 0 );
	hwndLastToolTip = (HWND)SendMessage( hTree, TVM_GETTOOLTIPS, 0, 0 );
	ok(hwndLastToolTip == NULL, "NULL tool tip, reported as 0x%p, expected 0.\n", hwndLastToolTip);

	/* TODO: Add a test of an actual tooltip */
	}

	static void TestGetSetUnicodeFormat(void)
	{
	BOOL bPreviousSetting = 0;
	BOOL bNewSetting = 0;

	/* Set to Unicode */
	bPreviousSetting = (BOOL)SendMessage( hTree, TVM_SETUNICODEFORMAT, 1, 0 );
	bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
	ok(bNewSetting == 1, "Unicode setting did not work.\n");

	/* Set to ANSI */
	SendMessage( hTree, TVM_SETUNICODEFORMAT, 0, 0 );
	bNewSetting = (BOOL)SendMessage( hTree, TVM_GETUNICODEFORMAT, 0, 0 );
	ok(bNewSetting == 0, "ANSI setting did not work.\n");

	/* Revert to original setting */
	SendMessage( hTree, TVM_SETUNICODEFORMAT, (LPARAM)bPreviousSetting, 0 );
	}

	static void TestGetSet(void)
	{
	TestGetSetBkColor(); /* TVM_GETBKCOLOR and TVM_SETBKCOLOR */
	TestGetSetImageList(); /* TVM_GETIMAGELIST and TVM_SETIMAGELIST */
	TestGetSetIndent(); /* TVM_SETINDENT and TVM_GETINDENT */
	TestGetSetInsertMarkColor(); /* TVM_GETINSERTMARKCOLOR and TVM_GETINSERTMARKCOLOR */
	TestGetSetItem(); /* TVM_GETITEM and TVM_SETITEM */
	TestGetSetItemHeight(); /* TVM_GETITEMHEIGHT and TVM_SETITEMHEIGHT*/
	TestGetSetScrollTime(); /* TVM_GETSCROLLTIME and TVM_SETSCROLLTIME */
	TestGetSetTextColor(); /* TVM_GETTEXTCOLOR and TVM_SETTEXTCOLOR */
	TestGetSetToolTips(); /* TVM_GETTOOLTIPS and TVM_SETTOOLTIPS */
	TestGetSetUnicodeFormat(); /* TVM_GETUNICODEFORMAT and TVM_SETUNICODEFORMAT */
	}

	static LRESULT CALLBACK MyWndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
	{
	switch(msg) {

	case WM_CREATE:
	{
	hTree = CreateWindowExA(WS_EX_CLIENTEDGE, WC_TREEVIEWA, NULL, WS_CHILD\|WS_VISIBLE\|
	TVS_LINESATROOT\|TVS_HASLINES\|TVS_HASBUTTONS\|TVS_EDITLABELS,
	0, 0, 120, 100, hWnd, (HMENU)100, GetModuleHandleA(0), 0);

	SetFocus(hTree);
	return 0;
	}
	case WM_NOTIFY:
	{
	NMHDR pHdr = (NMHDR )lParam;

	if (pHdr->idFrom == 100) {
	NMTREEVIEWA *pTreeView = (LPNMTREEVIEWA) lParam;
	switch(pHdr->code) {
	case TVN_SELCHANGINGA:
	AddItem('(');
	IdentifyItem(pTreeView->itemOld.hItem);
	IdentifyItem(pTreeView->itemNew.hItem);
	return 0;
	case TVN_SELCHANGEDA:
	AddItem(')');
	IdentifyItem(pTreeView->itemOld.hItem);
	IdentifyItem(pTreeView->itemNew.hItem);
	return 0;
	}
	}
	return 0;
	}

	case WM_SIZE:
	MoveWindow(hTree, 0, 0, LOWORD(lParam), HIWORD(lParam), TRUE);
	break;

	case WM_DESTROY:
	PostQuitMessage(0);
	break;

	default:
	return DefWindowProcA(hWnd, msg, wParam, lParam);
	}
	return 0L;
	}

	START_TEST(treeview)
	{
	WNDCLASSA wc;
	MSG msg;
	INITCOMMONCONTROLSEX icex;
	RECT rc;

	icex.dwSize = sizeof(INITCOMMONCONTROLSEX);
	icex.dwICC = ICC_TREEVIEW_CLASSES;
	InitCommonControlsEx(&icex);

	wc.style = CS_HREDRAW \| CS_VREDRAW;
	wc.cbClsExtra = 0;
	wc.cbWndExtra = 0;
	wc.hInstance = GetModuleHandleA(NULL);
	wc.hIcon = NULL;
	wc.hCursor = LoadCursorA(NULL, MAKEINTRESOURCEA(IDC_IBEAM));
	wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
	wc.lpszMenuName = NULL;
	wc.lpszClassName = "MyTestWnd";
	wc.lpfnWndProc = MyWndProc;
	RegisterClassA(&wc);


	hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW,
	CW_USEDEFAULT, CW_USEDEFAULT, 130, 105, NULL, NULL, GetModuleHandleA(NULL), 0);
	GetClientRect(hMainWnd, &rc);

	FillRoot();
	DoTest1();
	DoTest2();
	DoFocusTest();
	TestGetSet();

	PostMessageA(hMainWnd, WM_CLOSE, 0, 0);
	while(GetMessageA(&msg,0,0,0)) {
	TranslateMessage(&msg);
	DispatchMessageA(&msg);
	}
	}