comctl32/listview: Free ID array when removing all items.
[wine] / dlls / comctl32 / tests / toolbar.c
1 /* Unit tests for toolbar.
2  *
3  * Copyright 2005 Krzysztof Foltman
4  * Copyright 2007 Mikolaj Zalewski
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 "resources.h"
33
34 #include "wine/test.h"
35
36 static HWND hMainWnd;
37 static BOOL g_fBlockHotItemChange;
38 static BOOL g_fReceivedHotItemChange;
39 static BOOL g_fExpectedHotItemOld;
40 static BOOL g_fExpectedHotItemNew;
41 static DWORD g_dwExpectedDispInfoMask;
42
43 #define expect(EXPECTED,GOT) ok((GOT)==(EXPECTED), "Expected %d, got %d\n", (EXPECTED), (GOT))
44
45 #define check_rect(name, val, exp) ok(val.top == exp.top && val.bottom == exp.bottom && \
46     val.left == exp.left && val.right == exp.right, "invalid rect (" name ") (%d,%d) (%d,%d) - expected (%d,%d) (%d,%d)\n", \
47     val.left, val.top, val.right, val.bottom, exp.left, exp.top, exp.right, exp.bottom);
48  
49 #define compare(val, exp, format) ok((val) == (exp), #val " value " format " expected " format "\n", (val), (exp));
50
51 static void MakeButton(TBBUTTON *p, int idCommand, int fsStyle, int nString) {
52   p->iBitmap = -2;
53   p->idCommand = idCommand;
54   p->fsState = TBSTATE_ENABLED;
55   p->fsStyle = fsStyle;
56   p->iString = nString;
57 }
58
59 static LRESULT MyWnd_Notify(LPARAM lParam)
60 {
61     NMHDR *hdr = (NMHDR *)lParam;
62     NMTBHOTITEM *nmhi;
63     NMTBDISPINFO *nmdisp;
64     switch (hdr->code)
65     {
66         case TBN_HOTITEMCHANGE:
67             nmhi = (NMTBHOTITEM *)lParam;
68             g_fReceivedHotItemChange = TRUE;
69             if (g_fExpectedHotItemOld != g_fExpectedHotItemNew)
70             {
71                 compare(nmhi->idOld, g_fExpectedHotItemOld, "%d");
72                 compare(nmhi->idNew, g_fExpectedHotItemNew, "%d");
73             }
74             if (g_fBlockHotItemChange)
75                 return 1;
76             break;
77
78         case TBN_GETDISPINFOA:
79             ok(FALSE, "TBN_GETDISPINFOA received\n");
80             break;
81
82         case TBN_GETDISPINFOW:
83             nmdisp = (NMTBDISPINFOA *)lParam;
84
85             compare(nmdisp->dwMask, g_dwExpectedDispInfoMask, "%x");
86             ok(nmdisp->pszText == NULL, "pszText is not NULL\n");
87         break;
88     }
89     return 0;
90 }
91
92 static LRESULT CALLBACK MyWndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
93 {
94     switch (msg)
95     {
96         case WM_NOTIFY:
97             return MyWnd_Notify(lParam);
98     }
99     return DefWindowProcA(hWnd, msg, wParam, lParam);
100 }
101
102 static void basic_test(void)
103 {
104     TBBUTTON buttons[9];
105     HWND hToolbar;
106     int i;
107     for (i=0; i<9; i++)
108         MakeButton(buttons+i, 1000+i, TBSTYLE_CHECKGROUP, 0);
109     MakeButton(buttons+3, 1003, TBSTYLE_SEP|TBSTYLE_GROUP, 0);
110     MakeButton(buttons+6, 1006, TBSTYLE_SEP, 0);
111
112     hToolbar = CreateToolbarEx(hMainWnd,
113         WS_VISIBLE | WS_CLIPCHILDREN | CCS_TOP |
114         WS_CHILD | TBSTYLE_LIST,
115         100,
116         0, NULL, 0,
117         buttons, sizeof(buttons)/sizeof(buttons[0]),
118         0, 0, 20, 16, sizeof(TBBUTTON));
119     ok(hToolbar != NULL, "Toolbar creation\n");
120     SendMessage(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)"test\000");
121
122     /* test for exclusion working inside a separator-separated :-) group */
123     SendMessage(hToolbar, TB_CHECKBUTTON, 1000, 1); /* press A1 */
124     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 pressed\n");
125     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1001, 0), "A2 not pressed\n");
126
127     SendMessage(hToolbar, TB_CHECKBUTTON, 1004, 1); /* press A5, release A1 */
128     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1004, 0), "A5 pressed\n");
129     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 not pressed anymore\n");
130
131     SendMessage(hToolbar, TB_CHECKBUTTON, 1005, 1); /* press A6, release A5 */
132     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 pressed\n");
133     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1004, 0), "A5 not pressed anymore\n");
134
135     /* test for inter-group crosstalk, ie. two radio groups interfering with each other */
136     SendMessage(hToolbar, TB_CHECKBUTTON, 1007, 1); /* press B2 */
137     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 still pressed, no inter-group crosstalk\n");
138     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 still not pressed\n");
139     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1007, 0), "B2 pressed\n");
140
141     SendMessage(hToolbar, TB_CHECKBUTTON, 1000, 1); /* press A1 and ensure B group didn't suffer */
142     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 not pressed anymore\n");
143     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 pressed\n");
144     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1007, 0), "B2 still pressed\n");
145
146     SendMessage(hToolbar, TB_CHECKBUTTON, 1008, 1); /* press B3, and ensure A group didn't suffer */
147     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 pressed\n");
148     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 pressed\n");
149     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1007, 0), "B2 not pressed\n");
150     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1008, 0), "B3 pressed\n");
151
152     /* tests with invalid index */
153     compare(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 0xdeadbeef, 0), -1L, "%ld");
154     compare(SendMessage(hToolbar, TB_ISBUTTONPRESSED, 0xdeadbeef, 0), -1L, "%ld");
155     compare(SendMessage(hToolbar, TB_ISBUTTONENABLED, 0xdeadbeef, 0), -1L, "%ld");
156     compare(SendMessage(hToolbar, TB_ISBUTTONINDETERMINATE, 0xdeadbeef, 0), -1L, "%ld");
157     compare(SendMessage(hToolbar, TB_ISBUTTONHIGHLIGHTED, 0xdeadbeef, 0), -1L, "%ld");
158     compare(SendMessage(hToolbar, TB_ISBUTTONHIDDEN, 0xdeadbeef, 0), -1L, "%ld");
159
160     DestroyWindow(hToolbar);
161 }
162
163 static void rebuild_toolbar(HWND *hToolbar)
164 {
165     if (*hToolbar != NULL)
166         DestroyWindow(*hToolbar);
167     *hToolbar = CreateWindowEx(0, TOOLBARCLASSNAME, NULL, WS_CHILD | WS_VISIBLE, 0, 0, 0, 0,
168         hMainWnd, (HMENU)5, GetModuleHandle(NULL), NULL);
169     ok(*hToolbar != NULL, "Toolbar creation problem\n");
170     ok(SendMessage(*hToolbar, TB_BUTTONSTRUCTSIZE, sizeof(TBBUTTON), 0) == 0, "TB_BUTTONSTRUCTSIZE failed\n");
171     ok(SendMessage(*hToolbar, TB_AUTOSIZE, 0, 0) == 0, "TB_AUTOSIZE failed\n");
172     ok(SendMessage(*hToolbar, WM_SETFONT, (WPARAM)GetStockObject(SYSTEM_FONT), 0)==1, "WM_SETFONT\n");
173 }
174
175 static void rebuild_toolbar_with_buttons(HWND *hToolbar)
176 {
177     TBBUTTON buttons[5];
178     rebuild_toolbar(hToolbar);
179     
180     ZeroMemory(&buttons, sizeof(buttons));
181     buttons[0].idCommand = 1;
182     buttons[0].fsStyle = BTNS_BUTTON;
183     buttons[0].fsState = TBSTATE_ENABLED;
184     buttons[0].iString = -1;
185     buttons[1].idCommand = 3;
186     buttons[1].fsStyle = BTNS_BUTTON;
187     buttons[1].fsState = TBSTATE_ENABLED;
188     buttons[1].iString = -1;
189     buttons[2].idCommand = 5;
190     buttons[2].fsStyle = BTNS_SEP;
191     buttons[2].fsState = TBSTATE_ENABLED;
192     buttons[2].iString = -1;
193     buttons[3].idCommand = 7;
194     buttons[3].fsStyle = BTNS_BUTTON;
195     buttons[3].fsState = TBSTATE_ENABLED;
196     buttons[3].iString = -1;
197     buttons[4].idCommand = 9;
198     buttons[4].fsStyle = BTNS_BUTTON;
199     buttons[4].fsState = 0;  /* disabled */
200     buttons[4].iString = -1;
201     ok(SendMessage(*hToolbar, TB_ADDBUTTONS, 5, (LPARAM)buttons) == 1, "TB_ADDBUTTONS failed\n");
202     ok(SendMessage(*hToolbar, TB_AUTOSIZE, 0, 0) == 0, "TB_AUTOSIZE failed\n");
203 }
204
205 static void add_128x15_bitmap(HWND hToolbar, int nCmds)
206 {
207     TBADDBITMAP bmp128;
208     bmp128.hInst = GetModuleHandle(NULL);
209     bmp128.nID = IDB_BITMAP_128x15;
210     ok(SendMessageA(hToolbar, TB_ADDBITMAP, nCmds, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
211 }
212
213 #define CHECK_IMAGELIST(count, dx, dy) { \
214     int cx, cy; \
215     HIMAGELIST himl = (HIMAGELIST)SendMessageA(hToolbar, TB_GETIMAGELIST, 0, 0); \
216     ok(himl != NULL, "No image list\n"); \
217     if (himl != NULL) {\
218         ok(ImageList_GetImageCount(himl) == count, "Images count mismatch - %d vs %d\n", count, ImageList_GetImageCount(himl)); \
219         ImageList_GetIconSize(himl, &cx, &cy); \
220         ok(cx == dx && cy == dy, "Icon size mismatch - %dx%d vs %dx%d\n", dx, dy, cx, cy); \
221     } \
222 }
223
224 static void test_add_bitmap(void)
225 {
226     HWND hToolbar = NULL;
227     TBADDBITMAP bmp128;
228     TBADDBITMAP bmp80;
229     TBADDBITMAP stdsmall;
230     TBADDBITMAP addbmp;
231     HIMAGELIST himl;
232     INT ret;
233
234     /* empty 128x15 bitmap */
235     bmp128.hInst = GetModuleHandle(NULL);
236     bmp128.nID = IDB_BITMAP_128x15;
237
238     /* empty 80x15 bitmap */
239     bmp80.hInst = GetModuleHandle(NULL);
240     bmp80.nID = IDB_BITMAP_80x15;
241
242     /* standard bitmap - 240x15 pixels */
243     stdsmall.hInst = HINST_COMMCTRL;
244     stdsmall.nID = IDB_STD_SMALL_COLOR;
245
246     rebuild_toolbar(&hToolbar);
247     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 8, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
248     CHECK_IMAGELIST(8, 16, 16);
249     
250     /* adding more bitmaps */
251     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80) == 8, "TB_ADDBITMAP - unexpected return\n");
252     CHECK_IMAGELIST(13, 16, 16);
253     /* adding the same bitmap will simply return the index of the already loaded block */
254     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 8, (LPARAM)&bmp128);
255     ok(ret == 0, "TB_ADDBITMAP - unexpected return %d\n", ret);
256     CHECK_IMAGELIST(13, 16, 16);
257     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80);
258     ok(ret == 8, "TB_ADDBITMAP - unexpected return %d\n", ret);
259     CHECK_IMAGELIST(13, 16, 16);
260     /* even if we increase the wParam */
261     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 55, (LPARAM)&bmp80);
262     ok(ret == 8, "TB_ADDBITMAP - unexpected return %d\n", ret);
263     CHECK_IMAGELIST(13, 16, 16);
264
265     /* when the wParam is smaller than the bitmaps count but non-zero, all the bitmaps will be added*/
266     rebuild_toolbar(&hToolbar);
267     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 3, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
268     CHECK_IMAGELIST(8, 16, 16);
269     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80);
270     ok(ret == 3, "TB_ADDBITMAP - unexpected return %d\n", ret);
271     /* the returned value is misleading - id 8 is the id of the first icon from bmp80 */
272     CHECK_IMAGELIST(13, 16, 16);
273
274     /* the same for negative wParam */
275     rebuild_toolbar(&hToolbar);
276     ret = SendMessageA(hToolbar, TB_ADDBITMAP, -143, (LPARAM)&bmp128);
277     ok(ret == 0, "TB_ADDBITMAP - unexpected return %d\n", ret);
278     CHECK_IMAGELIST(8, 16, 16);
279     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp80);
280     ok(ret == -143, "TB_ADDBITMAP - unexpected return %d\n", ret);
281     CHECK_IMAGELIST(13, 16, 16);
282
283     /* for zero only one bitmap will be added */
284     rebuild_toolbar(&hToolbar);
285     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 0, (LPARAM)&bmp80);
286     ok(ret == 0, "TB_ADDBITMAP - unexpected return %d\n", ret);
287     CHECK_IMAGELIST(1, 16, 16);
288
289     /* if wParam is larger than the amount of icons, the list is grown */
290     rebuild_toolbar(&hToolbar);
291     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp80) == 0, "TB_ADDBITMAP - unexpected return\n");
292     CHECK_IMAGELIST(100, 16, 16);
293     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp128);
294     ok(ret == 100, "TB_ADDBITMAP - unexpected return %d\n", ret);
295     CHECK_IMAGELIST(200, 16, 16);
296
297     /* adding built-in items - the wParam is ignored */
298     rebuild_toolbar(&hToolbar);
299     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80) == 0, "TB_ADDBITMAP - unexpected return\n");
300     CHECK_IMAGELIST(5, 16, 16);
301     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 0, (LPARAM)&stdsmall) == 5, "TB_ADDBITMAP - unexpected return\n");
302     CHECK_IMAGELIST(20, 16, 16);
303     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp128) == 20, "TB_ADDBITMAP - unexpected return\n");
304     CHECK_IMAGELIST(28, 16, 16);
305
306     /* when we increase the bitmap size, less icons will be created */
307     rebuild_toolbar(&hToolbar);
308     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(20, 20)) == TRUE, "TB_SETBITMAPSIZE failed\n");
309     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
310     CHECK_IMAGELIST(6, 20, 20);
311     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp80);
312     ok(ret == 1, "TB_ADDBITMAP - unexpected return %d\n", ret);
313     CHECK_IMAGELIST(10, 20, 20);
314     /* the icons can be resized - an UpdateWindow is needed as this probably happens during WM_PAINT */
315     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(8, 8)) == TRUE, "TB_SETBITMAPSIZE failed\n");
316     UpdateWindow(hToolbar);
317     CHECK_IMAGELIST(26, 8, 8);
318     /* loading a standard bitmaps automatically resizes the icons */
319     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&stdsmall) == 2, "TB_ADDBITMAP - unexpected return\n");
320     UpdateWindow(hToolbar);
321     CHECK_IMAGELIST(28, 16, 16);
322
323     /* two more SETBITMAPSIZE tests */
324     rebuild_toolbar(&hToolbar);
325     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
326     CHECK_IMAGELIST(100, 16, 16);
327     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp80) == 100, "TB_ADDBITMAP - unexpected return\n");
328     CHECK_IMAGELIST(200, 16, 16);
329     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(8, 8)) == TRUE, "TB_SETBITMAPSIZE failed\n");
330     UpdateWindow(hToolbar);
331     CHECK_IMAGELIST(200, 8, 8);
332     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(30, 30)) == TRUE, "TB_SETBITMAPSIZE failed\n");
333     UpdateWindow(hToolbar);
334     CHECK_IMAGELIST(200, 30, 30);
335     rebuild_toolbar(&hToolbar);
336     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
337     CHECK_IMAGELIST(8, 16, 16);
338     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 3, (LPARAM)&bmp80) == 5, "TB_ADDBITMAP - unexpected return\n");
339     CHECK_IMAGELIST(13, 16, 16);
340     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(30, 30)) == TRUE, "TB_SETBITMAPSIZE failed\n");
341     UpdateWindow(hToolbar);
342     CHECK_IMAGELIST(8, 30, 30);
343     /* when the width or height is zero, set it to 1 */
344     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(0, 0)) == TRUE, "TB_SETBITMAPSIZE failed\n");
345     UpdateWindow(hToolbar);
346     CHECK_IMAGELIST(208, 1, 1);
347     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(0, 5)) == TRUE, "TB_SETBITMAPSIZE failed\n");
348     UpdateWindow(hToolbar);
349     CHECK_IMAGELIST(208, 1, 5);
350     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(5, 0)) == TRUE, "TB_SETBITMAPSIZE failed\n");
351     UpdateWindow(hToolbar);
352     CHECK_IMAGELIST(41, 5, 1);
353
354     /* the control can add bitmaps to an existing image list */
355     rebuild_toolbar(&hToolbar);
356     himl = ImageList_LoadImage(GetModuleHandle(NULL), MAKEINTRESOURCE(IDB_BITMAP_80x15), 20, 2, CLR_NONE, IMAGE_BITMAP, LR_DEFAULTCOLOR);
357     ok(himl != NULL, "failed to create imagelist\n");
358     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, (LPARAM)himl) == 0, "TB_SETIMAGELIST failed\n");
359     CHECK_IMAGELIST(4, 20, 15);
360     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
361     CHECK_IMAGELIST(10, 20, 15);
362     /* however TB_SETBITMAPSIZE/add std bitmap won't change the image size (the button size does change) */
363     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(8, 8)) == TRUE, "TB_SETBITMAPSIZE failed\n");
364     UpdateWindow(hToolbar);
365     compare((int)SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0), MAKELONG(15, 14), "%x");
366     CHECK_IMAGELIST(10, 20, 15);
367     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 0, (LPARAM)&stdsmall) == 1, "TB_SETBITMAPSIZE failed\n");
368     UpdateWindow(hToolbar);
369     compare((int)SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0), MAKELONG(23, 22), "%x");
370     CHECK_IMAGELIST(22, 20, 15);
371
372     /* check standard bitmaps */
373     addbmp.hInst = HINST_COMMCTRL;
374     addbmp.nID = IDB_STD_SMALL_COLOR;
375     rebuild_toolbar(&hToolbar);
376     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
377     CHECK_IMAGELIST(15, 16, 16);
378     compare((int)SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0), MAKELONG(23, 22), "%x");
379     addbmp.nID = IDB_STD_LARGE_COLOR;
380     rebuild_toolbar(&hToolbar);
381     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
382     CHECK_IMAGELIST(15, 24, 24);
383     compare((int)SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0), MAKELONG(31, 30), "%x");
384
385     addbmp.nID = IDB_VIEW_SMALL_COLOR;
386     rebuild_toolbar(&hToolbar);
387     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
388     CHECK_IMAGELIST(12, 16, 16);
389     addbmp.nID = IDB_VIEW_LARGE_COLOR;
390     rebuild_toolbar(&hToolbar);
391     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
392     CHECK_IMAGELIST(12, 24, 24);
393
394     addbmp.nID = IDB_HIST_SMALL_COLOR;
395     rebuild_toolbar(&hToolbar);
396     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
397     CHECK_IMAGELIST(5, 16, 16);
398     addbmp.nID = IDB_HIST_LARGE_COLOR;
399     rebuild_toolbar(&hToolbar);
400     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
401     CHECK_IMAGELIST(5, 24, 24);
402
403
404     DestroyWindow(hToolbar);
405 }
406
407 #define CHECK_STRING_TABLE(count, tab) { \
408         INT _i; \
409         CHAR _buf[260]; \
410         for (_i = 0; _i < (count); _i++) {\
411             ret = SendMessageA(hToolbar, TB_GETSTRING, MAKEWPARAM(260, _i), (LPARAM)_buf); \
412             ok(ret >= 0, "TB_GETSTRING - unexpected return %d while checking string %d\n", ret, _i); \
413             if (ret >= 0) \
414                 ok(strcmp(_buf, (tab)[_i]) == 0, "Invalid string #%d - '%s' vs '%s'\n", _i, (tab)[_i], _buf); \
415         } \
416         ok(SendMessageA(hToolbar, TB_GETSTRING, MAKEWPARAM(260, (count)), (LPARAM)_buf) == -1, \
417             "Too many strings in table\n"); \
418     }
419
420 static void test_add_string(void)
421 {
422     LPCSTR test1 = "a\0b\0";
423     LPCSTR test2 = "|a|b||\0";
424     LPCSTR ret1[] = {"a", "b"};
425     LPCSTR ret2[] = {"a", "b", "|a|b||"};
426     LPCSTR ret3[] = {"a", "b", "|a|b||", "p", "q"};
427     LPCSTR ret4[] = {"a", "b", "|a|b||", "p", "q", "p"};
428     LPCSTR ret5[] = {"a", "b", "|a|b||", "p", "q", "p", "p", "q"};
429     LPCSTR ret6[] = {"a", "b", "|a|b||", "p", "q", "p", "p", "q", "p", "", "q"};
430     LPCSTR ret7[] = {"a", "b", "|a|b||", "p", "q", "p", "p", "q", "p", "", "q", "br", "c", "d"};
431     HWND hToolbar = NULL;
432     TBBUTTON button;
433     int ret;
434     CHAR buf[260];
435
436     rebuild_toolbar(&hToolbar);
437     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)test1);
438     ok(ret == 0, "TB_ADDSTRINGA - unexpected return %d\n", ret);
439     ret = SendMessageA(hToolbar, TB_GETSTRING, MAKEWPARAM(260, 1), (LPARAM)buf);
440     if (ret == 0)
441     {
442         win_skip("TB_GETSTRING needs 5.80\n");
443         return;
444     }
445     CHECK_STRING_TABLE(2, ret1);
446     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)test2);
447     ok(ret == 2, "TB_ADDSTRINGA - unexpected return %d\n", ret);
448     CHECK_STRING_TABLE(3, ret2);
449
450     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD1);
451     ok(ret == 3, "TB_ADDSTRINGA - unexpected return %d\n", ret);
452     CHECK_STRING_TABLE(3, ret2);
453     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD2);
454     ok(ret == 3, "TB_ADDSTRINGA - unexpected return %d\n", ret);
455     CHECK_STRING_TABLE(5, ret3);
456     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD3);
457     ok(ret == 5, "TB_ADDSTRINGA - unexpected return %d\n", ret);
458     CHECK_STRING_TABLE(6, ret4);
459     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD4);
460     ok(ret == 6, "TB_ADDSTRINGA - unexpected return %d\n", ret);
461     CHECK_STRING_TABLE(8, ret5);
462     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD5);
463     ok(ret == 8, "TB_ADDSTRINGA - unexpected return %d\n", ret);
464     CHECK_STRING_TABLE(11, ret6);
465     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD7);
466     ok(ret == 11, "TB_ADDSTRINGA - unexpected return %d\n", ret);
467     CHECK_STRING_TABLE(14, ret7);
468
469     ZeroMemory(&button, sizeof(button));
470     button.iString = (UINT_PTR)"Test";
471     SendMessageA(hToolbar, TB_INSERTBUTTONA, 0, (LPARAM)&button);
472     CHECK_STRING_TABLE(14, ret7);
473     SendMessageA(hToolbar, TB_ADDBUTTONSA, 1, (LPARAM)&button);
474     CHECK_STRING_TABLE(14, ret7);
475
476     DestroyWindow(hToolbar);
477 }
478
479 static void expect_hot_notify(int idold, int idnew)
480 {
481     g_fExpectedHotItemOld = idold;
482     g_fExpectedHotItemNew = idnew;
483     g_fReceivedHotItemChange = FALSE;
484 }
485
486 #define check_hot_notify() \
487     ok(g_fReceivedHotItemChange, "TBN_HOTITEMCHANGE not received\n"); \
488     g_fExpectedHotItemOld = g_fExpectedHotItemNew = 0;
489
490 static void test_hotitem(void)
491 {
492     HWND hToolbar = NULL;
493     TBBUTTONINFO tbinfo;
494     LRESULT ret;
495
496     g_fBlockHotItemChange = FALSE;
497
498     rebuild_toolbar_with_buttons(&hToolbar);
499     /* set TBSTYLE_FLAT. comctl5 allows hot items only for such toolbars.
500      * comctl6 doesn't have this requirement even when theme == NULL */
501     SetWindowLong(hToolbar, GWL_STYLE, TBSTYLE_FLAT | GetWindowLong(hToolbar, GWL_STYLE));
502     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
503     ok(ret == -1, "Hot item: %ld, expected -1\n", ret);
504     ret = SendMessage(hToolbar, TB_SETHOTITEM, 1, 0);
505     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
506     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
507     ok(ret == 1, "Hot item: %ld, expected 1\n", ret);
508     ret = SendMessage(hToolbar, TB_SETHOTITEM, 2, 0);
509     ok(ret == 1, "TB_SETHOTITEM returned %ld, expected 1\n", ret);
510
511     ret = SendMessage(hToolbar, TB_SETHOTITEM, 0xbeef, 0);
512     ok(ret == 2, "TB_SETHOTITEM returned %ld, expected 2\n", ret);
513     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
514     ok(ret == 2, "Hot item: %lx, expected 2\n", ret);
515     ret = SendMessage(hToolbar, TB_SETHOTITEM, -0xbeef, 0);
516     ok(ret == 2, "TB_SETHOTITEM returned %ld, expected 2\n", ret);
517     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
518     ok(ret == -1, "Hot item: %lx, expected -1\n", ret);
519
520     expect_hot_notify(0, 7);
521     ret = SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
522     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
523     check_hot_notify();
524     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
525     ok(ret == 3, "Hot item: %lx, expected 3\n", ret);
526     g_fBlockHotItemChange = TRUE;
527     ret = SendMessage(hToolbar, TB_SETHOTITEM, 2, 0);
528     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 2\n", ret);
529     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
530     ok(ret == 3, "Hot item: %lx, expected 3\n", ret);
531     g_fBlockHotItemChange = FALSE;
532
533     g_fReceivedHotItemChange = FALSE;
534     ret = SendMessage(hToolbar, TB_SETHOTITEM, 0xbeaf, 0);
535     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
536     ok(g_fReceivedHotItemChange == FALSE, "TBN_HOTITEMCHANGE received for invalid parameter\n");
537
538     g_fReceivedHotItemChange = FALSE;
539     ret = SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
540     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
541     ok(g_fReceivedHotItemChange == FALSE, "TBN_HOTITEMCHANGE received after a duplication\n");
542
543     expect_hot_notify(7, 0);
544     ret = SendMessage(hToolbar, TB_SETHOTITEM, -0xbeaf, 0);
545     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
546     check_hot_notify();
547     SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
548
549     /* setting disabled buttons will generate a notify with the button id but no button will be hot */
550     expect_hot_notify(7, 9);
551     ret = SendMessage(hToolbar, TB_SETHOTITEM, 4, 0);
552     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
553     check_hot_notify();
554     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
555     ok(ret == -1, "Hot item: %lx, expected -1\n", ret);
556     /* enabling the button won't change that */
557     SendMessage(hToolbar, TB_ENABLEBUTTON, 9, TRUE);
558     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
559     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
560
561     /* disabling a hot button works */
562     ret = SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
563     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
564     g_fReceivedHotItemChange = FALSE;
565     SendMessage(hToolbar, TB_ENABLEBUTTON, 7, FALSE);
566     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
567     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
568     ok(g_fReceivedHotItemChange == FALSE, "Unexpected TBN_HOTITEMCHANGE\n");
569
570     SendMessage(hToolbar, TB_SETHOTITEM, 1, 0);
571     tbinfo.cbSize = sizeof(TBBUTTONINFO);
572     tbinfo.dwMask = TBIF_STATE;
573     tbinfo.fsState = 0;  /* disabled */
574     g_fReceivedHotItemChange = FALSE;
575     ok(SendMessage(hToolbar, TB_SETBUTTONINFO, 1, (LPARAM)&tbinfo) == TRUE, "TB_SETBUTTONINFO failed\n");
576     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
577     ok(ret == 1, "TB_SETHOTITEM returned %ld, expected 1\n", ret);
578     ok(g_fReceivedHotItemChange == FALSE, "Unexpected TBN_HOTITEMCHANGE\n");
579
580     DestroyWindow(hToolbar);
581 }
582
583 #if 0  /* use this to generate more tests*/
584
585 static void dump_sizes(HWND hToolbar)
586 {
587     SIZE sz;
588     RECT r;
589     int count = SendMessage(hToolbar, TB_BUTTONCOUNT, 0, 0);
590     int i;
591
592     GetClientRect(hToolbar, &r);
593     SendMessageA(hToolbar, TB_GETMAXSIZE, 0, &sz);
594     printf("  { {%d, %d, %d, %d}, {%d, %d}, %d, {", r.left, r.top, r.right, r.bottom,
595         sz.cx, sz.cy, count);
596     for (i=0; i<count; i++)
597     {
598         SendMessageA(hToolbar, TB_GETITEMRECT, i, &r);
599         printf("%s{%3d, %3d, %3d, %3d}, ", (i%3==0 ? "\n    " : ""), r.left, r.top, r.right, r.bottom);
600     }
601     printf("\n  }, },\n");
602 }
603
604 #define check_sizes() dump_sizes(hToolbar);
605 #define check_sizes_todo(todomask) dump_sizes(hToolbar);
606
607 #else
608
609 typedef struct
610 {
611     RECT rcClient;
612     SIZE szMin;
613     INT nButtons;
614     RECT rcButtons[100];
615 } tbsize_result_t;
616
617 static tbsize_result_t tbsize_results[] =
618 {
619   { {0, 0, 672, 26}, {100, 22}, 5, {
620     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
621     { 54,   2,  77,  24}, { 77,   2, 100,  24},
622   }, },
623   { {0, 0, 672, 26}, {146, 22}, 7, {
624     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
625     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
626     {  0,  24,  23,  46},
627   }, },
628   { {0, 0, 672, 48}, {146, 22}, 7, {
629     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
630     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
631     {  0,  24,  23,  46},
632   }, },
633   { {0, 0, 672, 26}, {146, 22}, 7, {
634     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
635     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
636     {123,   2, 146,  24},
637   }, },
638   { {0, 0, 672, 26}, {192, 22}, 9, {
639     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
640     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
641     {123,   2, 146,  24}, {146,   2, 169,  24}, {169,   2, 192,  24},
642   }, },
643   { {0, 0, 672, 92}, {882, 22}, 39, {
644     {  0,   2,  23,  24}, { 23,   2,  46,  24}, {  0,   2,   8,  29},
645     {  0,  29,  23,  51}, { 23,  29,  46,  51}, { 46,  29,  69,  51},
646     { 69,  29,  92,  51}, { 92,  29, 115,  51}, {115,  29, 138,  51},
647     {138,  29, 161,  51}, {161,  29, 184,  51}, {184,  29, 207,  51},
648     {207,  29, 230,  51}, {230,  29, 253,  51}, {253,  29, 276,  51},
649     {276,  29, 299,  51}, {299,  29, 322,  51}, {322,  29, 345,  51},
650     {345,  29, 368,  51}, {368,  29, 391,  51}, {391,  29, 414,  51},
651     {414,  29, 437,  51}, {437,  29, 460,  51}, {460,  29, 483,  51},
652     {483,  29, 506,  51}, {506,  29, 529,  51}, {529,  29, 552,  51},
653     {552,  29, 575,  51}, {575,  29, 598,  51}, {598,  29, 621,  51},
654     {621,  29, 644,  51}, {644,  29, 667,  51}, {  0,  51,  23,  73},
655     { 23,  51,  46,  73}, { 46,  51,  69,  73}, { 69,  51,  92,  73},
656     { 92,  51, 115,  73}, {115,  51, 138,  73}, {138,  51, 161,  73},
657   }, },
658   { {0, 0, 48, 226}, {23, 140}, 7, {
659     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  94,  24},
660     { 94,   2, 117,  24}, {117,   2, 140,  24}, {140,   2, 163,  24},
661     {  0,  24,  23,  46},
662   }, },
663   { {0, 0, 92, 226}, {23, 140}, 7, {
664     {  0,   2,  23,  24}, { 23,   2,  46,  24}, {  0,  24,  92,  32},
665     {  0,  32,  23,  54}, { 23,  32,  46,  54}, { 46,  32,  69,  54},
666     { 69,  32,  92,  54},
667   }, },
668   { {0, 0, 672, 26}, {194, 30}, 7, {
669     {  0,   2,  31,  32}, { 31,   2,  62,  32}, { 62,   2,  70,  32},
670     { 70,   2, 101,  32}, {101,   2, 132,  32}, {132,   2, 163,  32},
671     {  0,  32,  31,  62},
672   }, },
673   { {0, 0, 672, 64}, {194, 30}, 7, {
674     {  0,   2,  31,  32}, { 31,   2,  62,  32}, { 62,   2,  70,  32},
675     { 70,   2, 101,  32}, {101,   2, 132,  32}, {132,   2, 163,  32},
676     {  0,  32,  31,  62},
677   }, },
678   { {0, 0, 672, 64}, {194, 30}, 7, {
679     {  0,   0,  31,  30}, { 31,   0,  62,  30}, { 62,   0,  70,  30},
680     { 70,   0, 101,  30}, {101,   0, 132,  30}, {132,   0, 163,  30},
681     {  0,  30,  31,  60},
682   }, },
683   { {0, 0, 124, 226}, {31, 188}, 7, {
684     {  0,   0,  31,  30}, { 31,   0,  62,  30}, {  0,  30, 124,  38},
685     {  0,  38,  31,  68}, { 31,  38,  62,  68}, { 62,  38,  93,  68},
686     { 93,  38, 124,  68},
687   }, },
688   { {0, 0, 672, 26}, {146, 22}, 7, {
689     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
690     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
691     {123,   2, 146,  24},
692   }, },
693   { {0, 0, 672, 26}, {146, 100}, 7, {
694     {  0,   0,  23, 100}, { 23,   0,  46, 100}, { 46,   0,  54, 100},
695     { 54,   0,  77, 100}, { 77,   0, 100, 100}, {100,   0, 123, 100},
696     {123,   0, 146, 100},
697   }, },
698   { {0, 0, 672, 26}, {215, 100}, 10, {
699     {  0,   0,  23, 100}, { 23,   0,  46, 100}, { 46,   0,  54, 100},
700     { 54,   0,  77, 100}, { 77,   0, 100, 100}, {100,   0, 123, 100},
701     {123,   0, 146, 100}, {146,   0, 169, 100}, {169,   0, 192, 100},
702     {192,   0, 215, 100},
703   }, },
704   { {0, 0, 672, 26}, {238, 39}, 11, {
705     {  0,   0,  23,  39}, { 23,   0,  46,  39}, { 46,   0,  54,  39},
706     { 54,   0,  77,  39}, { 77,   0, 100,  39}, {100,   0, 123,  39},
707     {123,   0, 146,  39}, {146,   0, 169,  39}, {169,   0, 192,  39},
708     {192,   0, 215,  39}, {215,   0, 238,  39},
709   }, },
710   { {0, 0, 672, 26}, {238, 22}, 11, {
711     {  0,   0,  23,  22}, { 23,   0,  46,  22}, { 46,   0,  54,  22},
712     { 54,   0,  77,  22}, { 77,   0, 100,  22}, {100,   0, 123,  22},
713     {123,   0, 146,  22}, {146,   0, 169,  22}, {169,   0, 192,  22},
714     {192,   0, 215,  22}, {215,   0, 238,  22},
715   }, },
716   { {0, 0, 672, 26}, {489, 39}, 3, {
717     {  0,   2, 163,  41}, {163,   2, 330,  41}, {330,   2, 493,  41},
718   }, },
719   { {0, 0, 672, 104}, {978, 24}, 6, {
720     {  0,   2, 163,  26}, {163,   2, 326,  26}, {326,   2, 489,  26},
721     {489,   2, 652,  26}, {652,   2, 819,  26}, {819,   2, 850,  26},
722   }, },
723   { {0, 0, 672, 28}, {978, 38}, 6, {
724     {  0,   0, 163,  38}, {163,   0, 326,  38}, {326,   0, 489,  38},
725     {489,   0, 652,  38}, {652,   0, 819,  38}, {819,   0, 850,  38},
726   }, },
727   { {0, 0, 672, 100}, {239, 102}, 3, {
728     {  0,   2, 100,  102}, {100,   2, 139,  102}, {139, 2, 239,  102},
729   }, },
730   { {0, 0, 672, 42}, {185, 40}, 3, {
731       {  0,   2,  75,  40}, {75,   2, 118, 40}, {118, 2, 185, 40},
732   }, },
733   { {0, 0, 672, 42}, {67, 40}, 1, {
734       {  0,   2,  67,  40},
735   }, },
736   { {0, 0, 672, 42}, {67, 41}, 2, {
737       {  0,   2,  672,  41}, {  0,   41,  672,  80},
738   }, },
739 };
740
741 static int tbsize_numtests = 0;
742
743 typedef struct
744 {
745     int test_num;
746     int rect_index;
747     RECT rcButton;
748 } tbsize_alt_result_t;
749
750 static tbsize_alt_result_t tbsize_alt_results[] =
751 {
752   { 5, 2, { 0, 24, 8, 29 } },
753   { 20, 1, { 100, 2, 107, 102 } },
754   { 20, 2, { 107, 2, 207, 102 } }
755 };
756
757 static int tbsize_alt_numtests = 0;
758
759 #define check_sizes_todo(todomask) { \
760         RECT rc; \
761         int buttonCount, i, mask=(todomask); \
762         tbsize_result_t *res = &tbsize_results[tbsize_numtests]; \
763         assert(tbsize_numtests < sizeof(tbsize_results)/sizeof(tbsize_results[0])); \
764         GetClientRect(hToolbar, &rc); \
765         /*check_rect("client", rc, res->rcClient);*/ \
766         buttonCount = SendMessage(hToolbar, TB_BUTTONCOUNT, 0, 0); \
767         compare(buttonCount, res->nButtons, "%d"); \
768         for (i=0; i<min(buttonCount, res->nButtons); i++) { \
769             ok(SendMessageA(hToolbar, TB_GETITEMRECT, i, (LPARAM)&rc) == 1, "TB_GETITEMRECT\n"); \
770             if (broken(tbsize_alt_numtests < sizeof(tbsize_alt_results)/sizeof(tbsize_alt_results[0]) && \
771                        memcmp(&rc, &tbsize_alt_results[tbsize_alt_numtests].rcButton, sizeof(RECT)) == 0)) { \
772                 win_skip("Alternate rect found\n"); \
773                 tbsize_alt_numtests++; \
774             } else if (!(mask&1)) { \
775                 check_rect("button", rc, res->rcButtons[i]); \
776             } else {\
777                 todo_wine { check_rect("button", rc, res->rcButtons[i]); } \
778             } \
779             mask >>= 1; \
780         } \
781         tbsize_numtests++; \
782     }
783
784 #define check_sizes() check_sizes_todo(0)
785
786 #endif
787
788 static TBBUTTON buttons1[] = {
789     {0, 10, TBSTATE_WRAP|TBSTATE_ENABLED, 0, {0, }, 0, -1},
790     {0, 11, 0, 0, {0, }, 0, -1},
791 };
792 static TBBUTTON buttons2[] = {
793     {0, 20, TBSTATE_ENABLED, 0, {0, }, 0, -1},
794     {0, 21, TBSTATE_ENABLED, 0, {0, }, 0, -1},
795 };
796 static TBBUTTON buttons3[] = {
797     {0, 30, TBSTATE_ENABLED, 0, {0, }, 0, 0},
798     {0, 31, TBSTATE_ENABLED, 0, {0, }, 0, 1},
799     {0, 32, TBSTATE_ENABLED, BTNS_AUTOSIZE, {0, }, 0, 1},
800     {0, 33, TBSTATE_ENABLED, BTNS_AUTOSIZE, {0, }, 0, (UINT_PTR)"Tst"}
801 };
802
803 static void test_sizes(void)
804 {
805     HWND hToolbar = NULL;
806     HIMAGELIST himl, himl2;
807     TBBUTTONINFO tbinfo;
808     int style;
809     int i;
810
811     rebuild_toolbar_with_buttons(&hToolbar);
812     style = GetWindowLong(hToolbar, GWL_STYLE);
813     ok(style == (WS_CHILD|WS_VISIBLE|CCS_TOP), "Invalid style %x\n", style);
814     check_sizes();
815     /* the TBSTATE_WRAP makes a second row */
816     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
817     check_sizes();
818     SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0);
819     check_sizes();
820     /* after setting the TBSTYLE_WRAPABLE the TBSTATE_WRAP is ignored */
821     SetWindowLong(hToolbar, GWL_STYLE, style|TBSTYLE_WRAPABLE);
822     check_sizes();
823     /* adding new buttons with TBSTYLE_WRAPABLE doesn't add a new row */
824     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
825     check_sizes();
826     /* only after adding enough buttons the bar will be wrapped on a
827      * separator and then on the first button */
828     for (i=0; i<15; i++)
829         SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
830     check_sizes_todo(0x4);
831
832     rebuild_toolbar_with_buttons(&hToolbar);
833     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
834     /* setting the buttons vertical will only change the window client size */
835     SetWindowLong(hToolbar, GWL_STYLE, style | CCS_VERT);
836     SendMessage(hToolbar, TB_AUTOSIZE, 0, 0);
837     check_sizes_todo(0x3c);
838     /* with a TBSTYLE_WRAPABLE a wrapping will occur on the separator */
839     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_WRAPABLE | CCS_VERT);
840     SendMessage(hToolbar, TB_AUTOSIZE, 0, 0);
841     check_sizes_todo(0x7c);
842
843     rebuild_toolbar_with_buttons(&hToolbar);
844     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
845     /* a TB_SETBITMAPSIZE changes button sizes*/
846     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(24, 24));
847     check_sizes();
848
849     /* setting a TBSTYLE_FLAT doesn't change anything - even after a TB_AUTOSIZE */
850     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_FLAT);
851     SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0);
852     check_sizes();
853     /* but after a TB_SETBITMAPSIZE the top margins is changed */
854     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(20, 20));
855     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(24, 24));
856     check_sizes();
857     /* some vertical toolbar sizes */
858     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_FLAT | TBSTYLE_WRAPABLE | CCS_VERT);
859     check_sizes_todo(0x7c);
860
861     rebuild_toolbar_with_buttons(&hToolbar);
862     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_FLAT);
863     /* newly added buttons will be use the previous margin */
864     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons2);
865     check_sizes();
866     /* TB_SETBUTTONSIZE can't be used to reduce the size of a button below the default */
867     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
868     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(22, 21))==1, "TB_SETBUTTONSIZE\n");
869     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
870     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(5, 100))==1, "TB_SETBUTTONSIZE\n");
871     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 100), "Unexpected button size\n");
872     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3))==1, "TB_SETBUTTONSIZE\n");
873     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
874     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(5, 100))==1, "TB_SETBUTTONSIZE\n");
875     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 100), "Unexpected button size\n");
876     check_sizes();
877     /* add some buttons with non-default sizes */
878     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons2);
879     SendMessageA(hToolbar, TB_INSERTBUTTON, -1, (LPARAM)&buttons2[0]);
880     check_sizes();
881     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[0]);
882     /* TB_ADDSTRING resets the size */
883     SendMessageA(hToolbar, TB_ADDSTRING, 0, (LPARAM)"A\0MMMMMMMMMMMMM\0");
884     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 39), "Unexpected button size\n");
885     check_sizes();
886     /* TB_SETBUTTONSIZE can be used to crop the text */
887     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3));
888     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
889     check_sizes();
890     /* the default size is bitmap size + padding */
891     SendMessageA(hToolbar, TB_SETPADDING, 0, MAKELONG(1, 1));
892     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3));
893     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(17, 17), "Unexpected button size\n");
894     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(3, 3));
895     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3));
896     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(4, 4), "Unexpected button size\n");
897
898     rebuild_toolbar(&hToolbar);
899     /* sending a TB_SETBITMAPSIZE with the same sizes is enough to make the button smaller */
900     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
901     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(16, 15));
902     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 21), "Unexpected button size\n");
903     /* -1 in TB_SETBITMAPSIZE is a special code meaning that the coordinate shouldn't be changed */
904     add_128x15_bitmap(hToolbar, 16);
905     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(14, -1)), "TB_SETBITMAPSIZE failed\n");
906     compare((int)SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0), MAKELONG(21, 21), "%x");
907     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(-1, 12)), "TB_SETBITMAPSIZE failed\n");
908     compare((int)SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0), MAKELONG(21, 18), "%x");
909     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(-1, -1)), "TB_SETBITMAPSIZE failed\n");
910     compare((int)SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0), MAKELONG(21, 18), "%x");
911     /* check the imagelist */
912     InvalidateRect(hToolbar, NULL, TRUE);
913     UpdateWindow(hToolbar);
914     CHECK_IMAGELIST(16, 14, 12);
915
916     rebuild_toolbar(&hToolbar);
917     SendMessageA(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)"A\0MMMMMMMMMMMMM\0");
918     /* the height is increased after a TB_ADDSTRING */
919     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 39), "Unexpected button size\n");
920     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(100, 100));
921     /* if a string is in the pool, even adding a button without a string resets the size */
922     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons2[0]);
923     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
924     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(100, 100));
925     /* an BTNS_AUTOSIZE button is also considered when computing the new size */
926     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[2]);
927     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(163, 39), "Unexpected button size\n");
928     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[0]);
929     check_sizes();
930     /* delete button doesn't change the buttons size */
931     SendMessageA(hToolbar, TB_DELETEBUTTON, 2, 0);
932     SendMessageA(hToolbar, TB_DELETEBUTTON, 1, 0);
933     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(163, 39), "Unexpected button size\n");
934     /* TB_INSERTBUTTONS will */
935     SendMessageA(hToolbar, TB_INSERTBUTTON, 1, (LPARAM)&buttons2[0]);
936     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
937
938     /* TB_HIDEBUTTON and TB_MOVEBUTTON doesn't force a recalc */
939     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(100, 100));
940     ok(SendMessageA(hToolbar, TB_MOVEBUTTON, 0, 1), "TB_MOVEBUTTON failed\n");
941     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(100, 100), "Unexpected button size\n");
942     ok(SendMessageA(hToolbar, TB_HIDEBUTTON, 20, TRUE), "TB_HIDEBUTTON failed\n");
943     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(100, 100), "Unexpected button size\n");
944     /* however changing the hidden flag with TB_SETSTATE does */
945     ok(SendMessageA(hToolbar, TB_SETSTATE, 20, TBSTATE_ENABLED|TBSTATE_HIDDEN), "TB_SETSTATE failed\n");
946     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(100, 100), "Unexpected button size\n");
947     ok(SendMessageA(hToolbar, TB_SETSTATE, 20, TBSTATE_ENABLED), "TB_SETSTATE failed\n");
948     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
949
950     /* TB_SETIMAGELIST always changes the height but the width only if necessary */
951     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(100, 100));
952     himl = ImageList_LoadImage(GetModuleHandle(NULL), MAKEINTRESOURCE(IDB_BITMAP_80x15), 20, 2, CLR_NONE, IMAGE_BITMAP, LR_DEFAULTCOLOR);
953     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, (LPARAM)himl) == 0, "TB_SETIMAGELIST failed\n");
954     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(100, 21), "Unexpected button size\n");
955     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(100, 100));
956     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(100, 100), "Unexpected button size\n");
957     /* But there are no update when we change imagelist, and image sizes are the same */
958     himl2 = ImageList_LoadImage(GetModuleHandle(NULL), MAKEINTRESOURCE(IDB_BITMAP_128x15), 20, 2, CLR_NONE, IMAGE_BITMAP, LR_DEFAULTCOLOR);
959     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, (LRESULT)himl2) == (LRESULT)himl, "TB_SETIMAGELIST failed\n");
960     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(100, 100), "Unexpected button size\n");
961     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(1, 1));
962     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(27, 21), "Unexpected button size\n");
963     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, 0) == (LRESULT)himl2, "TB_SETIMAGELIST failed\n");
964     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(27, 7), "Unexpected button size\n");
965     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(1, 1));
966     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(8, 7), "Unexpected button size\n");
967     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, (LPARAM)himl) == 0, "TB_SETIMAGELIST failed\n");
968     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(27, 21), "Unexpected button size\n");
969     /* the text is taken into account */
970     SendMessageA(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)"A\0MMMMMMMMMMMMM\0");
971     SendMessageA(hToolbar, TB_ADDBUTTONS, 4, (LPARAM)buttons3);
972     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(163, 38), "Unexpected button size\n");
973     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, 0) == (LRESULT)himl, "TB_SETIMAGELIST failed\n");
974     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(163, 24), "Unexpected button size\n");
975     /* the style change also comes into effect */
976     check_sizes();
977     SetWindowLong(hToolbar, GWL_STYLE, GetWindowLong(hToolbar, GWL_STYLE) | TBSTYLE_FLAT);
978     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, (LPARAM)himl) == 0, "TB_SETIMAGELIST failed\n");
979     check_sizes_todo(0x30);     /* some small problems with BTNS_AUTOSIZE button sizes */
980
981     rebuild_toolbar(&hToolbar);
982     ImageList_Destroy(himl);
983
984     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[3]);
985     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(27, 39), "Unexpected button size\n");
986     SendMessageA(hToolbar, TB_DELETEBUTTON, 0, 0);
987     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(27, 39), "Unexpected button size\n");
988
989     rebuild_toolbar(&hToolbar);
990
991     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELPARAM(32, 32)) == 1, "TB_SETBITMAPSIZE failed\n");
992     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELPARAM(100, 100)) == 1, "TB_SETBUTTONSIZE failed\n");
993     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons2[0]) == 1, "TB_ADDBUTTONS failed\n");
994     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[2]) == 1, "TB_ADDBUTTONS failed\n");
995     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[0]) == 1, "TB_ADDBUTTONS failed\n");
996     SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0 );
997     check_sizes();
998
999     rebuild_toolbar(&hToolbar);
1000     SetWindowLong(hToolbar, GWL_STYLE, TBSTYLE_LIST | GetWindowLong(hToolbar, GWL_STYLE));
1001     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELPARAM(32, 32)) == 1, "TB_SETBITMAPSIZE failed\n");
1002     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELPARAM(100, 100)) == 1, "TB_SETBUTTONSIZE failed\n");
1003     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons2[0]) == 1, "TB_ADDBUTTONS failed\n");
1004     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[2]) == 1, "TB_ADDBUTTONS failed\n");
1005     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[3]) == 1, "TB_ADDBUTTONS failed\n");
1006     SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0 );
1007     check_sizes_todo(0xff);
1008
1009     rebuild_toolbar(&hToolbar);
1010     SetWindowLong(hToolbar, GWL_STYLE, TBSTYLE_LIST | GetWindowLong(hToolbar, GWL_STYLE));
1011     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELPARAM(32, 32)) == 1, "TB_SETBITMAPSIZE failed\n");
1012     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELPARAM(100, 100)) == 1, "TB_SETBUTTONSIZE failed\n");
1013     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[3]) == 1, "TB_ADDBUTTONS failed\n");
1014     SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0 );
1015     check_sizes();
1016
1017     rebuild_toolbar(&hToolbar);
1018     SetWindowLong(hToolbar, GWL_STYLE, TBSTYLE_WRAPABLE | GetWindowLong(hToolbar, GWL_STYLE));
1019     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[3]) == 1, "TB_ADDBUTTONS failed\n");
1020     ok(SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[3]) == 1, "TB_ADDBUTTONS failed\n");
1021     tbinfo.cx = 672;
1022     tbinfo.cbSize = sizeof(TBBUTTONINFO);
1023     tbinfo.dwMask = TBIF_SIZE | TBIF_BYINDEX;
1024     if (SendMessageA(hToolbar, TB_SETBUTTONINFO, 0, (LPARAM)&tbinfo))
1025     {
1026         ok(SendMessageA(hToolbar, TB_SETBUTTONINFO, 1, (LPARAM)&tbinfo) != 0, "TB_SETBUTTONINFO failed\n");
1027         SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0);
1028         check_sizes();
1029     }
1030     else  /* TBIF_BYINDEX probably not supported, confirm that this was the reason for the failure */
1031     {
1032         tbinfo.dwMask = TBIF_SIZE;
1033         ok(SendMessageA(hToolbar, TB_SETBUTTONINFO, 33, (LPARAM)&tbinfo) != 0, "TB_SETBUTTONINFO failed\n");
1034     }
1035
1036     DestroyWindow(hToolbar);
1037 }
1038
1039 /* Toolbar control has two ways of reacting to a change. We call them a
1040  * relayout and recalc. A recalc forces a recompute of values like button size
1041  * and top margin (the latter in comctl32 <v6), while a relayout uses the cached
1042  * values. This functions creates a flat toolbar with a top margin of a non-flat
1043  * toolbar. We will notice a recalc, as it will recompte the top margin and
1044  * change it to zero*/
1045 static void prepare_recalc_test(HWND *phToolbar)
1046 {
1047     RECT rect;
1048     rebuild_toolbar_with_buttons(phToolbar);
1049     SetWindowLong(*phToolbar, GWL_STYLE,
1050         GetWindowLong(*phToolbar, GWL_STYLE) | TBSTYLE_FLAT);
1051     SendMessage(*phToolbar, TB_GETITEMRECT, 1, (LPARAM)&rect);
1052     ok(rect.top == 2, "Test will make no sense because initial top is %d instead of 2\n",
1053         rect.top);
1054 }
1055
1056 static BOOL did_recalc(HWND hToolbar)
1057 {
1058     RECT rect;
1059     SendMessage(hToolbar, TB_GETITEMRECT, 1, (LPARAM)&rect);
1060     ok(rect.top == 2 || rect.top == 0, "Unexpected top margin %d in recalc test\n",
1061         rect.top);
1062     return (rect.top == 0);
1063 }
1064
1065 /* call after a recalc did happen to return to an unstable state */
1066 static void restore_recalc_state(HWND hToolbar)
1067 {
1068     RECT rect;
1069     /* return to style with a 2px top margin */
1070     SetWindowLong(hToolbar, GWL_STYLE,
1071         GetWindowLong(hToolbar, GWL_STYLE) & ~TBSTYLE_FLAT);
1072     /* recalc */
1073     SendMessage(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[3]);
1074     /* top margin will be 0px if a recalc occurs */
1075     SetWindowLong(hToolbar, GWL_STYLE,
1076         GetWindowLong(hToolbar, GWL_STYLE) | TBSTYLE_FLAT);
1077     /* safety check */
1078     SendMessage(hToolbar, TB_GETITEMRECT, 1, (LPARAM)&rect);
1079     ok(rect.top == 2, "Test will make no sense because initial top is %d instead of 2\n",
1080         rect.top);
1081 }
1082
1083 static void test_recalc(void)
1084 {
1085     HWND hToolbar;
1086     TBBUTTONINFO bi;
1087     CHAR test[] = "Test";
1088     const int EX_STYLES_COUNT = 5;
1089     int i;
1090     BOOL recalc;
1091
1092     /* Like TB_ADDBUTTONS tested in test_sized, inserting a button without text
1093      * results in a relayout, while adding one with text forces a recalc */
1094     prepare_recalc_test(&hToolbar);
1095     SendMessage(hToolbar, TB_INSERTBUTTON, 1, (LPARAM)&buttons3[0]);
1096     recalc = did_recalc(hToolbar);
1097     ok(!recalc, "Unexpected recalc - adding button without text\n");
1098
1099     prepare_recalc_test(&hToolbar);
1100     SendMessage(hToolbar, TB_INSERTBUTTON, 1, (LPARAM)&buttons3[3]);
1101     recalc = did_recalc(hToolbar);
1102     ok(recalc, "Expected a recalc - adding button with text\n");
1103
1104     /* TB_SETBUTTONINFO, even when adding a text, results only in a relayout */
1105     prepare_recalc_test(&hToolbar);
1106     bi.cbSize = sizeof(bi);
1107     bi.dwMask = TBIF_TEXT;
1108     bi.pszText = test;
1109     SendMessage(hToolbar, TB_SETBUTTONINFO, 1, (LPARAM)&bi);
1110     recalc = did_recalc(hToolbar);
1111     ok(!recalc, "Unexpected recalc - setting a button text\n");
1112
1113     /* most extended styled doesn't force a recalc (testing all the bits gives
1114      * the same results, but prints some ERRs while testing) */
1115     for (i = 0; i < EX_STYLES_COUNT; i++)
1116     {
1117         if (i == 1 || i == 3)  /* an undoc style and TBSTYLE_EX_MIXEDBUTTONS */
1118             continue;
1119         prepare_recalc_test(&hToolbar);
1120         expect(0, (int)SendMessage(hToolbar, TB_GETEXTENDEDSTYLE, 0, 0));
1121         SendMessage(hToolbar, TB_SETEXTENDEDSTYLE, 0, (1 << i));
1122         recalc = did_recalc(hToolbar);
1123         ok(!recalc, "Unexpected recalc - setting bit %d\n", i);
1124         SendMessage(hToolbar, TB_SETEXTENDEDSTYLE, 0, 0);
1125         recalc = did_recalc(hToolbar);
1126         ok(!recalc, "Unexpected recalc - clearing bit %d\n", i);
1127         expect(0, (int)SendMessage(hToolbar, TB_GETEXTENDEDSTYLE, 0, 0));
1128     }
1129
1130     /* TBSTYLE_EX_MIXEDBUTTONS does a recalc on change */
1131     prepare_recalc_test(&hToolbar);
1132     SendMessage(hToolbar, TB_SETEXTENDEDSTYLE, 0, TBSTYLE_EX_MIXEDBUTTONS);
1133     recalc = did_recalc(hToolbar);
1134     if (recalc)
1135     {
1136         ok(recalc, "Expected a recalc - setting TBSTYLE_EX_MIXEDBUTTONS\n");
1137         restore_recalc_state(hToolbar);
1138         SendMessage(hToolbar, TB_SETEXTENDEDSTYLE, 0, TBSTYLE_EX_MIXEDBUTTONS);
1139         recalc = did_recalc(hToolbar);
1140         ok(!recalc, "Unexpected recalc - setting TBSTYLE_EX_MIXEDBUTTONS again\n");
1141         restore_recalc_state(hToolbar);
1142         SendMessage(hToolbar, TB_SETEXTENDEDSTYLE, 0, 0);
1143         recalc = did_recalc(hToolbar);
1144         ok(recalc, "Expected a recalc - clearing TBSTYLE_EX_MIXEDBUTTONS\n");
1145     }
1146     else win_skip( "No recalc on TBSTYLE_EX_MIXEDBUTTONS\n" );
1147
1148     /* undocumented exstyle 0x2 seems to changes the top margin, what
1149      * interferes with these tests */
1150
1151     DestroyWindow(hToolbar);
1152 }
1153
1154 static void test_getbuttoninfo(void)
1155 {
1156     HWND hToolbar = NULL;
1157     int i;
1158
1159     rebuild_toolbar_with_buttons(&hToolbar);
1160     for (i = 0; i < 128; i++)
1161     {
1162         TBBUTTONINFO tbi;
1163         int ret;
1164
1165         tbi.cbSize = i;
1166         tbi.dwMask = TBIF_COMMAND;
1167         ret = (int)SendMessage(hToolbar, TB_GETBUTTONINFO, 1, (LPARAM)&tbi);
1168         if (i == sizeof(TBBUTTONINFO)) {
1169             compare(ret, 0, "%d");
1170         } else {
1171             compare(ret, -1, "%d");
1172         }
1173     }
1174     DestroyWindow(hToolbar);
1175 }
1176
1177 static void test_createtoolbarex(void)
1178 {
1179     HWND hToolbar;
1180     TBBUTTON btns[3];
1181     ZeroMemory(&btns, sizeof(btns));
1182
1183     hToolbar = CreateToolbarEx(hMainWnd, WS_VISIBLE, 1, 16, GetModuleHandle(NULL), IDB_BITMAP_128x15, btns,
1184         3, 20, 20, 16, 16, sizeof(TBBUTTON));
1185     CHECK_IMAGELIST(16, 20, 20);
1186     compare((int)SendMessage(hToolbar, TB_GETBUTTONSIZE, 0, 0), 0x1a001b, "%x");
1187     DestroyWindow(hToolbar);
1188
1189     hToolbar = CreateToolbarEx(hMainWnd, WS_VISIBLE, 1, 16, GetModuleHandle(NULL), IDB_BITMAP_128x15, btns,
1190         3, 4, 4, 16, 16, sizeof(TBBUTTON));
1191     CHECK_IMAGELIST(32, 4, 4);
1192     compare((int)SendMessage(hToolbar, TB_GETBUTTONSIZE, 0, 0), 0xa000b, "%x");
1193     DestroyWindow(hToolbar);
1194
1195     hToolbar = CreateToolbarEx(hMainWnd, WS_VISIBLE, 1, 16, GetModuleHandle(NULL), IDB_BITMAP_128x15, btns,
1196         3, 0, 8, 12, 12, sizeof(TBBUTTON));
1197     CHECK_IMAGELIST(16, 12, 12);
1198     compare((int)SendMessage(hToolbar, TB_GETBUTTONSIZE, 0, 0), 0x120013, "%x");
1199     DestroyWindow(hToolbar);
1200
1201     hToolbar = CreateToolbarEx(hMainWnd, WS_VISIBLE, 1, 16, GetModuleHandle(NULL), IDB_BITMAP_128x15, btns,
1202         3, -1, 8, 12, 12, sizeof(TBBUTTON));
1203     CHECK_IMAGELIST(16, 12, 8);
1204     compare((int)SendMessage(hToolbar, TB_GETBUTTONSIZE, 0, 0), 0xe0013, "%x");
1205     DestroyWindow(hToolbar);
1206
1207     hToolbar = CreateToolbarEx(hMainWnd, WS_VISIBLE, 1, 16, GetModuleHandle(NULL), IDB_BITMAP_128x15, btns,
1208         3, -1, 8, -1, 12, sizeof(TBBUTTON));
1209     CHECK_IMAGELIST(16, 16, 8);
1210     compare((int)SendMessage(hToolbar, TB_GETBUTTONSIZE, 0, 0), 0xe0017, "%x");
1211     DestroyWindow(hToolbar);
1212
1213     hToolbar = CreateToolbarEx(hMainWnd, WS_VISIBLE, 1, 16, GetModuleHandle(NULL), IDB_BITMAP_128x15, btns,
1214         3, 0, 0, 12, -1, sizeof(TBBUTTON));
1215     CHECK_IMAGELIST(16, 12, 16);
1216     compare((int)SendMessage(hToolbar, TB_GETBUTTONSIZE, 0, 0), 0x160013, "%x");
1217     DestroyWindow(hToolbar);
1218
1219     hToolbar = CreateToolbarEx(hMainWnd, WS_VISIBLE, 1, 16, GetModuleHandle(NULL), IDB_BITMAP_128x15, btns,
1220         3, 0, 0, 0, 12, sizeof(TBBUTTON));
1221     CHECK_IMAGELIST(16, 16, 16);
1222     compare((int)SendMessage(hToolbar, TB_GETBUTTONSIZE, 0, 0), 0x160017, "%x");
1223     DestroyWindow(hToolbar);
1224 }
1225
1226 static void test_dispinfo(void)
1227 {
1228     HWND hToolbar = NULL;
1229     const TBBUTTON buttons_disp[] = {
1230         {-1, 20, TBSTATE_ENABLED, 0, {0, }, 0, -1},
1231         {0,  21, TBSTATE_ENABLED, 0, {0, }, 0, -1},
1232     };
1233     BOOL ret;
1234
1235     rebuild_toolbar(&hToolbar);
1236     SendMessageA(hToolbar, TB_LOADIMAGES, IDB_HIST_SMALL_COLOR, (LPARAM)HINST_COMMCTRL);
1237     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons_disp);
1238     g_dwExpectedDispInfoMask = TBNF_IMAGE;
1239     /* Some TBN_GETDISPINFO tests will be done in MyWnd_Notify function.
1240      * We will receive TBN_GETDISPINFOW even if the control is ANSI */
1241     compare((BOOL)SendMessageA(hToolbar, CCM_GETUNICODEFORMAT, 0, 0), 0, "%d");
1242     ShowWindow(hToolbar, SW_SHOW);
1243     UpdateWindow(hToolbar);
1244
1245     ret = (BOOL)SendMessageA(hToolbar, CCM_SETUNICODEFORMAT, TRUE, 0);
1246     compare(ret, FALSE, "%d");
1247     compare(SendMessageA(hToolbar, CCM_GETUNICODEFORMAT, 0, 0), 1L, "%ld");
1248     InvalidateRect(hToolbar, NULL, FALSE);
1249     UpdateWindow(hToolbar);
1250
1251     ret = (BOOL)SendMessageA(hToolbar, CCM_SETUNICODEFORMAT, FALSE, 0);
1252     compare(ret, TRUE, "%d");
1253     compare(SendMessageA(hToolbar, CCM_GETUNICODEFORMAT, 0, 0), 0L, "%ld");
1254     InvalidateRect(hToolbar, NULL, FALSE);
1255     UpdateWindow(hToolbar);
1256
1257     DestroyWindow(hToolbar);
1258     g_dwExpectedDispInfoMask = 0;
1259 }
1260
1261 typedef struct
1262 {
1263     int  nRows;
1264     BOOL bLarger;
1265     int  expectedRows;
1266 } tbrows_result_t;
1267
1268 static tbrows_result_t tbrows_results[] =
1269 {
1270     {1, TRUE,  1}, /* 0: Simple case 9 in a row */
1271     {2, TRUE,  2}, /* 1: Another simple case 5 on one row, 4 on another*/
1272     {3, FALSE, 3}, /* 2: 3 lines - should be 3 lines of 3 buttons */
1273     {8, FALSE, 5}, /* 3: 8 lines - should be 5 lines of 2 buttons */
1274     {8, TRUE,  9}, /* 4: 8 lines but grow - should be 9 lines */
1275     {1, TRUE,  1}  /* 5: Back to simple case */
1276 };
1277
1278 static void test_setrows(void)
1279 {
1280     TBBUTTON buttons[9];
1281     HWND hToolbar;
1282     int i;
1283
1284     for (i=0; i<9; i++)
1285         MakeButton(buttons+i, 1000+i, TBSTYLE_FLAT | TBSTYLE_CHECKGROUP, 0);
1286
1287     /* Test 1 - 9 buttons */
1288     hToolbar = CreateToolbarEx(hMainWnd,
1289         WS_VISIBLE | WS_CLIPCHILDREN | WS_CHILD | CCS_NORESIZE | CCS_NOPARENTALIGN
1290         | CCS_NOMOVEY | CCS_TOP,
1291         0,
1292         0, NULL, 0,
1293         buttons, sizeof(buttons)/sizeof(buttons[0]),
1294         20, 20, 0, 0, sizeof(TBBUTTON));
1295     ok(hToolbar != NULL, "Toolbar creation\n");
1296     ok(SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0) == 0, "TB_AUTOSIZE failed\n");
1297
1298     /* test setting rows to each of 1-10 with bLarger true and false */
1299     for (i=0; i<(sizeof(tbrows_results) / sizeof(tbrows_result_t)); i++) {
1300         RECT rc;
1301         int rows;
1302
1303         memset(&rc, 0xCC, sizeof(rc));
1304         SendMessageA(hToolbar, TB_SETROWS,
1305                      MAKELONG(tbrows_results[i].nRows, tbrows_results[i].bLarger),
1306                      (LPARAM) &rc);
1307
1308         rows = SendMessageA(hToolbar, TB_GETROWS, MAKELONG(0,0), MAKELONG(0,0));
1309         ok(rows == tbrows_results[i].expectedRows,
1310                    "[%d] Unexpected number of rows %d (expected %d)\n", i, rows,
1311                    tbrows_results[i].expectedRows);
1312     }
1313
1314     DestroyWindow(hToolbar);
1315 }
1316
1317 static void test_getstring(void)
1318 {
1319     HWND hToolbar = NULL;
1320     char str[10];
1321     WCHAR strW[10];
1322     static const char answer[] = "STR";
1323     static const WCHAR answerW[] = { 'S','T','R',0 };
1324     INT r;
1325
1326     hToolbar = CreateWindowExA(0, TOOLBARCLASSNAME, NULL, WS_CHILD | WS_VISIBLE, 0, 0, 0, 0, hMainWnd, (HMENU)5, GetModuleHandle(NULL), NULL);
1327     ok(hToolbar != NULL, "Toolbar creation problem\n");
1328
1329     r = SendMessage(hToolbar, TB_GETSTRING, MAKEWPARAM(0, 0), 0);
1330     if (r == 0)
1331     {
1332         win_skip("TB_GETSTRING and TB_GETSTRINGW need 5.80\n");
1333         DestroyWindow(hToolbar);
1334         return;
1335     }
1336     expect(-1, r);
1337     r = SendMessage(hToolbar, TB_GETSTRINGW, MAKEWPARAM(0, 0), 0);
1338     expect(-1, r);
1339     r = SendMessage(hToolbar, TB_ADDSTRING, 0, (LPARAM)answer);
1340     expect(0, r);
1341     r = SendMessage(hToolbar, TB_GETSTRING, MAKEWPARAM(0, 0), 0);
1342     expect(lstrlenA(answer), r);
1343     r = SendMessage(hToolbar, TB_GETSTRINGW, MAKEWPARAM(0, 0), 0);
1344     expect(lstrlenA(answer), r);
1345     r = SendMessage(hToolbar, TB_GETSTRING, MAKEWPARAM(sizeof(str), 0), (LPARAM)str);
1346     expect(lstrlenA(answer), r);
1347     expect(0, lstrcmp(answer, str));
1348     r = SendMessage(hToolbar, TB_GETSTRINGW, MAKEWPARAM(sizeof(strW), 0), (LPARAM)strW);
1349     expect(lstrlenA(answer), r);
1350     expect(0, lstrcmpW(answerW, strW));
1351
1352     DestroyWindow(hToolbar);
1353 }
1354
1355 START_TEST(toolbar)
1356 {
1357     WNDCLASSA wc;
1358     MSG msg;
1359     RECT rc;
1360   
1361     InitCommonControls();
1362   
1363     wc.style = CS_HREDRAW | CS_VREDRAW;
1364     wc.cbClsExtra = 0;
1365     wc.cbWndExtra = 0;
1366     wc.hInstance = GetModuleHandleA(NULL);
1367     wc.hIcon = NULL;
1368     wc.hCursor = LoadCursorA(NULL, IDC_IBEAM);
1369     wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
1370     wc.lpszMenuName = NULL;
1371     wc.lpszClassName = "MyTestWnd";
1372     wc.lpfnWndProc = MyWndProc;
1373     RegisterClassA(&wc);
1374     
1375     hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW, 
1376       CW_USEDEFAULT, CW_USEDEFAULT, 680, 260, NULL, NULL, GetModuleHandleA(NULL), 0);
1377     GetClientRect(hMainWnd, &rc);
1378     ShowWindow(hMainWnd, SW_SHOW);
1379
1380     basic_test();
1381     test_add_bitmap();
1382     test_add_string();
1383     test_hotitem();
1384     test_sizes();
1385     test_recalc();
1386     test_getbuttoninfo();
1387     test_createtoolbarex();
1388     test_dispinfo();
1389     test_setrows();
1390     test_getstring();
1391
1392     PostQuitMessage(0);
1393     while(GetMessageA(&msg,0,0,0)) {
1394         TranslateMessage(&msg);
1395         DispatchMessageA(&msg);
1396     }
1397     DestroyWindow(hMainWnd);
1398 }