comctl32: Add missing '\n' to ok() call.
[wine] / dlls / comctl32 / tests / toolbar.c
1 /* Unit tests for treeview.
2  *
3  * Copyright 2005 Krzysztof Foltman
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18  */
19
20 #include <assert.h>
21 #include <stdarg.h>
22
23 #include "windef.h"
24 #include "winbase.h"
25 #include "wingdi.h"
26 #include "winuser.h"
27 #include "winnls.h"
28 #include "winreg.h"
29 #include "commctrl.h" 
30
31 #include "resources.h"
32
33 #include "wine/test.h"
34
35 HWND hMainWnd;
36 BOOL g_fBlockHotItemChange;
37 BOOL g_fReceivedHotItemChange;
38 BOOL g_fExpectedHotItemOld;
39 BOOL g_fExpectedHotItemNew;
40
41 #define check_rect(name, val, exp) ok(val.top == exp.top && val.bottom == exp.bottom && \
42     val.left == exp.left && val.right == exp.right, "invalid rect (" name ") (%d,%d) (%d,%d) - expected (%d,%d) (%d,%d)\n", \
43     val.left, val.top, val.right, val.bottom, exp.left, exp.top, exp.right, exp.bottom);
44  
45 #define compare(val, exp, format) ok((val) == (exp), #val " value " format " expected " format "\n", (val), (exp));
46
47 static void MakeButton(TBBUTTON *p, int idCommand, int fsStyle, int nString) {
48   p->iBitmap = -2;
49   p->idCommand = idCommand;
50   p->fsState = TBSTATE_ENABLED;
51   p->fsStyle = fsStyle;
52   p->iString = nString;
53 }
54
55 static LRESULT MyWnd_Notify(LPARAM lParam)
56 {
57     NMHDR *hdr = (NMHDR *)lParam;
58     NMTBHOTITEM *nmhi;
59     switch (hdr->code)
60     {
61         case TBN_HOTITEMCHANGE:
62             nmhi = (NMTBHOTITEM *)lParam;
63             g_fReceivedHotItemChange = TRUE;
64             if (g_fExpectedHotItemOld != g_fExpectedHotItemNew)
65             {
66                 compare(nmhi->idOld, g_fExpectedHotItemOld, "%d");
67                 compare(nmhi->idNew, g_fExpectedHotItemNew, "%d");
68             }
69             if (g_fBlockHotItemChange)
70                 return 1;
71             break;
72     }
73     return 0;
74 }
75
76 static LRESULT CALLBACK MyWndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
77 {
78     switch (msg)
79     {
80         case WM_NOTIFY:
81             return MyWnd_Notify(lParam);
82     }
83     return DefWindowProcA(hWnd, msg, wParam, lParam);
84 }
85
86 static void basic_test(void)
87 {
88     TBBUTTON buttons[9];
89     HWND hToolbar;
90     int i;
91     for (i=0; i<9; i++)
92         MakeButton(buttons+i, 1000+i, TBSTYLE_CHECKGROUP, 0);
93     MakeButton(buttons+3, 1003, TBSTYLE_SEP|TBSTYLE_GROUP, 0);
94     MakeButton(buttons+6, 1006, TBSTYLE_SEP, 0);
95
96     hToolbar = CreateToolbarEx(hMainWnd,
97         WS_VISIBLE | WS_CLIPCHILDREN | CCS_TOP |
98         WS_CHILD | TBSTYLE_LIST,
99         100,
100         0, NULL, (UINT)0,
101         buttons, sizeof(buttons)/sizeof(buttons[0]),
102         0, 0, 20, 16, sizeof(TBBUTTON));
103     ok(hToolbar != NULL, "Toolbar creation\n");
104     SendMessage(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)"test\000");
105
106     /* test for exclusion working inside a separator-separated :-) group */
107     SendMessage(hToolbar, TB_CHECKBUTTON, 1000, 1); /* press A1 */
108     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 pressed\n");
109     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1001, 0), "A2 not pressed\n");
110
111     SendMessage(hToolbar, TB_CHECKBUTTON, 1004, 1); /* press A5, release A1 */
112     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1004, 0), "A5 pressed\n");
113     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 not pressed anymore\n");
114
115     SendMessage(hToolbar, TB_CHECKBUTTON, 1005, 1); /* press A6, release A5 */
116     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 pressed\n");
117     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1004, 0), "A5 not pressed anymore\n");
118
119     /* test for inter-group crosstalk, ie. two radio groups interfering with each other */
120     SendMessage(hToolbar, TB_CHECKBUTTON, 1007, 1); /* press B2 */
121     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 still pressed, no inter-group crosstalk\n");
122     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 still not pressed\n");
123     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1007, 0), "B2 pressed\n");
124
125     SendMessage(hToolbar, TB_CHECKBUTTON, 1000, 1); /* press A1 and ensure B group didn't suffer */
126     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 not pressed anymore\n");
127     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 pressed\n");
128     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1007, 0), "B2 still pressed\n");
129
130     SendMessage(hToolbar, TB_CHECKBUTTON, 1008, 1); /* press B3, and ensure A group didn't suffer */
131     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1005, 0), "A6 pressed\n");
132     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1000, 0), "A1 pressed\n");
133     ok(!SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1007, 0), "B2 not pressed\n");
134     ok(SendMessage(hToolbar, TB_ISBUTTONCHECKED, 1008, 0), "B3 pressed\n");
135     DestroyWindow(hToolbar);
136 }
137
138 static void rebuild_toolbar(HWND *hToolbar)
139 {
140     if (*hToolbar != NULL)
141         DestroyWindow(*hToolbar);
142     *hToolbar = CreateWindowEx(0, TOOLBARCLASSNAME, NULL, WS_CHILD | WS_VISIBLE, 0, 0, 0, 0,
143         hMainWnd, (HMENU)5, GetModuleHandle(NULL), NULL);
144     ok(*hToolbar != NULL, "Toolbar creation problem\n");
145     ok(SendMessage(*hToolbar, TB_BUTTONSTRUCTSIZE, (WPARAM)sizeof(TBBUTTON), 0) == 0, "TB_BUTTONSTRUCTSIZE failed\n");
146     ok(SendMessage(*hToolbar, TB_AUTOSIZE, 0, 0) == 0, "TB_AUTOSIZE failed\n");
147     ok(SendMessage(*hToolbar, WM_SETFONT, (WPARAM)GetStockObject(SYSTEM_FONT), 0)==1, "WM_SETFONT\n");
148 }
149
150 void rebuild_toolbar_with_buttons(HWND *hToolbar)
151 {
152     TBBUTTON buttons[5];
153     rebuild_toolbar(hToolbar);
154     
155     ZeroMemory(&buttons, sizeof(buttons));
156     buttons[0].idCommand = 1;
157     buttons[0].fsStyle = BTNS_BUTTON;
158     buttons[0].fsState = TBSTATE_ENABLED;
159     buttons[0].iString = -1;
160     buttons[1].idCommand = 3;
161     buttons[1].fsStyle = BTNS_BUTTON;
162     buttons[1].fsState = TBSTATE_ENABLED;
163     buttons[1].iString = -1;
164     buttons[2].idCommand = 5;
165     buttons[2].fsStyle = BTNS_SEP;
166     buttons[2].fsState = TBSTATE_ENABLED;
167     buttons[2].iString = -1;
168     buttons[3].idCommand = 7;
169     buttons[3].fsStyle = BTNS_BUTTON;
170     buttons[3].fsState = TBSTATE_ENABLED;
171     buttons[3].iString = -1;
172     buttons[4].idCommand = 9;
173     buttons[4].fsStyle = BTNS_BUTTON;
174     buttons[4].fsState = 0;  /* disabled */
175     buttons[4].iString = -1;
176     ok(SendMessage(*hToolbar, TB_ADDBUTTONS, 5, (LPARAM)buttons) == 1, "TB_ADDBUTTONS failed\n");
177     ok(SendMessage(*hToolbar, TB_AUTOSIZE, 0, 0) == 0, "TB_AUTOSIZE failed\n");
178 }
179
180
181 #define CHECK_IMAGELIST(count, dx, dy) { \
182     int cx, cy; \
183     HIMAGELIST himl = (HIMAGELIST)SendMessageA(hToolbar, TB_GETIMAGELIST, 0, 0); \
184     ok(himl != NULL, "No image list\n"); \
185     if (himl != NULL) {\
186         ok(ImageList_GetImageCount(himl) == count, "Images count mismatch - %d vs %d\n", count, ImageList_GetImageCount(himl)); \
187         ImageList_GetIconSize(himl, &cx, &cy); \
188         ok(cx == dx && cy == cy, "Icon size mismatch - %dx%d vs %dx%d\n", dx, dy, cx, cy); \
189     } \
190 }
191
192 #define CHECK_IMAGELIST_TODO_COUNT(count, dx, dy) { \
193     int cx, cy; \
194     HIMAGELIST himl = (HIMAGELIST)SendMessageA(hToolbar, TB_GETIMAGELIST, 0, 0); \
195     ok(himl != NULL, "No image list\n"); \
196     if (himl != NULL) {\
197         todo_wine ok(ImageList_GetImageCount(himl) == count, "Images count mismatch - %d vs %d\n", count, ImageList_GetImageCount(himl)); \
198         ImageList_GetIconSize(himl, &cx, &cy); \
199         ok(cx == dx && cy == cy, "Icon size mismatch - %dx%d vs %dx%d\n", dx, dy, cx, cy); \
200     } \
201 }
202
203 #define CHECK_IMAGELIST_TODO_COUNT_SIZE(count, dx, dy) { \
204     int cx, cy; \
205     HIMAGELIST himl = (HIMAGELIST)SendMessageA(hToolbar, TB_GETIMAGELIST, 0, 0); \
206     ok(himl != NULL, "No image list\n"); \
207     if (himl != NULL) {\
208         todo_wine ok(ImageList_GetImageCount(himl) == count, "Images count mismatch - %d vs %d\n", count, ImageList_GetImageCount(himl)); \
209         ImageList_GetIconSize(himl, &cx, &cy); \
210         todo_wine ok(cx == dx && cy == cy, "Icon size mismatch - %dx%d vs %dx%d\n", dx, dy, cx, cy); \
211     } \
212 }
213
214 static void test_add_bitmap(void)
215 {
216     HWND hToolbar = NULL;
217     TBADDBITMAP bmp128;
218     TBADDBITMAP bmp80;
219     TBADDBITMAP stdsmall;
220     TBADDBITMAP addbmp;
221     HIMAGELIST himl;
222     INT ret;
223
224     /* empty 128x15 bitmap */
225     bmp128.hInst = GetModuleHandle(NULL);
226     bmp128.nID = IDB_BITMAP_128x15;
227
228     /* empty 80x15 bitmap */
229     bmp80.hInst = GetModuleHandle(NULL);
230     bmp80.nID = IDB_BITMAP_80x15;
231
232     /* standard bitmap - 240x15 pixels */
233     stdsmall.hInst = HINST_COMMCTRL;
234     stdsmall.nID = IDB_STD_SMALL_COLOR;
235
236     rebuild_toolbar(&hToolbar);
237     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 8, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
238     CHECK_IMAGELIST(8, 16, 15);
239     
240     /* adding more bitmaps */
241     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80) == 8, "TB_ADDBITMAP - unexpected return\n");
242     CHECK_IMAGELIST(13, 16, 15);
243     /* adding the same bitmap will simply return the index of the already loaded block */
244     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 8, (LPARAM)&bmp128);
245     ok(ret == 0, "TB_ADDBITMAP - unexpected return %d\n", ret);
246     CHECK_IMAGELIST(13, 16, 15);
247     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80);
248     ok(ret == 8, "TB_ADDBITMAP - unexpected return %d\n", ret);
249     CHECK_IMAGELIST(13, 16, 15);
250     /* even if we increase the wParam */
251     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 55, (LPARAM)&bmp80);
252     ok(ret == 8, "TB_ADDBITMAP - unexpected return %d\n", ret);
253     CHECK_IMAGELIST(13, 16, 15);
254
255     /* when the wParam is smaller than the bitmaps count but non-zero, all the bitmaps will be added*/
256     rebuild_toolbar(&hToolbar);
257     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 3, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
258     CHECK_IMAGELIST(8, 16, 15);
259     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80);
260     ok(ret == 3, "TB_ADDBITMAP - unexpected return %d\n", ret);
261     /* the returned value is misleading - id 8 is the id of the first icon from bmp80 */
262     CHECK_IMAGELIST(13, 16, 15);
263
264     /* the same for negative wParam */
265     rebuild_toolbar(&hToolbar);
266     ret = SendMessageA(hToolbar, TB_ADDBITMAP, -143, (LPARAM)&bmp128);
267     ok(ret == 0, "TB_ADDBITMAP - unexpected return %d\n", ret);
268     CHECK_IMAGELIST(8, 16, 15);
269     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp80);
270     ok(ret == -143, "TB_ADDBITMAP - unexpected return %d\n", ret);
271     CHECK_IMAGELIST(13, 16, 15);
272
273     /* for zero only one bitmap will be added */
274     rebuild_toolbar(&hToolbar);
275     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 0, (LPARAM)&bmp80);
276     ok(ret == 0, "TB_ADDBITMAP - unexpected return %d\n", ret);
277     CHECK_IMAGELIST(1, 16, 15);
278
279     /* if wParam is larger than the amount of icons, the list is grown */
280     rebuild_toolbar(&hToolbar);
281     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp80) == 0, "TB_ADDBITMAP - unexpected return\n");
282     CHECK_IMAGELIST(100, 16, 15);
283     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp128);
284     ok(ret == 100, "TB_ADDBITMAP - unexpected return %d\n", ret);
285     CHECK_IMAGELIST(200, 16, 15);
286
287     /* adding built-in items - the wParam is ignored */
288     rebuild_toolbar(&hToolbar);
289     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp80) == 0, "TB_ADDBITMAP - unexpected return\n");
290     CHECK_IMAGELIST(5, 16, 15);
291     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 0, (LPARAM)&stdsmall) == 5, "TB_ADDBITMAP - unexpected return\n");
292     CHECK_IMAGELIST(20, 16, 15);
293     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp128) == 20, "TB_ADDBITMAP - unexpected return\n");
294     CHECK_IMAGELIST(28, 16, 15);
295
296     /* when we increase the bitmap size, less icons will be created */
297     rebuild_toolbar(&hToolbar);
298     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(20, 20)) == TRUE, "TB_SETBITMAPSIZE failed\n");
299     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
300     CHECK_IMAGELIST(6, 20, 20);
301     ret = SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp80);
302     ok(ret == 1, "TB_ADDBITMAP - unexpected return %d\n", ret);
303     CHECK_IMAGELIST(10, 20, 20);
304     /* the icons can be resized - an UpdateWindow is needed as this probably happens during WM_PAINT */
305     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(8, 8)) == TRUE, "TB_SETBITMAPSIZE failed\n");
306     UpdateWindow(hToolbar);
307     CHECK_IMAGELIST(26, 8, 8);
308     /* loading a standard bitmaps automatically resizes the icons */
309     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&stdsmall) == 2, "TB_ADDBITMAP - unexpected return\n");
310     UpdateWindow(hToolbar);
311     CHECK_IMAGELIST(28, 16, 15);
312
313     /* two more SETBITMAPSIZE tests */
314     rebuild_toolbar(&hToolbar);
315     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
316     CHECK_IMAGELIST(100, 16, 15);
317     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 100, (LPARAM)&bmp80) == 100, "TB_ADDBITMAP - unexpected return\n");
318     CHECK_IMAGELIST(200, 16, 15);
319     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(8, 8)) == TRUE, "TB_SETBITMAPSIZE failed\n");
320     UpdateWindow(hToolbar);
321     CHECK_IMAGELIST(200, 8, 8);
322     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(30, 30)) == TRUE, "TB_SETBITMAPSIZE failed\n");
323     UpdateWindow(hToolbar);
324     CHECK_IMAGELIST(200, 30, 30);
325     rebuild_toolbar(&hToolbar);
326     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 5, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
327     CHECK_IMAGELIST(8, 16, 15);
328     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 3, (LPARAM)&bmp80) == 5, "TB_ADDBITMAP - unexpected return\n");
329     CHECK_IMAGELIST(13, 16, 15);
330     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(30, 30)) == TRUE, "TB_SETBITMAPSIZE failed\n");
331     UpdateWindow(hToolbar);
332     CHECK_IMAGELIST(8, 30, 30);
333
334     /* the control can add bitmaps to an existing image list */
335     rebuild_toolbar(&hToolbar);
336     himl = ImageList_LoadImage(GetModuleHandle(NULL), MAKEINTRESOURCE(IDB_BITMAP_80x15), 20, 2, CLR_NONE, IMAGE_BITMAP, LR_DEFAULTCOLOR);
337     ok(himl != NULL, "failed to create imagelist\n");
338     ok(SendMessageA(hToolbar, TB_SETIMAGELIST, 0, (LPARAM)himl) == 0, "TB_SETIMAGELIST failed\n");
339     CHECK_IMAGELIST(4, 20, 15);
340     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&bmp128) == 0, "TB_ADDBITMAP - unexpected return\n");
341     CHECK_IMAGELIST(10, 20, 15);
342     /* however TB_SETBITMAPSIZE/add std bitmap won't change the image size (the button size does change!) */
343     ok(SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(8, 8)) == TRUE, "TB_SETBITMAPSIZE failed\n");
344     UpdateWindow(hToolbar);
345     CHECK_IMAGELIST(10, 20, 15);
346     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 0, (LPARAM)&stdsmall) == 1, "TB_SETBITMAPSIZE failed\n");
347     UpdateWindow(hToolbar);
348     CHECK_IMAGELIST_TODO_COUNT(22, 20, 15);
349
350     /* check standard bitmaps */
351     addbmp.hInst = HINST_COMMCTRL;
352     addbmp.nID = IDB_STD_SMALL_COLOR;
353     rebuild_toolbar(&hToolbar);
354     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
355     CHECK_IMAGELIST(15, 16, 15);
356     addbmp.nID = IDB_STD_LARGE_COLOR;
357     rebuild_toolbar(&hToolbar);
358     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
359     CHECK_IMAGELIST(15, 24, 24);
360
361     addbmp.nID = IDB_VIEW_SMALL_COLOR;
362     rebuild_toolbar(&hToolbar);
363     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
364     CHECK_IMAGELIST(12, 16, 15);
365     addbmp.nID = IDB_VIEW_LARGE_COLOR;
366     rebuild_toolbar(&hToolbar);
367     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
368     CHECK_IMAGELIST(12, 24, 24);
369
370     addbmp.nID = IDB_HIST_SMALL_COLOR;
371     rebuild_toolbar(&hToolbar);
372     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
373     CHECK_IMAGELIST(5, 16, 15);
374     addbmp.nID = IDB_HIST_LARGE_COLOR;
375     rebuild_toolbar(&hToolbar);
376     ok(SendMessageA(hToolbar, TB_ADDBITMAP, 1, (LPARAM)&addbmp) == 0, "TB_ADDBITMAP - unexpected return\n");
377     CHECK_IMAGELIST(5, 24, 24);
378
379
380     DestroyWindow(hToolbar);
381 }
382
383 #define CHECK_STRING_TABLE(count, tab) { \
384         INT _i; \
385         CHAR _buf[260]; \
386         for (_i = 0; _i < (count); _i++) {\
387             ret = SendMessageA(hToolbar, TB_GETSTRING, MAKEWPARAM(260, _i), (LPARAM)_buf); \
388             ok(ret >= 0, "TB_GETSTRING - unexpected return %d while checking string %d\n", ret, _i); \
389             if (ret >= 0) \
390                 ok(strcmp(_buf, (tab)[_i]) == 0, "Invalid string #%d - '%s' vs '%s'\n", _i, (tab)[_i], _buf); \
391         } \
392         ok(SendMessageA(hToolbar, TB_GETSTRING, MAKEWPARAM(260, (count)), (LPARAM)_buf) == -1, \
393             "Too many string in table\n"); \
394     }
395
396 void test_add_string()
397 {
398     LPCSTR test1 = "a\0b\0";
399     LPCSTR test2 = "|a|b||\0";
400     LPCSTR ret1[] = {"a", "b"};
401     LPCSTR ret2[] = {"a", "b", "|a|b||"};
402     LPCSTR ret3[] = {"a", "b", "|a|b||", "p", "q"};
403     LPCSTR ret4[] = {"a", "b", "|a|b||", "p", "q", "p"};
404     LPCSTR ret5[] = {"a", "b", "|a|b||", "p", "q", "p", "p", "q"};
405     LPCSTR ret6[] = {"a", "b", "|a|b||", "p", "q", "p", "p", "q", "p", "", "q"};
406     LPCSTR ret7[] = {"a", "b", "|a|b||", "p", "q", "p", "p", "q", "p", "", "q", "br", "c", "d"};
407     HWND hToolbar = NULL;
408     TBBUTTON button;
409     int ret;
410
411     rebuild_toolbar(&hToolbar);
412     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)test1);
413     ok(ret == 0, "TB_ADDSTRINGA - unexpected return %d\n", ret);
414     CHECK_STRING_TABLE(2, ret1);
415     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)test2);
416     ok(ret == 2, "TB_ADDSTRINGA - unexpected return %d\n", ret);
417     CHECK_STRING_TABLE(3, ret2);
418
419     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD1);
420     ok(ret == 3, "TB_ADDSTRINGA - unexpected return %d\n", ret);
421     CHECK_STRING_TABLE(3, ret2);
422     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD2);
423     ok(ret == 3, "TB_ADDSTRINGA - unexpected return %d\n", ret);
424     CHECK_STRING_TABLE(5, ret3);
425     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD3);
426     ok(ret == 5, "TB_ADDSTRINGA - unexpected return %d\n", ret);
427     CHECK_STRING_TABLE(6, ret4);
428     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD4);
429     ok(ret == 6, "TB_ADDSTRINGA - unexpected return %d\n", ret);
430     CHECK_STRING_TABLE(8, ret5);
431     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD5);
432     ok(ret == 8, "TB_ADDSTRINGA - unexpected return %d\n", ret);
433     CHECK_STRING_TABLE(11, ret6);
434     ret = SendMessageA(hToolbar, TB_ADDSTRINGA, (WPARAM)GetModuleHandle(NULL), IDS_TBADD7);
435     ok(ret == 11, "TB_ADDSTRINGA - unexpected return %d\n", ret);
436     CHECK_STRING_TABLE(14, ret7);
437
438     ZeroMemory(&button, sizeof(button));
439     button.iString = (UINT_PTR)"Test";
440     SendMessageA(hToolbar, TB_INSERTBUTTONA, 0, (LPARAM)&button);
441     CHECK_STRING_TABLE(14, ret7);
442     SendMessageA(hToolbar, TB_ADDBUTTONSA, 1, (LPARAM)&button);
443     CHECK_STRING_TABLE(14, ret7);
444
445     DestroyWindow(hToolbar);
446 }
447
448 static void expect_hot_notify(int idold, int idnew)
449 {
450     g_fExpectedHotItemOld = idold;
451     g_fExpectedHotItemNew = idnew;
452     g_fReceivedHotItemChange = FALSE;
453 }
454
455 #define check_hot_notify() \
456     ok(g_fReceivedHotItemChange, "TBN_HOTITEMCHANGE not received\n"); \
457     g_fExpectedHotItemOld = g_fExpectedHotItemNew = 0;
458
459 void test_hotitem()
460 {
461     HWND hToolbar = NULL;
462     TBBUTTONINFO tbinfo;
463     LRESULT ret;
464
465     g_fBlockHotItemChange = FALSE;
466
467     rebuild_toolbar_with_buttons(&hToolbar);
468     /* set TBSTYLE_FLAT. comctl5 allows hot items only for such toolbars.
469      * comctl6 doesn't have this requirement even when theme == NULL */
470     SetWindowLong(hToolbar, GWL_STYLE, TBSTYLE_FLAT | GetWindowLong(hToolbar, GWL_STYLE));
471     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
472     ok(ret == -1, "Hot item: %ld, expected -1\n", ret);
473     ret = SendMessage(hToolbar, TB_SETHOTITEM, 1, 0);
474     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
475     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
476     ok(ret == 1, "Hot item: %ld, expected 1\n", ret);
477     ret = SendMessage(hToolbar, TB_SETHOTITEM, 2, 0);
478     ok(ret == 1, "TB_SETHOTITEM returned %ld, expected 1\n", ret);
479
480     ret = SendMessage(hToolbar, TB_SETHOTITEM, 0xbeef, 0);
481     ok(ret == 2, "TB_SETHOTITEM returned %ld, expected 2\n", ret);
482     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
483     ok(ret == 2, "Hot item: %lx, expected 2\n", ret);
484     ret = SendMessage(hToolbar, TB_SETHOTITEM, -0xbeef, 0);
485     ok(ret == 2, "TB_SETHOTITEM returned %ld, expected 2\n", ret);
486     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
487     ok(ret == -1, "Hot item: %lx, expected -1\n", ret);
488
489     expect_hot_notify(0, 7);
490     ret = SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
491     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
492     check_hot_notify();
493     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
494     ok(ret == 3, "Hot item: %lx, expected 3\n", ret);
495     g_fBlockHotItemChange = TRUE;
496     ret = SendMessage(hToolbar, TB_SETHOTITEM, 2, 0);
497     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 2\n", ret);
498     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
499     ok(ret == 3, "Hot item: %lx, expected 3\n", ret);
500     g_fBlockHotItemChange = FALSE;
501
502     g_fReceivedHotItemChange = FALSE;
503     ret = SendMessage(hToolbar, TB_SETHOTITEM, 0xbeaf, 0);
504     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
505     ok(g_fReceivedHotItemChange == FALSE, "TBN_HOTITEMCHANGE received for invalid parameter\n");
506
507     g_fReceivedHotItemChange = FALSE;
508     ret = SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
509     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
510     ok(g_fReceivedHotItemChange == FALSE, "TBN_HOTITEMCHANGE received after a duplication\n");
511
512     expect_hot_notify(7, 0);
513     ret = SendMessage(hToolbar, TB_SETHOTITEM, -0xbeaf, 0);
514     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
515     check_hot_notify();
516     SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
517
518     /* setting disabled buttons will generate a notify with the button id but no button will be hot */
519     expect_hot_notify(7, 9);
520     ret = SendMessage(hToolbar, TB_SETHOTITEM, 4, 0);
521     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
522     check_hot_notify();
523     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
524     ok(ret == -1, "Hot item: %lx, expected -1\n", ret);
525     /* enabling the button won't change that */
526     SendMessage(hToolbar, TB_ENABLEBUTTON, 9, TRUE);
527     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
528     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
529
530     /* disabling a hot button works */
531     ret = SendMessage(hToolbar, TB_SETHOTITEM, 3, 0);
532     ok(ret == -1, "TB_SETHOTITEM returned %ld, expected -1\n", ret);
533     g_fReceivedHotItemChange = FALSE;
534     SendMessage(hToolbar, TB_ENABLEBUTTON, 7, FALSE);
535     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
536     ok(ret == 3, "TB_SETHOTITEM returned %ld, expected 3\n", ret);
537     ok(g_fReceivedHotItemChange == FALSE, "Unexpected TBN_HOTITEMCHANGE\n");
538
539     SendMessage(hToolbar, TB_SETHOTITEM, 1, 0);
540     tbinfo.cbSize = sizeof(TBBUTTONINFO);
541     tbinfo.dwMask = TBIF_STATE;
542     tbinfo.fsState = 0;  /* disabled */
543     g_fReceivedHotItemChange = FALSE;
544     ok(SendMessage(hToolbar, TB_SETBUTTONINFO, 1, (LPARAM)&tbinfo) == TRUE, "TB_SETBUTTONINFO failed\n");
545     ret = SendMessage(hToolbar, TB_GETHOTITEM, 0, 0);
546     ok(ret == 1, "TB_SETHOTITEM returned %ld, expected 1\n", ret);
547     ok(g_fReceivedHotItemChange == FALSE, "Unexpected TBN_HOTITEMCHANGE\n");
548
549     DestroyWindow(hToolbar);
550 }
551
552 #if 0  /* use this to generate more tests*/
553
554 void dump_sizes(HWND hToolbar)
555 {
556     SIZE sz;
557     RECT r;
558     int count = SendMessage(hToolbar, TB_BUTTONCOUNT, 0, 0);
559     int i;
560
561     GetClientRect(hToolbar, &r);
562     SendMessageA(hToolbar, TB_GETMAXSIZE, 0, &sz);
563     printf("  { {%d, %d, %d, %d}, {%d, %d}, %d, {", r.left, r.top, r.right, r.bottom,
564         sz.cx, sz.cy, count);
565     for (i=0; i<count; i++)
566     {
567         SendMessageA(hToolbar, TB_GETITEMRECT, i, &r);
568         printf("%s{%3d, %3d, %3d, %3d}, ", (i%3==0 ? "\n    " : ""), r.left, r.top, r.right, r.bottom);
569     }
570     printf("\n  }, }, \n");
571 }
572
573 #define check_sizes() dump_sizes(hToolbar);
574 #define check_sizes_todo(todomask) dump_sizes(hToolbar);
575
576 #else
577
578 typedef struct
579 {
580     RECT rcClient;
581     SIZE szMin;
582     INT nButtons;
583     RECT rcButtons[100];
584 } tbsize_result_t;
585
586 tbsize_result_t tbsize_results[] = 
587 {
588   { {0, 0, 672, 26}, {100, 22}, 5, {
589     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
590     { 54,   2,  77,  24}, { 77,   2, 100,  24},
591   }, },
592   { {0, 0, 672, 26}, {146, 22}, 7, {
593     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
594     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
595     {  0,  24,  23,  46},
596   }, },
597   { {0, 0, 672, 48}, {146, 22}, 7, {
598     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
599     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
600     {  0,  24,  23,  46},
601   }, },
602   { {0, 0, 672, 26}, {146, 22}, 7, {
603     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
604     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
605     {123,   2, 146,  24},
606   }, },
607   { {0, 0, 672, 26}, {192, 22}, 9, {
608     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
609     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
610     {123,   2, 146,  24}, {146,   2, 169,  24}, {169,   2, 192,  24},
611   }, },
612   { {0, 0, 672, 92}, {882, 22}, 39, {
613     {  0,   2,  23,  24}, { 23,   2,  46,  24}, {  0,   2,   8,  29},
614     {  0,  29,  23,  51}, { 23,  29,  46,  51}, { 46,  29,  69,  51},
615     { 69,  29,  92,  51}, { 92,  29, 115,  51}, {115,  29, 138,  51},
616     {138,  29, 161,  51}, {161,  29, 184,  51}, {184,  29, 207,  51},
617     {207,  29, 230,  51}, {230,  29, 253,  51}, {253,  29, 276,  51},
618     {276,  29, 299,  51}, {299,  29, 322,  51}, {322,  29, 345,  51},
619     {345,  29, 368,  51}, {368,  29, 391,  51}, {391,  29, 414,  51},
620     {414,  29, 437,  51}, {437,  29, 460,  51}, {460,  29, 483,  51},
621     {483,  29, 506,  51}, {506,  29, 529,  51}, {529,  29, 552,  51},
622     {552,  29, 575,  51}, {575,  29, 598,  51}, {598,  29, 621,  51},
623     {621,  29, 644,  51}, {644,  29, 667,  51}, {  0,  51,  23,  73},
624     { 23,  51,  46,  73}, { 46,  51,  69,  73}, { 69,  51,  92,  73},
625     { 92,  51, 115,  73}, {115,  51, 138,  73}, {138,  51, 161,  73},
626   }, },
627   { {0, 0, 48, 226}, {23, 140}, 7, {
628     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  94,  24},
629     { 94,   2, 117,  24}, {117,   2, 140,  24}, {140,   2, 163,  24},
630     {  0,  24,  23,  46},
631   }, },
632   { {0, 0, 92, 226}, {23, 140}, 7, {
633     {  0,   2,  23,  24}, { 23,   2,  46,  24}, {  0,  24,  92,  32},
634     {  0,  32,  23,  54}, { 23,  32,  46,  54}, { 46,  32,  69,  54},
635     { 69,  32,  92,  54},
636   }, },
637   { {0, 0, 672, 26}, {194, 30}, 7, {
638     {  0,   2,  31,  32}, { 31,   2,  62,  32}, { 62,   2,  70,  32},
639     { 70,   2, 101,  32}, {101,   2, 132,  32}, {132,   2, 163,  32},
640     {  0,  32,  31,  62},
641   }, },
642   { {0, 0, 672, 64}, {194, 30}, 7, {
643     {  0,   2,  31,  32}, { 31,   2,  62,  32}, { 62,   2,  70,  32},
644     { 70,   2, 101,  32}, {101,   2, 132,  32}, {132,   2, 163,  32},
645     {  0,  32,  31,  62},
646   }, },
647   { {0, 0, 672, 64}, {194, 30}, 7, {
648     {  0,   0,  31,  30}, { 31,   0,  62,  30}, { 62,   0,  70,  30},
649     { 70,   0, 101,  30}, {101,   0, 132,  30}, {132,   0, 163,  30},
650     {  0,  30,  31,  60},
651   }, },
652   { {0, 0, 124, 226}, {31, 188}, 7, {
653     {  0,   0,  31,  30}, { 31,   0,  62,  30}, {  0,  30, 124,  38},
654     {  0,  38,  31,  68}, { 31,  38,  62,  68}, { 62,  38,  93,  68},
655     { 93,  38, 124,  68},
656   }, },
657   { {0, 0, 672, 26}, {146, 22}, 7, {
658     {  0,   2,  23,  24}, { 23,   2,  46,  24}, { 46,   2,  54,  24},
659     { 54,   2,  77,  24}, { 77,   2, 100,  24}, {100,   2, 123,  24},
660     {123,   2, 146,  24},
661   }, },
662   { {0, 0, 672, 26}, {146, 100}, 7, {
663     {  0,   0,  23, 100}, { 23,   0,  46, 100}, { 46,   0,  54, 100},
664     { 54,   0,  77, 100}, { 77,   0, 100, 100}, {100,   0, 123, 100},
665     {123,   0, 146, 100},
666   }, },
667   { {0, 0, 672, 26}, {215, 100}, 10, {
668     {  0,   0,  23, 100}, { 23,   0,  46, 100}, { 46,   0,  54, 100},
669     { 54,   0,  77, 100}, { 77,   0, 100, 100}, {100,   0, 123, 100},
670     {123,   0, 146, 100}, {146,   0, 169, 100}, {169,   0, 192, 100},
671     {192,   0, 215, 100},
672   }, },
673   { {0, 0, 672, 26}, {238, 39}, 11, {
674     {  0,   0,  23,  39}, { 23,   0,  46,  39}, { 46,   0,  54,  39},
675     { 54,   0,  77,  39}, { 77,   0, 100,  39}, {100,   0, 123,  39},
676     {123,   0, 146,  39}, {146,   0, 169,  39}, {169,   0, 192,  39},
677     {192,   0, 215,  39}, {215,   0, 238,  39},
678   }, },
679   { {0, 0, 672, 26}, {238, 22}, 11, {
680     {  0,   0,  23,  22}, { 23,   0,  46,  22}, { 46,   0,  54,  22},
681     { 54,   0,  77,  22}, { 77,   0, 100,  22}, {100,   0, 123,  22},
682     {123,   0, 146,  22}, {146,   0, 169,  22}, {169,   0, 192,  22},
683     {192,   0, 215,  22}, {215,   0, 238,  22},
684   }, },
685   { {0, 0, 672, 26}, {489, 39}, 3, {
686     {  0,   2, 163,  41}, {163,   2, 330,  41}, {330,   2, 493,  41},
687   }, },
688 };
689
690 int tbsize_numtests = 0;
691
692 #define check_sizes_todo(todomask) { \
693         RECT rc; \
694         int buttonCount, i, mask=(todomask); \
695         tbsize_result_t *res = &tbsize_results[tbsize_numtests]; \
696         assert(tbsize_numtests < sizeof(tbsize_results)/sizeof(tbsize_results[0])); \
697         GetClientRect(hToolbar, &rc); \
698         /*check_rect("client", rc, res->rcClient);*/ \
699         buttonCount = SendMessage(hToolbar, TB_BUTTONCOUNT, 0, 0); \
700         compare(buttonCount, res->nButtons, "%d"); \
701         for (i=0; i<min(buttonCount, res->nButtons); i++) { \
702             ok(SendMessageA(hToolbar, TB_GETITEMRECT, i, (LPARAM)&rc) == 1, "TB_GETITEMRECT\n"); \
703             if (!(mask&1)) { \
704                 check_rect("button", rc, res->rcButtons[i]); \
705             } else {\
706                 todo_wine { check_rect("button", rc, res->rcButtons[i]); } \
707             } \
708             mask >>= 1; \
709         } \
710         tbsize_numtests++; \
711     }
712
713 #define check_sizes() check_sizes_todo(0)
714
715 #endif
716
717 TBBUTTON buttons1[] = {
718     {0, 10, TBSTATE_WRAP|TBSTATE_ENABLED, 0, {0, }, 0, -1},
719     {0, 11, 0, 0, {0, }, 0, -1},
720 };
721 TBBUTTON buttons2[] = {
722     {0, 20, TBSTATE_ENABLED, 0, {0, }, 0, -1},
723     {0, 21, TBSTATE_ENABLED, 0, {0, }, 0, -1},
724 };
725 TBBUTTON buttons3[] = {
726     {0, 30, TBSTATE_ENABLED, 0, {0, }, 0, 0},
727     {0, 31, TBSTATE_ENABLED, 0, {0, }, 0, 1},
728     {0, 32, TBSTATE_ENABLED, BTNS_AUTOSIZE, {0, }, 0, 1},
729     {0, 33, TBSTATE_ENABLED, BTNS_AUTOSIZE, {0, }, 0, (UINT_PTR)"Tst"}
730 };
731
732 void test_sizes()
733 {
734     HWND hToolbar = NULL;
735     int style;
736     int i;
737
738     rebuild_toolbar_with_buttons(&hToolbar);
739     style = GetWindowLong(hToolbar, GWL_STYLE);
740     ok(style == (WS_CHILD|WS_VISIBLE|CCS_TOP), "Invalid style %x\n", style);
741     check_sizes();
742     /* the TBSTATE_WRAP makes a second row */
743     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
744     check_sizes();
745     SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0);
746     check_sizes();
747     /* after setting the TBSTYLE_WRAPABLE the TBSTATE_WRAP is ignored */
748     SetWindowLong(hToolbar, GWL_STYLE, style|TBSTYLE_WRAPABLE);
749     check_sizes();
750     /* adding new buttons with TBSTYLE_WRAPABLE doesn't add a new row */
751     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
752     check_sizes();
753     /* only after adding enough buttons the bar will be wrapped on a
754      * separator and then on the first button */
755     for (i=0; i<15; i++)
756         SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
757     check_sizes_todo(0x4);
758
759     rebuild_toolbar_with_buttons(&hToolbar);
760     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
761     /* setting the buttons vertical will only change the window client size */
762     SetWindowLong(hToolbar, GWL_STYLE, style | CCS_VERT);
763     SendMessage(hToolbar, TB_AUTOSIZE, 0, 0);
764     check_sizes_todo(0x3c);
765     /* with a TBSTYLE_WRAPABLE a wrapping will occure on the separator */
766     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_WRAPABLE | CCS_VERT);
767     SendMessage(hToolbar, TB_AUTOSIZE, 0, 0);
768     check_sizes_todo(0x7c);
769
770     rebuild_toolbar_with_buttons(&hToolbar);
771     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons1);
772     /* a TB_SETBITMAPSIZE changes button sizes*/
773     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(24, 24));
774     check_sizes();
775
776     /* setting a TBSTYLE_FLAT doesn't change anything - even after a TB_AUTOSIZE */
777     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_FLAT);
778     SendMessageA(hToolbar, TB_AUTOSIZE, 0, 0);
779     check_sizes();
780     /* but after a TB_SETBITMAPSIZE the top margins is changed */
781     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(20, 20));
782     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(24, 24));
783     check_sizes();
784     /* some vertical toolbar sizes */
785     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_FLAT | TBSTYLE_WRAPABLE | CCS_VERT);
786     check_sizes_todo(0x7c);
787
788     rebuild_toolbar_with_buttons(&hToolbar);
789     SetWindowLong(hToolbar, GWL_STYLE, style | TBSTYLE_FLAT);
790     /* newly added buttons will be use the previous margin */
791     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons2);
792     check_sizes();
793     /* TB_SETBUTTONSIZE can't be used to reduce the size of a button below the default */
794     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
795     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(22, 21))==1, "TB_SETBUTTONSIZE\n");
796     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
797     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(5, 100))==1, "TB_SETBUTTONSIZE\n");
798     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 100), "Unexpected button size\n");
799     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3))==1, "TB_SETBUTTONSIZE\n");
800     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
801     ok(SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(5, 100))==1, "TB_SETBUTTONSIZE\n");
802     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 100), "Unexpected button size\n");
803     check_sizes();
804     /* add some buttons with non-default sizes */
805     SendMessageA(hToolbar, TB_ADDBUTTONS, 2, (LPARAM)buttons2);
806     SendMessageA(hToolbar, TB_INSERTBUTTON, -1, (LPARAM)&buttons2[0]);
807     check_sizes();
808     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[0]);
809     /* TB_ADDSTRING resets the size */
810     SendMessageA(hToolbar, TB_ADDSTRING, 0, (LPARAM)"A\0MMMMMMMMMMMMM\0");
811     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 39), "Unexpected button size\n");
812     check_sizes();
813     /* TB_SETBUTTONSIZE can be used to crop the text */
814     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3));
815     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
816     check_sizes();
817     /* except for the first size, the default size is bitmap size + padding */
818     SendMessageA(hToolbar, TB_SETPADDING, 0, MAKELONG(1, 1));
819     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3));
820     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(17, 17), "Unexpected button size\n");
821     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(3, 3));
822     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(3, 3));
823     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(4, 4), "Unexpected button size\n");
824
825     rebuild_toolbar(&hToolbar);
826     /* sending a TB_SETBITMAPSIZE with the same sizes is enough to make the button smaller */
827     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
828     SendMessageA(hToolbar, TB_SETBITMAPSIZE, 0, MAKELONG(16, 15));
829     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 21), "Unexpected button size\n");
830
831     rebuild_toolbar(&hToolbar);
832     SendMessageA(hToolbar, TB_ADDSTRINGA, 0, (LPARAM)"A\0MMMMMMMMMMMMM\0");
833     /* the height is increased after a TB_ADDSTRING */
834     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 39), "Unexpected button size\n");
835     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(100, 100));
836     /* if a string is in the pool, even adding a button without a string resets the size */
837     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons2[0]);
838     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
839     SendMessageA(hToolbar, TB_SETBUTTONSIZE, 0, MAKELONG(100, 100));
840     /* an BTNS_AUTOSIZE button is also considered when computing the new size */
841     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[2]);
842     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(163, 39), "Unexpected button size\n");
843     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[0]);
844     check_sizes();
845     /* delete button doesn't change the buttons size */
846     SendMessageA(hToolbar, TB_DELETEBUTTON, 2, 0);
847     SendMessageA(hToolbar, TB_DELETEBUTTON, 1, 0);
848     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(163, 39), "Unexpected button size\n");
849     /* TB_INSERTBUTTONS will */
850     SendMessageA(hToolbar, TB_INSERTBUTTON, 1, (LPARAM)&buttons2[0]);
851     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(23, 22), "Unexpected button size\n");
852
853     rebuild_toolbar(&hToolbar);
854     SendMessageA(hToolbar, TB_ADDBUTTONS, 1, (LPARAM)&buttons3[3]);
855     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(27, 39), "Unexpected button size\n");
856     SendMessageA(hToolbar, TB_DELETEBUTTON, 0, 0);
857     ok(SendMessageA(hToolbar, TB_GETBUTTONSIZE, 0, 0) == MAKELONG(27, 39), "Unexpected button size\n");
858
859     DestroyWindow(hToolbar);
860 }
861
862 START_TEST(toolbar)
863 {
864     WNDCLASSA wc;
865     MSG msg;
866     RECT rc;
867   
868     InitCommonControls();
869   
870     wc.style = CS_HREDRAW | CS_VREDRAW;
871     wc.cbClsExtra = 0;
872     wc.cbWndExtra = 0;
873     wc.hInstance = GetModuleHandleA(NULL);
874     wc.hIcon = NULL;
875     wc.hCursor = LoadCursorA(NULL, MAKEINTRESOURCEA(IDC_IBEAM));
876     wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
877     wc.lpszMenuName = NULL;
878     wc.lpszClassName = "MyTestWnd";
879     wc.lpfnWndProc = MyWndProc;
880     RegisterClassA(&wc);
881     
882     hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW, 
883       CW_USEDEFAULT, CW_USEDEFAULT, 680, 260, NULL, NULL, GetModuleHandleA(NULL), 0);
884     GetClientRect(hMainWnd, &rc);
885     ShowWindow(hMainWnd, SW_SHOW);
886
887     basic_test();
888     test_add_bitmap();
889     test_add_string();
890     test_hotitem();
891     test_sizes();
892
893     PostQuitMessage(0);
894     while(GetMessageA(&msg,0,0,0)) {
895         TranslateMessage(&msg);
896         DispatchMessageA(&msg);
897     }
898     DestroyWindow(hMainWnd);
899 }