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 MultiByteToWideChar(CP_ACP
, 0, stra
, -1, buf
, sizeof(buf
)/sizeof(WCHAR
));
106 return lstrcmpW(strw
, buf
);
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 = %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 ok(!pvarRes
, "pvarRes != NULL\n");
251 ok(pei
!= NULL
, "pei == NULL\n");
253 ok(V_VT(pdp
->rgvarg
) == VT_BSTR
, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp
->rgvarg
));
254 ok(V_VT(pdp
->rgvarg
+1) == VT_BOOL
, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp
->rgvarg
));
255 ok(V_BOOL(pdp
->rgvarg
+1), "%s: %s\n", test_name
, debugstr_w(V_BSTR(pdp
->rgvarg
)));
259 case DISPID_GLOBAL_TRACE
:
260 ok(wFlags
== INVOKE_FUNC
, "wFlags = %x\n", wFlags
);
261 ok(pdp
!= NULL
, "pdp == NULL\n");
262 ok(pdp
->rgvarg
!= NULL
, "rgvarg == NULL\n");
263 ok(!pdp
->rgdispidNamedArgs
, "rgdispidNamedArgs != NULL\n");
264 ok(pdp
->cArgs
== 1, "cArgs = %d\n", pdp
->cArgs
);
265 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
266 ok(!pvarRes
, "pvarRes != NULL\n");
267 ok(pei
!= NULL
, "pei == NULL\n");
269 ok(V_VT(pdp
->rgvarg
) == VT_BSTR
, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp
->rgvarg
));
270 if(V_VT(pdp
->rgvarg
) == VT_BSTR
)
271 trace("%s: %s\n", test_name
, debugstr_w(V_BSTR(pdp
->rgvarg
)));
275 case DISPID_GLOBAL_REPORTSUCCESS
:
276 CHECK_EXPECT(global_success_i
);
278 ok(wFlags
== INVOKE_FUNC
, "wFlags = %x\n", wFlags
);
279 ok(pdp
!= NULL
, "pdp == NULL\n");
280 ok(!pdp
->rgdispidNamedArgs
, "rgdispidNamedArgs != NULL\n");
281 ok(pdp
->cArgs
== 0, "cArgs = %d\n", pdp
->cArgs
);
282 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
283 ok(!pvarRes
, "pvarRes != NULL\n");
284 ok(pei
!= NULL
, "pei == NULL\n");
288 case DISPID_GLOBAL_TESTPROPGET
:
289 CHECK_EXPECT(global_propget_i
);
291 ok(wFlags
== INVOKE_PROPERTYGET
, "wFlags = %x\n", wFlags
);
292 ok(pdp
!= NULL
, "pdp == NULL\n");
293 ok(!pdp
->rgvarg
, "rgvarg != NULL\n");
294 ok(!pdp
->rgdispidNamedArgs
, "rgdispidNamedArgs != NULL\n");
295 ok(!pdp
->cArgs
, "cArgs = %d\n", pdp
->cArgs
);
296 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
297 ok(pvarRes
!= NULL
, "pvarRes == NULL\n");
298 ok(V_VT(pvarRes
) == VT_EMPTY
, "V_VT(pvarRes) = %d\n", V_VT(pvarRes
));
299 ok(pei
!= NULL
, "pei == NULL\n");
301 V_VT(pvarRes
) = VT_I4
;
306 case DISPID_GLOBAL_TESTPROPPUT
:
307 CHECK_EXPECT(global_propput_i
);
309 ok(wFlags
== INVOKE_PROPERTYPUT
, "wFlags = %x\n", wFlags
);
310 ok(pdp
!= NULL
, "pdp == NULL\n");
311 ok(pdp
->rgvarg
!= NULL
, "rgvarg == NULL\n");
312 ok(pdp
->rgdispidNamedArgs
!= NULL
, "rgdispidNamedArgs == NULL\n");
313 ok(pdp
->cArgs
== 1, "cArgs = %d\n", pdp
->cArgs
);
314 ok(pdp
->cNamedArgs
== 1, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
315 ok(pdp
->rgdispidNamedArgs
[0] == DISPID_PROPERTYPUT
, "pdp->rgdispidNamedArgs[0] = %d\n", pdp
->rgdispidNamedArgs
[0]);
316 ok(!pvarRes
, "pvarRes != NULL\n");
318 ok(V_VT(pdp
->rgvarg
) == VT_I4
, "V_VT(pdp->rgvarg)=%d\n", V_VT(pdp
->rgvarg
));
319 ok(V_I4(pdp
->rgvarg
) == 1, "V_I4(pdp->rgvarg)=%d\n", V_I4(pdp
->rgvarg
));
322 case DISPID_GLOBAL_GETVT
:
323 ok(pdp
!= NULL
, "pdp == NULL\n");
324 ok(pdp
->rgvarg
!= NULL
, "rgvarg == NULL\n");
325 ok(!pdp
->rgdispidNamedArgs
, "rgdispidNamedArgs != NULL\n");
326 ok(pdp
->cArgs
== 1, "cArgs = %d\n", pdp
->cArgs
);
327 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
328 ok(pvarRes
!= NULL
, "pvarRes == NULL\n");
329 ok(V_VT(pvarRes
) == VT_EMPTY
, "V_VT(pvarRes) = %d\n", V_VT(pvarRes
));
330 ok(pei
!= NULL
, "pei == NULL\n");
332 V_VT(pvarRes
) = VT_BSTR
;
333 switch(V_VT(pdp
->rgvarg
)) {
335 V_BSTR(pvarRes
) = a2bstr("VT_EMPTY");
338 V_BSTR(pvarRes
) = a2bstr("VT_NULL");
341 V_BSTR(pvarRes
) = a2bstr("VT_I4");
344 V_BSTR(pvarRes
) = a2bstr("VT_R8");
347 V_BSTR(pvarRes
) = a2bstr("VT_BSTR");
350 V_BSTR(pvarRes
) = a2bstr("VT_DISPATCH");
353 V_BSTR(pvarRes
) = a2bstr("VT_BOOL");
356 ok(0, "unknown vt %d\n", V_VT(pdp
->rgvarg
));
363 ok(0, "unexpected call %x\n", id
);
364 return DISP_E_MEMBERNOTFOUND
;
367 static IDispatchExVtbl GlobalVtbl
= {
368 DispatchEx_QueryInterface
,
371 DispatchEx_GetTypeInfoCount
,
372 DispatchEx_GetTypeInfo
,
373 DispatchEx_GetIDsOfNames
,
377 DispatchEx_DeleteMemberByName
,
378 DispatchEx_DeleteMemberByDispID
,
379 DispatchEx_GetMemberProperties
,
380 DispatchEx_GetMemberName
,
381 DispatchEx_GetNextDispID
,
382 DispatchEx_GetNameSpaceParent
385 static IDispatchEx Global
= { &GlobalVtbl
};
387 static HRESULT WINAPI
ActiveScriptSite_QueryInterface(IActiveScriptSite
*iface
, REFIID riid
, void **ppv
)
391 if(IsEqualGUID(&IID_IUnknown
, riid
))
393 else if(IsEqualGUID(&IID_IActiveScriptSite
, riid
))
396 return E_NOINTERFACE
;
398 IUnknown_AddRef((IUnknown
*)*ppv
);
402 static ULONG WINAPI
ActiveScriptSite_AddRef(IActiveScriptSite
*iface
)
407 static ULONG WINAPI
ActiveScriptSite_Release(IActiveScriptSite
*iface
)
412 static HRESULT WINAPI
ActiveScriptSite_GetLCID(IActiveScriptSite
*iface
, LCID
*plcid
)
414 *plcid
= GetUserDefaultLCID();
418 static HRESULT WINAPI
ActiveScriptSite_GetItemInfo(IActiveScriptSite
*iface
, LPCOLESTR pstrName
,
419 DWORD dwReturnMask
, IUnknown
**ppiunkItem
, ITypeInfo
**ppti
)
421 ok(!lstrcmpW(pstrName
, testW
), "unexpected pstrName %s\n", debugstr_w(pstrName
));
422 ok(dwReturnMask
== SCRIPTINFO_IUNKNOWN
, "unexpected dwReturnMask %x\n", dwReturnMask
);
423 ok(!ppti
, "ppti != NULL\n");
425 *ppiunkItem
= (IUnknown
*)&Global
;
429 static HRESULT WINAPI
ActiveScriptSite_GetDocVersionString(IActiveScriptSite
*iface
, BSTR
*pbstrVersion
)
434 static HRESULT WINAPI
ActiveScriptSite_OnScriptTerminate(IActiveScriptSite
*iface
,
435 const VARIANT
*pvarResult
, const EXCEPINFO
*pexcepinfo
)
440 static HRESULT WINAPI
ActiveScriptSite_OnStateChange(IActiveScriptSite
*iface
, SCRIPTSTATE ssScriptState
)
445 static HRESULT WINAPI
ActiveScriptSite_OnScriptError(IActiveScriptSite
*iface
, IActiveScriptError
*pscripterror
)
450 static HRESULT WINAPI
ActiveScriptSite_OnEnterScript(IActiveScriptSite
*iface
)
455 static HRESULT WINAPI
ActiveScriptSite_OnLeaveScript(IActiveScriptSite
*iface
)
460 #undef ACTSCPSITE_THIS
462 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl
= {
463 ActiveScriptSite_QueryInterface
,
464 ActiveScriptSite_AddRef
,
465 ActiveScriptSite_Release
,
466 ActiveScriptSite_GetLCID
,
467 ActiveScriptSite_GetItemInfo
,
468 ActiveScriptSite_GetDocVersionString
,
469 ActiveScriptSite_OnScriptTerminate
,
470 ActiveScriptSite_OnStateChange
,
471 ActiveScriptSite_OnScriptError
,
472 ActiveScriptSite_OnEnterScript
,
473 ActiveScriptSite_OnLeaveScript
476 static IActiveScriptSite ActiveScriptSite
= { &ActiveScriptSiteVtbl
};
478 static IActiveScript
*create_script(void)
480 IActiveScript
*script
;
483 hres
= CoCreateInstance(&CLSID_JScript
, NULL
, CLSCTX_INPROC_SERVER
|CLSCTX_INPROC_HANDLER
,
484 &IID_IActiveScript
, (void**)&script
);
485 ok(hres
== S_OK
, "CoCreateInstance failed: %08x\n", hres
);
490 static void parse_script(BSTR script_str
)
492 IActiveScriptParse
*parser
;
493 IActiveScript
*engine
;
496 engine
= create_script();
500 hres
= IActiveScript_QueryInterface(engine
, &IID_IActiveScriptParse
, (void**)&parser
);
501 ok(hres
== S_OK
, "Could not get IActiveScriptParse: %08x\n", hres
);
503 hres
= IActiveScriptParse_InitNew(parser
);
504 ok(hres
== S_OK
, "InitNew failed: %08x\n", hres
);
506 hres
= IActiveScript_SetScriptSite(engine
, &ActiveScriptSite
);
507 ok(hres
== S_OK
, "SetScriptSite failed: %08x\n", hres
);
509 hres
= IActiveScript_AddNamedItem(engine
, testW
,
510 SCRIPTITEM_ISVISIBLE
|SCRIPTITEM_ISSOURCE
|SCRIPTITEM_GLOBALMEMBERS
);
511 ok(hres
== S_OK
, "AddNamedItem failed: %08x\n", hres
);
513 hres
= IActiveScript_SetScriptState(engine
, SCRIPTSTATE_STARTED
);
514 ok(hres
== S_OK
, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres
);
516 hres
= IActiveScriptParse_ParseScriptText(parser
, script_str
, NULL
, NULL
, NULL
, 0, 0, 0, NULL
, NULL
);
517 ok(hres
== S_OK
, "ParseScriptText failed: %08x\n", hres
);
519 IActiveScript_Release(engine
);
520 IActiveScriptParse_Release(parser
);
523 static void parse_script_a(const char *src
)
525 BSTR tmp
= a2bstr(src
);
530 static void run_from_res(const char *name
)
537 strict_dispid_check
= FALSE
;
540 src
= FindResourceA(NULL
, name
, (LPCSTR
)40);
541 ok(src
!= NULL
, "Could not find resource %s\n", name
);
543 size
= SizeofResource(NULL
, src
);
544 data
= LoadResource(NULL
, src
);
546 len
= MultiByteToWideChar(CP_ACP
, 0, data
, size
, NULL
, 0);
547 str
= SysAllocStringLen(NULL
, len
-1);
548 len
= MultiByteToWideChar(CP_ACP
, 0, data
, size
, str
, len
);
550 SET_EXPECT(global_success_d
);
551 SET_EXPECT(global_success_i
);
553 CHECK_CALLED(global_success_d
);
554 CHECK_CALLED(global_success_i
);
559 static void run_tests(void)
561 strict_dispid_check
= TRUE
;
564 parse_script_a("/* empty */ ;");
566 SET_EXPECT(global_propget_d
);
567 SET_EXPECT(global_propget_i
);
568 parse_script_a("testPropGet;");
569 CHECK_CALLED(global_propget_d
);
570 CHECK_CALLED(global_propget_i
);
572 SET_EXPECT(global_propput_d
);
573 SET_EXPECT(global_propput_i
);
574 parse_script_a("testPropPut = 1;");
575 CHECK_CALLED(global_propput_d
);
576 CHECK_CALLED(global_propput_i
);
578 SET_EXPECT(global_success_d
);
579 SET_EXPECT(global_success_i
);
580 parse_script_a("reportSuccess();");
581 CHECK_CALLED(global_success_d
);
582 CHECK_CALLED(global_success_i
);
584 run_from_res("lang.js");