2 * Copyright 2008 Jacek Caban for CodeWeavers
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
28 #include "wine/test.h"
30 static const CLSID CLSID_JScript =
31 {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
33 #define DEFINE_EXPECT(func) \
34 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
36 #define SET_EXPECT(func) \
37 expect_ ## func = TRUE
39 #define SET_CALLED(func) \
40 called_ ## func = TRUE
42 #define CHECK_EXPECT2(func) \
44 ok(expect_ ##func, "unexpected call " #func "\n"); \
45 called_ ## func = TRUE; \
48 #define CHECK_EXPECT(func) \
50 CHECK_EXPECT2(func); \
51 expect_ ## func = FALSE; \
54 #define CHECK_CALLED(func) \
56 ok(called_ ## func, "expected " #func "\n"); \
57 expect_ ## func = called_ ## func = FALSE; \
60 DEFINE_EXPECT(global_propget_d);
61 DEFINE_EXPECT(global_propget_i);
62 DEFINE_EXPECT(global_propput_d);
63 DEFINE_EXPECT(global_propput_i);
64 DEFINE_EXPECT(global_success_d);
65 DEFINE_EXPECT(global_success_i);
67 #define DISPID_GLOBAL_TESTPROPGET 0x1000
68 #define DISPID_GLOBAL_TESTPROPPUT 0x1001
69 #define DISPID_GLOBAL_REPORTSUCCESS 0x1002
70 #define DISPID_GLOBAL_TRACE 0x1003
71 #define DISPID_GLOBAL_OK 0x1004
72 #define DISPID_GLOBAL_GETVT 0x1005
74 static const WCHAR testW[] = {'t','e','s','t',0};
76 static BOOL strict_dispid_check;
77 static const char *test_name = "(null)";
79 static const char *debugstr_w(LPCWSTR str)
81 static char buf[1024];
86 WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL);
90 static BSTR a2bstr(const char *str)
95 len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
96 ret = SysAllocStringLen(NULL, len-1);
97 MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
102 static int strcmp_wa(LPCWSTR strw, const char *stra)
105 WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), 0, 0);
106 return lstrcmpA(buf, stra);
109 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
113 if(IsEqualGUID(riid, &IID_IUnknown)
114 || IsEqualGUID(riid, &IID_IDispatch)
115 || IsEqualGUID(riid, &IID_IDispatchEx))
118 return E_NOINTERFACE;
123 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
128 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
133 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
135 ok(0, "unexpected call\n");
139 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
140 LCID lcid, ITypeInfo **ppTInfo)
142 ok(0, "unexpected call\n");
146 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
147 LPOLESTR *rgszNames, UINT cNames,
148 LCID lcid, DISPID *rgDispId)
150 ok(0, "unexpected call\n");
154 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
155 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
156 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
158 ok(0, "unexpected call\n");
162 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
164 ok(0, "unexpected call %s %x\n", debugstr_w(bstrName), grfdex);
168 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
170 ok(0, "unexpected call\n");
174 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
176 ok(0, "unexpected call\n");
180 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
182 ok(0, "unexpected call\n");
186 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
188 ok(0, "unexpected call\n");
192 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
194 ok(0, "unexpected call\n");
198 static HRESULT WINAPI Global_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
200 if(!strcmp_wa(bstrName, "ok")) {
201 ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
202 *pid = DISPID_GLOBAL_OK;
205 if(!strcmp_wa(bstrName, "trace")) {
206 ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
207 *pid = DISPID_GLOBAL_TRACE;
210 if(!strcmp_wa(bstrName, "reportSuccess")) {
211 CHECK_EXPECT(global_success_d);
212 ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
213 *pid = DISPID_GLOBAL_REPORTSUCCESS;
216 if(!strcmp_wa(bstrName, "testPropGet")) {
217 CHECK_EXPECT(global_propget_d);
218 ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
219 *pid = DISPID_GLOBAL_TESTPROPGET;
222 if(!strcmp_wa(bstrName, "testPropPut")) {
223 CHECK_EXPECT(global_propput_d);
224 ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
225 *pid = DISPID_GLOBAL_TESTPROPPUT;
228 if(!strcmp_wa(bstrName, "getVT")) {
229 ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
230 *pid = DISPID_GLOBAL_GETVT;
234 if(strict_dispid_check)
235 ok(0, "unexpected call %s\n", debugstr_w(bstrName));
236 return DISP_E_UNKNOWNNAME;
239 static HRESULT WINAPI Global_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
240 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
243 case DISPID_GLOBAL_OK:
244 ok(wFlags == INVOKE_FUNC || wFlags == (INVOKE_FUNC|INVOKE_PROPERTYGET), "wFlags = %x\n", wFlags);
245 ok(pdp != NULL, "pdp == NULL\n");
246 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
247 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
248 ok(pdp->cArgs == 2, "cArgs = %d\n", pdp->cArgs);
249 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
250 if(wFlags & INVOKE_PROPERTYGET)
251 ok(pvarRes != NULL, "pvarRes == NULL\n");
253 ok(!pvarRes, "pvarRes != NULL\n");
254 ok(pei != NULL, "pei == NULL\n");
256 ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
257 ok(V_VT(pdp->rgvarg+1) == VT_BOOL, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp->rgvarg));
258 ok(V_BOOL(pdp->rgvarg+1), "%s: %s\n", test_name, debugstr_w(V_BSTR(pdp->rgvarg)));
262 case DISPID_GLOBAL_TRACE:
263 ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
264 ok(pdp != NULL, "pdp == NULL\n");
265 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
266 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
267 ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
268 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
269 ok(!pvarRes, "pvarRes != NULL\n");
270 ok(pei != NULL, "pei == NULL\n");
272 ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
273 if(V_VT(pdp->rgvarg) == VT_BSTR)
274 trace("%s: %s\n", test_name, debugstr_w(V_BSTR(pdp->rgvarg)));
278 case DISPID_GLOBAL_REPORTSUCCESS:
279 CHECK_EXPECT(global_success_i);
281 ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
282 ok(pdp != NULL, "pdp == NULL\n");
283 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
284 ok(pdp->cArgs == 0, "cArgs = %d\n", pdp->cArgs);
285 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
286 ok(!pvarRes, "pvarRes != NULL\n");
287 ok(pei != NULL, "pei == NULL\n");
291 case DISPID_GLOBAL_TESTPROPGET:
292 CHECK_EXPECT(global_propget_i);
294 ok(wFlags == INVOKE_PROPERTYGET, "wFlags = %x\n", wFlags);
295 ok(pdp != NULL, "pdp == NULL\n");
296 ok(!pdp->rgvarg, "rgvarg != NULL\n");
297 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
298 ok(!pdp->cArgs, "cArgs = %d\n", pdp->cArgs);
299 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
300 ok(pvarRes != NULL, "pvarRes == NULL\n");
301 ok(V_VT(pvarRes) == VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
302 ok(pei != NULL, "pei == NULL\n");
304 V_VT(pvarRes) = VT_I4;
309 case DISPID_GLOBAL_TESTPROPPUT:
310 CHECK_EXPECT(global_propput_i);
312 ok(wFlags == INVOKE_PROPERTYPUT, "wFlags = %x\n", wFlags);
313 ok(pdp != NULL, "pdp == NULL\n");
314 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
315 ok(pdp->rgdispidNamedArgs != NULL, "rgdispidNamedArgs == NULL\n");
316 ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
317 ok(pdp->cNamedArgs == 1, "cNamedArgs = %d\n", pdp->cNamedArgs);
318 ok(pdp->rgdispidNamedArgs[0] == DISPID_PROPERTYPUT, "pdp->rgdispidNamedArgs[0] = %d\n", pdp->rgdispidNamedArgs[0]);
319 ok(!pvarRes, "pvarRes != NULL\n");
321 ok(V_VT(pdp->rgvarg) == VT_I4, "V_VT(pdp->rgvarg)=%d\n", V_VT(pdp->rgvarg));
322 ok(V_I4(pdp->rgvarg) == 1, "V_I4(pdp->rgvarg)=%d\n", V_I4(pdp->rgvarg));
325 case DISPID_GLOBAL_GETVT:
326 ok(pdp != NULL, "pdp == NULL\n");
327 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
328 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
329 ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
330 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
331 ok(pvarRes != NULL, "pvarRes == NULL\n");
332 ok(V_VT(pvarRes) == VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
333 ok(pei != NULL, "pei == NULL\n");
335 V_VT(pvarRes) = VT_BSTR;
336 switch(V_VT(pdp->rgvarg)) {
338 V_BSTR(pvarRes) = a2bstr("VT_EMPTY");
341 V_BSTR(pvarRes) = a2bstr("VT_NULL");
344 V_BSTR(pvarRes) = a2bstr("VT_I4");
347 V_BSTR(pvarRes) = a2bstr("VT_R8");
350 V_BSTR(pvarRes) = a2bstr("VT_BSTR");
353 V_BSTR(pvarRes) = a2bstr("VT_DISPATCH");
356 V_BSTR(pvarRes) = a2bstr("VT_BOOL");
359 ok(0, "unknown vt %d\n", V_VT(pdp->rgvarg));
366 ok(0, "unexpected call %x\n", id);
367 return DISP_E_MEMBERNOTFOUND;
370 static IDispatchExVtbl GlobalVtbl = {
371 DispatchEx_QueryInterface,
374 DispatchEx_GetTypeInfoCount,
375 DispatchEx_GetTypeInfo,
376 DispatchEx_GetIDsOfNames,
380 DispatchEx_DeleteMemberByName,
381 DispatchEx_DeleteMemberByDispID,
382 DispatchEx_GetMemberProperties,
383 DispatchEx_GetMemberName,
384 DispatchEx_GetNextDispID,
385 DispatchEx_GetNameSpaceParent
388 static IDispatchEx Global = { &GlobalVtbl };
390 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
394 if(IsEqualGUID(&IID_IUnknown, riid))
396 else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
399 return E_NOINTERFACE;
401 IUnknown_AddRef((IUnknown*)*ppv);
405 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
410 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
415 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
417 *plcid = GetUserDefaultLCID();
421 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
422 DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
424 ok(!lstrcmpW(pstrName, testW), "unexpected pstrName %s\n", debugstr_w(pstrName));
425 ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
426 ok(!ppti, "ppti != NULL\n");
428 *ppiunkItem = (IUnknown*)&Global;
432 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
437 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
438 const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
443 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
448 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
453 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
458 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
463 #undef ACTSCPSITE_THIS
465 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
466 ActiveScriptSite_QueryInterface,
467 ActiveScriptSite_AddRef,
468 ActiveScriptSite_Release,
469 ActiveScriptSite_GetLCID,
470 ActiveScriptSite_GetItemInfo,
471 ActiveScriptSite_GetDocVersionString,
472 ActiveScriptSite_OnScriptTerminate,
473 ActiveScriptSite_OnStateChange,
474 ActiveScriptSite_OnScriptError,
475 ActiveScriptSite_OnEnterScript,
476 ActiveScriptSite_OnLeaveScript
479 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
481 static IActiveScript *create_script(void)
483 IActiveScript *script;
486 hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
487 &IID_IActiveScript, (void**)&script);
488 ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
493 static void parse_script(BSTR script_str)
495 IActiveScriptParse *parser;
496 IActiveScript *engine;
499 engine = create_script();
503 hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
504 ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
506 hres = IActiveScriptParse_InitNew(parser);
507 ok(hres == S_OK, "InitNew failed: %08x\n", hres);
509 hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
510 ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
512 hres = IActiveScript_AddNamedItem(engine, testW,
513 SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
514 ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
516 hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
517 ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
519 hres = IActiveScriptParse_ParseScriptText(parser, script_str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
520 ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
522 IActiveScript_Release(engine);
523 IActiveScriptParse_Release(parser);
526 static void parse_script_a(const char *src)
528 BSTR tmp = a2bstr(src);
533 static void run_from_res(const char *name)
540 strict_dispid_check = FALSE;
543 src = FindResourceA(NULL, name, (LPCSTR)40);
544 ok(src != NULL, "Could not find resource %s\n", name);
546 size = SizeofResource(NULL, src);
547 data = LoadResource(NULL, src);
549 len = MultiByteToWideChar(CP_ACP, 0, data, size, NULL, 0);
550 str = SysAllocStringLen(NULL, len);
551 len = MultiByteToWideChar(CP_ACP, 0, data, size, str, len);
553 SET_EXPECT(global_success_d);
554 SET_EXPECT(global_success_i);
556 CHECK_CALLED(global_success_d);
557 CHECK_CALLED(global_success_i);
562 static void run_tests(void)
564 strict_dispid_check = TRUE;
567 parse_script_a("/* empty */ ;");
569 SET_EXPECT(global_propget_d);
570 SET_EXPECT(global_propget_i);
571 parse_script_a("testPropGet;");
572 CHECK_CALLED(global_propget_d);
573 CHECK_CALLED(global_propget_i);
575 SET_EXPECT(global_propput_d);
576 SET_EXPECT(global_propput_i);
577 parse_script_a("testPropPut = 1;");
578 CHECK_CALLED(global_propput_d);
579 CHECK_CALLED(global_propput_i);
581 SET_EXPECT(global_success_d);
582 SET_EXPECT(global_success_i);
583 parse_script_a("reportSuccess();");
584 CHECK_CALLED(global_success_d);
585 CHECK_CALLED(global_success_i);
587 run_from_res("lang.js");
588 run_from_res("api.js");
589 run_from_res("regexp.js");