| /* 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); |
| } |
| } |