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 #define test_state(s,ss) _test_state(__LINE__,s,ss)
  89 static void _test_state(unsigned line, IActiveScript *script, SCRIPTSTATE exstate)
  90 {
  91     SCRIPTSTATE state = -1;
  92     HRESULT hres;
  93
  94     hres = IActiveScript_GetScriptState(script, &state);
  95     ok_(__FILE__,line) (hres == S_OK, "GetScriptState failed: %08x\n", hres);
  96     ok_(__FILE__,line) (state == exstate, "state=%d, expected %d\n", state, exstate);
  97 }
  98
  99 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
 100 {
 101     *ppv = NULL;
 102
 103     if(IsEqualGUID(&IID_IUnknown, riid))
 104         *ppv = iface;
 105     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
 106         *ppv = iface;
 107     else
 108         return E_NOINTERFACE;
 109
 110     IUnknown_AddRef((IUnknown*)*ppv);
 111     return S_OK;
 112 }
 113
 114 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
 115 {
 116     return 2;
 117 }
 118
 119 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
 120 {
 121     return 1;
 122 }
 123
 124 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 125 {
 126     CHECK_EXPECT(GetLCID);
 127     return E_NOTIMPL;
 128 }
 129
 130 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 131         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 132 {
 133     ok(0, "unexpected call\n");
 134     return E_NOTIMPL;
 135 }
 136
 137 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 138 {
 139     ok(0, "unexpected call\n");
 140     return E_NOTIMPL;
 141 }
 142
 143 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 144         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 145 {
 146     ok(0, "unexpected call\n");
 147     return E_NOTIMPL;
 148 }
 149
 150 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 151 {
 152     switch(ssScriptState) {
 153     case SCRIPTSTATE_UNINITIALIZED:
 154         CHECK_EXPECT(OnStateChange_UNINITIALIZED);
 155         return S_OK;
 156     case SCRIPTSTATE_STARTED:
 157         CHECK_EXPECT(OnStateChange_STARTED);
 158         return S_OK;
 159     case SCRIPTSTATE_CONNECTED:
 160         CHECK_EXPECT(OnStateChange_CONNECTED);
 161         return S_OK;
 162     case SCRIPTSTATE_DISCONNECTED:
 163         CHECK_EXPECT(OnStateChange_DISCONNECTED);
 164         return S_OK;
 165     case SCRIPTSTATE_CLOSED:
 166         CHECK_EXPECT(OnStateChange_CLOSED);
 167         return S_OK;
 168     case SCRIPTSTATE_INITIALIZED:
 169         CHECK_EXPECT(OnStateChange_INITIALIZED);
 170         return S_OK;
 171     default:
 172         ok(0, "unexpected call %d\n", ssScriptState);
 173     }
 174
 175     return E_NOTIMPL;
 176 }
 177
 178 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 179 {
 180     ok(0, "unexpected call\n");
 181     return E_NOTIMPL;
 182 }
 183
 184 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 185 {
 186     CHECK_EXPECT(OnEnterScript);
 187     return S_OK;
 188 }
 189
 190 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 191 {
 192     CHECK_EXPECT(OnLeaveScript);
 193     return S_OK;
 194 }
 195
 196 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 197     ActiveScriptSite_QueryInterface,
 198     ActiveScriptSite_AddRef,
 199     ActiveScriptSite_Release,
 200     ActiveScriptSite_GetLCID,
 201     ActiveScriptSite_GetItemInfo,
 202     ActiveScriptSite_GetDocVersionString,
 203     ActiveScriptSite_OnScriptTerminate,
 204     ActiveScriptSite_OnStateChange,
 205     ActiveScriptSite_OnScriptError,
 206     ActiveScriptSite_OnEnterScript,
 207     ActiveScriptSite_OnLeaveScript
 208 };
 209
 210 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
 211
 212 static void test_safety(IActiveScript *script)
 213 {
 214     IObjectSafety *safety;
 215     DWORD supported, enabled;
 216     HRESULT hres;
 217
 218     hres = IActiveScript_QueryInterface(script, &IID_IObjectSafety, (void**)&safety);
 219     ok(hres == S_OK, "Could not get IObjectSafety: %08x\n", hres);
 220     if(FAILED(hres))
 221         return;
 222
 223     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, NULL);
 224     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
 225     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, NULL, &enabled);
 226     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
 227
 228     supported = enabled = 0xdeadbeef;
 229     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, &enabled);
 230     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 231     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 232        "supported=%x\n", supported);
 233     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 234
 235     supported = enabled = 0xdeadbeef;
 236     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScript, &supported, &enabled);
 237     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 238     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 239        "supported=%x\n", supported);
 240     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 241
 242     supported = enabled = 0xdeadbeef;
 243     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 244     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 245     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 246        "supported=%x\n", supported);
 247     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 248
 249     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 250             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER
 251                 |INTERFACESAFE_FOR_UNTRUSTED_CALLER,
 252             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
 253     ok(hres == E_FAIL, "SetInterfaceSafetyOptions failed: %08x, expected E_FAIL\n", hres);
 254
 255     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 256             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER,
 257             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
 258     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 259
 260     supported = enabled = 0xdeadbeef;
 261     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 262     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 263     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 264        "supported=%x\n", supported);
 265     ok(enabled == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 266        "enabled=%x\n", enabled);
 267
 268     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, INTERFACESAFE_FOR_UNTRUSTED_DATA, 0);
 269     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 270
 271     supported = enabled = 0xdeadbeef;
 272     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 273     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 274     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 275        "supported=%x\n", supported);
 276     ok(enabled == (INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER), "enabled=%x\n", enabled);
 277
 278     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 279             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER, 0);
 280     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 281
 282     supported = enabled = 0xdeadbeef;
 283     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 284     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 285     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 286        "supported=%x\n", supported);
 287     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 288
 289     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 290             INTERFACE_USES_DISPEX, 0);
 291     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 292
 293     supported = enabled = 0xdeadbeef;
 294     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 295     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 296     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 297        "supported=%x\n", supported);
 298     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 299
 300     IObjectSafety_Release(safety);
 301 }
 302
 303 static IDispatchEx *get_script_dispatch(IActiveScript *script)
 304 {
 305     IDispatchEx *dispex;
 306     IDispatch *disp;
 307     HRESULT hres;
 308
 309     disp = (void*)0xdeadbeef;
 310     hres = IActiveScript_GetScriptDispatch(script, NULL, &disp);
 311     ok(hres == S_OK, "GetScriptDispatch failed: %08x\n", hres);
 312     if(FAILED(hres))
 313         return NULL;
 314
 315     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 316     IDispatch_Release(disp);
 317     ok(hres == S_OK, "Could not get IDispatchEx iface: %08x\n", hres);
 318     return dispex;
 319 }
 320
 321 static void test_no_script_dispatch(IActiveScript *script)
 322 {
 323     IDispatch *disp;
 324     HRESULT hres;
 325
 326     disp = (void*)0xdeadbeef;
 327     hres = IActiveScript_GetScriptDispatch(script, NULL, &disp);
 328     ok(hres == E_UNEXPECTED, "hres = %08x, expected E_UNEXPECTED\n", hres);
 329     ok(!disp, "disp != NULL\n");
 330 }
 331
 332 static IActiveScript *create_vbscript(void)
 333 {
 334     IActiveScript *ret;
 335     HRESULT hres;
 336
 337     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 338             &IID_IActiveScript, (void**)&ret);
 339     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
 340
 341     return ret;
 342 }
 343
 344 static void test_vbscript(void)
 345 {
 346     IActiveScriptParseProcedure2 *parse_proc;
 347     IActiveScriptParse *parser;
 348     IActiveScript *vbscript;
 349     ULONG ref;
 350     HRESULT hres;
 351
 352     vbscript = create_vbscript();
 353
 354     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
 355     ok(hres == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hres);
 356
 357     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 358     test_safety(vbscript);
 359
 360     SET_EXPECT(GetLCID);
 361     hres = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
 362     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 363     CHECK_CALLED(GetLCID);
 364
 365     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 366
 367     SET_EXPECT(OnStateChange_INITIALIZED);
 368     hres = IActiveScriptParse_InitNew(parser);
 369     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 370     CHECK_CALLED(OnStateChange_INITIALIZED);
 371
 372     test_state(vbscript, SCRIPTSTATE_INITIALIZED);
 373
 374     hres = IActiveScriptParse_InitNew(parser);
 375     ok(hres == E_UNEXPECTED, "InitNew failed: %08x, expected E_UNEXPECTED\n", hres);
 376
 377     SET_EXPECT(OnStateChange_CONNECTED);
 378     hres = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
 379     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 380     CHECK_CALLED(OnStateChange_CONNECTED);
 381
 382     test_state(vbscript, SCRIPTSTATE_CONNECTED);
 383
 384     SET_EXPECT(OnStateChange_DISCONNECTED);
 385     SET_EXPECT(OnStateChange_INITIALIZED);
 386     SET_EXPECT(OnStateChange_CLOSED);
 387     hres = IActiveScript_Close(vbscript);
 388     ok(hres == S_OK, "Close failed: %08x\n", hres);
 389     CHECK_CALLED(OnStateChange_DISCONNECTED);
 390     CHECK_CALLED(OnStateChange_INITIALIZED);
 391     CHECK_CALLED(OnStateChange_CLOSED);
 392
 393     test_state(vbscript, SCRIPTSTATE_CLOSED);
 394     test_no_script_dispatch(vbscript);
 395
 396     IActiveScriptParse_Release(parser);
 397
 398     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParseProcedure, (void**)&parse_proc);
 399     ok(hres == E_NOINTERFACE, "Got IActiveScriptParseProcedure interface, expected E_NOTIMPL\n");
 400
 401     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParseProcedure2, (void**)&parse_proc);
 402     ok(hres == S_OK, "Could not get IActiveScriptParseProcedure2 interface\n");
 403     IActiveScriptParseProcedure2_Release(parse_proc);
 404
 405     ref = IActiveScript_Release(vbscript);
 406     ok(!ref, "ref = %d\n", ref);
 407 }
 408
 409 static void test_vbscript_uninitializing(void)
 410 {
 411     IActiveScriptParse *parse;
 412     IActiveScript *script;
 413     IDispatchEx *dispex;
 414     ULONG ref;
 415     HRESULT hres;
 416
 417     static const WCHAR script_textW[] =
 418         {'F','u','n','c','t','i','o','n',' ','f','\n','E','n','d',' ','F','u','n','c','t','i','o','n','\n',0};
 419
 420     script = create_vbscript();
 421
 422     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parse);
 423     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 424
 425     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 426
 427     hres = IActiveScriptParse_InitNew(parse);
 428     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 429
 430     SET_EXPECT(GetLCID);
 431     SET_EXPECT(OnStateChange_INITIALIZED);
 432     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 433     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 434     CHECK_CALLED(GetLCID);
 435     CHECK_CALLED(OnStateChange_INITIALIZED);
 436
 437     test_state(script, SCRIPTSTATE_INITIALIZED);
 438
 439     hres = IActiveScriptParse_ParseScriptText(parse, script_textW, NULL, NULL, NULL, 0, 1, 0x42, NULL, NULL);
 440     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 441
 442     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 443     ok(hres == E_UNEXPECTED, "SetScriptSite failed: %08x, expected E_UNEXPECTED\n", hres);
 444
 445     SET_EXPECT(OnStateChange_UNINITIALIZED);
 446     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 447     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 448     CHECK_CALLED(OnStateChange_UNINITIALIZED);
 449
 450     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 451
 452     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 453     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 454
 455     SET_EXPECT(GetLCID);
 456     SET_EXPECT(OnStateChange_INITIALIZED);
 457     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 458     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 459     CHECK_CALLED(GetLCID);
 460     CHECK_CALLED(OnStateChange_INITIALIZED);
 461
 462     SET_EXPECT(OnStateChange_CONNECTED);
 463     SET_EXPECT(OnEnterScript);
 464     SET_EXPECT(OnLeaveScript);
 465     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_CONNECTED);
 466     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 467     CHECK_CALLED(OnStateChange_CONNECTED);
 468     CHECK_CALLED(OnEnterScript);
 469     CHECK_CALLED(OnLeaveScript);
 470
 471     test_state(script, SCRIPTSTATE_CONNECTED);
 472
 473     dispex = get_script_dispatch(script);
 474     ok(dispex != NULL, "dispex == NULL\n");
 475     if(dispex)
 476         IDispatchEx_Release(dispex);
 477
 478     SET_EXPECT(OnStateChange_DISCONNECTED);
 479     SET_EXPECT(OnStateChange_INITIALIZED);
 480     SET_EXPECT(OnStateChange_UNINITIALIZED);
 481     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 482     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 483     CHECK_CALLED(OnStateChange_DISCONNECTED);
 484     CHECK_CALLED(OnStateChange_INITIALIZED);
 485     CHECK_CALLED(OnStateChange_UNINITIALIZED);
 486
 487     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 488
 489     hres = IActiveScript_Close(script);
 490     ok(hres == S_OK, "Close failed: %08x\n", hres);
 491
 492     test_state(script, SCRIPTSTATE_CLOSED);
 493
 494     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 495     ok(hres == E_UNEXPECTED, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x, expected E_UNEXPECTED\n", hres);
 496
 497     test_state(script, SCRIPTSTATE_CLOSED);
 498
 499     SET_EXPECT(GetLCID);
 500     SET_EXPECT(OnStateChange_INITIALIZED);
 501     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 502     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 503     CHECK_CALLED(GetLCID);
 504     CHECK_CALLED(OnStateChange_INITIALIZED);
 505
 506     test_state(script, SCRIPTSTATE_INITIALIZED);
 507
 508     SET_EXPECT(OnStateChange_CLOSED);
 509     hres = IActiveScript_Close(script);
 510     ok(hres == S_OK, "Close failed: %08x\n", hres);
 511     CHECK_CALLED(OnStateChange_CLOSED);
 512
 513     test_state(script, SCRIPTSTATE_CLOSED);
 514
 515     IActiveScriptParse_Release(parse);
 516
 517     ref = IActiveScript_Release(script);
 518     ok(!ref, "ref = %d\n", ref);
 519 }
 520
 521 static void test_vbscript_release(void)
 522 {
 523     IActiveScriptParse *parser;
 524     IActiveScript *vbscript;
 525     ULONG ref;
 526     HRESULT hres;
 527
 528     vbscript = create_vbscript();
 529
 530     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
 531     ok(hres == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hres);
 532
 533     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 534     test_safety(vbscript);
 535
 536     SET_EXPECT(GetLCID);
 537     hres = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
 538     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 539     CHECK_CALLED(GetLCID);
 540
 541     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 542
 543     SET_EXPECT(OnStateChange_INITIALIZED);
 544     hres = IActiveScriptParse_InitNew(parser);
 545     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 546     CHECK_CALLED(OnStateChange_INITIALIZED);
 547
 548     test_state(vbscript, SCRIPTSTATE_INITIALIZED);
 549
 550     SET_EXPECT(OnStateChange_CONNECTED);
 551     hres = IActiveScript_SetScriptState(vbscript, SCRIPTSTATE_CONNECTED);
 552     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_CONNECTED) failed: %08x\n", hres);
 553     CHECK_CALLED(OnStateChange_CONNECTED);
 554
 555     test_state(vbscript, SCRIPTSTATE_CONNECTED);
 556
 557     IActiveScriptParse_Release(parser);
 558
 559     SET_EXPECT(OnStateChange_DISCONNECTED);
 560     SET_EXPECT(OnStateChange_INITIALIZED);
 561     SET_EXPECT(OnStateChange_CLOSED);
 562     ref = IActiveScript_Release(vbscript);
 563     ok(!ref, "ref = %d\n", ref);
 564     CHECK_CALLED(OnStateChange_DISCONNECTED);
 565     CHECK_CALLED(OnStateChange_INITIALIZED);
 566     CHECK_CALLED(OnStateChange_CLOSED);
 567 }
 568
 569 static void test_vbscript_simplecreate(void)
 570 {
 571     IActiveScript *script;
 572     ULONG ref;
 573     HRESULT hres;
 574
 575     script = create_vbscript();
 576
 577     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_UNINITIALIZED);
 578     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_UNINITIALIZED) failed: %08x\n", hres);
 579
 580     ref = IActiveScript_Release(script);
 581     ok(!ref, "ref = %d\n", ref);
 582 }
 583
 584 static void test_vbscript_initializing(void)
 585 {
 586     IActiveScriptParse *parse;
 587     IActiveScript *script;
 588     ULONG ref;
 589     HRESULT hres;
 590
 591     script = create_vbscript();
 592
 593     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parse);
 594     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 595
 596     test_state(script, SCRIPTSTATE_UNINITIALIZED);
 597
 598     SET_EXPECT(GetLCID);
 599     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 600     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 601     CHECK_CALLED(GetLCID);
 602
 603     SET_EXPECT(OnStateChange_INITIALIZED);
 604     hres = IActiveScriptParse_InitNew(parse);
 605     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 606     CHECK_CALLED(OnStateChange_INITIALIZED);
 607
 608     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 609     ok(hres == E_UNEXPECTED, "SetScriptSite failed: %08x, expected E_UNEXPECTED\n", hres);
 610
 611     SET_EXPECT(OnStateChange_CLOSED);
 612     hres = IActiveScript_Close(script);
 613     ok(hres == S_OK, "Close failed: %08x\n", hres);
 614     CHECK_CALLED(OnStateChange_CLOSED);
 615
 616     test_state(script, SCRIPTSTATE_CLOSED);
 617
 618     IActiveScriptParse_Release(parse);
 619
 620     ref = IActiveScript_Release(script);
 621     ok(!ref, "ref = %d\n", ref);
 622 }
 623
 624 static BOOL check_vbscript(void)
 625 {
 626     IActiveScript *vbscript;
 627     HRESULT hres;
 628
 629     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 630             &IID_IActiveScript, (void**)&vbscript);
 631     if(SUCCEEDED(hres))
 632         IActiveScript_Release(vbscript);
 633
 634     return hres == S_OK;
 635 }
 636
 637 START_TEST(vbscript)
 638 {
 639     CoInitialize(NULL);
 640
 641     if(check_vbscript()) {
 642         test_vbscript();
 643         test_vbscript_uninitializing();
 644         test_vbscript_release();
 645         test_vbscript_simplecreate();
 646         test_vbscript_initializing();
 647     }else {
 648         win_skip("VBScript engine not available\n");
 649     }
 650
 651     CoUninitialize();
 652 }