jscript: Correctly handle empty matches in String.replace.
[wine] / dlls / jscript / tests / jscript.c
1 /*
2  * Copyright 2008 Jacek Caban for CodeWeavers
3  *
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.
8  *
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.
13  *
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
17  */
18
19 #define COBJMACROS
20 #define CONST_VTABLE
21
22 #include <initguid.h>
23 #include <ole2.h>
24 #include <activscp.h>
25 #include <objsafe.h>
26 #include <dispex.h>
27
28 #include "wine/test.h"
29
30 DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
31
32 static const CLSID CLSID_JScript =
33     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
34 static const CLSID CLSID_JScriptEncode =
35     {0xf414c262,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
36
37 #define DEFINE_EXPECT(func) \
38     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
39
40 #define SET_EXPECT(func) \
41     expect_ ## func = TRUE
42
43 #define CHECK_EXPECT2(func) \
44     do { \
45         ok(expect_ ##func, "unexpected call " #func "\n"); \
46         called_ ## func = TRUE; \
47     }while(0)
48
49 #define CHECK_EXPECT(func) \
50     do { \
51         CHECK_EXPECT2(func); \
52         expect_ ## func = FALSE; \
53     }while(0)
54
55 #define CHECK_CALLED(func) \
56     do { \
57         ok(called_ ## func, "expected " #func "\n"); \
58         expect_ ## func = called_ ## func = FALSE; \
59     }while(0)
60
61 DEFINE_EXPECT(GetLCID);
62 DEFINE_EXPECT(OnStateChange_UNINITIALIZED);
63 DEFINE_EXPECT(OnStateChange_STARTED);
64 DEFINE_EXPECT(OnStateChange_CONNECTED);
65 DEFINE_EXPECT(OnStateChange_DISCONNECTED);
66 DEFINE_EXPECT(OnStateChange_CLOSED);
67 DEFINE_EXPECT(OnStateChange_INITIALIZED);
68 DEFINE_EXPECT(OnEnterScript);
69 DEFINE_EXPECT(OnLeaveScript);
70
71 static const CLSID *engine_clsid = &CLSID_JScript;
72
73 static BSTR a2bstr(const char *str)
74 {
75     BSTR ret;
76     int len;
77
78     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
79     ret = SysAllocStringLen(NULL, len-1);
80     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
81
82     return ret;
83 }
84
85 #define test_state(s,ss) _test_state(__LINE__,s,ss)
86 static void _test_state(unsigned line, IActiveScript *script, SCRIPTSTATE exstate)
87 {
88     SCRIPTSTATE state = -1;
89     HRESULT hres;
90
91     hres = IActiveScript_GetScriptState(script, &state);
92     ok_(__FILE__,line) (hres == S_OK, "GetScriptState failed: %08x\n", hres);
93     ok_(__FILE__,line) (state == exstate, "state=%d, expected %d\n", state, exstate);
94 }
95
96 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
97 {
98     *ppv = NULL;
99
100     if(IsEqualGUID(&IID_IUnknown, riid))
101         *ppv = iface;
102     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
103         *ppv = iface;
104     else
105         return E_NOINTERFACE;
106
107     IUnknown_AddRef((IUnknown*)*ppv);
108     return S_OK;
109 }
110
111 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
112 {
113     return 2;
114 }
115
116 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
117 {
118     return 1;
119 }
120
121 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
122 {
123     CHECK_EXPECT(GetLCID);
124     return E_NOTIMPL;
125 }
126
127 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
128         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
129 {
130     ok(0, "unexpected call\n");
131     return E_NOTIMPL;
132 }
133
134 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
135 {
136     ok(0, "unexpected call\n");
137     return E_NOTIMPL;
138 }
139
140 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
141         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
142 {
143     ok(0, "unexpected call\n");
144     return E_NOTIMPL;
145 }
146
147 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
148 {
149     switch(ssScriptState) {
150     case SCRIPTSTATE_UNINITIALIZED:
151         CHECK_EXPECT(OnStateChange_UNINITIALIZED);
152         return S_OK;
153     case SCRIPTSTATE_STARTED:
154         CHECK_EXPECT(OnStateChange_STARTED);
155         return S_OK;
156     case SCRIPTSTATE_CONNECTED:
157         CHECK_EXPECT(OnStateChange_CONNECTED);
158         return S_OK;
159     case SCRIPTSTATE_DISCONNECTED:
160         CHECK_EXPECT(OnStateChange_DISCONNECTED);
161         return S_OK;
162     case SCRIPTSTATE_CLOSED:
163         CHECK_EXPECT(OnStateChange_CLOSED);
164         return S_OK;
165     case SCRIPTSTATE_INITIALIZED:
166         CHECK_EXPECT(OnStateChange_INITIALIZED);
167         return S_OK;
168     default:
169         ok(0, "unexpected call %d\n", ssScriptState);
170     }
171
172     return E_NOTIMPL;
173 }
174
175 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
176 {
177     ok(0, "unexpected call\n");
178     return E_NOTIMPL;
179 }
180
181 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
182 {
183     CHECK_EXPECT(OnEnterScript);
184     return S_OK;
185 }
186
187 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
188 {
189     CHECK_EXPECT(OnLeaveScript);
190     return S_OK;
191 }
192
193 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
194     ActiveScriptSite_QueryInterface,
195     ActiveScriptSite_AddRef,
196     ActiveScriptSite_Release,
197     ActiveScriptSite_GetLCID,
198     ActiveScriptSite_GetItemInfo,
199     ActiveScriptSite_GetDocVersionString,
200     ActiveScriptSite_OnScriptTerminate,
201     ActiveScriptSite_OnStateChange,
202     ActiveScriptSite_OnScriptError,
203     ActiveScriptSite_OnEnterScript,
204     ActiveScriptSite_OnLeaveScript
205 };
206
207 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
208
209 static void test_script_dispatch(IDispatchEx *dispex)
210 {
211     DISPPARAMS dp = {NULL,NULL,0,0};
212     EXCEPINFO ei;
213     BSTR str;
214     DISPID id;
215     VARIANT v;
216     HRESULT hres;
217
218     str = a2bstr("ActiveXObject");
219     hres = IDispatchEx_GetDispID(dispex, str, fdexNameCaseSensitive, &id);
220     SysFreeString(str);
221     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
222
223     str = a2bstr("Math");
224     hres = IDispatchEx_GetDispID(dispex, str, fdexNameCaseSensitive, &id);
225     SysFreeString(str);
226     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
227
228     memset(&ei, 0, sizeof(ei));
229     hres = IDispatchEx_InvokeEx(dispex, id, 0, DISPATCH_PROPERTYGET, &dp, &v, &ei, NULL);
230     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
231     ok(V_VT(&v) == VT_DISPATCH, "V_VT(v) = %d\n", V_VT(&v));
232     ok(V_DISPATCH(&v) != NULL, "V_DISPATCH(v) = NULL\n");
233     VariantClear(&v);
234
235     str = a2bstr("String");
236     hres = IDispatchEx_GetDispID(dispex, str, fdexNameCaseSensitive, &id);
237     SysFreeString(str);
238     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
239
240     memset(&ei, 0, sizeof(ei));
241     hres = IDispatchEx_InvokeEx(dispex, id, 0, DISPATCH_PROPERTYGET, &dp, &v, &ei, NULL);
242     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
243     ok(V_VT(&v) == VT_DISPATCH, "V_VT(v) = %d\n", V_VT(&v));
244     ok(V_DISPATCH(&v) != NULL, "V_DISPATCH(v) = NULL\n");
245     VariantClear(&v);
246 }
247
248 static IDispatchEx *get_script_dispatch(IActiveScript *script)
249 {
250     IDispatchEx *dispex;
251     IDispatch *disp;
252     HRESULT hres;
253
254     disp = (void*)0xdeadbeef;
255     hres = IActiveScript_GetScriptDispatch(script, NULL, &disp);
256     ok(hres == S_OK, "GetScriptDispatch failed: %08x\n", hres);
257
258     IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
259     IDispatch_Release(disp);
260     ok(hres == S_OK, "Could not get IDispatch iface: %08x\n", hres);
261     return dispex;
262 }
263
264 static void test_no_script_dispatch(IActiveScript *script)
265 {
266     IDispatch *disp;
267     HRESULT hres;
268
269     disp = (void*)0xdeadbeef;
270     hres = IActiveScript_GetScriptDispatch(script, NULL, &disp);
271     ok(hres == E_UNEXPECTED, "hres = %08x, expected E_UNEXPECTED\n", hres);
272     ok(!disp, "disp != NULL\n");
273 }
274
275 static void test_safety(IUnknown *unk)
276 {
277     IObjectSafety *safety;
278     DWORD supported, enabled;
279     HRESULT hres;
280
281     hres = IUnknown_QueryInterface(unk, &IID_IObjectSafety, (void**)&safety);
282     ok(hres == S_OK, "Could not get IObjectSafety: %08x\n", hres);
283     if(FAILED(hres))
284         return;
285
286     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, NULL);
287     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
288     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, NULL, &enabled);
289     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
290
291     supported = enabled = 0xdeadbeef;
292     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, &enabled);
293     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
294     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
295        "supported=%x\n", supported);
296     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
297
298     supported = enabled = 0xdeadbeef;
299     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScript, &supported, &enabled);
300     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
301     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
302        "supported=%x\n", supported);
303     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
304
305     supported = enabled = 0xdeadbeef;
306     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
307     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
308     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
309        "supported=%x\n", supported);
310     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
311
312     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
313             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER
314                 |INTERFACESAFE_FOR_UNTRUSTED_CALLER,
315             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
316     ok(hres == E_FAIL, "SetInterfaceSafetyOptions failed: %08x, expected E_FAIL\n", hres);
317
318     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
319             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER,
320             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
321     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
322
323     supported = enabled = 0xdeadbeef;
324     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
325     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
326     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
327        "supported=%x\n", supported);
328     ok(enabled == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
329        "enabled=%x\n", enabled);
330
331     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, INTERFACESAFE_FOR_UNTRUSTED_DATA, 0);
332     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
333
334     supported = enabled = 0xdeadbeef;
335     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
336     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
337     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
338        "supported=%x\n", supported);
339     ok(enabled == (INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER), "enabled=%x\n", enabled);
340
341     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
342             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER, 0);
343     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
344
345     supported = enabled = 0xdeadbeef;
346     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
347     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
348     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
349        "supported=%x\n", supported);
350     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
351
352     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
353             INTERFACE_USES_DISPEX, 0);
354     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
355
356     supported = enabled = 0xdeadbeef;
357     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
358     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
359     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
360        "supported=%x\n", supported);
361     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
362
363     IObjectSafety_Release(safety);
364 }
365
366 static HRESULT set_script_prop(IActiveScript *engine, DWORD property, VARIANT *val)
367 {
368     IActiveScriptProperty *script_prop;
369     HRESULT hres;
370
371     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptProperty,
372             (void**)&script_prop);
373     ok(hres == S_OK, "Could not get IActiveScriptProperty: %08x\n", hres);
374     if(FAILED(hres))
375         return hres;
376
377     hres = IActiveScriptProperty_SetProperty(script_prop, property, NULL, val);
378     IActiveScriptProperty_Release(script_prop);
379     return hres;
380 }
381
382 static void test_invoke_versioning(IActiveScript *script)
383 {
384     VARIANT v;
385     HRESULT hres;
386
387     V_VT(&v) = VT_NULL;
388     hres = set_script_prop(script, SCRIPTPROP_INVOKEVERSIONING, &v);
389     if(hres == E_NOTIMPL) {
390         win_skip("SCRIPTPROP_INVOKESTRING not supported\n");
391         return;
392     }
393     ok(hres == E_INVALIDARG, "SetProperty(SCRIPTPROP_INVOKEVERSIONING) failed: %08x\n", hres);
394
395     V_VT(&v) = VT_I2;
396     V_I2(&v) = 0;
397     hres = set_script_prop(script, SCRIPTPROP_INVOKEVERSIONING, &v);
398     ok(hres == E_INVALIDARG, "SetProperty(SCRIPTPROP_INVOKEVERSIONING) failed: %08x\n", hres);
399
400     V_VT(&v) = VT_I4;
401     V_I4(&v) = 16;
402     hres = set_script_prop(script, SCRIPTPROP_INVOKEVERSIONING, &v);
403     ok(hres == E_INVALIDARG, "SetProperty(SCRIPTPROP_INVOKEVERSIONING) failed: %08x\n", hres);
404
405     V_VT(&v) = VT_I4;
406     V_I4(&v) = 2;
407     hres = set_script_prop(script, SCRIPTPROP_INVOKEVERSIONING, &v);
408     ok(hres == S_OK, "SetProperty(SCRIPTPROP_INVOKEVERSIONING) failed: %08x\n", hres);
409 }
410
411 static IActiveScript *create_jscript(void)
412 {
413     IActiveScript *ret;
414     HRESULT hres;
415
416     hres = CoCreateInstance(engine_clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
417             &IID_IActiveScript, (void**)&ret);
418     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
419
420     return ret;
421 }
422
423 static void test_jscript(void)
424 {
425     IActiveScriptParse *parse;
426     IActiveScript *script;
427     IDispatchEx *dispex;
428     ULONG ref;
429     HRESULT hres;
430
431     script = create_jscript();
432
433     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parse);
434     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
435
436     test_state(script, SCRIPTSTATE_UNINITIALIZED);
437     test_safety((IUnknown*)script);
438     test_invoke_versioning(script);
439
440     hres = IActiveScriptParse64_InitNew(parse);
441     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
442
443     hres = IActiveScriptParse64_InitNew(parse);
444     ok(hres == E_UNEXPECTED, "InitNew failed: %08x, expected E_UNEXPECTED\n", hres);
445
446     hres = IActiveScript_SetScriptSite(script, NULL);
447     ok(hres == E_POINTER, "SetScriptSite failed: %08x, expected E_POINTER\n", hres);
448
449     test_state(script, SCRIPTSTATE_UNINITIALIZED);
450     test_no_script_dispatch(script);
451
452     SET_EXPECT(GetLCID);
453     SET_EXPECT(OnStateChange_INITIALIZED);
454     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
455     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
456     CHECK_CALLED(GetLCID);
457     CHECK_CALLED(OnStateChange_INITIALIZED);
458
459     test_state(script, SCRIPTSTATE_INITIALIZED);
460
461     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
462     ok(hres == E_UNEXPECTED, "SetScriptSite failed: %08x, expected E_UNEXPECTED\n", hres);
463
464     dispex = get_script_dispatch(script);
465     test_script_dispatch(dispex);
466
467     SET_EXPECT(OnStateChange_STARTED);
468     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
469     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
470     CHECK_CALLED(OnStateChange_STARTED);
471
472     test_state(script, SCRIPTSTATE_STARTED);
473
474     SET_EXPECT(OnStateChange_CLOSED);
475     hres = IActiveScript_Close(script);
476     ok(hres == S_OK, "Close failed: %08x\n", hres);
477     CHECK_CALLED(OnStateChange_CLOSED);
478
479     test_state(script, SCRIPTSTATE_CLOSED);
480     test_no_script_dispatch(script);
481     test_script_dispatch(dispex);
482     IDispatchEx_Release(dispex);
483
484     IUnknown_Release(parse);
485
486     ref = IActiveScript_Release(script);
487     ok(!ref, "ref = %d\n", ref);
488 }
489
490 static void test_jscript2(void)
491 {
492     IActiveScriptParse *parse;
493     IActiveScript *script;
494     ULONG ref;
495     HRESULT hres;
496
497     script = create_jscript();
498
499     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parse);
500     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
501
502     test_state(script, SCRIPTSTATE_UNINITIALIZED);
503
504     SET_EXPECT(GetLCID);
505     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
506     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
507     CHECK_CALLED(GetLCID);
508
509     test_state(script, SCRIPTSTATE_UNINITIALIZED);
510
511     SET_EXPECT(OnStateChange_INITIALIZED);
512     hres = IActiveScriptParse64_InitNew(parse);
513     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
514     CHECK_CALLED(OnStateChange_INITIALIZED);
515
516     hres = IActiveScriptParse64_InitNew(parse);
517     ok(hres == E_UNEXPECTED, "InitNew failed: %08x, expected E_UNEXPECTED\n", hres);
518
519     SET_EXPECT(OnStateChange_CONNECTED);
520     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_CONNECTED);
521     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
522     CHECK_CALLED(OnStateChange_CONNECTED);
523
524     test_state(script, SCRIPTSTATE_CONNECTED);
525
526     SET_EXPECT(OnStateChange_DISCONNECTED);
527     SET_EXPECT(OnStateChange_INITIALIZED);
528     SET_EXPECT(OnStateChange_CLOSED);
529     hres = IActiveScript_Close(script);
530     ok(hres == S_OK, "Close failed: %08x\n", hres);
531     CHECK_CALLED(OnStateChange_DISCONNECTED);
532     CHECK_CALLED(OnStateChange_INITIALIZED);
533     CHECK_CALLED(OnStateChange_CLOSED);
534
535     test_state(script, SCRIPTSTATE_CLOSED);
536     test_no_script_dispatch(script);
537
538     IUnknown_Release(parse);
539
540     ref = IActiveScript_Release(script);
541     ok(!ref, "ref = %d\n", ref);
542 }
543
544 static void test_jscript_uninitializing(void)
545 {
546     IActiveScriptParse *parse;
547     IActiveScript *script;
548     IDispatchEx *dispex;
549     ULONG ref;
550     HRESULT hres;
551
552     static const WCHAR script_textW[] =
553         {'f','u','n','c','t','i','o','n',' ','f','(',')',' ','{','}',0};
554
555     script = create_jscript();
556
557     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parse);
558     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
559
560     test_state(script, SCRIPTSTATE_UNINITIALIZED);
561
562     hres = IActiveScriptParse64_InitNew(parse);
563     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
564
565     SET_EXPECT(GetLCID);
566     SET_EXPECT(OnStateChange_INITIALIZED);
567     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
568     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
569     CHECK_CALLED(GetLCID);
570     CHECK_CALLED(OnStateChange_INITIALIZED);
571
572     test_state(script, SCRIPTSTATE_INITIALIZED);
573
574     hres = IActiveScriptParse64_ParseScriptText(parse, script_textW, NULL, NULL, NULL, 0, 1, 0x42, NULL, NULL);
575     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
576
577     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
578     ok(hres == E_UNEXPECTED, "SetScriptSite failed: %08x, expected E_UNEXPECTED\n", hres);
579
580     SET_EXPECT(OnStateChange_UNINITIALIZED);
581     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
582     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
583     CHECK_CALLED(OnStateChange_UNINITIALIZED);
584
585     test_state(script, SCRIPTSTATE_UNINITIALIZED);
586
587     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
588     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
589
590     SET_EXPECT(GetLCID);
591     SET_EXPECT(OnStateChange_INITIALIZED);
592     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
593     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
594     CHECK_CALLED(GetLCID);
595     CHECK_CALLED(OnStateChange_INITIALIZED);
596
597     SET_EXPECT(OnStateChange_CONNECTED);
598     SET_EXPECT(OnEnterScript);
599     SET_EXPECT(OnLeaveScript);
600     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_CONNECTED);
601     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
602     CHECK_CALLED(OnStateChange_CONNECTED);
603     CHECK_CALLED(OnEnterScript);
604     CHECK_CALLED(OnLeaveScript);
605
606     test_state(script, SCRIPTSTATE_CONNECTED);
607
608     dispex = get_script_dispatch(script);
609     ok(dispex != NULL, "dispex == NULL\n");
610     IDispatchEx_Release(dispex);
611
612     SET_EXPECT(OnStateChange_DISCONNECTED);
613     SET_EXPECT(OnStateChange_INITIALIZED);
614     SET_EXPECT(OnStateChange_UNINITIALIZED);
615     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
616     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
617     CHECK_CALLED(OnStateChange_DISCONNECTED);
618     CHECK_CALLED(OnStateChange_INITIALIZED);
619     CHECK_CALLED(OnStateChange_UNINITIALIZED);
620
621     test_state(script, SCRIPTSTATE_UNINITIALIZED);
622
623     hres = IActiveScript_Close(script);
624     ok(hres == S_OK, "Close failed: %08x\n", hres);
625
626     test_state(script, SCRIPTSTATE_CLOSED);
627
628     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
629     ok(hres == E_UNEXPECTED, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x, expected E_UNEXPECTED\n", hres);
630
631     test_state(script, SCRIPTSTATE_CLOSED);
632
633     IUnknown_Release(parse);
634
635     ref = IActiveScript_Release(script);
636     ok(!ref, "ref = %d\n", ref);
637 }
638
639 static void test_aggregation(void)
640 {
641     IUnknown *unk = (IUnknown*)0xdeadbeef;
642     HRESULT hres;
643
644     hres = CoCreateInstance(&CLSID_JScript, (IUnknown*)0xdeadbeef, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
645             &IID_IUnknown, (void**)&unk);
646     ok(hres == CLASS_E_NOAGGREGATION || broken(E_INVALIDARG) /* win2k */,
647        "CoCreateInstance failed: %08x, expected CLASS_E_NOAGGREGATION\n", hres);
648     ok(!unk || broken(unk != NULL), "unk = %p\n", unk);
649 }
650
651 static BOOL check_jscript(void)
652 {
653     IActiveScriptProperty *script_prop;
654     HRESULT hres;
655
656     hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
657             &IID_IActiveScriptProperty, (void**)&script_prop);
658     if(SUCCEEDED(hres))
659         IActiveScriptProperty_Release(script_prop);
660
661     return hres == S_OK;
662 }
663
664 START_TEST(jscript)
665 {
666     CoInitialize(NULL);
667
668     if(check_jscript()) {
669         trace("Testing JScript object...\n");
670         test_jscript();
671         test_jscript2();
672         test_jscript_uninitializing();
673         test_aggregation();
674
675         trace("Testing JScriptEncode object...\n");
676         engine_clsid = &CLSID_JScriptEncode;
677         test_jscript();
678     }else {
679         win_skip("Broken engine, probably too old\n");
680     }
681
682     CoUninitialize();
683 }