dlls/comctl32/tests/propsheet.c

   1 /* Unit test suite for property sheet control.
   2  *
   3  * Copyright 2006 Huw Davies
   4  * Copyright 2009 Jan de Mooij
   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 <windows.h>
  22 #include <commctrl.h>
  23 #include "msg.h"
  24
  25 #include "resources.h"
  26
  27 #include "wine/test.h"
  28
  29 static HWND parenthwnd;
  30 static HWND sheethwnd;
  31
  32 static LONG active_page = -1;
  33
  34 #define IDC_APPLY_BUTTON 12321
  35
  36
  37 /* try to make sure pending X events have been processed before continuing */
  38 static void flush_events(void)
  39 {
  40     MSG msg;
  41     int diff = 200;
  42     int min_timeout = 100;
  43     DWORD time = GetTickCount() + diff;
  44
  45     while (diff > 0)
  46     {
  47         if (MsgWaitForMultipleObjects( 0, NULL, FALSE, min_timeout, QS_ALLINPUT ) == WAIT_TIMEOUT) break;
  48         while (PeekMessageA( &msg, 0, 0, 0, PM_REMOVE )) DispatchMessageA( &msg );
  49         diff = time - GetTickCount();
  50     }
  51 }
  52
  53
  54 static int CALLBACK sheet_callback(HWND hwnd, UINT msg, LPARAM lparam)
  55 {
  56     switch(msg)
  57     {
  58     case PSCB_INITIALIZED:
  59       {
  60         char caption[256];
  61         GetWindowTextA(hwnd, caption, sizeof(caption));
  62         ok(!strcmp(caption,"test caption"), "caption: %s\n", caption);
  63         sheethwnd = hwnd;
  64         return 0;
  65       }
  66     }
  67     return 0;
  68 }
  69
  70 static INT_PTR CALLBACK page_dlg_proc(HWND hwnd, UINT msg, WPARAM wparam,
  71                                       LPARAM lparam)
  72 {
  73     switch(msg)
  74     {
  75     case WM_INITDIALOG:
  76       {
  77         HWND sheet = GetParent(hwnd);
  78         char caption[256];
  79         GetWindowTextA(sheet, caption, sizeof(caption));
  80         ok(!strcmp(caption,"test caption"), "caption: %s\n", caption);
  81         return TRUE;
  82       }
  83
  84     case WM_NOTIFY:
  85       {
  86         NMHDR *nmhdr = (NMHDR *)lparam;
  87         switch(nmhdr->code)
  88         {
  89         case PSN_APPLY:
  90             return TRUE;
  91         default:
  92             return FALSE;
  93         }
  94       }
  95     case WM_NCDESTROY:
  96         ok(!SendMessageA(sheethwnd, PSM_INDEXTOHWND, 400, 0),"Should always be 0\n");
  97         return TRUE;
  98
  99     default:
 100         return FALSE;
 101     }
 102 }
 103
 104 static void test_title(void)
 105 {
 106     HPROPSHEETPAGE hpsp[1];
 107     PROPSHEETPAGEA psp;
 108     PROPSHEETHEADERA psh;
 109     HWND hdlg;
 110     DWORD style;
 111
 112     memset(&psp, 0, sizeof(psp));
 113     psp.dwSize = sizeof(psp);
 114     psp.dwFlags = 0;
 115     psp.hInstance = GetModuleHandleA(NULL);
 116     U(psp).pszTemplate = "prop_page1";
 117     U2(psp).pszIcon = NULL;
 118     psp.pfnDlgProc = page_dlg_proc;
 119     psp.lParam = 0;
 120
 121     hpsp[0] = CreatePropertySheetPageA(&psp);
 122
 123     memset(&psh, 0, sizeof(psh));
 124     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 125     psh.dwFlags = PSH_MODELESS | PSH_USECALLBACK;
 126     psh.pszCaption = "test caption";
 127     psh.nPages = 1;
 128     psh.hwndParent = GetDesktopWindow();
 129     U3(psh).phpage = hpsp;
 130     psh.pfnCallback = sheet_callback;
 131
 132     hdlg = (HWND)PropertySheetA(&psh);
 133     ok(hdlg != INVALID_HANDLE_VALUE, "got invalid handle value %p\n", hdlg);
 134
 135     style = GetWindowLongA(hdlg, GWL_STYLE);
 136     ok(style == (WS_POPUP|WS_VISIBLE|WS_CLIPSIBLINGS|WS_CAPTION|WS_SYSMENU|
 137                  DS_CONTEXTHELP|DS_MODALFRAME|DS_SETFONT|DS_3DLOOK),
 138        "got unexpected style: %x\n", style);
 139
 140     DestroyWindow(hdlg);
 141 }
 142
 143 static void test_nopage(void)
 144 {
 145     HPROPSHEETPAGE hpsp[1];
 146     PROPSHEETPAGEA psp;
 147     PROPSHEETHEADERA psh;
 148     HWND hdlg, hpage;
 149     MSG msg;
 150
 151     memset(&psp, 0, sizeof(psp));
 152     psp.dwSize = sizeof(psp);
 153     psp.dwFlags = 0;
 154     psp.hInstance = GetModuleHandleA(NULL);
 155     U(psp).pszTemplate = "prop_page1";
 156     U2(psp).pszIcon = NULL;
 157     psp.pfnDlgProc = page_dlg_proc;
 158     psp.lParam = 0;
 159
 160     hpsp[0] = CreatePropertySheetPageA(&psp);
 161
 162     memset(&psh, 0, sizeof(psh));
 163     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 164     psh.dwFlags = PSH_MODELESS | PSH_USECALLBACK;
 165     psh.pszCaption = "test caption";
 166     psh.nPages = 1;
 167     psh.hwndParent = GetDesktopWindow();
 168     U3(psh).phpage = hpsp;
 169     psh.pfnCallback = sheet_callback;
 170
 171     hdlg = (HWND)PropertySheetA(&psh);
 172     ok(hdlg != INVALID_HANDLE_VALUE, "got invalid handle value %p\n", hdlg);
 173
 174     ShowWindow(hdlg,SW_NORMAL);
 175     SendMessageA(hdlg, PSM_REMOVEPAGE, 0, 0);
 176     hpage = /* PropSheet_GetCurrentPageHwnd(hdlg); */
 177         (HWND)SendMessageA(hdlg, PSM_GETCURRENTPAGEHWND, 0, 0);
 178     active_page = /* PropSheet_HwndToIndex(hdlg, hpage)); */
 179         (int)SendMessageA(hdlg, PSM_HWNDTOINDEX, (WPARAM)hpage, 0);
 180     ok(hpage == NULL, "expected no current page, got %p, index=%d\n", hpage, active_page);
 181     flush_events();
 182     RedrawWindow(hdlg,NULL,NULL,RDW_UPDATENOW|RDW_ERASENOW);
 183
 184     /* Check that the property sheet was fully redrawn */
 185     ok(!PeekMessageA(&msg, 0, WM_PAINT, WM_PAINT, PM_NOREMOVE),
 186        "expected no pending WM_PAINT messages\n");
 187     DestroyWindow(hdlg);
 188 }
 189
 190 static int CALLBACK disableowner_callback(HWND hwnd, UINT msg, LPARAM lparam)
 191 {
 192     switch(msg)
 193     {
 194     case PSCB_INITIALIZED:
 195       {
 196         ok(IsWindowEnabled(parenthwnd) == 0, "parent window should be disabled\n");
 197         PostQuitMessage(0);
 198         return FALSE;
 199       }
 200     }
 201     return FALSE;
 202 }
 203
 204 static void register_parent_wnd_class(void)
 205 {
 206     WNDCLASSA cls;
 207
 208     cls.style = 0;
 209     cls.lpfnWndProc = DefWindowProcA;
 210     cls.cbClsExtra = 0;
 211     cls.cbWndExtra = 0;
 212     cls.hInstance = GetModuleHandleA(NULL);
 213     cls.hIcon = 0;
 214     cls.hCursor = LoadCursorA(0, (LPCSTR)IDC_ARROW);
 215     cls.hbrBackground = GetStockObject(WHITE_BRUSH);
 216     cls.lpszMenuName = NULL;
 217     cls.lpszClassName = "parent class";
 218     RegisterClassA(&cls);
 219 }
 220
 221 static void test_disableowner(void)
 222 {
 223     HPROPSHEETPAGE hpsp[1];
 224     PROPSHEETPAGEA psp;
 225     PROPSHEETHEADERA psh;
 226     INT_PTR p;
 227
 228     register_parent_wnd_class();
 229     parenthwnd = CreateWindowA("parent class", "", WS_CAPTION | WS_SYSMENU | WS_VISIBLE, 100, 100, 100, 100, GetDesktopWindow(), NULL, GetModuleHandleA(NULL), 0);
 230
 231     memset(&psp, 0, sizeof(psp));
 232     psp.dwSize = sizeof(psp);
 233     psp.dwFlags = 0;
 234     psp.hInstance = GetModuleHandleA(NULL);
 235     U(psp).pszTemplate = "prop_page1";
 236     U2(psp).pszIcon = NULL;
 237     psp.pfnDlgProc = NULL;
 238     psp.lParam = 0;
 239
 240     hpsp[0] = CreatePropertySheetPageA(&psp);
 241
 242     memset(&psh, 0, sizeof(psh));
 243     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 244     psh.dwFlags = PSH_USECALLBACK;
 245     psh.pszCaption = "test caption";
 246     psh.nPages = 1;
 247     psh.hwndParent = parenthwnd;
 248     U3(psh).phpage = hpsp;
 249     psh.pfnCallback = disableowner_callback;
 250
 251     p = PropertySheetA(&psh);
 252     todo_wine
 253     ok(p == 0, "Expected 0, got %ld\n", p);
 254     ok(IsWindowEnabled(parenthwnd) != 0, "parent window should be enabled\n");
 255     DestroyWindow(parenthwnd);
 256 }
 257
 258 static INT_PTR CALLBACK nav_page_proc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
 259 {
 260     switch(msg){
 261     case WM_NOTIFY:
 262         {
 263             LPNMHDR hdr = (LPNMHDR)lparam;
 264             switch(hdr->code){
 265             case PSN_SETACTIVE:
 266                 active_page = /* PropSheet_HwndToIndex(hdr->hwndFrom, hwnd); */
 267                     (int)SendMessageA(hdr->hwndFrom, PSM_HWNDTOINDEX, (WPARAM)hwnd, 0);
 268                 return TRUE;
 269             case PSN_KILLACTIVE:
 270                 /* prevent navigation away from the fourth page */
 271                 if(active_page == 3){
 272                     SetWindowLongPtrA(hwnd, DWLP_MSGRESULT, TRUE);
 273                     return TRUE;
 274                 }
 275             }
 276             break;
 277         }
 278     }
 279     return FALSE;
 280 }
 281
 282 static void test_wiznavigation(void)
 283 {
 284     HPROPSHEETPAGE hpsp[4];
 285     PROPSHEETPAGEA psp[4];
 286     PROPSHEETHEADERA psh;
 287     HWND hdlg, control;
 288     LONG_PTR controlID;
 289     DWORD style;
 290     LRESULT defidres;
 291     BOOL hwndtoindex_supported = TRUE;
 292     const INT nextID = 12324;
 293     const INT backID = 12323;
 294
 295     /* create the property sheet pages */
 296     memset(psp, 0, sizeof(PROPSHEETPAGEA) * 4);
 297
 298     psp[0].dwSize = sizeof(PROPSHEETPAGEA);
 299     psp[0].hInstance = GetModuleHandleA(NULL);
 300     U(psp[0]).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_INTRO);
 301     psp[0].pfnDlgProc = nav_page_proc;
 302     hpsp[0] = CreatePropertySheetPageA(&psp[0]);
 303
 304     psp[1].dwSize = sizeof(PROPSHEETPAGEA);
 305     psp[1].hInstance = GetModuleHandleA(NULL);
 306     U(psp[1]).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_EDIT);
 307     psp[1].pfnDlgProc = nav_page_proc;
 308     hpsp[1] = CreatePropertySheetPageA(&psp[1]);
 309
 310     psp[2].dwSize = sizeof(PROPSHEETPAGEA);
 311     psp[2].hInstance = GetModuleHandleA(NULL);
 312     U(psp[2]).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_RADIO);
 313     psp[2].pfnDlgProc = nav_page_proc;
 314     hpsp[2] = CreatePropertySheetPageA(&psp[2]);
 315
 316     psp[3].dwSize = sizeof(PROPSHEETPAGEA);
 317     psp[3].hInstance = GetModuleHandleA(NULL);
 318     U(psp[3]).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_EXIT);
 319     psp[3].pfnDlgProc = nav_page_proc;
 320     hpsp[3] = CreatePropertySheetPageA(&psp[3]);
 321
 322     /* set up the property sheet dialog */
 323     memset(&psh, 0, sizeof(psh));
 324     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 325     psh.dwFlags = PSH_MODELESS | PSH_WIZARD;
 326     psh.pszCaption = "A Wizard";
 327     psh.nPages = 4;
 328     psh.hwndParent = GetDesktopWindow();
 329     U3(psh).phpage = hpsp;
 330     hdlg = (HWND)PropertySheetA(&psh);
 331     ok(hdlg != INVALID_HANDLE_VALUE, "got invalid handle %p\n", hdlg);
 332
 333     ok(active_page == 0, "Active page should be 0. Is: %d\n", active_page);
 334
 335     style = GetWindowLongA(hdlg, GWL_STYLE) & ~(DS_CONTEXTHELP|WS_SYSMENU);
 336     ok(style == (WS_POPUP|WS_VISIBLE|WS_CLIPSIBLINGS|WS_CAPTION|
 337                  DS_MODALFRAME|DS_SETFONT|DS_3DLOOK),
 338        "got unexpected style: %x\n", style);
 339
 340     control = GetFocus();
 341     controlID = GetWindowLongPtrA(control, GWLP_ID);
 342     ok(controlID == nextID, "Focus should have been set to the Next button. Expected: %d, Found: %ld\n", nextID, controlID);
 343
 344     /* simulate pressing the Next button */
 345     SendMessageA(hdlg, PSM_PRESSBUTTON, PSBTN_NEXT, 0);
 346     if (!active_page) hwndtoindex_supported = FALSE;
 347     if (hwndtoindex_supported)
 348         ok(active_page == 1, "Active page should be 1 after pressing Next. Is: %d\n", active_page);
 349
 350     control = GetFocus();
 351     controlID = GetWindowLongPtrA(control, GWLP_ID);
 352     ok(controlID == IDC_PS_EDIT1, "Focus should be set to the first item on the second page. Expected: %d, Found: %ld\n", IDC_PS_EDIT1, controlID);
 353
 354     defidres = SendMessageA(hdlg, DM_GETDEFID, 0, 0);
 355     ok(defidres == MAKELRESULT(nextID, DC_HASDEFID), "Expected default button ID to be %d, is %d\n", nextID, LOWORD(defidres));
 356
 357     /* set the focus to the second edit box on this page */
 358     SetFocus(GetNextDlgTabItem(hdlg, control, FALSE));
 359
 360     /* press next again */
 361     SendMessageA(hdlg, PSM_PRESSBUTTON, PSBTN_NEXT, 0);
 362     if (hwndtoindex_supported)
 363         ok(active_page == 2, "Active page should be 2 after pressing Next. Is: %d\n", active_page);
 364
 365     control = GetFocus();
 366     controlID = GetWindowLongPtrA(control, GWLP_ID);
 367     ok(controlID == IDC_PS_RADIO1, "Focus should have been set to item on third page. Expected: %d, Found %ld\n", IDC_PS_RADIO1, controlID);
 368
 369     /* back button */
 370     SendMessageA(hdlg, PSM_PRESSBUTTON, PSBTN_BACK, 0);
 371     if (hwndtoindex_supported)
 372         ok(active_page == 1, "Active page should be 1 after pressing Back. Is: %d\n", active_page);
 373
 374     control = GetFocus();
 375     controlID = GetWindowLongPtrA(control, GWLP_ID);
 376     ok(controlID == IDC_PS_EDIT1, "Focus should have been set to the first item on second page. Expected: %d, Found %ld\n", IDC_PS_EDIT1, controlID);
 377
 378     defidres = SendMessageA(hdlg, DM_GETDEFID, 0, 0);
 379     ok(defidres == MAKELRESULT(backID, DC_HASDEFID), "Expected default button ID to be %d, is %d\n", backID, LOWORD(defidres));
 380
 381     /* press next twice */
 382     SendMessageA(hdlg, PSM_PRESSBUTTON, PSBTN_NEXT, 0);
 383     if (hwndtoindex_supported)
 384         ok(active_page == 2, "Active page should be 2 after pressing Next. Is: %d\n", active_page);
 385     SendMessageA(hdlg, PSM_PRESSBUTTON, PSBTN_NEXT, 0);
 386     if (hwndtoindex_supported)
 387         ok(active_page == 3, "Active page should be 3 after pressing Next. Is: %d\n", active_page);
 388     else
 389         active_page = 3;
 390
 391     control = GetFocus();
 392     controlID = GetWindowLongPtrA(control, GWLP_ID);
 393     ok(controlID == nextID, "Focus should have been set to the Next button. Expected: %d, Found: %ld\n", nextID, controlID);
 394
 395     /* try to navigate away, but shouldn't be able to */
 396     SendMessageA(hdlg, PSM_PRESSBUTTON, PSBTN_BACK, 0);
 397     ok(active_page == 3, "Active page should still be 3 after pressing Back. Is: %d\n", active_page);
 398
 399     defidres = SendMessageA(hdlg, DM_GETDEFID, 0, 0);
 400     ok(defidres == MAKELRESULT(nextID, DC_HASDEFID), "Expected default button ID to be %d, is %d\n", nextID, LOWORD(defidres));
 401
 402     DestroyWindow(hdlg);
 403 }
 404
 405 static void test_buttons(void)
 406 {
 407     HPROPSHEETPAGE hpsp[1];
 408     PROPSHEETPAGEA psp;
 409     PROPSHEETHEADERA psh;
 410     HWND hdlg;
 411     HWND button;
 412     RECT rc;
 413     int prevRight, top;
 414
 415     memset(&psp, 0, sizeof(psp));
 416     psp.dwSize = sizeof(psp);
 417     psp.dwFlags = 0;
 418     psp.hInstance = GetModuleHandleA(NULL);
 419     U(psp).pszTemplate = "prop_page1";
 420     U2(psp).pszIcon = NULL;
 421     psp.pfnDlgProc = page_dlg_proc;
 422     psp.lParam = 0;
 423
 424     hpsp[0] = CreatePropertySheetPageA(&psp);
 425
 426     memset(&psh, 0, sizeof(psh));
 427     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 428     psh.dwFlags = PSH_MODELESS | PSH_USECALLBACK;
 429     psh.pszCaption = "test caption";
 430     psh.nPages = 1;
 431     psh.hwndParent = GetDesktopWindow();
 432     U3(psh).phpage = hpsp;
 433     psh.pfnCallback = sheet_callback;
 434
 435     hdlg = (HWND)PropertySheetA(&psh);
 436     ok(hdlg != INVALID_HANDLE_VALUE, "got null handle\n");
 437
 438     /* OK button */
 439     button = GetDlgItem(hdlg, IDOK);
 440     GetWindowRect(button, &rc);
 441     prevRight = rc.right;
 442     top = rc.top;
 443
 444     /* Cancel button */
 445     button = GetDlgItem(hdlg, IDCANCEL);
 446     GetWindowRect(button, &rc);
 447     ok(rc.top == top, "Cancel button should have same top as OK button\n");
 448     ok(rc.left > prevRight, "Cancel button should be to the right of OK button\n");
 449     prevRight = rc.right;
 450
 451     button = GetDlgItem(hdlg, IDC_APPLY_BUTTON);
 452     GetWindowRect(button, &rc);
 453     ok(rc.top == top, "Apply button should have same top as OK button\n");
 454     ok(rc.left > prevRight, "Apply button should be to the right of Cancel button\n");
 455     prevRight = rc.right;
 456
 457     button = GetDlgItem(hdlg, IDHELP);
 458     GetWindowRect(button, &rc);
 459     ok(rc.top == top, "Help button should have same top as OK button\n");
 460     ok(rc.left > prevRight, "Help button should be to the right of Apply button\n");
 461
 462     DestroyWindow(hdlg);
 463 }
 464
 465 static BOOL add_button_has_been_pressed;
 466
 467 static INT_PTR CALLBACK
 468 page_with_custom_default_button_dlg_proc(HWND hdlg, UINT msg, WPARAM wparam, LPARAM lparam)
 469 {
 470     switch (msg)
 471     {
 472     case WM_COMMAND:
 473         switch(LOWORD(wparam))
 474         {
 475         case IDC_PS_PUSHBUTTON1:
 476             switch(HIWORD(wparam))
 477             {
 478             case BN_CLICKED:
 479                 add_button_has_been_pressed = TRUE;
 480                 return TRUE;
 481             }
 482             break;
 483         }
 484         break;
 485     }
 486     return FALSE;
 487 }
 488
 489 static void test_custom_default_button(void)
 490 {
 491     HWND hdlg;
 492     PROPSHEETPAGEA psp[1];
 493     PROPSHEETHEADERA psh;
 494     MSG msg;
 495     LRESULT result;
 496
 497     psp[0].dwSize = sizeof (PROPSHEETPAGEA);
 498     psp[0].dwFlags = PSP_USETITLE;
 499     psp[0].hInstance = GetModuleHandleA(NULL);
 500     U(psp[0]).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_WITH_CUSTOM_DEFAULT_BUTTON);
 501     U2(psp[0]).pszIcon = NULL;
 502     psp[0].pfnDlgProc = page_with_custom_default_button_dlg_proc;
 503     psp[0].pszTitle = "Page1";
 504     psp[0].lParam = 0;
 505
 506     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 507     psh.dwFlags = PSH_PROPSHEETPAGE | PSH_MODELESS;
 508     psh.hwndParent = GetDesktopWindow();
 509     psh.hInstance = GetModuleHandleA(NULL);
 510     U(psh).pszIcon = NULL;
 511     psh.pszCaption =  "PropertySheet1";
 512     psh.nPages = 1;
 513     U3(psh).ppsp = psp;
 514     U2(psh).nStartPage = 0;
 515
 516     /* The goal of the test is to make sure that the Add button is pressed
 517      * when the ENTER key is pressed and a different control, a combobox,
 518      * has the keyboard focus. */
 519     add_button_has_been_pressed = FALSE;
 520
 521     /* Create the modeless property sheet. */
 522     hdlg = (HWND)PropertySheetA(&psh);
 523     ok(hdlg != INVALID_HANDLE_VALUE, "Cannot create the property sheet\n");
 524
 525     /* Set the Add button as the default button. */
 526     SendMessageA(hdlg, DM_SETDEFID, (WPARAM)IDC_PS_PUSHBUTTON1, 0);
 527
 528     /* Make sure the default button is the Add button. */
 529     result = SendMessageA(hdlg, DM_GETDEFID, 0, 0);
 530     ok(DC_HASDEFID == HIWORD(result), "The property sheet does not have a default button\n");
 531     ok(IDC_PS_PUSHBUTTON1 == LOWORD(result), "The default button is not the Add button\n");
 532
 533     /* At this point, the combobox should have keyboard focus, so we press ENTER.
 534      * Pull the lever, Kronk! */
 535     keybd_event(VK_RETURN, 0, 0, 0);
 536
 537     /* Process all the messages in the queue for this thread. */
 538     while (PeekMessageA(&msg, NULL, 0, 0, PM_REMOVE))
 539     {
 540         /* (!PropSheet_IsDialogMessage(hdlg, &msg)) */
 541         if (!((BOOL)SendMessageA(hdlg, PSM_ISDIALOGMESSAGE, 0, (LPARAM)&msg)))
 542         {
 543             TranslateMessage(&msg);
 544             DispatchMessageA(&msg);
 545         }
 546     }
 547
 548     ok(add_button_has_been_pressed, "The Add button has not been pressed!\n");
 549
 550     DestroyWindow(hdlg);
 551 }
 552
 553 #define RECEIVER_SHEET_CALLBACK 0
 554 #define RECEIVER_SHEET_WINPROC  1
 555 #define RECEIVER_PAGE           2
 556
 557 #define NUM_MSG_SEQUENCES   1
 558 #define PROPSHEET_SEQ_INDEX 0
 559
 560 static struct msg_sequence *sequences[NUM_MSG_SEQUENCES];
 561 static WNDPROC oldWndProc;
 562
 563 static const struct message property_sheet_seq[] = {
 564     { PSCB_PRECREATE, sent|id, 0, 0, RECEIVER_SHEET_CALLBACK },
 565     { PSCB_INITIALIZED, sent|id, 0, 0, RECEIVER_SHEET_CALLBACK },
 566     { WM_WINDOWPOSCHANGING, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 567     /*{ WM_NCCALCSIZE, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 568     { WM_WINDOWPOSCHANGED, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 569     { WM_MOVE, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 570     { WM_SIZE, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 571     /*{ WM_GETTEXT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 572     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 573     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 574     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 575     { WM_NCCALCSIZE, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 576     { DM_REPOSITION, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 577     { WM_WINDOWPOSCHANGING, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 578     { WM_WINDOWPOSCHANGING, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 579     { WM_ACTIVATEAPP, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 580     /*{ WM_NCACTIVATE, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 581     { WM_GETTEXT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 582     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 583     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 584     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 585     { WM_GETTEXT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 586     { WM_ACTIVATE, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 587     /*{ WM_IME_SETCONTEXT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 588     { WM_IME_NOTIFY, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 589     { WM_SETFOCUS, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 590     { WM_KILLFOCUS, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 591     /*{ WM_IME_SETCONTEXT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 592     { WM_PARENTNOTIFY, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 593     { WM_INITDIALOG, sent|id, 0, 0, RECEIVER_PAGE },
 594     { WM_WINDOWPOSCHANGING, sent|id, 0, 0, RECEIVER_PAGE },
 595     /*{ WM_NCCALCSIZE, sent|id, 0, 0, RECEIVER_PAGE },*/
 596     { WM_CHILDACTIVATE, sent|id, 0, 0, RECEIVER_PAGE },
 597     { WM_WINDOWPOSCHANGED, sent|id, 0, 0, RECEIVER_PAGE },
 598     { WM_MOVE, sent|id, 0, 0, RECEIVER_PAGE },
 599     { WM_SIZE, sent|id, 0, 0, RECEIVER_PAGE },
 600     { WM_NOTIFY, sent|id, 0, 0, RECEIVER_PAGE },
 601     { WM_STYLECHANGING, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 602     { WM_STYLECHANGED, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 603     /*{ WM_GETTEXT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 604     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 605     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 606     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 607     { WM_SETTEXT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 608     { WM_SHOWWINDOW, sent|id, 0, 0, RECEIVER_PAGE },
 609     /*{ 0x00000401, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 610     { 0x00000400, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 611     { WM_CHANGEUISTATE, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 612     { WM_UPDATEUISTATE, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 613     { WM_UPDATEUISTATE, sent|id|optional, 0, 0, RECEIVER_PAGE },
 614     { WM_SHOWWINDOW, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 615     { WM_WINDOWPOSCHANGING, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 616     /*{ WM_NCPAINT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 617     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 618     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 619     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 620     { WM_ERASEBKGND, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 621     { WM_CTLCOLORDLG, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 622     { WM_WINDOWPOSCHANGED, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 623     /*{ WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 624     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 625     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 626     { WM_PAINT, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 627     { WM_PAINT, sent|id, 0, 0, RECEIVER_PAGE },
 628     { WM_NCPAINT, sent|id, 0, 0, RECEIVER_PAGE },
 629     { WM_ERASEBKGND, sent|id, 0, 0, RECEIVER_PAGE },*/
 630     { WM_CTLCOLORDLG, sent|id, 0, 0, RECEIVER_PAGE },
 631     { WM_CTLCOLORSTATIC, sent|id, 0, 0, RECEIVER_PAGE },
 632     { WM_CTLCOLORSTATIC, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 633     { WM_CTLCOLORBTN, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 634     { WM_CTLCOLORBTN, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 635     { WM_CTLCOLORBTN, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 636     /*{ WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 637     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 638     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 639     { WM_COMMAND, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 640     { WM_NOTIFY, sent|id|optional, 0, 0, RECEIVER_PAGE },
 641     { WM_NOTIFY, sent|id|optional, 0, 0, RECEIVER_PAGE },
 642     { WM_WINDOWPOSCHANGING, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 643     { WM_WINDOWPOSCHANGED, sent|id|optional, 0, 0, RECEIVER_SHEET_WINPROC },
 644     /*{ WM_NCACTIVATE, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 645     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 646     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 647     { WM_GETICON, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 648     /*{ WM_ACTIVATE, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 649     { WM_ACTIVATE, sent|id, 0, 0, RECEIVER_PAGE },
 650     { WM_ACTIVATEAPP, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 651     { WM_ACTIVATEAPP, sent|id, 0, 0, RECEIVER_PAGE },
 652     { WM_DESTROY, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },
 653     { WM_DESTROY, sent|id, 0, 0, RECEIVER_PAGE },*/
 654     /*{ WM_NCDESTROY, sent|id, 0, 0, RECEIVER_PAGE },
 655     { WM_NCDESTROY, sent|id, 0, 0, RECEIVER_SHEET_WINPROC },*/
 656     { 0 }
 657 };
 658
 659 static void save_message(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam, INT receiver)
 660 {
 661     struct message msg;
 662
 663     if (message < WM_USER &&
 664         message != WM_GETICON &&
 665         message != WM_GETTEXT &&
 666         message != WM_IME_SETCONTEXT &&
 667         message != WM_IME_NOTIFY &&
 668         message != WM_PAINT &&
 669         message != WM_ERASEBKGND &&
 670         message != WM_SETCURSOR &&
 671         (message < WM_NCCREATE || message > WM_NCMBUTTONDBLCLK) &&
 672         (message < WM_MOUSEFIRST || message > WM_MOUSEHWHEEL) &&
 673         message != 0x90)
 674     {
 675         msg.message = message;
 676         msg.flags = sent|wparam|lparam|id;
 677         msg.wParam = wParam;
 678         msg.lParam = lParam;
 679         msg.id = receiver;
 680         add_message(sequences, PROPSHEET_SEQ_INDEX, &msg);
 681     }
 682 }
 683
 684 static LRESULT CALLBACK sheet_callback_messages_proc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
 685 {
 686     save_message(hwnd, msg, wParam, lParam, RECEIVER_SHEET_WINPROC);
 687
 688     return CallWindowProcA(oldWndProc, hwnd, msg, wParam, lParam);
 689 }
 690
 691 static int CALLBACK sheet_callback_messages(HWND hwnd, UINT msg, LPARAM lParam)
 692 {
 693     save_message(hwnd, msg, 0, lParam, RECEIVER_SHEET_CALLBACK);
 694
 695     switch (msg)
 696     {
 697     case PSCB_INITIALIZED:
 698         oldWndProc = (WNDPROC)GetWindowLongPtrA(hwnd, GWLP_WNDPROC);
 699         SetWindowLongPtrA(hwnd, GWLP_WNDPROC, (LONG_PTR)&sheet_callback_messages_proc);
 700         return TRUE;
 701     }
 702
 703     return TRUE;
 704 }
 705
 706 static INT_PTR CALLBACK page_dlg_proc_messages(HWND hwnd, UINT msg, WPARAM wParam,
 707                                                LPARAM lParam)
 708 {
 709     save_message(hwnd, msg, wParam, lParam, RECEIVER_PAGE);
 710
 711     return FALSE;
 712 }
 713
 714 static void test_messages(void)
 715 {
 716     HPROPSHEETPAGE hpsp[1];
 717     PROPSHEETPAGEA psp;
 718     PROPSHEETHEADERA psh;
 719     HWND hdlg;
 720
 721     init_msg_sequences(sequences, NUM_MSG_SEQUENCES);
 722
 723     memset(&psp, 0, sizeof(psp));
 724     psp.dwSize = sizeof(psp);
 725     psp.dwFlags = 0;
 726     psp.hInstance = GetModuleHandleA(NULL);
 727     U(psp).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_MESSAGE_TEST);
 728     U2(psp).pszIcon = NULL;
 729     psp.pfnDlgProc = page_dlg_proc_messages;
 730     psp.lParam = 0;
 731
 732     hpsp[0] = CreatePropertySheetPageA(&psp);
 733
 734     memset(&psh, 0, sizeof(psh));
 735     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 736     psh.dwFlags = PSH_NOAPPLYNOW | PSH_WIZARD | PSH_USECALLBACK
 737                   | PSH_MODELESS | PSH_USEICONID;
 738     psh.pszCaption = "test caption";
 739     psh.nPages = 1;
 740     psh.hwndParent = GetDesktopWindow();
 741     U3(psh).phpage = hpsp;
 742     psh.pfnCallback = sheet_callback_messages;
 743
 744     hdlg = (HWND)PropertySheetA(&psh);
 745     ok(hdlg != INVALID_HANDLE_VALUE, "got invalid handle %p\n", hdlg);
 746
 747     ShowWindow(hdlg,SW_NORMAL);
 748
 749     ok_sequence(sequences, PROPSHEET_SEQ_INDEX, property_sheet_seq, "property sheet with custom window proc", TRUE);
 750
 751     DestroyWindow(hdlg);
 752 }
 753
 754 static void test_PSM_ADDPAGE(void)
 755 {
 756     HPROPSHEETPAGE hpsp[5];
 757     PROPSHEETPAGEA psp;
 758     PROPSHEETHEADERA psh;
 759     HWND hdlg, tab;
 760     BOOL ret;
 761     DWORD r;
 762
 763     memset(&psp, 0, sizeof(psp));
 764     psp.dwSize = sizeof(psp);
 765     psp.dwFlags = 0;
 766     psp.hInstance = GetModuleHandleA(NULL);
 767     U(psp).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_MESSAGE_TEST);
 768     U2(psp).pszIcon = NULL;
 769     psp.pfnDlgProc = page_dlg_proc_messages;
 770     psp.lParam = 0;
 771
 772     /* two page with the same data */
 773     hpsp[0] = CreatePropertySheetPageA(&psp);
 774     hpsp[1] = CreatePropertySheetPageA(&psp);
 775     hpsp[2] = CreatePropertySheetPageA(&psp);
 776
 777     U(psp).pszTemplate = (LPCSTR)MAKEINTRESOURCE(IDD_PROP_PAGE_ERROR);
 778     hpsp[3] = CreatePropertySheetPageA(&psp);
 779
 780     psp.dwFlags = PSP_PREMATURE;
 781     hpsp[4] = CreatePropertySheetPageA(&psp);
 782
 783     memset(&psh, 0, sizeof(psh));
 784     psh.dwSize = PROPSHEETHEADERA_V1_SIZE;
 785     psh.dwFlags = PSH_MODELESS;
 786     psh.pszCaption = "test caption";
 787     psh.nPages = 1;
 788     psh.hwndParent = GetDesktopWindow();
 789     U3(psh).phpage = hpsp;
 790
 791     hdlg = (HWND)PropertySheetA(&psh);
 792     ok(hdlg != INVALID_HANDLE_VALUE, "got invalid handle %p\n", hdlg);
 793
 794     /* add pages one by one */
 795     ret = SendMessageA(hdlg, PSM_ADDPAGE, 0, (LPARAM)hpsp[1]);
 796     ok(ret == TRUE, "got %d\n", ret);
 797
 798     /* try with null and invalid value */
 799     ret = SendMessageA(hdlg, PSM_ADDPAGE, 0, 0);
 800     ok(ret == FALSE, "got %d\n", ret);
 801
 802 if (0)
 803 {
 804     /* crashes on native */
 805     ret = SendMessageA(hdlg, PSM_ADDPAGE, 0, (LPARAM)INVALID_HANDLE_VALUE);
 806 }
 807     /* check item count */
 808     tab = (HWND)SendMessageA(hdlg, PSM_GETTABCONTROL, 0, 0);
 809
 810     r = SendMessageA(tab, TCM_GETITEMCOUNT, 0, 0);
 811     ok(r == 2, "got %d\n", r);
 812
 813     ret = SendMessageA(hdlg, PSM_ADDPAGE, 0, (LPARAM)hpsp[2]);
 814     ok(ret == TRUE, "got %d\n", ret);
 815
 816     r = SendMessageA(tab, TCM_GETITEMCOUNT, 0, 0);
 817     ok(r == 3, "got %d\n", r);
 818
 819     /* add property sheet page that can't be created */
 820     ret = SendMessageA(hdlg, PSM_ADDPAGE, 0, (LPARAM)hpsp[3]);
 821     ok(ret == TRUE, "got %d\n", ret);
 822
 823     r = SendMessageA(tab, TCM_GETITEMCOUNT, 0, 0);
 824     ok(r == 4, "got %d\n", r);
 825
 826     /* select page that can't be created */
 827     ret = SendMessageA(hdlg, PSM_SETCURSEL, 3, 1);
 828     ok(ret == TRUE, "got %d\n", ret);
 829
 830     r = SendMessageA(tab, TCM_GETITEMCOUNT, 0, 0);
 831     ok(r == 3, "got %d\n", r);
 832
 833     /* test PSP_PREMATURE flag with incorrect property sheet page */
 834     ret = SendMessageA(hdlg, PSM_ADDPAGE, 0, (LPARAM)hpsp[4]);
 835     ok(ret == FALSE, "got %d\n", ret);
 836
 837     r = SendMessageA(tab, TCM_GETITEMCOUNT, 0, 0);
 838     ok(r == 3, "got %d\n", r);
 839
 840     DestroyWindow(hdlg);
 841 }
 842
 843 START_TEST(propsheet)
 844 {
 845     test_title();
 846     test_nopage();
 847     test_disableowner();
 848     test_wiznavigation();
 849     test_buttons();
 850     test_custom_default_button();
 851     test_messages();
 852     test_PSM_ADDPAGE();
 853 }