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 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
  68 {
  69     *ppv = NULL;
  70
  71     if(IsEqualGUID(&IID_IUnknown, riid))
  72         *ppv = iface;
  73     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
  74         *ppv = iface;
  75     else
  76         return E_NOINTERFACE;
  77
  78     IUnknown_AddRef((IUnknown*)*ppv);
  79     return S_OK;
  80 }
  81
  82 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
  83 {
  84     return 2;
  85 }
  86
  87 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
  88 {
  89     return 1;
  90 }
  91
  92 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
  93 {
  94     CHECK_EXPECT(GetLCID);
  95     return E_NOTIMPL;
  96 }
  97
  98 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
  99         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 100 {
 101     ok(0, "unexpected call\n");
 102     return E_NOTIMPL;
 103 }
 104
 105 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 106 {
 107     ok(0, "unexpected call\n");
 108     return E_NOTIMPL;
 109 }
 110
 111 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 112         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 113 {
 114     ok(0, "unexpected call\n");
 115     return E_NOTIMPL;
 116 }
 117
 118 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 119 {
 120     switch(ssScriptState) {
 121     case SCRIPTSTATE_UNINITIALIZED:
 122         CHECK_EXPECT(OnStateChange_UNINITIALIZED);
 123         return S_OK;
 124     case SCRIPTSTATE_STARTED:
 125         CHECK_EXPECT(OnStateChange_STARTED);
 126         return S_OK;
 127     case SCRIPTSTATE_CONNECTED:
 128         CHECK_EXPECT(OnStateChange_CONNECTED);
 129         return S_OK;
 130     case SCRIPTSTATE_DISCONNECTED:
 131         CHECK_EXPECT(OnStateChange_DISCONNECTED);
 132         return S_OK;
 133     case SCRIPTSTATE_CLOSED:
 134         CHECK_EXPECT(OnStateChange_CLOSED);
 135         return S_OK;
 136     case SCRIPTSTATE_INITIALIZED:
 137         CHECK_EXPECT(OnStateChange_INITIALIZED);
 138         return S_OK;
 139     default:
 140         ok(0, "unexpected call %d\n", ssScriptState);
 141     }
 142
 143     return E_NOTIMPL;
 144 }
 145
 146 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 147 {
 148     ok(0, "unexpected call\n");
 149     return E_NOTIMPL;
 150 }
 151
 152 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 153 {
 154     CHECK_EXPECT(OnEnterScript);
 155     return S_OK;
 156 }
 157
 158 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 159 {
 160     CHECK_EXPECT(OnLeaveScript);
 161     return S_OK;
 162 }
 163
 164 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 165     ActiveScriptSite_QueryInterface,
 166     ActiveScriptSite_AddRef,
 167     ActiveScriptSite_Release,
 168     ActiveScriptSite_GetLCID,
 169     ActiveScriptSite_GetItemInfo,
 170     ActiveScriptSite_GetDocVersionString,
 171     ActiveScriptSite_OnScriptTerminate,
 172     ActiveScriptSite_OnStateChange,
 173     ActiveScriptSite_OnScriptError,
 174     ActiveScriptSite_OnEnterScript,
 175     ActiveScriptSite_OnLeaveScript
 176 };
 177
 178 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
 179
 180 static IActiveScript *create_vbscript(void)
 181 {
 182     IActiveScript *ret;
 183     HRESULT hres;
 184
 185     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 186             &IID_IActiveScript, (void**)&ret);
 187     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
 188
 189     return ret;
 190 }
 191
 192 static void test_vbscript(void)
 193 {
 194     IActiveScriptParse *parser;
 195     IActiveScript *vbscript;
 196     ULONG ref;
 197     HRESULT hres;
 198
 199     vbscript = create_vbscript();
 200
 201     hres = IActiveScript_QueryInterface(vbscript, &IID_IActiveScriptParse, (void**)&parser);
 202     ok(hres == S_OK, "Could not get IActiveScriptParse iface: %08x\n", hres);
 203
 204     SET_EXPECT(GetLCID);
 205     hres = IActiveScript_SetScriptSite(vbscript, &ActiveScriptSite);
 206     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 207     CHECK_CALLED(GetLCID);
 208
 209     SET_EXPECT(OnStateChange_INITIALIZED);
 210     hres = IActiveScriptParse64_InitNew(parser);
 211     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 212     CHECK_CALLED(OnStateChange_INITIALIZED);
 213
 214     hres = IActiveScriptParse64_InitNew(parser);
 215     ok(hres == E_UNEXPECTED, "InitNew failed: %08x, expected E_UNEXPECTED\n", hres);
 216
 217     SET_EXPECT(OnStateChange_CLOSED);
 218     hres = IActiveScript_Close(vbscript);
 219     ok(hres == S_OK, "Close failed: %08x\n", hres);
 220     CHECK_CALLED(OnStateChange_CLOSED);
 221
 222     IActiveScriptParse64_Release(parser);
 223
 224     ref = IActiveScript_Release(vbscript);
 225     ok(!ref, "ref = %d\n", ref);
 226 }
 227
 228
 229 static BOOL check_vbscript(void)
 230 {
 231     IActiveScript *vbscript;
 232     HRESULT hres;
 233
 234     hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 235             &IID_IActiveScript, (void**)&vbscript);
 236     if(SUCCEEDED(hres))
 237         IActiveScript_Release(vbscript);
 238
 239     return hres == S_OK;
 240 }
 241
 242 START_TEST(vbscript)
 243 {
 244     CoInitialize(NULL);
 245
 246     if(check_vbscript())
 247         test_vbscript();
 248     else
 249         win_skip("VBScript engine not available\n");
 250
 251     CoUninitialize();
 252 }