dlls/msdasql/tests/provider.c

   1 /*
   2  * Copyright 2019 Alistair Leslie-Hughes
   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 #define COBJMACROS
  21
  22 #include "msdasc.h"
  23 #include "oledb.h"
  24 #include "odbcinst.h"
  25 #include "wtypes.h"
  26
  27 #include "initguid.h"
  28
  29 #include "msdasql.h"
  30
  31 #include "wine/test.h"
  32
  33 DEFINE_GUID(DBPROPSET_DBINITALL, 0xc8b522ca, 0x5cf3, 0x11ce, 0xad, 0xe5, 0x00, 0xaa, 0x00, 0x44, 0x77, 0x3d);
  34 DEFINE_GUID(DBPROPSET_DBINIT,    0xc8b522bc, 0x5cf3, 0x11ce, 0xad, 0xe5, 0x00, 0xaa, 0x00, 0x44, 0x77, 0x3d);
  35
  36 static BOOL db_created;
  37 static char mdbpath[MAX_PATH];
  38
  39 static const VARTYPE intptr_vartype = (sizeof(void *) == 8 ? VT_I8 : VT_I4);
  40
  41 static void test_msdasql(void)
  42 {
  43     HRESULT hr;
  44     IUnknown *unk;
  45     IPersist *persist;
  46     CLSID classid;
  47
  48     hr = CoCreateInstance( &CLSID_MSDASQL, NULL, CLSCTX_ALL, &IID_IUnknown, (void **)&unk);
  49     ok(hr == S_OK, "Failed to create object 0x%08x\n", hr);
  50     if (FAILED(hr))
  51     {
  52         return;
  53     }
  54
  55     hr = IUnknown_QueryInterface(unk, &IID_IPersist, (void**)&persist);
  56     ok(hr == S_OK, "got 0x%08x\n", hr);
  57
  58     hr = IPersist_GetClassID(persist, &classid);
  59     ok(hr == S_OK, "got 0x%08x\n", hr);
  60     ok(IsEqualGUID(&classid, &CLSID_MSDASQL), "got %s\n", debugstr_guid(&classid));
  61
  62     IPersist_Release(persist);
  63
  64     IUnknown_Release(unk);
  65 }
  66
  67 static void test_Properties(void)
  68 {
  69     HRESULT hr;
  70     IDBProperties *props;
  71     DBPROPIDSET propidset;
  72     ULONG infocount;
  73     DBPROPINFOSET *propinfoset;
  74     WCHAR *desc;
  75
  76     hr = CoCreateInstance( &CLSID_MSDASQL, NULL, CLSCTX_ALL, &IID_IDBProperties, (void **)&props);
  77     ok(hr == S_OK, "Failed to create object 0x%08x\n", hr);
  78
  79     propidset.rgPropertyIDs = NULL;
  80     propidset.cPropertyIDs = 0;
  81     propidset.guidPropertySet = DBPROPSET_DBINITALL;
  82
  83     infocount = 0;
  84     hr = IDBProperties_GetPropertyInfo(props, 1, &propidset, &infocount, &propinfoset, &desc);
  85     ok(hr == S_OK, "got 0x%08x\n", hr);
  86     if (hr == S_OK)
  87     {
  88         ULONG i;
  89         VARTYPE types[14] = { VT_BSTR, VT_BOOL, VT_BSTR, VT_BSTR, intptr_vartype, VT_BSTR, VT_I4, VT_I2 , VT_I4, VT_BSTR, VT_I4, VT_BSTR, VT_I4, VT_I4 };
  90
  91         ok(IsEqualGUID(&propinfoset->guidPropertySet, &DBPROPSET_DBINIT), "got %s\n", debugstr_guid(&propinfoset->guidPropertySet));
  92         ok(propinfoset->cPropertyInfos == 14, "got %d\n", propinfoset->cPropertyInfos);
  93
  94         for (i = 0; i < propinfoset->cPropertyInfos; i++)
  95         {
  96             trace("%d: pwszDescription: %s\n", i, debugstr_w(propinfoset->rgPropertyInfos[i].pwszDescription) );
  97             ok(propinfoset->rgPropertyInfos[i].vtType == types[i], "got %d\n", propinfoset->rgPropertyInfos[i].vtType);
  98             ok(propinfoset->rgPropertyInfos[i].dwFlags == (DBPROPFLAGS_DBINIT | DBPROPFLAGS_READ | DBPROPFLAGS_WRITE),
  99                 "got %d\n", propinfoset->rgPropertyInfos[i].dwFlags);
 100         }
 101
 102         for (i = 0; i < propinfoset->cPropertyInfos; i++)
 103             VariantClear(&propinfoset->rgPropertyInfos[i].vValues);
 104
 105         CoTaskMemFree(propinfoset->rgPropertyInfos);
 106         CoTaskMemFree(propinfoset);
 107     }
 108
 109     IDBProperties_Release(props);
 110 }
 111
 112 static void test_command_interfaces(IUnknown *cmd)
 113 {
 114     HRESULT hr;
 115     ICommandProperties *commandProp;
 116     ICommandText *comand_text;
 117     IConvertType *convertype;
 118     ICommandPrepare *commandprepare;
 119     ICommandStream *commandstream;
 120     IColumnsInfo *colinfo;
 121     IMultipleResults *multiple;
 122     IUnknown *unk;
 123
 124     hr = IUnknown_QueryInterface(cmd, &IID_ICommandProperties, (void**)&commandProp);
 125     ok(hr == S_OK, "got 0x%08x\n", hr);
 126     ICommandProperties_Release(commandProp);
 127
 128     hr = IUnknown_QueryInterface(cmd, &IID_ICommandText, (void**)&comand_text);
 129     ok(hr == S_OK, "got 0x%08x\n", hr);
 130     ICommandText_Release(comand_text);
 131
 132     hr = IUnknown_QueryInterface(cmd, &IID_IConvertType, (void**)&convertype);
 133     todo_wine ok(hr == S_OK, "got 0x%08x\n", hr);
 134     if (hr == S_OK)
 135         IConvertType_Release(convertype);
 136
 137     hr = IUnknown_QueryInterface(cmd, &IID_ICommandPrepare, (void**)&commandprepare);
 138     todo_wine ok(hr == S_OK, "got 0x%08x\n", hr);
 139     if (hr == S_OK)
 140         ICommandPrepare_Release(commandprepare);
 141
 142     hr = IUnknown_QueryInterface(cmd, &IID_IColumnsInfo, (void**)&colinfo);
 143     ok(hr == S_OK, "got 0x%08x\n", hr);
 144     IColumnsInfo_Release(colinfo);
 145
 146     hr = IUnknown_QueryInterface(cmd, &IID_ICommandStream, (void**)&commandstream);
 147     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
 148
 149     hr = IUnknown_QueryInterface(cmd, &IID_IMultipleResults, (void**)&multiple);
 150     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
 151
 152     hr = IUnknown_QueryInterface(cmd, &IID_IRowsetChange, (void**)&unk);
 153     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
 154
 155     hr = IUnknown_QueryInterface(cmd, &IID_IRowsetUpdate, (void**)&unk);
 156     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
 157
 158     hr = IUnknown_QueryInterface(cmd, &IID_IRowsetLocate, (void**)&unk);
 159     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
 160 }
 161
 162 static void test_sessions(void)
 163 {
 164     IDBProperties *props;
 165     IDBInitialize *dbinit = NULL;
 166     IDataInitialize *datainit;
 167     IDBCreateSession *dbsession = NULL;
 168     IUnknown *session = NULL;
 169     IOpenRowset *openrowset = NULL;
 170     IDBCreateCommand *create_command = NULL;
 171     IGetDataSource *datasource = NULL;
 172     ISessionProperties *session_props = NULL;
 173     IUnknown *cmd = NULL;
 174     HRESULT hr;
 175     BSTR connect_str;
 176
 177     if (!db_created)
 178     {
 179         skip("ODBC source wine_test not available.\n");
 180         return;
 181     }
 182
 183     connect_str = SysAllocString(L"Provider=MSDASQL.1;Persist Security Info=False;Data Source=wine_test");
 184
 185     hr = CoCreateInstance( &CLSID_MSDAINITIALIZE, NULL, CLSCTX_INPROC_SERVER, &IID_IDataInitialize,
 186                                 (void **)&datainit );
 187     ok(hr == S_OK, "Failed to create object 0x%08x\n", hr);
 188     hr = IDataInitialize_GetDataSource( datainit, NULL, CLSCTX_INPROC_SERVER, connect_str, &IID_IDBInitialize,
 189                                              (IUnknown **)&dbinit );
 190     SysFreeString(connect_str);
 191     todo_wine ok(hr == S_OK, "got 0x%08x\n", hr);
 192     if(FAILED(hr))
 193     {
 194         IDataInitialize_Release( datainit );
 195         return;
 196     }
 197
 198     hr = IDBInitialize_QueryInterface( dbinit, &IID_IDBProperties, (void **)&props );
 199     ok(hr == S_OK, "got 0x%08x\n", hr);
 200     IDBProperties_Release(props);
 201
 202     hr = IDBInitialize_Initialize( dbinit );
 203     ok(hr == S_OK, "got 0x%08x\n", hr);
 204     if(FAILED(hr))
 205     {
 206         IDBInitialize_Release( dbinit );
 207         IDataInitialize_Release( datainit );
 208         return;
 209     }
 210
 211     hr = IDBInitialize_QueryInterface( dbinit, &IID_IDBCreateSession, (void **)&dbsession );
 212     ok(hr == S_OK, "got 0x%08x\n", hr);
 213
 214     hr = IDBCreateSession_CreateSession( dbsession, NULL, &IID_IUnknown, &session );
 215     ok(hr == S_OK, "got 0x%08x\n", hr);
 216
 217     hr = IUnknown_QueryInterface(session, &IID_IGetDataSource, (void**)&datasource);
 218     ok(hr == S_OK, "got 0x%08x\n", hr);
 219     IGetDataSource_Release(datasource);
 220
 221     hr = IUnknown_QueryInterface(session, &IID_ISessionProperties, (void**)&session_props);
 222     ok(hr == S_OK, "got 0x%08x\n", hr);
 223     ISessionProperties_Release(session_props);
 224
 225     hr = IUnknown_QueryInterface(session, &IID_IOpenRowset, (void**)&openrowset);
 226     ok(hr == S_OK, "got 0x%08x\n", hr);
 227
 228     hr = IOpenRowset_QueryInterface(openrowset, &IID_IDBCreateCommand, (void**)&create_command);
 229     ok(hr == S_OK, "got 0x%08x\n", hr);
 230
 231     hr = IDBCreateCommand_CreateCommand(create_command, NULL, &IID_IUnknown, (IUnknown **)&cmd);
 232     ok(hr == S_OK, "got 0x%08x\n", hr);
 233     if (hr == S_OK)
 234     {
 235         test_command_interfaces(cmd);
 236         IUnknown_Release(cmd);
 237     }
 238
 239     IDBCreateCommand_Release(create_command);
 240     IOpenRowset_Release(openrowset);
 241     IUnknown_Release( session );
 242     IDBCreateSession_Release(dbsession);
 243     IDBInitialize_Uninitialize( dbinit );
 244     IDBInitialize_Release( dbinit );
 245     IDataInitialize_Release( datainit );
 246 }
 247
 248 static void setup_database(void)
 249 {
 250     char *driver;
 251     DWORD code;
 252     char buffer[1024];
 253     WORD size;
 254
 255     if (winetest_interactive)
 256     {
 257         trace("assuming odbc 'wine_test' is available\n");
 258         db_created = TRUE;
 259         return;
 260     }
 261
 262     /*
 263      * 32 bit Windows has a default driver for "Microsoft Access Driver" (Windows 7+)
 264      *  and has the ability to create files on the fly.
 265      *
 266      * 64 bit Windows ONLY has a driver for "SQL Server", which we cannot use since we don't have a
 267      *   server to connect to.
 268      *
 269      * The filename passed to CREATE_DB must end in mdb.
 270      */
 271     GetTempPathA(sizeof(mdbpath), mdbpath);
 272     strcat(mdbpath, "wine_test.mdb");
 273
 274     driver = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof("DSN=wine_test\0CREATE_DB=") + strlen(mdbpath) + 2);
 275     memcpy(driver, "DSN=wine_test\0CREATE_DB=", sizeof("DSN=wine_test\0CREATE_DB="));
 276     strcat(driver+sizeof("DSN=wine_test\0CREATE_DB=")-1, mdbpath);
 277
 278     db_created = SQLConfigDataSource(NULL, ODBC_ADD_DSN, "Microsoft Access Driver (*.mdb)", driver);
 279     if (!db_created)
 280     {
 281         SQLInstallerError(1, &code, buffer, sizeof(buffer), &size);
 282         trace("code  %d, buffer %s, size %d\n", code, debugstr_a(buffer), size);
 283
 284         HeapFree(GetProcessHeap(), 0, driver);
 285
 286         return;
 287     }
 288
 289     memcpy(driver, "DSN=wine_test\0DBQ=", sizeof("DSN=wine_test\0DBQ="));
 290     strcat(driver+sizeof("DSN=wine_test\0DBQ=")-1, mdbpath);
 291     db_created = SQLConfigDataSource(NULL, ODBC_ADD_DSN, "Microsoft Access Driver (*.mdb)", driver);
 292
 293     HeapFree(GetProcessHeap(), 0, driver);
 294 }
 295
 296 static void cleanup_database(void)
 297 {
 298     BOOL ret;
 299
 300     if (winetest_interactive)
 301         return;
 302
 303     ret = SQLConfigDataSource(NULL, ODBC_REMOVE_DSN, "Microsoft Access Driver (*.mdb)", "DSN=wine_test\0\0");
 304     if (!ret)
 305     {
 306         DWORD code;
 307         char buffer[1024];
 308         WORD size;
 309
 310         SQLInstallerError(1, &code, buffer, sizeof(buffer), &size);
 311         trace("code  %d, buffer %s, size %d\n", code, debugstr_a(buffer), size);
 312     }
 313
 314     DeleteFileA(mdbpath);
 315 }
 316
 317 START_TEST(provider)
 318 {
 319     CoInitialize(0);
 320
 321     setup_database();
 322
 323     test_msdasql();
 324     test_Properties();
 325     test_sessions();
 326
 327     cleanup_database();
 328
 329     CoUninitialize();
 330 }