uiautomationcore/tests: Add tests for ReservedValue COM interfaces.

Signed-off-by: Connor McAdams <cmcadams@codeweavers.com>
Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Connor McAdams 2021-11-04 15:14:13 -04:00 committed by Alexandre Julliard
parent 67add3f22a
commit 475dce071c
2 changed files with 134 additions and 1 deletions

View File

@ -1,5 +1,5 @@
TESTDLL = uiautomationcore.dll
IMPORTS = uiautomationcore user32
IMPORTS = uiautomationcore user32 ole32
C_SRCS = \
uiautomation.c

View File

@ -86,7 +86,140 @@ if (hr == S_OK)
UnregisterClassA("HostProviderFromHwnd class", NULL);
}
static DWORD WINAPI uia_reserved_val_iface_marshal_thread(LPVOID param)
{
IStream **stream = param;
IUnknown *unk_ns, *unk_ns2, *unk_ma, *unk_ma2;
HRESULT hr;
CoInitializeEx(NULL, COINIT_MULTITHREADED);
hr = CoGetInterfaceAndReleaseStream(stream[0], &IID_IUnknown, (void **)&unk_ns);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
hr = CoGetInterfaceAndReleaseStream(stream[1], &IID_IUnknown, (void **)&unk_ma);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
hr = UiaGetReservedNotSupportedValue(&unk_ns2);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
hr = UiaGetReservedMixedAttributeValue(&unk_ma2);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
ok(unk_ns2 == unk_ns, "UiaGetReservedNotSupported pointer mismatch, unk_ns2 %p, unk_ns %p\n", unk_ns2, unk_ns);
ok(unk_ma2 == unk_ma, "UiaGetReservedMixedAttribute pointer mismatch, unk_ma2 %p, unk_ma %p\n", unk_ma2, unk_ma);
CoUninitialize();
return 0;
}
static void test_uia_reserved_value_ifaces(void)
{
IUnknown *unk_ns, *unk_ns2, *unk_ma, *unk_ma2;
IStream *stream[2];
IMarshal *marshal;
HANDLE thread;
ULONG refcnt;
HRESULT hr;
/* ReservedNotSupportedValue. */
hr = UiaGetReservedNotSupportedValue(NULL);
ok(hr == E_INVALIDARG, "Unexpected hr %#x.\n", hr);
hr = UiaGetReservedNotSupportedValue(&unk_ns);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
ok(unk_ns != NULL, "UiaGetReservedNotSupportedValue returned NULL interface.\n");
refcnt = IUnknown_AddRef(unk_ns);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
refcnt = IUnknown_AddRef(unk_ns);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
refcnt = IUnknown_Release(unk_ns);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
hr = UiaGetReservedNotSupportedValue(&unk_ns2);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
ok(unk_ns2 != NULL, "UiaGetReservedNotSupportedValue returned NULL interface.");
ok(unk_ns2 == unk_ns, "UiaGetReservedNotSupported pointer mismatch, unk_ns2 %p, unk_ns %p\n", unk_ns2, unk_ns);
marshal = NULL;
hr = IUnknown_QueryInterface(unk_ns, &IID_IMarshal, (void **)&marshal);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
ok(marshal != NULL, "Failed to get IMarshal interface.\n");
refcnt = IMarshal_AddRef(marshal);
ok(refcnt == 2, "Expected refcnt %d, got %d\n", 2, refcnt);
refcnt = IMarshal_Release(marshal);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
refcnt = IMarshal_Release(marshal);
ok(refcnt == 0, "Expected refcnt %d, got %d\n", 0, refcnt);
/* ReservedMixedAttributeValue. */
hr = UiaGetReservedMixedAttributeValue(NULL);
ok(hr == E_INVALIDARG, "Unexpected hr %#x.\n", hr);
hr = UiaGetReservedMixedAttributeValue(&unk_ma);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
ok(unk_ma != NULL, "UiaGetReservedMixedAttributeValue returned NULL interface.");
refcnt = IUnknown_AddRef(unk_ma);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
refcnt = IUnknown_AddRef(unk_ma);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
refcnt = IUnknown_Release(unk_ma);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
hr = UiaGetReservedMixedAttributeValue(&unk_ma2);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
ok(unk_ma2 != NULL, "UiaGetReservedMixedAttributeValue returned NULL interface.");
ok(unk_ma2 == unk_ma, "UiaGetReservedMixedAttribute pointer mismatch, unk_ma2 %p, unk_ma %p\n", unk_ma2, unk_ma);
marshal = NULL;
hr = IUnknown_QueryInterface(unk_ma, &IID_IMarshal, (void **)&marshal);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
ok(marshal != NULL, "Failed to get IMarshal interface.\n");
refcnt = IMarshal_AddRef(marshal);
ok(refcnt == 2, "Expected refcnt %d, got %d\n", 2, refcnt);
refcnt = IMarshal_Release(marshal);
ok(refcnt == 1, "Expected refcnt %d, got %d\n", 1, refcnt);
refcnt = IMarshal_Release(marshal);
ok(refcnt == 0, "Expected refcnt %d, got %d\n", 0, refcnt);
/* Test cross-thread marshaling behavior. */
CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
hr = CoMarshalInterThreadInterfaceInStream(&IID_IUnknown, unk_ns, &stream[0]);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
hr = CoMarshalInterThreadInterfaceInStream(&IID_IUnknown, unk_ma, &stream[1]);
ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
thread = CreateThread(NULL, 0, uia_reserved_val_iface_marshal_thread, (void *)stream, 0, NULL);
while (MsgWaitForMultipleObjects(1, &thread, FALSE, INFINITE, QS_ALLINPUT) != WAIT_OBJECT_0)
{
MSG msg;
while(PeekMessageW(&msg, 0, 0, 0, PM_REMOVE))
{
TranslateMessage(&msg);
DispatchMessageW(&msg);
}
}
CloseHandle(thread);
CoUninitialize();
}
START_TEST(uiautomation)
{
test_UiaHostProviderFromHwnd();
test_uia_reserved_value_ifaces();
}