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
  26 #include "wine/test.h"
  27
  28 #define DEFINE_EXPECT(func) \
  29     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  30
  31 #define SET_EXPECT(func) \
  32     expect_ ## func = TRUE
  33
  34 #define SET_CALLED(func) \
  35     called_ ## func = TRUE
  36
  37 #define CHECK_EXPECT2(func) \
  38     do { \
  39         ok(expect_ ##func, "unexpected call " #func "\n"); \
  40         called_ ## func = TRUE; \
  41     }while(0)
  42
  43 #define CHECK_EXPECT(func) \
  44     do { \
  45         CHECK_EXPECT2(func); \
  46         expect_ ## func = FALSE; \
  47     }while(0)
  48
  49 #define CHECK_CALLED(func) \
  50     do { \
  51         ok(called_ ## func, "expected " #func "\n"); \
  52         expect_ ## func = called_ ## func = FALSE; \
  53     }while(0)
  54
  55 DEFINE_EXPECT(GetLCID);
  56 DEFINE_EXPECT(OnStateChange_UNINITIALIZED);
  57 DEFINE_EXPECT(OnStateChange_STARTED);
  58 DEFINE_EXPECT(OnStateChange_CONNECTED);
  59 DEFINE_EXPECT(OnStateChange_DISCONNECTED);
  60 DEFINE_EXPECT(OnStateChange_CLOSED);
  61 DEFINE_EXPECT(OnStateChange_INITIALIZED);
  62 DEFINE_EXPECT(OnEnterScript);
  63 DEFINE_EXPECT(OnLeaveScript);
  64
  65 DEFINE_GUID(CLSID_VBScript, 0xb54f3741, 0x5b07, 0x11cf, 0xa4,0xb0, 0x00,0xaa,0x00,0x4a,0x55,0xe8);
  66
  67 #define test_state(s,ss) _test_state(__LINE__,s,ss)
  68 static void _test_state(unsigned line, IActiveScript *script, SCRIPTSTATE exstate)
  69 {
  70     SCRIPTSTATE state = -1;
  71     HRESULT hres;
  72
  73     hres = IActiveScript_GetScriptState(script, &state);
  74     ok_(__FILE__,line) (hres == S_OK, "GetScriptState failed: %08x\n", hres);
  75     ok_(__FILE__,line) (state == exstate, "state=%d, expected %d\n", state, exstate);
  76 }
  77
  78 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
  79 {
  80     *ppv = NULL;
  81
  82     if(IsEqualGUID(&IID_IUnknown, riid))
  83         *ppv = iface;
  84     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
  85         *ppv = iface;
  86     else
  87         return E_NOINTERFACE;
  88
  89     IUnknown_AddRef((IUnknown*)*ppv);
  90     return S_OK;
  91 }
  92
  93 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
  94 {
  95     return 2;
  96 }
  97
  98 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
  99 {
 100     return 1;
 101 }
 102
 103 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 104 {
 105     CHECK_EXPECT(GetLCID);
 106     return E_NOTIMPL;
 107 }
 108
 109 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 110         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 111 {
 112     ok(0, "unexpected call\n");
 113     return E_NOTIMPL;
 114 }
 115
 116 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 117 {
 118     ok(0, "unexpected call\n");
 119     return E_NOTIMPL;
 120 }
 121
 122 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 123         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 124 {
 125     ok(0, "unexpected call\n");
 126     return E_NOTIMPL;
 127 }
 128
 129 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 130 {
 131     switch(ssScriptState) {
 132     case SCRIPTSTATE_UNINITIALIZED:
 133         CHECK_EXPECT(OnStateChange_UNINITIALIZED);
 134         return S_OK;
 135     case SCRIPTSTATE_STARTED:
 136         CHECK_EXPECT(OnStateChange_STARTED);
 137         return S_OK;
 138     case SCRIPTSTATE_CONNECTED:
 139         CHECK_EXPECT(OnStateChange_CONNECTED);
 140         return S_OK;
 141     case SCRIPTSTATE_DISCONNECTED:
 142         CHECK_EXPECT(OnStateChange_DISCONNECTED);
 143         return S_OK;
 144     case SCRIPTSTATE_CLOSED:
 145         CHECK_EXPECT(OnStateChange_CLOSED);
 146         return S_OK;
 147     case SCRIPTSTATE_INITIALIZED:
 148         CHECK_EXPECT(OnStateChange_INITIALIZED);
 149         return S_OK;
 150     default:
 151         ok(0, "unexpected call %d\n", ssScriptState);
 152     }
 153
 154     return E_NOTIMPL;
 155 }
 156
 157 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 158 {
 159     ok(0, "unexpected call\n");
 160     return E_NOTIMPL;
 161 }
 162
 163 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 164 {
 165     CHECK_EXPECT(OnEnterScript);
 166     return S_OK;
 167 }
 168
 169 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 170 {
 171     CHECK_EXPECT(OnLeaveScript);
 172     return S_OK;
 173 }
 174
 175 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 176     ActiveScriptSite_QueryInterface,
 177     ActiveScriptSite_AddRef,
 178     ActiveScriptSite_Release,
 179     ActiveScriptSite_GetLCID,
 180     ActiveScriptSite_GetItemInfo,
 181     ActiveScriptSite_GetDocVersionString,
 182     ActiveScriptSite_OnScriptTerminate,
 183     ActiveScriptSite_OnStateChange,
 184     ActiveScriptSite_OnScriptError,
 185     ActiveScriptSite_OnEnterScript,
 186     ActiveScriptSite_OnLeaveScript
 187 };
 188
 189 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
 190
 191 static IActiveScript *create_vbscript(void)
 192 {
 193     IActiveScript *ret;
 194     HRESULT hres;
 195
 196     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 197             &IID_IActiveScript, (void**)&ret);
 198     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
 199
 200     return ret;
 201 }
 202
 203 static void test_vbscript(void)
 204 {
 205     IActiveScriptParse *parser;
 206     IActiveScript *vbscript;
 207     ULONG ref;
 208     HRESULT hres;
 209
 210     vbscript = create_vbscript();
 211
 212     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
 213     ok(hres == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hres);
 214
 215     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 216
 217     SET_EXPECT(GetLCID);
 218     hres = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
 219     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 220     CHECK_CALLED(GetLCID);
 221
 222     test_state(vbscript, SCRIPTSTATE_UNINITIALIZED);
 223
 224     SET_EXPECT(OnStateChange_INITIALIZED);
 225     hres = IActiveScriptParse64_InitNew(parser);
 226     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 227     CHECK_CALLED(OnStateChange_INITIALIZED);
 228
 229     test_state(vbscript, SCRIPTSTATE_INITIALIZED);
 230
 231     hres = IActiveScriptParse64_InitNew(parser);
 232     ok(hres == E_UNEXPECTED, "InitNew failed: %08x, expected E_UNEXPECTED\n", hres);
 233
 234     SET_EXPECT(OnStateChange_CLOSED);
 235     hres = IActiveScript_Close(vbscript);
 236     ok(hres == S_OK, "Close failed: %08x\n", hres);
 237     CHECK_CALLED(OnStateChange_CLOSED);
 238
 239     test_state(vbscript, SCRIPTSTATE_CLOSED);
 240
 241     IActiveScriptParse64_Release(parser);
 242
 243     ref = IActiveScript_Release(vbscript);
 244     ok(!ref, "ref = %d\n", ref);
 245 }
 246
 247
 248 static BOOL check_vbscript(void)
 249 {
 250     IActiveScript *vbscript;
 251     HRESULT hres;
 252
 253     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 254             &IID_IActiveScript, (void**)&vbscript);
 255     if(SUCCEEDED(hres))
 256         IActiveScript_Release(vbscript);
 257
 258     return hres == S_OK;
 259 }
 260
 261 START_TEST(vbscript)
 262 {
 263     CoInitialize(NULL);
 264
 265     if(check_vbscript())
 266         test_vbscript();
 267     else
 268         win_skip("VBScript engine not available\n");
 269
 270     CoUninitialize();
 271 }