Dmitry Timoshkov : vbscript/tests: Add some tests for IRegExp2::Replace().

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 1333ea8482311169a6bab3504b42767137df46e6
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=1333ea8482311169a6bab3504...

Author: Dmitry Timoshkov <dmitry(a)baikal.ru>
Date:   Wed Oct 23 14:20:21 2019 +0800

vbscript/tests: Add some tests for IRegExp2::Replace().

Signed-off-by: Dmitry Timoshkov <dmitry(a)baikal.ru>
Signed-off-by: Jacek Caban <jacek(a)codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>

---

 dlls/vbscript/tests/vbscript.c | 60 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/dlls/vbscript/tests/vbscript.c b/dlls/vbscript/tests/vbscript.c
index 08edfc1be7..fdbfc4465b 100644
--- a/dlls/vbscript/tests/vbscript.c
+++ b/dlls/vbscript/tests/vbscript.c
@@ -1,5 +1,6 @@
 /*
  * Copyright 2011 Jacek Caban for CodeWeavers
+ * Copyright 2019 Dmitry Timoshkov
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
@@ -1131,6 +1132,64 @@ static void test_RegExp(void)
     IRegExp2_Release(regexp);
 }
 
+static void test_RegExp_Replace(void)
+{
+    static const struct
+    {
+        const char *pattern;
+        const char *replace;
+        const char *source;
+        const char *result;
+        BOOL global;
+    } test[] =
+    {
+        { "abc", "", "123abc456", "123456", FALSE },
+        { "abc", "dcba", "123abc456", "123dcba456", FALSE },
+        { "[\r\n\t\f]+", " ", "\nHello\rNew\fWorld\t!", " Hello\rNew\fWorld\t!", FALSE },
+        { "[\r\n\t\f]+", " ", "\nHello\rNew\fWorld\t!", " Hello New World !", TRUE },
+    };
+    HRESULT hr;
+    IRegExp2 *regexp;
+    VARIANT var;
+    BSTR str, ret, result;
+    int i;
+
+    hr = CoCreateInstance(&CLSID_VBScriptRegExp, NULL,
+                          CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER,
+                          &IID_IRegExp2, (void **)®exp);
+    if (hr == REGDB_E_CLASSNOTREG)
+    {
+        win_skip("VBScriptRegExp is not registered\n");
+        return;
+    }
+    ok(hr == S_OK, "got %#x\n", hr);
+
+    for (i = 0; i < ARRAY_SIZE(test); i++)
+    {
+        hr = IRegExp2_put_Global(regexp, test[i].global ? VARIANT_TRUE : VARIANT_FALSE);
+        ok(hr == S_OK, "got %#x\n", hr);
+
+        str = a2bstr(test[i].pattern);
+        hr = IRegExp2_put_Pattern(regexp, str);
+        ok(hr == S_OK, "got %#x\n", hr);
+        SysFreeString(str);
+
+        str = a2bstr(test[i].source);
+        V_VT(&var) = VT_BSTR;
+        V_BSTR(&var) = a2bstr(test[i].replace);
+        hr = IRegExp2_Replace(regexp, str, var, &ret);
+        ok(hr == S_OK, "got %#x\n", hr);
+        result = a2bstr(test[i].result);
+        ok(!wcscmp(ret, result), "got %s, expected %s\n", wine_dbgstr_w(ret), wine_dbgstr_w(result));
+        SysFreeString(result);
+        SysFreeString(ret);
+        SysFreeString(V_BSTR(&var));
+        SysFreeString(str);
+    }
+
+    IRegExp2_Release(regexp);
+}
+
 static BOOL check_vbscript(void)
 {
     IActiveScriptParseProcedure2 *vbscript;
@@ -1157,6 +1216,7 @@ START_TEST(vbscript)
         test_named_items();
         test_scriptdisp();
         test_RegExp();
+        test_RegExp_Replace();
     }else {
         win_skip("VBScript engine not available or too old\n");
     }