uiautomationcore/tests: Add tests for ReservedValue COM interfaces.

dlls/uiautomationcore/tests/Makefile.in

 TESTDLL = uiautomationcore.dll
-IMPORTS = uiautomationcore user32
+IMPORTS = uiautomationcore user32 ole32
 
 C_SRCS = \
 	uiautomation.c

dlls/uiautomationcore/tests/uiautomation.c

@@ -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();
 }