msftedit: Add tests for interfaces.

Signed-off-by: Jactry Zeng <jzeng@codeweavers.com>
Signed-off-by: Huw Davies <huw@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>

configure

@@ -19594,6 +19594,7 @@ wine_fn_config_makefile dlls/msdmo enable_msdmo
 wine_fn_config_makefile dlls/msdmo/tests enable_tests
 wine_fn_config_makefile dlls/msdrm enable_msdrm
 wine_fn_config_makefile dlls/msftedit enable_msftedit
+wine_fn_config_makefile dlls/msftedit/tests enable_tests
 wine_fn_config_makefile dlls/msg711.acm enable_msg711_acm
 wine_fn_config_makefile dlls/msgsm32.acm enable_msgsm32_acm
 wine_fn_config_makefile dlls/mshtml.tlb enable_mshtml_tlb

configure.ac

@@ -3450,6 +3450,7 @@ WINE_CONFIG_MAKEFILE(dlls/msdmo)
 WINE_CONFIG_MAKEFILE(dlls/msdmo/tests)
 WINE_CONFIG_MAKEFILE(dlls/msdrm)
 WINE_CONFIG_MAKEFILE(dlls/msftedit)
+WINE_CONFIG_MAKEFILE(dlls/msftedit/tests)
 WINE_CONFIG_MAKEFILE(dlls/msg711.acm)
 WINE_CONFIG_MAKEFILE(dlls/msgsm32.acm)
 WINE_CONFIG_MAKEFILE(dlls/mshtml.tlb)

dlls/msftedit/tests/Makefile.in

@@ -0,0 +1,5 @@
+TESTDLL   = msftedit.dll
+IMPORTS   = ole32 oleaut32 user32
+
+C_SRCS = \
+	richole.c

dlls/msftedit/tests/richole.c

@@ -0,0 +1,190 @@
+/*
+ * Tests for IRichEditOle and friends.
+ *
+ * Copyright 2008 Google (Dan Hipschman)
+ * Copyright 2018 Jactry Zeng for CodeWeavers
+ *
+ * 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
+ */
+
+#define COBJMACROS
+
+#include <stdarg.h>
+
+#include <windef.h>
+#include <winbase.h>
+#include <initguid.h>
+#include <ole2.h>
+#include <richedit.h>
+#include <richole.h>
+#include <tom.h>
+#include <wine/test.h>
+
+static HMODULE msftedit_hmodule;
+
+#define EXPECT_REF(obj,ref) _expect_ref((IUnknown*)obj, ref, __LINE__)
+static void _expect_ref(IUnknown* obj, ULONG ref, int line)
+{
+    ULONG rc;
+    IUnknown_AddRef(obj);
+    rc = IUnknown_Release(obj);
+    ok_(__FILE__,line)(rc == ref, "expected refcount %d, got %d.\n", ref, rc);
+}
+
+static HWND new_window(LPCWSTR classname, DWORD dwstyle, HWND parent)
+{
+    HWND hwnd = CreateWindowW(classname, NULL,
+                              dwstyle | WS_POPUP | WS_HSCROLL | WS_VSCROLL | WS_VISIBLE,
+                              0, 0, 200, 60, parent, NULL, msftedit_hmodule, NULL);
+    ok(hwnd != NULL, "class: %s, error: %d.\n", wine_dbgstr_w(classname), (int) GetLastError());
+    return hwnd;
+}
+
+static void test_Interfaces(void)
+{
+    IRichEditOle *reole = NULL, *reole1 = NULL;
+    ITextDocument *txtdoc = NULL;
+    ITextDocument2Old *txtdoc2old = NULL;
+    ITextDocument2 *txtdoc2 = NULL;
+    ITextSelection *txtsel = NULL, *txtsel2;
+    IUnknown *punk;
+    HRESULT hres;
+    LRESULT res;
+    HWND hwnd;
+    ULONG refcount;
+
+    hwnd = new_window(MSFTEDIT_CLASS, ES_MULTILINE, NULL);
+    if (!hwnd)
+    {
+        skip("Couldn't create window.\n");
+        return;
+    }
+
+    res = SendMessageA(hwnd, EM_GETOLEINTERFACE, 0, (LPARAM)&reole);
+    ok(res, "SendMessage\n");
+    ok(reole != NULL, "EM_GETOLEINTERFACE\n");
+    EXPECT_REF(reole, 2);
+
+    res = SendMessageA(hwnd, EM_GETOLEINTERFACE, 0, (LPARAM)&reole1);
+    ok(res == 1, "SendMessage\n");
+    ok(reole1 == reole, "Should not return a new IRichEditOle interface.\n");
+    EXPECT_REF(reole, 3);
+
+    hres = IRichEditOle_QueryInterface(reole, &IID_ITextDocument, (void **)&txtdoc);
+    ok(hres == S_OK, "IRichEditOle_QueryInterface failed: 0x%08x.\n", hres);
+    ok(txtdoc != NULL, "IRichEditOle_QueryInterface\n");
+
+    hres = ITextDocument_GetSelection(txtdoc, NULL);
+    ok(hres == E_INVALIDARG, "ITextDocument_GetSelection: 0x%08x.\n", hres);
+
+    EXPECT_REF(txtdoc, 4);
+
+    hres = ITextDocument_GetSelection(txtdoc, &txtsel);
+    ok(hres == S_OK, "ITextDocument_GetSelection failed 0x%08x.\n", hres);
+
+    EXPECT_REF(txtdoc, 4);
+    EXPECT_REF(txtsel, 2);
+
+    hres = ITextDocument_GetSelection(txtdoc, &txtsel2);
+    ok(hres == S_OK, "ITextDocument_GetSelection failed: 0x%08x.\n", hres);
+    ok(txtsel2 == txtsel, "got %p, %p\n", txtsel, txtsel2);
+
+    EXPECT_REF(txtdoc, 4);
+    EXPECT_REF(txtsel, 3);
+
+    ITextSelection_Release(txtsel2);
+
+    punk = NULL;
+    hres = ITextSelection_QueryInterface(txtsel, &IID_ITextSelection, (void **)&punk);
+    ok(hres == S_OK, "ITextSelection_QueryInterface failed: 0x%08x.\n", hres);
+    ok(punk != NULL, "ITextSelection_QueryInterface\n");
+    IUnknown_Release(punk);
+
+    punk = NULL;
+    hres = ITextSelection_QueryInterface(txtsel, &IID_ITextRange, (void **)&punk);
+    ok(hres == S_OK, "ITextSelection_QueryInterface failed: 0x%08x.\n", hres);
+    ok(punk != NULL, "ITextSelection_QueryInterface\n");
+    IUnknown_Release(punk);
+
+    punk = NULL;
+    hres = ITextSelection_QueryInterface(txtsel, &IID_IDispatch, (void **)&punk);
+    ok(hres == S_OK, "ITextSelection_QueryInterface failed: 0x%08x.\n", hres);
+    ok(punk != NULL, "ITextSelection_QueryInterface\n");
+    IUnknown_Release(punk);
+
+    punk = NULL;
+    hres = IRichEditOle_QueryInterface(reole, &IID_IOleClientSite, (void **)&punk);
+    ok(hres == E_NOINTERFACE, "IRichEditOle_QueryInterface: 0x%08x.\n", hres);
+
+    punk = NULL;
+    hres = IRichEditOle_QueryInterface(reole, &IID_IOleWindow, (void **)&punk);
+    ok(hres == E_NOINTERFACE, "IRichEditOle_QueryInterface: 0x%08x.\n", hres);
+
+    punk = NULL;
+    hres = IRichEditOle_QueryInterface(reole, &IID_IOleInPlaceSite, (void **)&punk);
+    ok(hres == E_NOINTERFACE, "IRichEditOle_QueryInterface: 0x%08x.\n", hres);
+
+    /* ITextDocument2 is implemented on msftedit after win8 for superseding ITextDocument2Old  */
+    hres = IRichEditOle_QueryInterface(reole, &IID_ITextDocument2, (void **)&txtdoc2);
+    ok(hres == S_OK ||
+       hres == E_NOINTERFACE /* before win8 */, "IRichEditOle_QueryInterface: 0x%08x.\n", hres);
+    if (hres != E_NOINTERFACE)
+    {
+        ok(txtdoc2 != NULL, "IRichEditOle_QueryInterface\n");
+        ok((ITextDocument *)txtdoc2 == txtdoc, "Interface pointer isn't equal.\n");
+        EXPECT_REF(txtdoc2, 5);
+        EXPECT_REF(reole, 5);
+
+        hres = ITextDocument2_QueryInterface(txtdoc2, &IID_ITextDocument2Old, (void **)&txtdoc2old);
+        ok(hres == S_OK, "ITextDocument2_QueryInterface failed: 0x%08x.\n", hres);
+        ok((ITextDocument *)txtdoc2old != txtdoc, "Interface pointer is equal.\n");
+        EXPECT_REF(txtdoc2, 5);
+        EXPECT_REF(reole, 5);
+        EXPECT_REF(txtdoc2old, 1);
+        ITextDocument2Old_Release(txtdoc2old);
+        ITextDocument2_Release(txtdoc2);
+    }
+    else
+    {
+        hres = IRichEditOle_QueryInterface(reole, &IID_ITextDocument2Old, (void **)&txtdoc2old);
+        ok(hres == S_OK, "IRichEditOle_QueryInterface failed: 0x%08x.\n", hres);
+        ok(txtdoc2old != NULL, "IRichEditOle_QueryInterface\n");
+        ok((ITextDocument *)txtdoc2old == txtdoc, "Interface pointer is equal.\n");
+        EXPECT_REF(txtdoc2old, 5);
+        EXPECT_REF(reole, 5);
+        ITextDocument2Old_Release(txtdoc2old);
+    }
+
+    ITextDocument_Release(txtdoc);
+    IRichEditOle_Release(reole);
+    refcount = IRichEditOle_Release(reole);
+    ok(refcount == 1, "Got wrong ref count: %d.\n", refcount);
+    DestroyWindow(hwnd);
+
+    /* Methods should return CO_E_RELEASED if the backing document has
+       been released.  One test should suffice.  */
+    hres = ITextSelection_CanEdit(txtsel, NULL);
+    ok(hres == CO_E_RELEASED, "ITextSelection after ITextDocument destroyed\n");
+
+    ITextSelection_Release(txtsel);
+}
+
+START_TEST(richole)
+{
+    msftedit_hmodule = LoadLibraryA("msftedit.dll");
+    ok(msftedit_hmodule != NULL, "error: %d\n", (int) GetLastError());
+
+    test_Interfaces();
+}