[PATCH v2 8/8] vbscript/tests: Add tests for pending and executed code persistence.
Gabriel Ivăncescu
gabrielopcode at gmail.com
Wed Oct 30 07:57:26 CDT 2019
Signed-off-by: Gabriel Ivăncescu <gabrielopcode at gmail.com>
---
dlls/vbscript/tests/vbscript.c | 194 +++++++++++++++++++++++++++++++++
1 file changed, 194 insertions(+)
diff --git a/dlls/vbscript/tests/vbscript.c b/dlls/vbscript/tests/vbscript.c
index fdbfc44..a678710 100644
--- a/dlls/vbscript/tests/vbscript.c
+++ b/dlls/vbscript/tests/vbscript.c
@@ -624,6 +624,199 @@ static void test_scriptdisp(void)
ok(!ref, "ref = %d\n", ref);
}
+static void test_code_persistence(void)
+{
+ IActiveScriptParse *parser;
+ IDispatchEx *script_disp;
+ IActiveScript *vbscript;
+ HRESULT hr;
+ DISPID id;
+ ULONG ref;
+ BSTR str;
+
+ vbscript = create_vbscript();
+
+ hr = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
+ ok(hr == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hr);
+ test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
+ test_safety(vbscript);
+
+ SET_EXPECT(GetLCID);
+ hr = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
+ ok(hr == S_OK, "SetScriptSite failed: %08x\n", hr);
+ CHECK_CALLED(GetLCID);
+
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ hr = IActiveScriptParse_InitNew(parser);
+ ok(hr == S_OK, "InitNew failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+ test_state(vbscript, SCRIPTSTATE_INITIALIZED);
+
+ str = a2bstr(
+ "x = 1\n"
+ "dim y\ny = 2\n");
+ hr = IActiveScriptParse_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
+ ok(hr == S_OK, "ParseScriptText failed: %08x\n", hr);
+ SysFreeString(str);
+
+ str = a2bstr("dim z\nz = 3\n");
+ hr = IActiveScriptParse_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, SCRIPTTEXT_ISPERSISTENT, NULL, NULL);
+ ok(hr == S_OK, "ParseScriptText failed: %08x\n", hr);
+ SysFreeString(str);
+
+ /* Pending code does not add identifiers to the global scope */
+ script_disp = get_script_dispatch(vbscript);
+ id = 0;
+ get_disp_id(script_disp, "x", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ id = 0;
+ get_disp_id(script_disp, "y", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ id = 0;
+ get_disp_id(script_disp, "z", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ IDispatchEx_Release(script_disp);
+
+ /* Uninitialized state removes code without SCRIPTTEXT_ISPERSISTENT */
+ SET_EXPECT(OnStateChange_UNINITIALIZED);
+ hr = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_UNINITIALIZED);
+ ok(hr == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_UNINITIALIZED);
+ test_no_script_dispatch(vbscript);
+
+ SET_EXPECT(GetLCID);
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ hr = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
+ ok(hr == S_OK, "SetScriptSite failed: %08x\n", hr);
+ CHECK_CALLED(GetLCID);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+
+ SET_EXPECT(OnStateChange_CONNECTED);
+ SET_EXPECT(OnEnterScript);
+ SET_EXPECT(OnLeaveScript);
+ hr = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
+ ok(hr == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_CONNECTED);
+ CHECK_CALLED(OnEnterScript);
+ CHECK_CALLED(OnLeaveScript);
+ test_state(vbscript, SCRIPTSTATE_CONNECTED);
+
+ script_disp = get_script_dispatch(vbscript);
+ id = 0;
+ get_disp_id(script_disp, "x", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ id = 0;
+ get_disp_id(script_disp, "y", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ id = 0;
+ get_disp_id(script_disp, "z", S_OK, &id);
+ ok(id != -1, "id = -1\n");
+ IDispatchEx_Release(script_disp);
+
+ /* Uninitialized state does not remove persistent code, even if it was executed */
+ SET_EXPECT(OnStateChange_DISCONNECTED);
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ SET_EXPECT(OnStateChange_UNINITIALIZED);
+ hr = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_UNINITIALIZED);
+ ok(hr == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_DISCONNECTED);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+ CHECK_CALLED(OnStateChange_UNINITIALIZED);
+ test_no_script_dispatch(vbscript);
+
+ SET_EXPECT(GetLCID);
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ hr = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
+ ok(hr == S_OK, "SetScriptSite failed: %08x\n", hr);
+ CHECK_CALLED(GetLCID);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+
+ script_disp = get_script_dispatch(vbscript);
+ id = 0;
+ get_disp_id(script_disp, "z", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ IDispatchEx_Release(script_disp);
+
+ SET_EXPECT(OnStateChange_CONNECTED);
+ SET_EXPECT(OnEnterScript);
+ SET_EXPECT(OnLeaveScript);
+ hr = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
+ ok(hr == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_CONNECTED);
+ CHECK_CALLED(OnEnterScript);
+ CHECK_CALLED(OnLeaveScript);
+ test_state(vbscript, SCRIPTSTATE_CONNECTED);
+
+ script_disp = get_script_dispatch(vbscript);
+ id = 0;
+ get_disp_id(script_disp, "z", S_OK, &id);
+ ok(id != -1, "id = -1\n");
+ IDispatchEx_Release(script_disp);
+
+ SET_EXPECT(OnStateChange_DISCONNECTED);
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ SET_EXPECT(OnStateChange_UNINITIALIZED);
+ hr = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_UNINITIALIZED);
+ ok(hr == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_DISCONNECTED);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+ CHECK_CALLED(OnStateChange_UNINITIALIZED);
+
+ SET_EXPECT(GetLCID);
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ hr = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
+ ok(hr == S_OK, "SetScriptSite failed: %08x\n", hr);
+ CHECK_CALLED(GetLCID);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+
+ str = a2bstr("dim y\ny = 2\n");
+ hr = IActiveScriptParse_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, SCRIPTTEXT_ISPERSISTENT, NULL, NULL);
+ ok(hr == S_OK, "ParseScriptText failed: %08x\n", hr);
+ SysFreeString(str);
+
+ /* Closing the script engine removes all code (even if it's pending and persistent) */
+ SET_EXPECT(OnStateChange_CLOSED);
+ hr = IActiveScript_Close(vbscript);
+ ok(hr == S_OK, "Close failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_CLOSED);
+ test_state(vbscript, SCRIPTSTATE_CLOSED);
+ test_no_script_dispatch(vbscript);
+
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ SET_EXPECT(GetLCID);
+ hr = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
+ ok(hr == S_OK, "SetScriptSite failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+ CHECK_CALLED(GetLCID);
+ test_state(vbscript, SCRIPTSTATE_INITIALIZED);
+
+ SET_EXPECT(OnStateChange_CONNECTED);
+ hr = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
+ ok(hr == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hr);
+ CHECK_CALLED(OnStateChange_CONNECTED);
+ test_state(vbscript, SCRIPTSTATE_CONNECTED);
+
+ script_disp = get_script_dispatch(vbscript);
+ id = 0;
+ get_disp_id(script_disp, "y", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ id = 0;
+ get_disp_id(script_disp, "z", DISP_E_UNKNOWNNAME, &id);
+ ok(id == -1, "id = %d, expected -1\n", id);
+ IDispatchEx_Release(script_disp);
+
+ IActiveScriptParse_Release(parser);
+
+ SET_EXPECT(OnStateChange_DISCONNECTED);
+ SET_EXPECT(OnStateChange_INITIALIZED);
+ SET_EXPECT(OnStateChange_CLOSED);
+ ref = IActiveScript_Release(vbscript);
+ ok(!ref, "ref = %d\n", ref);
+ CHECK_CALLED(OnStateChange_DISCONNECTED);
+ CHECK_CALLED(OnStateChange_INITIALIZED);
+ CHECK_CALLED(OnStateChange_CLOSED);
+}
+
static void test_vbscript(void)
{
IActiveScriptParseProcedure2 *parse_proc;
@@ -1215,6 +1408,7 @@ START_TEST(vbscript)
test_vbscript_initializing();
test_named_items();
test_scriptdisp();
+ test_code_persistence();
test_RegExp();
test_RegExp_Replace();
}else {
--
2.21.0
More information about the wine-devel
mailing list