git.oblomov.eu Git - wine/blob - dlls/vbscript/tests/vbscript.c

   1 /*
   2  * Copyright 2011 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 #ifdef _WIN64
  31
  32 #define IActiveScriptParse_QueryInterface IActiveScriptParse64_QueryInterface
  33 #define IActiveScriptParse_Release IActiveScriptParse64_Release
  34 #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
  35 #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
  36 #define IActiveScriptParseProcedure2_Release \
  37     IActiveScriptParseProcedure2_64_Release
  38
  39 #else
  40
  41 #define IActiveScriptParse_QueryInterface IActiveScriptParse32_QueryInterface
  42 #define IActiveScriptParse_Release IActiveScriptParse32_Release
  43 #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
  44 #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
  45 #define IActiveScriptParseProcedure2_Release \
  46     IActiveScriptParseProcedure2_32_Release
  47
  48 #endif
  49
  50 DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
  51
  52 #define DEFINE_EXPECT(func) \
  53     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  54
  55 #define SET_EXPECT(func) \
  56     expect_ ## func = TRUE
  57
  58 #define CHECK_EXPECT2(func) \
  59     do { \
  60         ok(expect_ ##func, "unexpected call " #func "\n"); \
  61         called_ ## func = TRUE; \
  62     }while(0)
  63
  64 #define CHECK_EXPECT(func) \
  65     do { \
  66         CHECK_EXPECT2(func); \
  67         expect_ ## func = FALSE; \
  68     }while(0)
  69
  70 #define CHECK_CALLED(func) \
  71     do { \
  72         ok(called_ ## func, "expected " #func "\n"); \
  73         expect_ ## func = called_ ## func = FALSE; \
  74     }while(0)
  75
  76 DEFINE_EXPECT(GetLCID);
  77 DEFINE_EXPECT(OnStateChange_UNINITIALIZED);
  78 DEFINE_EXPECT(OnStateChange_STARTED);
  79 DEFINE_EXPECT(OnStateChange_CONNECTED);
  80 DEFINE_EXPECT(OnStateChange_DISCONNECTED);
  81 DEFINE_EXPECT(OnStateChange_CLOSED);
  82 DEFINE_EXPECT(OnStateChange_INITIALIZED);
  83 DEFINE_EXPECT(OnEnterScript);
  84 DEFINE_EXPECT(OnLeaveScript);
  85
  86 DEFINE_GUID(CLSID_VBScript, 0xb54f3741, 0x5b07, 0x11cf, 0xa4,0xb0, 0x00,0xaa,0x00,0x4a,0x55,0xe8);
  87
  88 static BSTR a2bstr(const char *str)
  89 {
  90     BSTR ret;
  91     int len;
  92
  93     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
  94     ret = SysAllocStringLen(NULL, len-1);
  95     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
  96
  97     return ret;
  98 }
  99
 100 #define test_state(s,ss) _test_state(__LINE__,s,ss)
 101 static void _test_state(unsigned line, IActiveScript *script, SCRIPTSTATE exstate)
 102 {
 103     SCRIPTSTATE state = -1;
 104     HRESULT hres;
 105
 106     hres = IActiveScript_GetScriptState(script, &state);
 107     ok_(__FILE__,line) (hres == S_OK, "GetScriptState failed: %08x\n", hres);
 108     ok_(__FILE__,line) (state == exstate, "state=%d, expected %d\n", state, exstate);
 109 }
 110
 111 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
 112 {
 113     *ppv = NULL;
 114
 115     if(IsEqualGUID(&IID_IUnknown, riid))
 116         *ppv = iface;
 117     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
 118         *ppv = iface;
 119     else
 120         return E_NOINTERFACE;
 121
 122     IUnknown_AddRef((IUnknown*)*ppv);
 123     return S_OK;
 124 }
 125
 126 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
 127 {
 128     return 2;
 129 }
 130
 131 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
 132 {
 133     return 1;
 134 }
 135
 136 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 137 {
 138     CHECK_EXPECT(GetLCID);
 139     return E_NOTIMPL;
 140 }
 141
 142 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 143         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 144 {
 145     ok(0, "unexpected call\n");
 146     return E_NOTIMPL;
 147 }
 148
 149 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 150 {
 151     ok(0, "unexpected call\n");
 152     return E_NOTIMPL;
 153 }
 154
 155 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 156         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 157 {
 158     ok(0, "unexpected call\n");
 159     return E_NOTIMPL;
 160 }
 161
 162 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 163 {
 164     switch(ssScriptState) {
 165     case SCRIPTSTATE_UNINITIALIZED:
 166         CHECK_EXPECT(OnStateChange_UNINITIALIZED);
 167         return S_OK;
 168     case SCRIPTSTATE_STARTED:
 169         CHECK_EXPECT(OnStateChange_STARTED);
 170         return S_OK;
 171     case SCRIPTSTATE_CONNECTED:
 172         CHECK_EXPECT(OnStateChange_CONNECTED);
 173         return S_OK;
 174     case SCRIPTSTATE_DISCONNECTED:
 175         CHECK_EXPECT(OnStateChange_DISCONNECTED);
 176         return S_OK;
 177     case SCRIPTSTATE_CLOSED:
 178         CHECK_EXPECT(OnStateChange_CLOSED);
 179         return S_OK;
 180     case SCRIPTSTATE_INITIALIZED:
 181         CHECK_EXPECT(OnStateChange_INITIALIZED);
 182         return S_OK;
 183     default:
 184         ok(0, "unexpected call %d\n", ssScriptState);
 185     }
 186
 187     return E_NOTIMPL;
 188 }
 189
 190 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 191 {
 192     ok(0, "unexpected call\n");
 193     return E_NOTIMPL;
 194 }
 195
 196 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 197 {
 198     CHECK_EXPECT(OnEnterScript);
 199     return S_OK;
 200 }
 201
 202 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 203 {
 204     CHECK_EXPECT(OnLeaveScript);
 205     return S_OK;
 206 }
 207
 208 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 209     ActiveScriptSite_QueryInterface,
 210     ActiveScriptSite_AddRef,
 211     ActiveScriptSite_Release,
 212     ActiveScriptSite_GetLCID,
 213     ActiveScriptSite_GetItemInfo,
 214     ActiveScriptSite_GetDocVersionString,
 215     ActiveScriptSite_OnScriptTerminate,
 216     ActiveScriptSite_OnStateChange,
 217     ActiveScriptSite_OnScriptError,
 218     ActiveScriptSite_OnEnterScript,
 219     ActiveScriptSite_OnLeaveScript
 220 };
 221
 222 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
 223
 224 static void test_safety(IActiveScript *script)
 225 {
 226     IObjectSafety *safety;
 227     DWORD supported, enabled;
 228     HRESULT hres;
 229
 230     hres = IActiveScript_QueryInterface(script, &IID_IObjectSafety, (void**)&safety);
 231     ok(hres == S_OK, "Could not get IObjectSafety: %08x\n", hres);
 232     if(FAILED(hres))
 233         return;
 234
 235     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, NULL);
 236     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
 237     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, NULL, &enabled);
 238     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
 239
 240     supported = enabled = 0xdeadbeef;
 241     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, &enabled);
 242     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 243     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 244        "supported=%x\n", supported);
 245     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 246
 247     supported = enabled = 0xdeadbeef;
 248     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScript, &supported, &enabled);
 249     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 250     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 251        "supported=%x\n", supported);
 252     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 253
 254     supported = enabled = 0xdeadbeef;
 255     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 256     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 257     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 258        "supported=%x\n", supported);
 259     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 260
 261     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 262             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER
 263                 |INTERFACESAFE_FOR_UNTRUSTED_CALLER,
 264             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
 265     ok(hres == E_FAIL, "SetInterfaceSafetyOptions failed: %08x, expected E_FAIL\n", hres);
 266
 267     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 268             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER,
 269             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
 270     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 271
 272     supported = enabled = 0xdeadbeef;
 273     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 274     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 275     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 276        "supported=%x\n", supported);
 277     ok(enabled == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 278        "enabled=%x\n", enabled);
 279
 280     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, INTERFACESAFE_FOR_UNTRUSTED_DATA, 0);
 281     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 282
 283     supported = enabled = 0xdeadbeef;
 284     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 285     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 286     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 287        "supported=%x\n", supported);
 288     ok(enabled == (INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER), "enabled=%x\n", enabled);
 289
 290     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 291             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER, 0);
 292     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 293
 294     supported = enabled = 0xdeadbeef;
 295     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 296     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 297     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 298        "supported=%x\n", supported);
 299     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 300
 301     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 302             INTERFACE_USES_DISPEX, 0);
 303     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 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     IObjectSafety_Release(safety);
 313 }
 314
 315 static IDispatchEx *get_script_dispatch(IActiveScript *script)
 316 {
 317     IDispatchEx *dispex;
 318     IDispatch *disp;
 319     HRESULT hres;
 320
 321     disp = (void*)0xdeadbeef;
 322     hres = IActiveScript_GetScriptDispatch(script, NULL, &disp);
 323     ok(hres == S_OK, "GetScriptDispatch failed: %08x\n", hres);
 324     if(FAILED(hres))
 325         return NULL;
 326
 327     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 328     IDispatch_Release(disp);
 329     ok(hres == S_OK, "Could not get IDispatchEx iface: %08x\n", hres);
 330     return dispex;
 331 }
 332
 333 static void parse_script(IActiveScriptParse *parse, const char *src)
 334 {
 335     BSTR str;
 336     HRESULT hres;
 337
 338     SET_EXPECT(OnEnterScript);
 339     SET_EXPECT(OnLeaveScript);
 340
 341     str = a2bstr(src);
 342     hres = IActiveScriptParse_ParseScriptText(parse, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
 343     SysFreeString(str);
 344     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 345
 346     CHECK_CALLED(OnEnterScript);
 347     CHECK_CALLED(OnLeaveScript);
 348 }
 349
 350 #define get_disp_id(a,b,c,d) _get_disp_id(__LINE__,a,b,c,d)
 351 static void _get_disp_id(unsigned line, IDispatchEx *dispex, const char *name, HRESULT exhres, DISPID *id)
 352 {
 353     BSTR str;
 354     HRESULT hres;
 355
 356     str = a2bstr(name);
 357     hres = IDispatchEx_GetDispID(dispex, str, 0, id);
 358     SysFreeString(str);
 359     ok_(__FILE__,line)(hres == exhres, "GetDispID(%s) returned %08x, expected %08x\n", name, hres, exhres);
 360 }
 361
 362 static void test_no_script_dispatch(IActiveScript *script)
 363 {
 364     IDispatch *disp;
 365     HRESULT hres;
 366
 367     disp = (void*)0xdeadbeef;
 368     hres = IActiveScript_GetScriptDispatch(script, NULL, &disp);
 369     ok(hres == E_UNEXPECTED, "hres = %08x, expected E_UNEXPECTED\n", hres);
 370     ok(!disp, "disp != NULL\n");
 371 }
 372
 373 static IActiveScript *create_vbscript(void)
 374 {
 375     IActiveScript *ret;
 376     HRESULT hres;
 377
 378     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 379             &IID_IActiveScript, (void**)&ret);
 380     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
 381
 382     return ret;
 383 }
 384
 385 static void test_scriptdisp(void)
 386 {
 387     IActiveScriptParse *parser;
 388     IDispatchEx *script_disp;
 389     IActiveScript *vbscript;
 390     DISPID id, id2;
 391     DISPPARAMS dp;
 392     EXCEPINFO ei;
 393     VARIANT v;
 394     ULONG ref;
 395     HRESULT hres;
 396
 397     vbscript = create_vbscript();
 398
 399     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
 400     ok(hres == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hres);
 401
 402     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 403     test_safety(vbscript);
 404
 405     SET_EXPECT(GetLCID);
 406     hres = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
 407     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 408     CHECK_CALLED(GetLCID);
 409
 410     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 411
 412     SET_EXPECT(OnStateChange_INITIALIZED);
 413     hres = IActiveScriptParse_InitNew(parser);
 414     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 415     CHECK_CALLED(OnStateChange_INITIALIZED);
 416
 417     test_state(vbscript, SCRIPTSTATE_INITIALIZED);
 418
 419     SET_EXPECT(OnStateChange_CONNECTED);
 420     hres = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
 421     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 422     CHECK_CALLED(OnStateChange_CONNECTED);
 423
 424     test_state(vbscript, SCRIPTSTATE_CONNECTED);
 425
 426     script_disp = get_script_dispatch(vbscript);
 427
 428     id = 100;
 429     get_disp_id(script_disp, "LCase", DISP_E_UNKNOWNNAME, &id);
 430     ok(id == -1, "id = %d, expected -1\n", id);
 431
 432     get_disp_id(script_disp, "globalVariable", DISP_E_UNKNOWNNAME, &id);
 433     parse_script(parser, "dim globalVariable\nglobalVariable = 3");
 434     get_disp_id(script_disp, "globalVariable", S_OK, &id);
 435
 436     memset(&dp, 0, sizeof(dp));
 437     memset(&ei, 0, sizeof(ei));
 438     V_VT(&v) = VT_EMPTY;
 439     hres = IDispatchEx_InvokeEx(script_disp, id, 0, DISPATCH_PROPERTYGET|DISPATCH_METHOD, &dp, &v, &ei, NULL);
 440     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
 441     ok(V_VT(&v) == VT_I2, "V_VT(v) = %d\n", V_VT(&v));
 442     ok(V_I2(&v) == 3, "V_I2(v) = %d\n", V_I2(&v));
 443
 444     get_disp_id(script_disp, "globalVariable2", DISP_E_UNKNOWNNAME, &id);
 445     parse_script(parser, "globalVariable2 = 4");
 446     get_disp_id(script_disp, "globalVariable2", S_OK, &id);
 447
 448     get_disp_id(script_disp, "globalFunction", DISP_E_UNKNOWNNAME, &id);
 449     parse_script(parser, "function globalFunction()\nglobalFunction=5\nend function");
 450     get_disp_id(script_disp, "globalFunction", S_OK, &id);
 451
 452     SET_EXPECT(OnEnterScript);
 453     SET_EXPECT(OnLeaveScript);
 454
 455     memset(&dp, 0, sizeof(dp));
 456     memset(&ei, 0, sizeof(ei));
 457     V_VT(&v) = VT_EMPTY;
 458     hres = IDispatchEx_InvokeEx(script_disp, id, 0, DISPATCH_PROPERTYGET|DISPATCH_METHOD, &dp, &v, &ei, NULL);
 459     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
 460     ok(V_VT(&v) == VT_I2, "V_VT(v) = %d\n", V_VT(&v));
 461     ok(V_I2(&v) == 5, "V_I2(v) = %d\n", V_I2(&v));
 462
 463     CHECK_CALLED(OnEnterScript);
 464     CHECK_CALLED(OnLeaveScript);
 465
 466     get_disp_id(script_disp, "globalSub", DISP_E_UNKNOWNNAME, &id);
 467     parse_script(parser, "sub globalSub()\nend sub");
 468     get_disp_id(script_disp, "globalSub", S_OK, &id);
 469     get_disp_id(script_disp, "globalSub", S_OK, &id2);
 470     ok(id == id2, "id != id2\n");
 471
 472     get_disp_id(script_disp, "constVariable", DISP_E_UNKNOWNNAME, &id);
 473     parse_script(parser, "const constVariable = 6");
 474     get_disp_id(script_disp, "ConstVariable", S_OK, &id);
 475     get_disp_id(script_disp, "Constvariable", S_OK, &id2);
 476     ok(id == id2, "id != id2\n");
 477
 478     IDispatchEx_Release(script_disp);
 479
 480     IActiveScriptParse_Release(parser);
 481
 482     SET_EXPECT(OnStateChange_DISCONNECTED);
 483     SET_EXPECT(OnStateChange_INITIALIZED);
 484     SET_EXPECT(OnStateChange_CLOSED);
 485     hres = IActiveScript_Close(vbscript);
 486     ok(hres == S_OK, "Close failed: %08x\n", hres);
 487     CHECK_CALLED(OnStateChange_DISCONNECTED);
 488     CHECK_CALLED(OnStateChange_INITIALIZED);
 489     CHECK_CALLED(OnStateChange_CLOSED);
 490
 491     ref = IActiveScript_Release(vbscript);
 492     ok(!ref, "ref = %d\n", ref);
 493 }
 494
 495 static void test_vbscript(void)
 496 {
 497     IActiveScriptParseProcedure2 *parse_proc;
 498     IActiveScriptParse *parser;
 499     IActiveScript *vbscript;
 500     ULONG ref;
 501     HRESULT hres;
 502
 503     vbscript = create_vbscript();
 504
 505     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
 506     ok(hres == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hres);
 507
 508     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 509     test_safety(vbscript);
 510
 511     SET_EXPECT(GetLCID);
 512     hres = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
 513     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 514     CHECK_CALLED(GetLCID);
 515
 516     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 517
 518     SET_EXPECT(OnStateChange_INITIALIZED);
 519     hres = IActiveScriptParse_InitNew(parser);
 520     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 521     CHECK_CALLED(OnStateChange_INITIALIZED);
 522
 523     test_state(vbscript, SCRIPTSTATE_INITIALIZED);
 524
 525     hres = IActiveScriptParse_InitNew(parser);
 526     ok(hres == E_UNEXPECTED, "InitNew failed: %08x, expected E_UNEXPECTED\n", hres);
 527
 528     SET_EXPECT(OnStateChange_CONNECTED);
 529     hres = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
 530     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 531     CHECK_CALLED(OnStateChange_CONNECTED);
 532
 533     test_state(vbscript, SCRIPTSTATE_CONNECTED);
 534
 535     SET_EXPECT(OnStateChange_DISCONNECTED);
 536     SET_EXPECT(OnStateChange_INITIALIZED);
 537     SET_EXPECT(OnStateChange_CLOSED);
 538     hres = IActiveScript_Close(vbscript);
 539     ok(hres == S_OK, "Close failed: %08x\n", hres);
 540     CHECK_CALLED(OnStateChange_DISCONNECTED);
 541     CHECK_CALLED(OnStateChange_INITIALIZED);
 542     CHECK_CALLED(OnStateChange_CLOSED);
 543
 544     test_state(vbscript, SCRIPTSTATE_CLOSED);
 545     test_no_script_dispatch(vbscript);
 546
 547     IActiveScriptParse_Release(parser);
 548
 549     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParseProcedure, (void**)&parse_proc);
 550     ok(hres == E_NOINTERFACE, "Got IActiveScriptParseProcedure interface, expected E_NOTIMPL\n");
 551
 552     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParseProcedure2, (void**)&parse_proc);
 553     ok(hres == S_OK, "Could not get IActiveScriptParseProcedure2 interface\n");
 554     IActiveScriptParseProcedure2_Release(parse_proc);
 555
 556     ref = IActiveScript_Release(vbscript);
 557     ok(!ref, "ref = %d\n", ref);
 558 }
 559
 560 static void test_vbscript_uninitializing(void)
 561 {
 562     IActiveScriptParse *parse;
 563     IActiveScript *script;
 564     IDispatchEx *dispex;
 565     ULONG ref;
 566     HRESULT hres;
 567
 568     static const WCHAR script_textW[] =
 569         {'F','u','n','c','t','i','o','n',' ','f','\n','E','n','d',' ','F','u','n','c','t','i','o','n','\n',0};
 570
 571     script = create_vbscript();
 572
 573     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parse);
 574     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 575
 576     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 577
 578     hres = IActiveScriptParse_InitNew(parse);
 579     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 580
 581     SET_EXPECT(GetLCID);
 582     SET_EXPECT(OnStateChange_INITIALIZED);
 583     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 584     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 585     CHECK_CALLED(GetLCID);
 586     CHECK_CALLED(OnStateChange_INITIALIZED);
 587
 588     test_state(script, SCRIPTSTATE_INITIALIZED);
 589
 590     hres = IActiveScriptParse_ParseScriptText(parse, script_textW, NULL, NULL, NULL, 0, 1, 0x42, NULL, NULL);
 591     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 592
 593     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 594     ok(hres == E_UNEXPECTED, "SetScriptSite failed: %08x, expected E_UNEXPECTED\n", hres);
 595
 596     SET_EXPECT(OnStateChange_UNINITIALIZED);
 597     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 598     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 599     CHECK_CALLED(OnStateChange_UNINITIALIZED);
 600
 601     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 602
 603     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 604     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 605
 606     SET_EXPECT(GetLCID);
 607     SET_EXPECT(OnStateChange_INITIALIZED);
 608     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 609     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 610     CHECK_CALLED(GetLCID);
 611     CHECK_CALLED(OnStateChange_INITIALIZED);
 612
 613     SET_EXPECT(OnStateChange_CONNECTED);
 614     SET_EXPECT(OnEnterScript);
 615     SET_EXPECT(OnLeaveScript);
 616     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_CONNECTED);
 617     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 618     CHECK_CALLED(OnStateChange_CONNECTED);
 619     CHECK_CALLED(OnEnterScript);
 620     CHECK_CALLED(OnLeaveScript);
 621
 622     test_state(script, SCRIPTSTATE_CONNECTED);
 623
 624     dispex = get_script_dispatch(script);
 625     ok(dispex != NULL, "dispex == NULL\n");
 626     if(dispex)
 627         IDispatchEx_Release(dispex);
 628
 629     SET_EXPECT(OnStateChange_DISCONNECTED);
 630     SET_EXPECT(OnStateChange_INITIALIZED);
 631     SET_EXPECT(OnStateChange_UNINITIALIZED);
 632     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 633     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 634     CHECK_CALLED(OnStateChange_DISCONNECTED);
 635     CHECK_CALLED(OnStateChange_INITIALIZED);
 636     CHECK_CALLED(OnStateChange_UNINITIALIZED);
 637
 638     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 639
 640     hres = IActiveScript_Close(script);
 641     ok(hres == S_OK, "Close failed: %08x\n", hres);
 642
 643     test_state(script, SCRIPTSTATE_CLOSED);
 644
 645     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 646     ok(hres == E_UNEXPECTED, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x, expected E_UNEXPECTED\n", hres);
 647
 648     test_state(script, SCRIPTSTATE_CLOSED);
 649
 650     SET_EXPECT(GetLCID);
 651     SET_EXPECT(OnStateChange_INITIALIZED);
 652     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 653     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 654     CHECK_CALLED(GetLCID);
 655     CHECK_CALLED(OnStateChange_INITIALIZED);
 656
 657     test_state(script, SCRIPTSTATE_INITIALIZED);
 658
 659     SET_EXPECT(OnStateChange_CLOSED);
 660     hres = IActiveScript_Close(script);
 661     ok(hres == S_OK, "Close failed: %08x\n", hres);
 662     CHECK_CALLED(OnStateChange_CLOSED);
 663
 664     test_state(script, SCRIPTSTATE_CLOSED);
 665
 666     IActiveScriptParse_Release(parse);
 667
 668     ref = IActiveScript_Release(script);
 669     ok(!ref, "ref = %d\n", ref);
 670 }
 671
 672 static void test_vbscript_release(void)
 673 {
 674     IActiveScriptParse *parser;
 675     IActiveScript *vbscript;
 676     ULONG ref;
 677     HRESULT hres;
 678
 679     vbscript = create_vbscript();
 680
 681     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
 682     ok(hres == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hres);
 683
 684     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 685     test_safety(vbscript);
 686
 687     SET_EXPECT(GetLCID);
 688     hres = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
 689     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 690     CHECK_CALLED(GetLCID);
 691
 692     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 693
 694     SET_EXPECT(OnStateChange_INITIALIZED);
 695     hres = IActiveScriptParse_InitNew(parser);
 696     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 697     CHECK_CALLED(OnStateChange_INITIALIZED);
 698
 699     test_state(vbscript, SCRIPTSTATE_INITIALIZED);
 700
 701     SET_EXPECT(OnStateChange_CONNECTED);
 702     hres = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
 703     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 704     CHECK_CALLED(OnStateChange_CONNECTED);
 705
 706     test_state(vbscript, SCRIPTSTATE_CONNECTED);
 707
 708     IActiveScriptParse_Release(parser);
 709
 710     SET_EXPECT(OnStateChange_DISCONNECTED);
 711     SET_EXPECT(OnStateChange_INITIALIZED);
 712     SET_EXPECT(OnStateChange_CLOSED);
 713     ref = IActiveScript_Release(vbscript);
 714     ok(!ref, "ref = %d\n", ref);
 715     CHECK_CALLED(OnStateChange_DISCONNECTED);
 716     CHECK_CALLED(OnStateChange_INITIALIZED);
 717     CHECK_CALLED(OnStateChange_CLOSED);
 718 }
 719
 720 static void test_vbscript_simplecreate(void)
 721 {
 722     IActiveScript *script;
 723     ULONG ref;
 724     HRESULT hres;
 725
 726     script = create_vbscript();
 727
 728     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 729     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 730
 731     ref = IActiveScript_Release(script);
 732     ok(!ref, "ref = %d\n", ref);
 733 }
 734
 735 static void test_vbscript_initializing(void)
 736 {
 737     IActiveScriptParse *parse;
 738     IActiveScript *script;
 739     ULONG ref;
 740     HRESULT hres;
 741
 742     script = create_vbscript();
 743
 744     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parse);
 745     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 746
 747     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 748
 749     SET_EXPECT(GetLCID);
 750     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 751     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 752     CHECK_CALLED(GetLCID);
 753
 754     SET_EXPECT(OnStateChange_INITIALIZED);
 755     hres = IActiveScriptParse_InitNew(parse);
 756     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 757     CHECK_CALLED(OnStateChange_INITIALIZED);
 758
 759     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 760     ok(hres == E_UNEXPECTED, "SetScriptSite failed: %08x, expected E_UNEXPECTED\n", hres);
 761
 762     SET_EXPECT(OnStateChange_CLOSED);
 763     hres = IActiveScript_Close(script);
 764     ok(hres == S_OK, "Close failed: %08x\n", hres);
 765     CHECK_CALLED(OnStateChange_CLOSED);
 766
 767     test_state(script, SCRIPTSTATE_CLOSED);
 768
 769     IActiveScriptParse_Release(parse);
 770
 771     ref = IActiveScript_Release(script);
 772     ok(!ref, "ref = %d\n", ref);
 773 }
 774
 775 static BOOL check_vbscript(void)
 776 {
 777     IActiveScriptParseProcedure2 *vbscript;
 778     HRESULT hres;
 779
 780     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 781             &IID_IActiveScriptParseProcedure2, (void**)&vbscript);
 782     if(SUCCEEDED(hres))
 783         IActiveScriptParseProcedure2_Release(vbscript);
 784
 785     return hres == S_OK;
 786 }
 787
 788 START_TEST(vbscript)
 789 {
 790     CoInitialize(NULL);
 791
 792     if(check_vbscript()) {
 793         test_vbscript();
 794         test_vbscript_uninitializing();
 795         test_vbscript_release();
 796         test_vbscript_simplecreate();
 797         test_vbscript_initializing();
 798         test_scriptdisp();
 799     }else {
 800         win_skip("VBScript engine not available or too old\n");
 801     }
 802
 803     CoUninitialize();
 804 }