284 lines
11 KiB
C
284 lines
11 KiB
C
/*
|
|
* Copyright 2022 Piotr Caban
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <windows.h>
|
|
#include <prsht.h>
|
|
#include <ddk/compstui.h>
|
|
|
|
#include "wine/test.h"
|
|
|
|
#define DEFINE_EXPECT(func) \
|
|
static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
|
|
|
|
#define SET_EXPECT(func) \
|
|
do { called_ ## func = FALSE; expect_ ## func = TRUE; } while(0)
|
|
|
|
#define CHECK_EXPECT2(func) \
|
|
do { \
|
|
ok(expect_ ##func, "unexpected call " #func "\n"); \
|
|
called_ ## func = TRUE; \
|
|
}while(0)
|
|
|
|
#define CHECK_EXPECT(func) \
|
|
do { \
|
|
CHECK_EXPECT2(func); \
|
|
expect_ ## func = FALSE; \
|
|
}while(0)
|
|
|
|
#define CHECK_CALLED(func) \
|
|
do { \
|
|
ok(called_ ## func, "expected " #func "\n"); \
|
|
expect_ ## func = called_ ## func = FALSE; \
|
|
}while(0)
|
|
|
|
#define CLEAR_CALLED(func) \
|
|
expect_ ## func = called_ ## func = FALSE
|
|
|
|
DEFINE_EXPECT(page_proc_WM_INITDIALOG);
|
|
DEFINE_EXPECT(page_proc2_WM_INITDIALOG);
|
|
DEFINE_EXPECT(device_PROPSHEETUI_REASON_INIT);
|
|
DEFINE_EXPECT(device_PROPSHEETUI_REASON_DESTROY);
|
|
DEFINE_EXPECT(callback_PROPSHEETUI_REASON_BEFORE_INIT);
|
|
DEFINE_EXPECT(callback_PROPSHEETUI_REASON_INIT);
|
|
DEFINE_EXPECT(callback_PROPSHEETUI_REASON_GET_INFO_HEADER);
|
|
DEFINE_EXPECT(callback_PROPSHEETUI_REASON_DESTROY);
|
|
|
|
typedef struct {
|
|
WORD cbSize;
|
|
WORD Flags;
|
|
LPWSTR pTitle;
|
|
HWND hWndParent;
|
|
HINSTANCE hInst;
|
|
union {
|
|
HICON hIcon;
|
|
ULONG_PTR IconID;
|
|
};
|
|
} PROPSHEETUI_INFO_HEADERW;
|
|
|
|
static INT_PTR CALLBACK prop_page_proc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
|
|
{
|
|
if (msg == WM_INITDIALOG)
|
|
{
|
|
HWND dlg = GetParent(hwnd);
|
|
|
|
CHECK_EXPECT(page_proc_WM_INITDIALOG);
|
|
|
|
PostMessageW(dlg, PSM_PRESSBUTTON, PSBTN_OK, 0);
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
static INT_PTR CALLBACK prop_page_proc2(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
|
|
{
|
|
if (msg == WM_INITDIALOG)
|
|
{
|
|
PROPSHEETPAGEW *psp = (PROPSHEETPAGEW*)lparam;
|
|
PSPINFO *info = (PSPINFO*)((BYTE*)lparam + psp->dwSize);
|
|
HWND dlg = GetParent(hwnd);
|
|
|
|
CHECK_EXPECT(page_proc2_WM_INITDIALOG);
|
|
|
|
ok(psp->dwSize == sizeof(PROPSHEETPAGEW), "psp->dwSize = %ld\n", psp->dwSize);
|
|
ok(psp->pfnDlgProc == prop_page_proc2, "psp->pfnDlgProc != prop_page_proc2\n");
|
|
ok(!psp->lParam, "psp->lParam = %Ix\n", psp->lParam);
|
|
|
|
ok(info->cbSize == sizeof(*info), "info->cbSize = %hd\n", info->cbSize);
|
|
ok(!info->wReserved, "info->wReserved = %hd\n", info->wReserved);
|
|
ok(info->hComPropSheet != NULL, "info->hComPropSheet = NULL\n");
|
|
ok(info->hCPSUIPage != NULL, "info->hCPSUIPage = NULL\n");
|
|
ok(info->pfnComPropSheet != NULL, "info->pfnComPropSheet = NULL\n");
|
|
|
|
PostMessageW(dlg, PSM_SETCURSEL, 1, 0);
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
static LONG WINAPI device_property_sheets(PROPSHEETUI_INFO *info, LPARAM lparam)
|
|
{
|
|
if (info->Reason == PROPSHEETUI_REASON_INIT)
|
|
CHECK_EXPECT(device_PROPSHEETUI_REASON_INIT);
|
|
else
|
|
CHECK_EXPECT(device_PROPSHEETUI_REASON_DESTROY);
|
|
|
|
ok(info->cbSize == sizeof(*info), "info->cbSize = %hd\n", info->cbSize);
|
|
ok(info->Version == PROPSHEETUI_INFO_VERSION, "info->Version = %hd\n", info->Version);
|
|
ok(info->Flags == info->lParamInit ? PSUIINFO_UNICODE : 0, "info->Flags = %hd\n", info->Flags);
|
|
ok(info->Reason == PROPSHEETUI_REASON_INIT || info->Reason == PROPSHEETUI_REASON_DESTROY,
|
|
"info->Reason = %hx\n", info->Reason);
|
|
ok(info->hComPropSheet != NULL, "info->hComPropSheet = NULL\n");
|
|
ok(info->pfnComPropSheet != NULL, "info->pfnComPropSheet = NULL\n");
|
|
if (info->Reason == PROPSHEETUI_REASON_INIT)
|
|
{
|
|
ok(info->lParamInit == lparam, "info->lParamInit = %Ix, lparam = %Ix\n",
|
|
info->lParamInit, lparam);
|
|
}
|
|
ok(!info->UserData, "info->UserData = %Ix\n", info->UserData);
|
|
ok(!info->Result, "info->Result = %Ix\n", info->Result);
|
|
return lparam;
|
|
}
|
|
|
|
#define ADD_PAGES 0x1
|
|
|
|
static LONG WINAPI propsheetui_callback(PROPSHEETUI_INFO *info, LPARAM lparam)
|
|
{
|
|
PROPSHEETUI_INFO_HEADERW *info_header;
|
|
|
|
ok(info->cbSize == sizeof(*info), "info->cbSize = %hd\n", info->cbSize);
|
|
|
|
switch(info->Reason)
|
|
{
|
|
case PROPSHEETUI_REASON_BEFORE_INIT:
|
|
CHECK_EXPECT(callback_PROPSHEETUI_REASON_BEFORE_INIT);
|
|
ok(!info->Version, "info->Version = %hd\n", info->Version);
|
|
ok(!info->Flags, "info->Flags = %hd\n", info->Flags);
|
|
ok(!info->hComPropSheet, "info->hComPropSheet = %p\n", info->hComPropSheet);
|
|
ok(!info->pfnComPropSheet, "info->pfnComPropSheet = %p\n", info->pfnComPropSheet);
|
|
ok(!info->Result, "info->Result = %Ix\n", info->Result);
|
|
break;
|
|
case PROPSHEETUI_REASON_INIT:
|
|
CHECK_EXPECT(callback_PROPSHEETUI_REASON_INIT);
|
|
ok(info->Version == PROPSHEETUI_INFO_VERSION, "info->Version = %hd\n", info->Version);
|
|
ok(info->Flags == PSUIINFO_UNICODE, "info->Flags = %hd\n", info->Flags);
|
|
ok(info->hComPropSheet != NULL, "info->hComPropSheet = NULL\n");
|
|
ok(info->pfnComPropSheet != NULL, "info->pfnComPropSheet = NULL\n");
|
|
ok(!info->UserData, "info->UserData = %Ix\n", info->UserData);
|
|
ok(!info->Result, "info->Result = %Ix\n", info->Result);
|
|
break;
|
|
case PROPSHEETUI_REASON_GET_INFO_HEADER:
|
|
CHECK_EXPECT(callback_PROPSHEETUI_REASON_GET_INFO_HEADER);
|
|
ok(info->Version == PROPSHEETUI_INFO_VERSION, "info->Version = %hd\n", info->Version);
|
|
ok(info->Flags == PSUIINFO_UNICODE, "info->Flags = %hd\n", info->Flags);
|
|
ok(info->hComPropSheet != NULL, "info->hComPropSheet = NULL\n");
|
|
ok(info->pfnComPropSheet != NULL, "info->pfnComPropSheet = NULL\n");
|
|
ok(!info->UserData, "info->UserData = %Ix\n", info->UserData);
|
|
ok(!info->Result, "info->Result = %Ix\n", info->Result);
|
|
|
|
info_header = (PROPSHEETUI_INFO_HEADERW*)lparam;
|
|
ok(info_header->cbSize == sizeof(*info_header), "info_header->cbSize = %hd\n", info_header->cbSize);
|
|
ok(!info_header->Flags, "info_header->Flags = %hx\n", info_header->Flags);
|
|
ok(!info_header->pTitle, "info_header->pTitle = %p\n", info_header->pTitle);
|
|
ok(info_header->hWndParent == (HWND)0x4321, "info_header->hWndParent = %p\n", info_header->hWndParent);
|
|
ok(!info_header->hInst, "info_header->hInst = %p\n", info_header->hInst);
|
|
|
|
info_header->hWndParent = 0;
|
|
|
|
if (info->lParamInit & ADD_PAGES)
|
|
{
|
|
HPROPSHEETPAGE hpsp;
|
|
PROPSHEETPAGEW psp;
|
|
LONG_PTR ret;
|
|
|
|
SET_EXPECT(device_PROPSHEETUI_REASON_INIT);
|
|
SET_EXPECT(device_PROPSHEETUI_REASON_DESTROY);
|
|
ret = info->pfnComPropSheet(info->hComPropSheet, CPSFUNC_ADD_PFNPROPSHEETUIA, (LPARAM)device_property_sheets, 0);
|
|
ok(!ret, "ret = %Ix\n", ret);
|
|
CHECK_CALLED(device_PROPSHEETUI_REASON_INIT);
|
|
CHECK_CALLED(device_PROPSHEETUI_REASON_DESTROY);
|
|
|
|
SET_EXPECT(device_PROPSHEETUI_REASON_INIT);
|
|
ret = info->pfnComPropSheet(info->hComPropSheet, CPSFUNC_ADD_PFNPROPSHEETUIW, (LPARAM)device_property_sheets, 1);
|
|
ok(ret, "ret = 0\n");
|
|
CHECK_CALLED(device_PROPSHEETUI_REASON_INIT);
|
|
SET_EXPECT(device_PROPSHEETUI_REASON_DESTROY);
|
|
|
|
memset(&psp, 0, sizeof(psp));
|
|
psp.dwSize = sizeof(psp);
|
|
psp.pszTemplate = L"prop_page1";
|
|
psp.pfnDlgProc = prop_page_proc2;
|
|
ret = info->pfnComPropSheet(info->hComPropSheet, CPSFUNC_ADD_PROPSHEETPAGEW, (LPARAM)&psp, 0);
|
|
ok(ret, "ret = 0\n");
|
|
|
|
psp.pfnDlgProc = prop_page_proc;
|
|
hpsp = CreatePropertySheetPageW(&psp);
|
|
ok(hpsp != NULL, "hpsp = %p\n", hpsp);
|
|
ret = info->pfnComPropSheet(info->hComPropSheet, CPSFUNC_ADD_HPROPSHEETPAGE, (LPARAM)hpsp, 0);
|
|
ok(ret, "ret = 0\n");
|
|
}
|
|
break;
|
|
case PROPSHEETUI_REASON_DESTROY:
|
|
CHECK_EXPECT(callback_PROPSHEETUI_REASON_DESTROY);
|
|
if (info->lParamInit & ADD_PAGES)
|
|
{
|
|
CHECK_CALLED(device_PROPSHEETUI_REASON_DESTROY);
|
|
ok(lparam, "lparam = 0\n");
|
|
}
|
|
else
|
|
ok(!lparam, "lparam = %Ix\n", lparam);
|
|
|
|
ok(info->Version == PROPSHEETUI_INFO_VERSION, "info->Version = %hd\n", info->Version);
|
|
ok(info->Flags == PSUIINFO_UNICODE, "info->Flags = %hd\n", info->Flags);
|
|
ok(info->hComPropSheet != NULL, "info->hComPropSheet = NULL\n");
|
|
ok(info->pfnComPropSheet != NULL, "info->pfnComPropSheet = NULL\n");
|
|
ok(!info->UserData, "info->UserData = %Ix\n", info->UserData);
|
|
ok(!info->Result, "info->Result = %Ix\n", info->Result);
|
|
break;
|
|
default:
|
|
ok(0, "unhandled info->Reason = %hd\n", info->Reason);
|
|
break;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
static void test_CommonPropertySheetUI(void)
|
|
{
|
|
DWORD res;
|
|
LONG ret;
|
|
|
|
SetLastError(0xdeadbeef);
|
|
res = 0xdeadbeef;
|
|
ret = CommonPropertySheetUIW(0, NULL, 0, &res);
|
|
ok(ret == ERR_CPSUI_GETLASTERROR, "ret = %ld\n", ret);
|
|
ok(res == 0xdeadbeef, "res = %lx\n", res);
|
|
ok(!GetLastError(), "CommonPropertySheetUIW error %ld\n", GetLastError());
|
|
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_BEFORE_INIT);
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_INIT);
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_GET_INFO_HEADER);
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_DESTROY);
|
|
res = 0xdeadbeef;
|
|
ret = CommonPropertySheetUIW((HWND)0x4321, propsheetui_callback, 0, &res);
|
|
ok(ret == ERR_CPSUI_NO_PROPSHEETPAGE, "CommonPropertySheetUIW returned %ld\n", ret);
|
|
ok(!res, "res = %lx\n", res);
|
|
CLEAR_CALLED(callback_PROPSHEETUI_REASON_BEFORE_INIT);
|
|
CHECK_CALLED(callback_PROPSHEETUI_REASON_INIT);
|
|
CHECK_CALLED(callback_PROPSHEETUI_REASON_GET_INFO_HEADER);
|
|
CHECK_CALLED(callback_PROPSHEETUI_REASON_DESTROY);
|
|
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_BEFORE_INIT);
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_INIT);
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_GET_INFO_HEADER);
|
|
SET_EXPECT(page_proc2_WM_INITDIALOG);
|
|
SET_EXPECT(page_proc_WM_INITDIALOG);
|
|
SET_EXPECT(callback_PROPSHEETUI_REASON_DESTROY);
|
|
res = 0xdeadbeef;
|
|
ret = CommonPropertySheetUIW((HWND)0x4321, propsheetui_callback, ADD_PAGES, &res);
|
|
ok(ret == CPSUI_OK, "CommonPropertySheetUIW returned %ld\n", ret);
|
|
ok(!res, "res = %lx\n", res);
|
|
CLEAR_CALLED(callback_PROPSHEETUI_REASON_BEFORE_INIT);
|
|
CHECK_CALLED(callback_PROPSHEETUI_REASON_INIT);
|
|
CHECK_CALLED(callback_PROPSHEETUI_REASON_GET_INFO_HEADER);
|
|
CHECK_CALLED(page_proc2_WM_INITDIALOG);
|
|
CHECK_CALLED(page_proc_WM_INITDIALOG);
|
|
CHECK_CALLED(callback_PROPSHEETUI_REASON_DESTROY);
|
|
}
|
|
|
|
START_TEST(compstui)
|
|
{
|
|
test_CommonPropertySheetUI();
|
|
}
|