wbemprox: Add support for enumerating class properties.
[wine] / dlls / jscript / tests / caller.c
1 /*
2  * Copyright 2012 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 #include <objsafe.h>
28
29 #include "wine/test.h"
30
31 #ifdef _WIN64
32
33 #define IActiveScriptParse_QueryInterface IActiveScriptParse64_QueryInterface
34 #define IActiveScriptParse_Release IActiveScriptParse64_Release
35 #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
36 #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
37
38 #else
39
40 #define IActiveScriptParse_QueryInterface IActiveScriptParse32_QueryInterface
41 #define IActiveScriptParse_Release IActiveScriptParse32_Release
42 #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
43 #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
44
45 #endif
46
47 static const CLSID CLSID_JScript =
48     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
49
50 #define DEFINE_EXPECT(func) \
51     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
52
53 #define SET_EXPECT(func) \
54     expect_ ## func = TRUE
55
56 #define CHECK_EXPECT2(func) \
57     do { \
58         ok(expect_ ##func, "unexpected call " #func "\n"); \
59         called_ ## func = TRUE; \
60     }while(0)
61
62 #define CHECK_EXPECT(func) \
63     do { \
64         CHECK_EXPECT2(func); \
65         expect_ ## func = FALSE; \
66     }while(0)
67
68 #define CHECK_CALLED(func) \
69     do { \
70         ok(called_ ## func, "expected " #func "\n"); \
71         expect_ ## func = called_ ## func = FALSE; \
72     }while(0)
73
74 DEFINE_EXPECT(testArgConv);
75
76 static const WCHAR testW[] = {'t','e','s','t',0};
77
78 static IVariantChangeType *script_change_type;
79 static IDispatch *stored_obj;
80
81 #define DISPID_TEST_TESTARGCONV      0x1000
82
83 static BSTR a2bstr(const char *str)
84 {
85     BSTR ret;
86     int len;
87
88     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
89     ret = SysAllocStringLen(NULL, len-1);
90     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
91
92     return ret;
93 }
94
95 static int strcmp_wa(LPCWSTR strw, const char *stra)
96 {
97     CHAR buf[512];
98     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), 0, 0);
99     return lstrcmpA(buf, stra);
100 }
101
102 typedef struct {
103     int int_result;
104     const char *str_result;
105     VARIANT_BOOL bool_result;
106     int test_double;
107     double double_result;
108 } conv_results_t;
109
110 #define call_change_type(a,b,c,d) _call_change_type(__LINE__,a,b,c,d)
111 static void _call_change_type(unsigned line, IVariantChangeType *change_type, VARIANT *dst, VARIANT *src, VARTYPE vt)
112 {
113     HRESULT hres;
114
115     VariantInit(dst);
116     hres = IVariantChangeType_ChangeType(change_type, dst, src, 0, vt);
117     ok_(__FILE__,line)(hres == S_OK, "ChangeType(%d) failed: %08x\n", vt, hres);
118     ok_(__FILE__,line)(V_VT(dst) == vt, "V_VT(dst) = %d\n", V_VT(dst));
119 }
120
121 #define change_type_fail(a,b,c,d) _change_type_fail(__LINE__,a,b,c,d)
122 static void _change_type_fail(unsigned line, IVariantChangeType *change_type, VARIANT *src, VARTYPE vt, HRESULT exhres)
123 {
124     VARIANT v;
125     HRESULT hres;
126
127     V_VT(&v) = VT_EMPTY;
128     hres = IVariantChangeType_ChangeType(change_type, &v, src, 0, vt);
129     ok_(__FILE__,line)(hres == exhres, "ChangeType failed: %08x, expected %08x\n", hres, exhres);
130 }
131
132 static void test_change_type(IVariantChangeType *change_type, VARIANT *src, const conv_results_t *ex)
133 {
134     VARIANT v;
135
136     call_change_type(change_type, &v, src, VT_I4);
137     ok(V_I4(&v) == ex->int_result, "V_I4(v) = %d, expected %d\n", V_I4(&v), ex->int_result);
138
139     call_change_type(change_type, &v, src, VT_BSTR);
140     ok(!strcmp_wa(V_BSTR(&v), ex->str_result), "V_BSTR(v) = %s, expected %s\n", wine_dbgstr_w(V_BSTR(&v)), ex->str_result);
141     VariantClear(&v);
142
143     call_change_type(change_type, &v, src, VT_BOOL);
144     ok(V_BOOL(&v) == ex->bool_result, "V_BOOL(v) = %x, expected %x\n", V_BOOL(&v), ex->bool_result);
145
146     if(ex->test_double) {
147         call_change_type(change_type, &v, src, VT_R8);
148         ok(V_R8(&v) == ex->double_result, "V_R8(v) = %lf, expected %lf\n", V_R8(&v), ex->double_result);
149
150         call_change_type(change_type, &v, src, VT_R4);
151         ok(V_R4(&v) == (float)ex->double_result, "V_R4(v) = %f, expected %f\n", V_R4(&v), (float)ex->double_result);
152     }
153
154     if(V_VT(src) == VT_NULL)
155         call_change_type(change_type, &v, src, VT_NULL);
156     else
157         change_type_fail(change_type, src, VT_NULL, E_NOTIMPL);
158
159     if(V_VT(src) == VT_EMPTY)
160         call_change_type(change_type, &v, src, VT_EMPTY);
161     else
162         change_type_fail(change_type, src, VT_EMPTY, E_NOTIMPL);
163
164     call_change_type(change_type, &v, src, VT_I2);
165     ok(V_I2(&v) == (INT16)ex->int_result, "V_I2(v) = %d, expected %d\n", V_I2(&v), ex->int_result);
166 }
167
168 static void test_change_types(IVariantChangeType *change_type, IDispatch *obj_disp)
169 {
170     VARIANT v, dst;
171     HRESULT hres;
172
173     static const conv_results_t bool_results[] = {
174         {0, "false", VARIANT_FALSE, 1,0.0},
175         {1, "true", VARIANT_TRUE, 1,1.0}};
176     static const conv_results_t int_results[] = {
177         {0, "0", VARIANT_FALSE, 1,0.0},
178         {-100, "-100", VARIANT_TRUE, 1,-100.0},
179         {0x10010, "65552", VARIANT_TRUE, 1,65552.0}};
180     static const conv_results_t empty_results =
181         {0, "undefined", VARIANT_FALSE, 0,0};
182     static const conv_results_t null_results =
183         {0, "null", VARIANT_FALSE, 0,0};
184     static const conv_results_t obj_results =
185         {10, "strval", VARIANT_TRUE, 1,10.0};
186
187     V_VT(&v) = VT_BOOL;
188     V_BOOL(&v) = VARIANT_FALSE;
189     test_change_type(change_type, &v, bool_results);
190     V_BOOL(&v) = VARIANT_TRUE;
191     test_change_type(change_type, &v, bool_results+1);
192
193     V_VT(&v) = VT_I4;
194     V_I4(&v) = 0;
195     test_change_type(change_type, &v, int_results);
196     V_I4(&v) = -100;
197     test_change_type(change_type, &v, int_results+1);
198     V_I4(&v) = 0x10010;
199     test_change_type(change_type, &v, int_results+2);
200
201     V_VT(&v) = VT_EMPTY;
202     test_change_type(change_type, &v, &empty_results);
203
204     V_VT(&v) = VT_NULL;
205     test_change_type(change_type, &v, &null_results);
206
207     V_VT(&v) = VT_DISPATCH;
208     V_DISPATCH(&v) = obj_disp;
209     test_change_type(change_type, &v, &obj_results);
210
211     V_VT(&v) = VT_BOOL;
212     V_BOOL(&v) = VARIANT_FALSE;
213     V_VT(&dst) = 0xdead;
214     hres = IVariantChangeType_ChangeType(change_type, &dst, &v, 0, VT_I4);
215     ok(hres == DISP_E_BADVARTYPE, "ChangeType failed: %08x, expected DISP_E_BADVARTYPE\n", hres);
216     ok(V_VT(&dst) == 0xdead, "V_VT(dst) = %d\n", V_VT(&dst));
217 }
218
219 static void test_caller(IServiceProvider *caller, IDispatch *arg_obj)
220 {
221     IVariantChangeType *change_type;
222     HRESULT hres;
223
224     hres = IServiceProvider_QueryService(caller, &SID_VariantConversion, &IID_IVariantChangeType, (void**)&change_type);
225     ok(hres == S_OK, "Could not get SID_VariantConversion service: %08x\n", hres);
226
227     ok(change_type == script_change_type, "change_type != script_change_type\n");
228     test_change_types(change_type, arg_obj);
229
230     IVariantChangeType_Release(change_type);
231 }
232
233 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
234 {
235     if(IsEqualGUID(riid, &IID_IUnknown)) {
236         *ppv = iface;
237     }else if(IsEqualGUID(riid, &IID_IDispatch) || IsEqualGUID(riid, &IID_IDispatchEx)) {
238         *ppv = iface;
239     }else if(IsEqualGUID(&IID_IObjectSafety, riid)) {
240         ok(0, "Unexpected IID_IObjectSafety query\n");
241     }else {
242         *ppv = NULL;
243         return E_NOINTERFACE;
244     }
245
246     return S_OK;
247 }
248
249 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
250 {
251     return 2;
252 }
253
254 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
255 {
256     return 1;
257 }
258
259 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
260 {
261     ok(0, "unexpected call\n");
262     return E_NOTIMPL;
263 }
264
265 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
266                                               LCID lcid, ITypeInfo **ppTInfo)
267 {
268     ok(0, "unexpected call\n");
269     return E_NOTIMPL;
270 }
271
272 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
273                                                 LPOLESTR *rgszNames, UINT cNames,
274                                                 LCID lcid, DISPID *rgDispId)
275 {
276     ok(0, "unexpected call\n");
277     return E_NOTIMPL;
278 }
279
280 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
281                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
282                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
283 {
284     ok(0, "unexpected call\n");
285     return E_NOTIMPL;
286 }
287
288 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
289 {
290     ok(0, "unexpected call %s %x\n", wine_dbgstr_w(bstrName), grfdex);
291     return E_NOTIMPL;
292 }
293
294 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
295 {
296     ok(0, "unexpected call\n");
297     return E_NOTIMPL;
298 }
299
300 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
301 {
302     ok(0, "unexpected call\n");
303     return E_NOTIMPL;
304 }
305
306 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
307 {
308     ok(0, "unexpected call\n");
309     return E_NOTIMPL;
310 }
311
312 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
313 {
314     ok(0, "unexpected call\n");
315     return E_NOTIMPL;
316 }
317
318 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
319 {
320     ok(0, "unexpected call\n");
321     return E_NOTIMPL;
322 }
323
324 static HRESULT WINAPI Test_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
325 {
326     if(!strcmp_wa(bstrName, "testArgConv")) {
327         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
328         *pid = DISPID_TEST_TESTARGCONV;
329         return S_OK;
330     }
331
332     return E_NOTIMPL;
333 }
334
335 static HRESULT WINAPI Test_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
336         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
337 {
338     ok(pspCaller != NULL, "pspCaller == NULL\n");
339
340     switch(id) {
341     case DISPID_TEST_TESTARGCONV:
342         CHECK_EXPECT(testArgConv);
343
344         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
345         ok(pdp != NULL, "pdp == NULL\n");
346         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
347         ok(!pvarRes, "pvarRes != NULL\n");
348         ok(pei != NULL, "pei == NULL\n");
349
350         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
351         ok(V_VT(pdp->rgvarg) == VT_DISPATCH, "V_VT(rgvarg) = %d\n", V_VT(pdp->rgvarg));
352
353         test_caller(pspCaller, V_DISPATCH(pdp->rgvarg));
354
355         stored_obj = V_DISPATCH(pdp->rgvarg);
356         IDispatch_AddRef(stored_obj);
357         break;
358
359     default:
360         ok(0, "unexpected call\n");
361         return E_NOTIMPL;
362     }
363
364     return S_OK;
365 }
366
367 static IDispatchExVtbl testObjVtbl = {
368     DispatchEx_QueryInterface,
369     DispatchEx_AddRef,
370     DispatchEx_Release,
371     DispatchEx_GetTypeInfoCount,
372     DispatchEx_GetTypeInfo,
373     DispatchEx_GetIDsOfNames,
374     DispatchEx_Invoke,
375     Test_GetDispID,
376     Test_InvokeEx,
377     DispatchEx_DeleteMemberByName,
378     DispatchEx_DeleteMemberByDispID,
379     DispatchEx_GetMemberProperties,
380     DispatchEx_GetMemberName,
381     DispatchEx_GetNextDispID,
382     DispatchEx_GetNameSpaceParent
383 };
384
385 static IDispatchEx testObj = { &testObjVtbl };
386
387 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
388 {
389     if(IsEqualGUID(&IID_IUnknown, riid)) {
390         *ppv = iface;
391     }else if(IsEqualGUID(&IID_IActiveScriptSite, riid)) {
392         *ppv = iface;
393     }else {
394         *ppv = NULL;
395         return E_NOINTERFACE;
396     }
397
398     IUnknown_AddRef((IUnknown*)*ppv);
399     return S_OK;
400 }
401
402 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
403 {
404     return 2;
405 }
406
407 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
408 {
409     return 1;
410 }
411
412 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
413 {
414     *plcid = GetUserDefaultLCID();
415     return S_OK;
416 }
417
418 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
419         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
420 {
421     ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
422     ok(!ppti, "ppti != NULL\n");
423     ok(!strcmp_wa(pstrName, "test"), "pstrName = %s\n", wine_dbgstr_w(pstrName));
424
425     *ppiunkItem = (IUnknown*)&testObj;
426     return S_OK;
427 }
428
429 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
430 {
431     return E_NOTIMPL;
432 }
433
434 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
435         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
436 {
437     return E_NOTIMPL;
438 }
439
440 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
441 {
442     return E_NOTIMPL;
443 }
444
445 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
446 {
447     return E_NOTIMPL;
448 }
449
450 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
451 {
452     return E_NOTIMPL;
453 }
454
455 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
456 {
457     return E_NOTIMPL;
458 }
459
460 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
461     ActiveScriptSite_QueryInterface,
462     ActiveScriptSite_AddRef,
463     ActiveScriptSite_Release,
464     ActiveScriptSite_GetLCID,
465     ActiveScriptSite_GetItemInfo,
466     ActiveScriptSite_GetDocVersionString,
467     ActiveScriptSite_OnScriptTerminate,
468     ActiveScriptSite_OnStateChange,
469     ActiveScriptSite_OnScriptError,
470     ActiveScriptSite_OnEnterScript,
471     ActiveScriptSite_OnLeaveScript
472 };
473
474 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
475
476 #define parse_script_a(p,s) _parse_script_a(__LINE__,p,s)
477 static void _parse_script_a(unsigned line, IActiveScriptParse *parser, const char *script)
478 {
479     BSTR str;
480     HRESULT hres;
481
482     str = a2bstr(script);
483     hres = IActiveScriptParse_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
484     SysFreeString(str);
485     ok_(__FILE__,line)(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
486 }
487
488 static IActiveScriptParse *create_script(void)
489 {
490     IActiveScriptParse *parser;
491     IActiveScript *script;
492     HRESULT hres;
493
494     hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
495             &IID_IActiveScript, (void**)&script);
496     if(FAILED(hres))
497         return NULL;
498
499     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parser);
500     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
501
502     hres = IActiveScriptParse_InitNew(parser);
503     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
504
505     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
506     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
507
508     hres = IActiveScript_AddNamedItem(script, testW,
509             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
510     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
511
512     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
513     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
514
515     IActiveScript_Release(script);
516
517     return parser;
518 }
519
520 static void run_scripts(void)
521 {
522     IActiveScriptParse *parser;
523     HRESULT hres;
524
525     parser = create_script();
526
527     hres = IActiveScriptParse_QueryInterface(parser, &IID_IVariantChangeType, (void**)&script_change_type);
528     ok(hres == S_OK, "Could not get IVariantChangeType iface: %08x\n", hres);
529
530     SET_EXPECT(testArgConv);
531     parse_script_a(parser,
532                    "var obj = {"
533                    "    toString: function() { return 'strval'; },"
534                    "    valueOf: function()  { return 10; }"
535                    "};"
536                    "testArgConv(obj);");
537     CHECK_CALLED(testArgConv);
538
539     test_change_types(script_change_type, stored_obj);
540     IDispatch_Release(stored_obj);
541     IVariantChangeType_Release(script_change_type);
542
543     IActiveScriptParse_Release(parser);
544 }
545
546 static BOOL check_jscript(void)
547 {
548     IActiveScriptProperty *script_prop;
549     IActiveScriptParse *parser;
550     BSTR str;
551     HRESULT hres;
552
553     parser = create_script();
554     if(!parser)
555         return FALSE;
556
557     str = a2bstr("if(!('localeCompare' in String.prototype)) throw 1;");
558     hres = IActiveScriptParse_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
559     SysFreeString(str);
560
561     if(hres == S_OK)
562         hres = IActiveScriptParse_QueryInterface(parser, &IID_IActiveScriptProperty, (void**)&script_prop);
563     IActiveScriptParse_Release(parser);
564     if(hres == S_OK)
565         IActiveScriptProperty_Release(script_prop);
566
567     return hres == S_OK;
568 }
569
570 START_TEST(caller)
571 {
572     CoInitialize(NULL);
573
574     if(check_jscript())
575         run_scripts();
576     else
577         win_skip("Broken (too old) jscript\n");
578
579     CoUninitialize();
580 }