comctl32/treeview: Use separate pointer for currently edited item.
[wine] / dlls / jscript / tests / run.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 #include <stdio.h>
20
21 #define COBJMACROS
22 #define CONST_VTABLE
23
24 #include <ole2.h>
25 #include <dispex.h>
26 #include <activscp.h>
27
28 #include "wine/test.h"
29
30 static const CLSID CLSID_JScript =
31     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
32
33 #define DEFINE_EXPECT(func) \
34     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
35
36 #define SET_EXPECT(func) \
37     expect_ ## func = TRUE
38
39 #define SET_CALLED(func) \
40     called_ ## func = TRUE
41
42 #define CHECK_EXPECT2(func) \
43     do { \
44         ok(expect_ ##func, "unexpected call " #func "\n"); \
45         called_ ## func = TRUE; \
46     }while(0)
47
48 #define CHECK_EXPECT(func) \
49     do { \
50         CHECK_EXPECT2(func); \
51         expect_ ## func = FALSE; \
52     }while(0)
53
54 #define CHECK_CALLED(func) \
55     do { \
56         ok(called_ ## func, "expected " #func "\n"); \
57         expect_ ## func = called_ ## func = FALSE; \
58     }while(0)
59
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);
66 DEFINE_EXPECT(testobj_delete);
67 DEFINE_EXPECT(GetItemInfo_testVal);
68
69 #define DISPID_GLOBAL_TESTPROPGET   0x1000
70 #define DISPID_GLOBAL_TESTPROPPUT   0x1001
71 #define DISPID_GLOBAL_REPORTSUCCESS 0x1002
72 #define DISPID_GLOBAL_TRACE         0x1003
73 #define DISPID_GLOBAL_OK            0x1004
74 #define DISPID_GLOBAL_GETVT         0x1005
75 #define DISPID_GLOBAL_TESTOBJ       0x1006
76 #define DISPID_GLOBAL_NULL_BSTR     0x1007
77
78 static const WCHAR testW[] = {'t','e','s','t',0};
79 static const CHAR testA[] = "test";
80 static const WCHAR test_valW[] = {'t','e','s','t','V','a','l',0};
81 static const CHAR test_valA[] = "testVal";
82
83 static BOOL strict_dispid_check;
84 static const char *test_name = "(null)";
85
86 static const char *debugstr_w(LPCWSTR str)
87 {
88     static char buf[1024];
89
90     if(!str)
91         return "(null)";
92
93     WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL);
94     return buf;
95 }
96
97 static BSTR a2bstr(const char *str)
98 {
99     BSTR ret;
100     int len;
101
102     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
103     ret = SysAllocStringLen(NULL, len-1);
104     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
105
106     return ret;
107 }
108
109 static int strcmp_wa(LPCWSTR strw, const char *stra)
110 {
111     CHAR buf[512];
112     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), 0, 0);
113     return lstrcmpA(buf, stra);
114 }
115
116 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
117 {
118     *ppv = NULL;
119
120     if(IsEqualGUID(riid, &IID_IUnknown)
121        || IsEqualGUID(riid, &IID_IDispatch)
122        || IsEqualGUID(riid, &IID_IDispatchEx))
123         *ppv = iface;
124     else
125         return E_NOINTERFACE;
126
127     return S_OK;
128 }
129
130 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
131 {
132     return 2;
133 }
134
135 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
136 {
137     return 1;
138 }
139
140 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
141 {
142     ok(0, "unexpected call\n");
143     return E_NOTIMPL;
144 }
145
146 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
147                                               LCID lcid, ITypeInfo **ppTInfo)
148 {
149     ok(0, "unexpected call\n");
150     return E_NOTIMPL;
151 }
152
153 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
154                                                 LPOLESTR *rgszNames, UINT cNames,
155                                                 LCID lcid, DISPID *rgDispId)
156 {
157     ok(0, "unexpected call\n");
158     return E_NOTIMPL;
159 }
160
161 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
162                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
163                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
164 {
165     ok(0, "unexpected call\n");
166     return E_NOTIMPL;
167 }
168
169 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
170 {
171     ok(0, "unexpected call %s %x\n", debugstr_w(bstrName), grfdex);
172     return E_NOTIMPL;
173 }
174
175 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
176 {
177     ok(0, "unexpected call\n");
178     return E_NOTIMPL;
179 }
180
181 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
182 {
183     ok(0, "unexpected call\n");
184     return E_NOTIMPL;
185 }
186
187 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
188 {
189     ok(0, "unexpected call\n");
190     return E_NOTIMPL;
191 }
192
193 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
194 {
195     ok(0, "unexpected call\n");
196     return E_NOTIMPL;
197 }
198
199 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
200 {
201     ok(0, "unexpected call\n");
202     return E_NOTIMPL;
203 }
204
205 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
206 {
207     ok(0, "unexpected call\n");
208     return E_NOTIMPL;
209 }
210
211 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
212         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
213 {
214     ok(0, "unexpected call\n");
215     return E_NOTIMPL;
216 }
217
218 static HRESULT WINAPI testObj_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
219 {
220     CHECK_EXPECT(testobj_delete);
221
222     ok(!strcmp_wa(bstrName, "deleteTest"), "unexpected name %s\n", debugstr_w(bstrName));
223     ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
224     return S_OK;
225 }
226
227 static IDispatchExVtbl testObjVtbl = {
228     DispatchEx_QueryInterface,
229     DispatchEx_AddRef,
230     DispatchEx_Release,
231     DispatchEx_GetTypeInfoCount,
232     DispatchEx_GetTypeInfo,
233     DispatchEx_GetIDsOfNames,
234     DispatchEx_Invoke,
235     DispatchEx_GetDispID,
236     DispatchEx_InvokeEx,
237     testObj_DeleteMemberByName,
238     DispatchEx_DeleteMemberByDispID,
239     DispatchEx_GetMemberProperties,
240     DispatchEx_GetMemberName,
241     DispatchEx_GetNextDispID,
242     DispatchEx_GetNameSpaceParent
243 };
244
245 static IDispatchEx testObj = { &testObjVtbl };
246
247 static HRESULT WINAPI Global_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
248 {
249     if(!strcmp_wa(bstrName, "ok")) {
250         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
251         *pid = DISPID_GLOBAL_OK;
252         return S_OK;
253     }
254     if(!strcmp_wa(bstrName, "trace")) {
255         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
256         *pid = DISPID_GLOBAL_TRACE;
257         return S_OK;
258     }
259     if(!strcmp_wa(bstrName, "reportSuccess")) {
260         CHECK_EXPECT(global_success_d);
261         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
262         *pid = DISPID_GLOBAL_REPORTSUCCESS;
263         return S_OK;
264     }
265     if(!strcmp_wa(bstrName, "testPropGet")) {
266         CHECK_EXPECT(global_propget_d);
267         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
268         *pid = DISPID_GLOBAL_TESTPROPGET;
269         return S_OK;
270     }
271     if(!strcmp_wa(bstrName, "testPropPut")) {
272         CHECK_EXPECT(global_propput_d);
273         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
274         *pid = DISPID_GLOBAL_TESTPROPPUT;
275         return S_OK;
276     }
277     if(!strcmp_wa(bstrName, "getVT")) {
278         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
279         *pid = DISPID_GLOBAL_GETVT;
280         return S_OK;
281     }
282     if(!strcmp_wa(bstrName, "testObj")) {
283         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
284         *pid = DISPID_GLOBAL_TESTOBJ;
285         return S_OK;
286     }
287     if(!strcmp_wa(bstrName, "createNullBSTR")) {
288         *pid = DISPID_GLOBAL_NULL_BSTR;
289         return S_OK;
290     }
291
292     if(strict_dispid_check)
293         ok(0, "unexpected call %s\n", debugstr_w(bstrName));
294     return DISP_E_UNKNOWNNAME;
295 }
296
297 static HRESULT WINAPI Global_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
298         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
299 {
300     switch(id) {
301     case DISPID_GLOBAL_OK:
302         ok(wFlags == INVOKE_FUNC || wFlags == (INVOKE_FUNC|INVOKE_PROPERTYGET), "wFlags = %x\n", wFlags);
303         ok(pdp != NULL, "pdp == NULL\n");
304         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
305         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
306         ok(pdp->cArgs == 2, "cArgs = %d\n", pdp->cArgs);
307         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
308         if(wFlags & INVOKE_PROPERTYGET)
309             ok(pvarRes != NULL, "pvarRes == NULL\n");
310         else
311             ok(!pvarRes, "pvarRes != NULL\n");
312         ok(pei != NULL, "pei == NULL\n");
313
314         ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
315         ok(V_VT(pdp->rgvarg+1) == VT_BOOL, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp->rgvarg));
316         ok(V_BOOL(pdp->rgvarg+1), "%s: %s\n", test_name, debugstr_w(V_BSTR(pdp->rgvarg)));
317
318         return S_OK;
319
320      case DISPID_GLOBAL_TRACE:
321         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
322         ok(pdp != NULL, "pdp == NULL\n");
323         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
324         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
325         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
326         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
327         ok(!pvarRes, "pvarRes != NULL\n");
328         ok(pei != NULL, "pei == NULL\n");
329
330         ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
331         if(V_VT(pdp->rgvarg) == VT_BSTR)
332             trace("%s: %s\n", test_name, debugstr_w(V_BSTR(pdp->rgvarg)));
333
334         return S_OK;
335
336     case DISPID_GLOBAL_REPORTSUCCESS:
337         CHECK_EXPECT(global_success_i);
338
339         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
340         ok(pdp != NULL, "pdp == NULL\n");
341         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
342         ok(pdp->cArgs == 0, "cArgs = %d\n", pdp->cArgs);
343         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
344         ok(!pvarRes, "pvarRes != NULL\n");
345         ok(pei != NULL, "pei == NULL\n");
346
347         return S_OK;
348
349      case DISPID_GLOBAL_TESTPROPGET:
350         CHECK_EXPECT(global_propget_i);
351
352         ok(wFlags == INVOKE_PROPERTYGET, "wFlags = %x\n", wFlags);
353         ok(pdp != NULL, "pdp == NULL\n");
354         ok(!pdp->rgvarg, "rgvarg != NULL\n");
355         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
356         ok(!pdp->cArgs, "cArgs = %d\n", pdp->cArgs);
357         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
358         ok(pvarRes != NULL, "pvarRes == NULL\n");
359         ok(V_VT(pvarRes) ==  VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
360         ok(pei != NULL, "pei == NULL\n");
361
362         V_VT(pvarRes) = VT_I4;
363         V_I4(pvarRes) = 1;
364
365         return S_OK;
366
367     case DISPID_GLOBAL_TESTPROPPUT:
368         CHECK_EXPECT(global_propput_i);
369
370         ok(wFlags == INVOKE_PROPERTYPUT, "wFlags = %x\n", wFlags);
371         ok(pdp != NULL, "pdp == NULL\n");
372         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
373         ok(pdp->rgdispidNamedArgs != NULL, "rgdispidNamedArgs == NULL\n");
374         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
375         ok(pdp->cNamedArgs == 1, "cNamedArgs = %d\n", pdp->cNamedArgs);
376         ok(pdp->rgdispidNamedArgs[0] == DISPID_PROPERTYPUT, "pdp->rgdispidNamedArgs[0] = %d\n", pdp->rgdispidNamedArgs[0]);
377         ok(!pvarRes, "pvarRes != NULL\n");
378
379         ok(V_VT(pdp->rgvarg) == VT_I4, "V_VT(pdp->rgvarg)=%d\n", V_VT(pdp->rgvarg));
380         ok(V_I4(pdp->rgvarg) == 1, "V_I4(pdp->rgvarg)=%d\n", V_I4(pdp->rgvarg));
381         return S_OK;
382
383      case DISPID_GLOBAL_GETVT:
384         ok(pdp != NULL, "pdp == NULL\n");
385         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
386         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
387         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
388         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
389         ok(pvarRes != NULL, "pvarRes == NULL\n");
390         ok(V_VT(pvarRes) ==  VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
391         ok(pei != NULL, "pei == NULL\n");
392
393         V_VT(pvarRes) = VT_BSTR;
394         switch(V_VT(pdp->rgvarg)) {
395         case VT_EMPTY:
396             V_BSTR(pvarRes) = a2bstr("VT_EMPTY");
397             break;
398         case VT_NULL:
399             V_BSTR(pvarRes) = a2bstr("VT_NULL");
400             break;
401         case VT_I4:
402             V_BSTR(pvarRes) = a2bstr("VT_I4");
403             break;
404         case VT_R8:
405             V_BSTR(pvarRes) = a2bstr("VT_R8");
406             break;
407         case VT_BSTR:
408             V_BSTR(pvarRes) = a2bstr("VT_BSTR");
409             break;
410         case VT_DISPATCH:
411             V_BSTR(pvarRes) = a2bstr("VT_DISPATCH");
412             break;
413         case VT_BOOL:
414             V_BSTR(pvarRes) = a2bstr("VT_BOOL");
415             break;
416         default:
417             ok(0, "unknown vt %d\n", V_VT(pdp->rgvarg));
418             return E_FAIL;
419         }
420
421         return S_OK;
422
423     case DISPID_GLOBAL_TESTOBJ:
424         ok(wFlags == INVOKE_PROPERTYGET, "wFlags = %x\n", wFlags);
425         ok(pdp != NULL, "pdp == NULL\n");
426         ok(!pdp->rgvarg, "rgvarg != NULL\n");
427         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
428         ok(!pdp->cArgs, "cArgs = %d\n", pdp->cArgs);
429         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
430         ok(pvarRes != NULL, "pvarRes == NULL\n");
431         ok(V_VT(pvarRes) ==  VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
432         ok(pei != NULL, "pei == NULL\n");
433
434         V_VT(pvarRes) = VT_DISPATCH;
435         V_DISPATCH(pvarRes) = (IDispatch*)&testObj;
436         return S_OK;
437
438     case DISPID_GLOBAL_NULL_BSTR:
439         if(pvarRes) {
440             V_VT(pvarRes) = VT_BSTR;
441             V_BSTR(pvarRes) = NULL;
442         }
443         return S_OK;
444     }
445
446     ok(0, "unexpected call %x\n", id);
447     return DISP_E_MEMBERNOTFOUND;
448 }
449
450 static IDispatchExVtbl GlobalVtbl = {
451     DispatchEx_QueryInterface,
452     DispatchEx_AddRef,
453     DispatchEx_Release,
454     DispatchEx_GetTypeInfoCount,
455     DispatchEx_GetTypeInfo,
456     DispatchEx_GetIDsOfNames,
457     DispatchEx_Invoke,
458     Global_GetDispID,
459     Global_InvokeEx,
460     DispatchEx_DeleteMemberByName,
461     DispatchEx_DeleteMemberByDispID,
462     DispatchEx_GetMemberProperties,
463     DispatchEx_GetMemberName,
464     DispatchEx_GetNextDispID,
465     DispatchEx_GetNameSpaceParent
466 };
467
468 static IDispatchEx Global = { &GlobalVtbl };
469
470 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
471 {
472     *ppv = NULL;
473
474     if(IsEqualGUID(&IID_IUnknown, riid))
475         *ppv = iface;
476     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
477         *ppv = iface;
478     else
479         return E_NOINTERFACE;
480
481     IUnknown_AddRef((IUnknown*)*ppv);
482     return S_OK;
483 }
484
485 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
486 {
487     return 2;
488 }
489
490 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
491 {
492     return 1;
493 }
494
495 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
496 {
497     *plcid = GetUserDefaultLCID();
498     return S_OK;
499 }
500
501 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
502         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
503 {
504     ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
505     ok(!ppti, "ppti != NULL\n");
506
507     if(!strcmp_wa(pstrName, test_valA))
508         CHECK_EXPECT(GetItemInfo_testVal);
509     else if(strcmp_wa(pstrName, testA))
510         ok(0, "unexpected pstrName %s\n", debugstr_w(pstrName));
511
512     *ppiunkItem = (IUnknown*)&Global;
513     return S_OK;
514 }
515
516 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
517 {
518     return E_NOTIMPL;
519 }
520
521 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
522         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
523 {
524     return E_NOTIMPL;
525 }
526
527 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
528 {
529     return E_NOTIMPL;
530 }
531
532 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
533 {
534     return E_NOTIMPL;
535 }
536
537 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
538 {
539     return E_NOTIMPL;
540 }
541
542 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
543 {
544     return E_NOTIMPL;
545 }
546
547 #undef ACTSCPSITE_THIS
548
549 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
550     ActiveScriptSite_QueryInterface,
551     ActiveScriptSite_AddRef,
552     ActiveScriptSite_Release,
553     ActiveScriptSite_GetLCID,
554     ActiveScriptSite_GetItemInfo,
555     ActiveScriptSite_GetDocVersionString,
556     ActiveScriptSite_OnScriptTerminate,
557     ActiveScriptSite_OnStateChange,
558     ActiveScriptSite_OnScriptError,
559     ActiveScriptSite_OnEnterScript,
560     ActiveScriptSite_OnLeaveScript
561 };
562
563 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
564
565 static IActiveScript *create_script(void)
566 {
567     IActiveScript *script;
568     HRESULT hres;
569
570     hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
571             &IID_IActiveScript, (void**)&script);
572     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
573
574     return script;
575 }
576
577 static void parse_script(BSTR script_str)
578 {
579     IActiveScriptParse *parser;
580     IActiveScript *engine;
581     HRESULT hres;
582
583     engine = create_script();
584     if(!engine)
585         return;
586
587     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
588     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
589     if (FAILED(hres))
590     {
591         IActiveScript_Release(engine);
592         return;
593     }
594
595     hres = IActiveScriptParse64_InitNew(parser);
596     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
597
598     hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
599     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
600
601     hres = IActiveScript_AddNamedItem(engine, testW,
602             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
603     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
604
605     hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
606     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
607
608     hres = IActiveScriptParse64_ParseScriptText(parser, script_str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
609     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
610
611     IActiveScript_Release(engine);
612     IUnknown_Release(parser);
613 }
614
615 static HRESULT parse_htmlscript(BSTR script_str)
616 {
617     IActiveScriptParse *parser;
618     IActiveScript *engine;
619     HRESULT hres;
620     BSTR tmp = a2bstr("</SCRIPT>");
621
622     engine = create_script();
623     if(!engine)
624         return E_FAIL;
625
626     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
627     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
628     if (FAILED(hres))
629     {
630         IActiveScript_Release(engine);
631         return E_FAIL;
632     }
633
634     hres = IActiveScriptParse64_InitNew(parser);
635     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
636
637     hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
638     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
639
640     hres = IActiveScript_AddNamedItem(engine, testW,
641             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
642     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
643
644     hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
645     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
646
647     hres = IActiveScriptParse64_ParseScriptText(parser, script_str, NULL, NULL, tmp, 0, 0, 0, NULL, NULL);
648
649     IActiveScript_Release(engine);
650     IUnknown_Release(parser);
651     SysFreeString(tmp);
652
653     return hres;
654 }
655
656 static void parse_script_a(const char *src)
657 {
658     BSTR tmp = a2bstr(src);
659     parse_script(tmp);
660     SysFreeString(tmp);
661 }
662
663 static HRESULT parse_htmlscript_a(const char *src)
664 {
665     HRESULT hres;
666     BSTR tmp = a2bstr(src);
667     hres = parse_htmlscript(tmp);
668     SysFreeString(tmp);
669
670     return hres;
671 }
672
673 static BSTR get_script_from_file(const char *filename)
674 {
675     DWORD size, len;
676     HANDLE file, map;
677     const char *file_map;
678     BSTR ret;
679
680     file = CreateFileA(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
681     if(file == INVALID_HANDLE_VALUE) {
682         trace("Could not open file: %u\n", GetLastError());
683         return NULL;
684     }
685
686     size = GetFileSize(file, NULL);
687
688     map = CreateFileMapping(file, NULL, PAGE_READONLY, 0, 0, NULL);
689     CloseHandle(file);
690     if(map == INVALID_HANDLE_VALUE) {
691         trace("Could not create file mapping: %u\n", GetLastError());
692         return NULL;
693     }
694
695     file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
696     CloseHandle(map);
697     if(!file_map) {
698         trace("MapViewOfFile failed: %u\n", GetLastError());
699         return NULL;
700     }
701
702     len = MultiByteToWideChar(CP_ACP, 0, file_map, size, NULL, 0);
703     ret = SysAllocStringLen(NULL, len);
704     MultiByteToWideChar(CP_ACP, 0, file_map, size, ret, len);
705
706     UnmapViewOfFile(file_map);
707
708     return ret;
709 }
710
711 static void run_from_file(const char *filename)
712 {
713     BSTR script_str = get_script_from_file(filename);
714
715     strict_dispid_check = FALSE;
716
717     if(script_str)
718         parse_script(script_str);
719
720     SysFreeString(script_str);
721 }
722
723 static void run_from_res(const char *name)
724 {
725     const char *data;
726     DWORD size, len;
727     BSTR str;
728     HRSRC src;
729
730     strict_dispid_check = FALSE;
731     test_name = name;
732
733     src = FindResourceA(NULL, name, (LPCSTR)40);
734     ok(src != NULL, "Could not find resource %s\n", name);
735
736     size = SizeofResource(NULL, src);
737     data = LoadResource(NULL, src);
738
739     len = MultiByteToWideChar(CP_ACP, 0, data, size, NULL, 0);
740     str = SysAllocStringLen(NULL, len);
741     len = MultiByteToWideChar(CP_ACP, 0, data, size, str, len);
742
743     SET_EXPECT(global_success_d);
744     SET_EXPECT(global_success_i);
745     parse_script(str);
746     CHECK_CALLED(global_success_d);
747     CHECK_CALLED(global_success_i);
748
749     SysFreeString(str);
750 }
751
752 static void test_isvisible(BOOL global_members)
753 {
754     IActiveScriptParse *parser;
755     IActiveScript *engine;
756     HRESULT hres;
757
758     static const WCHAR script_textW[] =
759         {'v','a','r',' ','v',' ','=',' ','t','e','s','t','V','a','l',';',0};
760
761     engine = create_script();
762     if(!engine)
763         return;
764
765     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
766     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
767     if (FAILED(hres))
768     {
769         IActiveScript_Release(engine);
770         return;
771     }
772
773     hres = IActiveScriptParse64_InitNew(parser);
774     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
775
776     hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
777     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
778
779     if(global_members)
780         SET_EXPECT(GetItemInfo_testVal);
781     hres = IActiveScript_AddNamedItem(engine, test_valW,
782             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|
783             (global_members ? SCRIPTITEM_GLOBALMEMBERS : 0));
784     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
785     if(global_members)
786         CHECK_CALLED(GetItemInfo_testVal);
787
788     hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
789     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
790
791     if(!global_members)
792         SET_EXPECT(GetItemInfo_testVal);
793     hres = IActiveScriptParse64_ParseScriptText(parser, script_textW, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
794     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
795     if(!global_members)
796         CHECK_CALLED(GetItemInfo_testVal);
797
798     hres = IActiveScriptParse64_ParseScriptText(parser, script_textW, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
799     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
800
801     IActiveScript_Release(engine);
802     IUnknown_Release(parser);
803 }
804
805 static void run_tests(void)
806 {
807     HRESULT hres;
808
809     strict_dispid_check = TRUE;
810
811     parse_script_a("");
812     parse_script_a("/* empty */ ;");
813
814     SET_EXPECT(global_propget_d);
815     SET_EXPECT(global_propget_i);
816     parse_script_a("testPropGet;");
817     CHECK_CALLED(global_propget_d);
818     CHECK_CALLED(global_propget_i);
819
820     SET_EXPECT(global_propput_d);
821     SET_EXPECT(global_propput_i);
822     parse_script_a("testPropPut = 1;");
823     CHECK_CALLED(global_propput_d);
824     CHECK_CALLED(global_propput_i);
825
826     SET_EXPECT(global_success_d);
827     SET_EXPECT(global_success_i);
828     parse_script_a("reportSuccess();");
829     CHECK_CALLED(global_success_d);
830     CHECK_CALLED(global_success_i);
831
832     SET_EXPECT(testobj_delete);
833     parse_script_a("delete testObj.deleteTest;");
834     CHECK_CALLED(testobj_delete);
835
836     parse_script_a("ok(typeof(test) === 'object', \"typeof(test) != 'object'\");");
837
838     run_from_res("lang.js");
839     run_from_res("api.js");
840     run_from_res("regexp.js");
841
842     test_isvisible(FALSE);
843     test_isvisible(TRUE);
844
845     hres = parse_htmlscript_a("<!--");
846     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
847     hres = parse_htmlscript_a("-->");
848     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
849     hres = parse_htmlscript_a("<!--\nvar a=1;\n-->\n");
850     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
851     hres = parse_htmlscript_a("<!--\n<!-- ignore this\n-->\n");
852     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
853     hres = parse_htmlscript_a("var a=1;\nif(a-->0) a=5;\n");
854     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
855     hres = parse_htmlscript_a("var a=1;\nif(a\n-->0) a=5;\n");
856     ok(hres != S_OK, "ParseScriptText have not failed\n");
857 }
858
859 START_TEST(run)
860 {
861     int argc;
862     char **argv;
863
864     argc = winetest_get_mainargs(&argv);
865
866     CoInitialize(NULL);
867
868     if(argc > 2)
869         run_from_file(argv[2]);
870     else
871         run_tests();
872
873     CoUninitialize();
874 }