git.oblomov.eu Git - wine/blob - 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
  63 #define DISPID_GLOBAL_TESTPROPGET   0x1000
  64
  65 static const WCHAR testW[] = {'t','e','s','t',0};
  66
  67 static const char *debugstr_w(LPCWSTR str)
  68 {
  69     static char buf[1024];
  70
  71     if(!str)
  72         return "(null)";
  73
  74     WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL);
  75     return buf;
  76 }
  77
  78 static BSTR a2bstr(const char *str)
  79 {
  80     BSTR ret;
  81     int len;
  82
  83     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
  84     ret = SysAllocStringLen(NULL, len-1);
  85     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
  86
  87     return ret;
  88 }
  89
  90 static int strcmp_wa(LPCWSTR strw, const char *stra)
  91 {
  92     WCHAR buf[512];
  93     MultiByteToWideChar(CP_ACP, 0, stra, -1, buf, sizeof(buf)/sizeof(WCHAR));
  94     return lstrcmpW(strw, buf);
  95 }
  96
  97 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
  98 {
  99     *ppv = NULL;
 100
 101     if(IsEqualGUID(riid, &IID_IUnknown)
 102        || IsEqualGUID(riid, &IID_IDispatch)
 103        || IsEqualGUID(riid, &IID_IDispatchEx))
 104         *ppv = iface;
 105     else
 106         return E_NOINTERFACE;
 107
 108     return S_OK;
 109 }
 110
 111 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
 112 {
 113     return 2;
 114 }
 115
 116 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
 117 {
 118     return 1;
 119 }
 120
 121 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
 122 {
 123     ok(0, "unexpected call\n");
 124     return E_NOTIMPL;
 125 }
 126
 127 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
 128                                               LCID lcid, ITypeInfo **ppTInfo)
 129 {
 130     ok(0, "unexpected call\n");
 131     return E_NOTIMPL;
 132 }
 133
 134 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
 135                                                 LPOLESTR *rgszNames, UINT cNames,
 136                                                 LCID lcid, DISPID *rgDispId)
 137 {
 138     ok(0, "unexpected call\n");
 139     return E_NOTIMPL;
 140 }
 141
 142 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
 143                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 144                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 145 {
 146     ok(0, "unexpected call\n");
 147     return E_NOTIMPL;
 148 }
 149
 150 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 151 {
 152     ok(0, "unexpected call %s %x\n", debugstr_w(bstrName), grfdex);
 153     return E_NOTIMPL;
 154 }
 155
 156 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 157 {
 158     ok(0, "unexpected call\n");
 159     return E_NOTIMPL;
 160 }
 161
 162 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 163 {
 164     ok(0, "unexpected call\n");
 165     return E_NOTIMPL;
 166 }
 167
 168 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
 169 {
 170     ok(0, "unexpected call\n");
 171     return E_NOTIMPL;
 172 }
 173
 174 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
 175 {
 176     ok(0, "unexpected call\n");
 177     return E_NOTIMPL;
 178 }
 179
 180 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
 181 {
 182     ok(0, "unexpected call\n");
 183     return E_NOTIMPL;
 184 }
 185
 186 static HRESULT WINAPI Global_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 187 {
 188     if(!strcmp_wa(bstrName, "testPropGet")) {
 189         CHECK_EXPECT(global_propget_d);
 190         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 191         *pid = DISPID_GLOBAL_TESTPROPGET;
 192         return S_OK;
 193     }
 194
 195     ok(0, "unexpected call %s\n", debugstr_w(bstrName));
 196     return DISP_E_UNKNOWNNAME;
 197 }
 198
 199 static HRESULT WINAPI Global_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 200         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 201 {
 202      switch(id) {
 203      case DISPID_GLOBAL_TESTPROPGET:
 204         CHECK_EXPECT(global_propget_i);
 205
 206         ok(wFlags == INVOKE_PROPERTYGET, "wFlags = %x\n", wFlags);
 207         ok(pdp != NULL, "pdp == NULL\n");
 208         ok(!pdp->rgvarg, "rgvarg != NULL\n");
 209         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 210         ok(!pdp->cArgs, "cArgs = %d\n", pdp->cArgs);
 211         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 212         ok(pvarRes != NULL, "pvarRes == NULL\n");
 213         ok(V_VT(pvarRes) ==  VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
 214         ok(pei != NULL, "pei == NULL\n");
 215
 216         V_VT(pvarRes) = VT_I4;
 217         V_I4(pvarRes) = 1;
 218
 219         return S_OK;
 220      }
 221
 222      ok(0, "unexpected call %x\n", id);
 223     return DISP_E_MEMBERNOTFOUND;
 224 }
 225
 226 static IDispatchExVtbl GlobalVtbl = {
 227     DispatchEx_QueryInterface,
 228     DispatchEx_AddRef,
 229     DispatchEx_Release,
 230     DispatchEx_GetTypeInfoCount,
 231     DispatchEx_GetTypeInfo,
 232     DispatchEx_GetIDsOfNames,
 233     DispatchEx_Invoke,
 234     Global_GetDispID,
 235     Global_InvokeEx,
 236     DispatchEx_DeleteMemberByName,
 237     DispatchEx_DeleteMemberByDispID,
 238     DispatchEx_GetMemberProperties,
 239     DispatchEx_GetMemberName,
 240     DispatchEx_GetNextDispID,
 241     DispatchEx_GetNameSpaceParent
 242 };
 243
 244 static IDispatchEx Global = { &GlobalVtbl };
 245
 246 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
 247 {
 248     *ppv = NULL;
 249
 250     if(IsEqualGUID(&IID_IUnknown, riid))
 251         *ppv = iface;
 252     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
 253         *ppv = iface;
 254     else
 255         return E_NOINTERFACE;
 256
 257     IUnknown_AddRef((IUnknown*)*ppv);
 258     return S_OK;
 259 }
 260
 261 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
 262 {
 263     return 2;
 264 }
 265
 266 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
 267 {
 268     return 1;
 269 }
 270
 271 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 272 {
 273     *plcid = GetUserDefaultLCID();
 274     return S_OK;
 275 }
 276
 277 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 278         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 279 {
 280     ok(!lstrcmpW(pstrName, testW), "unexpected pstrName %s\n", debugstr_w(pstrName));
 281     ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
 282     ok(!ppti, "ppti != NULL\n");
 283
 284     *ppiunkItem = (IUnknown*)&Global;
 285     return S_OK;
 286 }
 287
 288 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 289 {
 290     return E_NOTIMPL;
 291 }
 292
 293 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 294         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 295 {
 296     return E_NOTIMPL;
 297 }
 298
 299 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 300 {
 301     return E_NOTIMPL;
 302 }
 303
 304 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 305 {
 306     return E_NOTIMPL;
 307 }
 308
 309 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 310 {
 311     return E_NOTIMPL;
 312 }
 313
 314 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 315 {
 316     return E_NOTIMPL;
 317 }
 318
 319 #undef ACTSCPSITE_THIS
 320
 321 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 322     ActiveScriptSite_QueryInterface,
 323     ActiveScriptSite_AddRef,
 324     ActiveScriptSite_Release,
 325     ActiveScriptSite_GetLCID,
 326     ActiveScriptSite_GetItemInfo,
 327     ActiveScriptSite_GetDocVersionString,
 328     ActiveScriptSite_OnScriptTerminate,
 329     ActiveScriptSite_OnStateChange,
 330     ActiveScriptSite_OnScriptError,
 331     ActiveScriptSite_OnEnterScript,
 332     ActiveScriptSite_OnLeaveScript
 333 };
 334
 335 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
 336
 337 static IActiveScript *create_script(void)
 338 {
 339     IActiveScript *script;
 340     HRESULT hres;
 341
 342     hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 343             &IID_IActiveScript, (void**)&script);
 344     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
 345
 346     return script;
 347 }
 348
 349 static void parse_script(BSTR script_str)
 350 {
 351     IActiveScriptParse *parser;
 352     IActiveScript *engine;
 353     HRESULT hres;
 354
 355     engine = create_script();
 356     if(!engine)
 357         return;
 358
 359     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
 360     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 361
 362     hres = IActiveScriptParse_InitNew(parser);
 363     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 364
 365     hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
 366     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 367
 368     hres = IActiveScript_AddNamedItem(engine, testW,
 369             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
 370     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
 371
 372     hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
 373     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 374
 375     hres = IActiveScriptParse_ParseScriptText(parser, script_str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
 376     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 377
 378     IActiveScript_Release(engine);
 379     IActiveScriptParse_Release(parser);
 380 }
 381
 382 static void parse_script_a(const char *src)
 383 {
 384     BSTR tmp = a2bstr(src);
 385     parse_script(tmp);
 386     SysFreeString(tmp);
 387 }
 388
 389 static void run_tests(void)
 390 {
 391     parse_script_a("");
 392     parse_script_a("/* empty */ ;");
 393
 394     SET_EXPECT(global_propget_d);
 395     SET_EXPECT(global_propget_i);
 396     parse_script_a("testPropGet;");
 397     CHECK_CALLED(global_propget_d);
 398     CHECK_CALLED(global_propget_i);
 399 }
 400
 401 START_TEST(run)
 402 {
 403     CoInitialize(NULL);
 404
 405     run_tests();
 406
 407     CoUninitialize();
 408 }