| /* OLEDB Database tests |
| * |
| * Copyright 2012 Alistair Leslie-Hughes |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| #include <stdarg.h> |
| #include <stdio.h> |
| |
| #define COBJMACROS |
| #define CONST_VTABLE |
| #define NONAMELESSUNION |
| #define NONAMELESSSTRUCT |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "ole2.h" |
| #include "msdadc.h" |
| #include "msdasc.h" |
| #include "shlobj.h" |
| #include "msdaguid.h" |
| #include "initguid.h" |
| #include "oledberr.h" |
| |
| #include "wine/test.h" |
| |
| DEFINE_GUID(CSLID_MSDAER, 0xc8b522cf,0x5cf3,0x11ce,0xad,0xe5,0x00,0xaa,0x00,0x44,0x77,0x3d); |
| |
| static WCHAR initstring_default[] = {'D','a','t','a',' ','S','o','u','r','c','e','=','d','u','m','m','y',';',0}; |
| |
| static const char *debugstr_guid(REFIID riid) |
| { |
| static char buf[50]; |
| |
| if(!riid) |
| return "(null)"; |
| |
| sprintf(buf, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}", |
| riid->Data1, riid->Data2, riid->Data3, riid->Data4[0], |
| riid->Data4[1], riid->Data4[2], riid->Data4[3], riid->Data4[4], |
| riid->Data4[5], riid->Data4[6], riid->Data4[7]); |
| |
| return buf; |
| } |
| |
| static void test_GetDataSource(WCHAR *initstring) |
| { |
| IDataInitialize *datainit = NULL; |
| IDBInitialize *dbinit = NULL; |
| HRESULT hr; |
| |
| trace("Data Source: %s\n", wine_dbgstr_w(initstring)); |
| |
| hr = CoCreateInstance(&CLSID_MSDAINITIALIZE, NULL, CLSCTX_INPROC_SERVER, &IID_IDataInitialize,(void**)&datainit); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| /* a failure to create data source here may indicate provider is simply not present */ |
| hr = IDataInitialize_GetDataSource(datainit, NULL, CLSCTX_INPROC_SERVER, initstring, &IID_IDBInitialize, (IUnknown**)&dbinit); |
| if(SUCCEEDED(hr)) |
| { |
| IDBProperties *props = NULL; |
| IMalloc *ppM = NULL; |
| |
| hr = SHGetMalloc(&ppM); |
| if (FAILED(hr)) |
| { |
| ok(0, "Couldn't get IMalloc object.\n"); |
| goto end; |
| } |
| |
| hr = IDBInitialize_QueryInterface(dbinit, &IID_IDBProperties, (void**)&props); |
| ok(hr == S_OK, "got %08x\n", hr); |
| if(SUCCEEDED(hr)) |
| { |
| ULONG cnt; |
| DBPROPINFOSET *pInfoset; |
| OLECHAR *ary; |
| |
| hr = IDBProperties_GetPropertyInfo(props, 0, NULL, &cnt, &pInfoset, &ary); |
| todo_wine ok(hr == S_OK, "got %08x\n", hr); |
| if(hr == S_OK) |
| { |
| ULONG i; |
| for(i =0; i < pInfoset->cPropertyInfos; i++) |
| { |
| trace("(0x%04x) '%s' %d\n", pInfoset->rgPropertyInfos[i].dwPropertyID, wine_dbgstr_w(pInfoset->rgPropertyInfos[i].pwszDescription), |
| pInfoset->rgPropertyInfos[i].vtType); |
| } |
| |
| IMalloc_Free(ppM, ary); |
| } |
| |
| IDBProperties_Release(props); |
| } |
| |
| IMalloc_Release(ppM); |
| |
| end: |
| IDBInitialize_Release(dbinit); |
| } |
| |
| IDataInitialize_Release(datainit); |
| } |
| |
| static void test_database(void) |
| { |
| static WCHAR initstring_jet[] = {'P','r','o','v','i','d','e','r','=','M','i','c','r','o','s','o','f','t','.', |
| 'J','e','t','.','O','L','E','D','B','.','4','.','0',';', |
| 'D','a','t','a',' ','S','o','u','r','c','e','=','d','u','m','m','y',';', |
| 'P','e','r','s','i','s','t',' ','S','e','c','u','r','i','t','y',' ','I','n','f','o','=','F','a','l','s','e',';',0}; |
| static WCHAR initstring_lower[] = {'d','a','t','a',' ','s','o','u','r','c','e','=','d','u','m','m','y',';',0}; |
| IDataInitialize *datainit = NULL; |
| HRESULT hr; |
| |
| hr = CoCreateInstance(&CLSID_MSDAINITIALIZE, NULL, CLSCTX_INPROC_SERVER, &IID_IDataInitialize, (void**)&datainit); |
| if (FAILED(hr)) |
| { |
| win_skip("Unable to load oledb library\n"); |
| return; |
| } |
| IDataInitialize_Release(datainit); |
| |
| test_GetDataSource(NULL); |
| test_GetDataSource(initstring_jet); |
| test_GetDataSource(initstring_default); |
| test_GetDataSource(initstring_lower); |
| } |
| |
| static void test_errorinfo(void) |
| { |
| HRESULT hr; |
| IUnknown *unk = NULL; |
| |
| hr = CoCreateInstance(&CSLID_MSDAER, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown,(void**)&unk); |
| ok(hr == S_OK, "got %08x\n", hr); |
| if(hr == S_OK) |
| { |
| IErrorInfo *errorinfo; |
| IErrorRecords *errrecs; |
| |
| hr = IUnknown_QueryInterface(unk, &IID_IErrorInfo, (void**)&errorinfo); |
| ok(hr == S_OK, "got %08x\n", hr); |
| if(hr == S_OK) |
| { |
| IErrorInfo_Release(errorinfo); |
| } |
| |
| hr = IUnknown_QueryInterface(unk, &IID_IErrorRecords, (void**)&errrecs); |
| ok(hr == S_OK, "got %08x\n", hr); |
| if(hr == S_OK) |
| { |
| ERRORINFO info, info2, info3; |
| ULONG cnt = 0; |
| |
| memset(&info, 0, sizeof(ERRORINFO)); |
| info.dwMinor = 1; |
| memset(&info2, 0, sizeof(ERRORINFO)); |
| info2.dwMinor = 2; |
| memset(&info3, 0, sizeof(ERRORINFO)); |
| |
| hr = IErrorRecords_AddErrorRecord(errrecs, NULL, 268435456, NULL, NULL, 0); |
| ok(hr == E_INVALIDARG, "got %08x\n", hr); |
| |
| hr = IErrorRecords_AddErrorRecord(errrecs, &info, 1, NULL, NULL, 0); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| hr = IErrorRecords_GetRecordCount(errrecs, &cnt); |
| ok(hr == S_OK, "got %08x\n", hr); |
| ok(cnt == 1, "expected 1 got %d\n", cnt); |
| |
| hr = IErrorRecords_AddErrorRecord(errrecs, &info2, 2, NULL, NULL, 0); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| hr = IErrorRecords_GetRecordCount(errrecs, &cnt); |
| ok(hr == S_OK, "got %08x\n", hr); |
| ok(cnt == 2, "expected 2 got %d\n", cnt); |
| |
| hr = IErrorRecords_GetBasicErrorInfo(errrecs, 0, NULL); |
| ok(hr == E_INVALIDARG, "got %08x\n", hr); |
| |
| hr = IErrorRecords_GetBasicErrorInfo(errrecs, 100, &info3); |
| ok(hr == DB_E_BADRECORDNUM, "got %08x\n", hr); |
| |
| hr = IErrorRecords_GetBasicErrorInfo(errrecs, 0, &info3); |
| todo_wine ok(hr == S_OK, "got %08x\n", hr); |
| if(hr == S_OK) |
| { |
| ok(info3.dwMinor == 2, "expected 2 got %d\n", info3.dwMinor); |
| } |
| |
| IErrorRecords_Release(errrecs); |
| } |
| |
| IUnknown_Release(unk); |
| } |
| } |
| |
| static void test_initializationstring(void) |
| { |
| static WCHAR initstring_msdasql[] = {'P','r','o','v','i','d','e','r','=','M','S','D','A','S','Q','L','.','1',';', |
| 'D','a','t','a',' ','S','o','u','r','c','e','=','d','u','m','m','y', 0}; |
| static WCHAR initstring_sqloledb[] = {'P','r','o','v','i','d','e','r','=','S','Q','L','O','L','E','D','B','.','1',';', |
| 'D','a','t','a',' ','S','o','u','r','c','e','=','d','u','m','m','y', 0}; |
| IDataInitialize *datainit = NULL; |
| IDBInitialize *dbinit = NULL; |
| HRESULT hr; |
| WCHAR *initstring = NULL; |
| |
| hr = CoCreateInstance(&CLSID_MSDAINITIALIZE, NULL, CLSCTX_INPROC_SERVER, &IID_IDataInitialize,(void**)&datainit); |
| ok(hr == S_OK, "got %08x\n", hr); |
| if(SUCCEEDED(hr)) |
| { |
| hr = IDataInitialize_GetDataSource(datainit, NULL, CLSCTX_INPROC_SERVER, initstring_default, |
| &IID_IDBInitialize, (IUnknown**)&dbinit); |
| if(SUCCEEDED(hr)) |
| { |
| hr = IDataInitialize_GetInitializationString(datainit, (IUnknown*)dbinit, 0, &initstring); |
| ok(hr == S_OK, "got %08x\n", hr); |
| if(hr == S_OK) |
| { |
| trace("Init String: %s\n", wine_dbgstr_w(initstring)); |
| todo_wine ok(!lstrcmpW(initstring_msdasql, initstring) || |
| !lstrcmpW(initstring_sqloledb, initstring), "got %s\n", wine_dbgstr_w(initstring)); |
| CoTaskMemFree(initstring); |
| } |
| |
| IDBInitialize_Release(dbinit); |
| } |
| |
| IDataInitialize_Release(datainit); |
| } |
| } |
| |
| static void test_rowposition(void) |
| { |
| IEnumConnectionPoints *enum_points; |
| IConnectionPointContainer *cpc; |
| IConnectionPoint *cp; |
| IRowPosition *rowpos; |
| HRESULT hr; |
| IID iid; |
| |
| hr = CoCreateInstance(&CLSID_OLEDB_ROWPOSITIONLIBRARY, NULL, CLSCTX_INPROC_SERVER, &IID_IRowPosition, (void**)&rowpos); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| hr = IRowPosition_QueryInterface(rowpos, &IID_IConnectionPointContainer, (void**)&cpc); |
| ok(hr == S_OK, "got 0x%08x\n", hr); |
| |
| hr = IConnectionPointContainer_EnumConnectionPoints(cpc, &enum_points); |
| todo_wine |
| ok(hr == S_OK, "got 0x%08x\n", hr); |
| if (hr == S_OK) { |
| hr = IEnumConnectionPoints_Next(enum_points, 1, &cp, NULL); |
| ok(hr == S_OK, "got 0x%08x\n", hr); |
| hr = IConnectionPoint_GetConnectionInterface(cp, &iid); |
| ok(hr == S_OK, "got 0x%08x\n", hr); |
| ok(IsEqualIID(&iid, &IID_IRowPositionChange), "got %s\n", debugstr_guid(&iid)); |
| IConnectionPoint_Release(cp); |
| |
| hr = IEnumConnectionPoints_Next(enum_points, 1, &cp, NULL); |
| ok(hr == S_FALSE, "got 0x%08x\n", hr); |
| |
| IEnumConnectionPoints_Release(enum_points); |
| } |
| IConnectionPointContainer_Release(cpc); |
| IRowPosition_Release(rowpos); |
| } |
| |
| typedef struct |
| { |
| IRowset IRowset_iface; |
| IChapteredRowset IChapteredRowset_iface; |
| } test_rset_t; |
| |
| static test_rset_t test_rset; |
| |
| static HRESULT WINAPI rset_QI(IRowset *iface, REFIID riid, void **obj) |
| { |
| if (IsEqualIID(riid, &IID_IUnknown) || |
| IsEqualIID(riid, &IID_IRowset)) |
| { |
| *obj = &test_rset.IRowset_iface; |
| return S_OK; |
| } |
| else if (IsEqualIID(riid, &IID_IChapteredRowset)) |
| { |
| *obj = &test_rset.IChapteredRowset_iface; |
| return S_OK; |
| } |
| |
| ok(0, "unexpected riid %s\n", debugstr_guid(riid)); |
| return E_NOINTERFACE; |
| } |
| |
| static ULONG WINAPI rset_AddRef(IRowset *iface) |
| { |
| return 2; |
| } |
| |
| static ULONG WINAPI rset_Release(IRowset *iface) |
| { |
| return 1; |
| } |
| |
| static HRESULT WINAPI rset_AddRefRows(IRowset *iface, DBCOUNTITEM cRows, |
| const HROW rghRows[], DBREFCOUNT rgRefCounts[], DBROWSTATUS rgRowStatus[]) |
| { |
| trace("AddRefRows: %ld\n", rghRows[0]); |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI rset_GetData(IRowset *iface, HROW hRow, HACCESSOR hAccessor, void *pData) |
| { |
| ok(0, "unexpected call\n"); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI rset_GetNextRows(IRowset *iface, HCHAPTER hReserved, DBROWOFFSET lRowsOffset, |
| DBROWCOUNT cRows, DBCOUNTITEM *pcRowObtained, HROW **prghRows) |
| { |
| ok(0, "unexpected call\n"); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI rset_ReleaseRows(IRowset *iface, DBCOUNTITEM cRows, const HROW rghRows[], DBROWOPTIONS rgRowOptions[], |
| DBREFCOUNT rgRefCounts[], DBROWSTATUS rgRowStatus[]) |
| { |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI rset_RestartPosition(IRowset *iface, HCHAPTER hReserved) |
| { |
| ok(0, "unexpected call\n"); |
| return E_NOTIMPL; |
| } |
| |
| static const IRowsetVtbl rset_vtbl = { |
| rset_QI, |
| rset_AddRef, |
| rset_Release, |
| rset_AddRefRows, |
| rset_GetData, |
| rset_GetNextRows, |
| rset_ReleaseRows, |
| rset_RestartPosition |
| }; |
| |
| static HRESULT WINAPI chrset_QI(IChapteredRowset *iface, REFIID riid, void **obj) |
| { |
| return IRowset_QueryInterface(&test_rset.IRowset_iface, riid, obj); |
| } |
| |
| static ULONG WINAPI chrset_AddRef(IChapteredRowset *iface) |
| { |
| return IRowset_AddRef(&test_rset.IRowset_iface); |
| } |
| |
| static ULONG WINAPI chrset_Release(IChapteredRowset *iface) |
| { |
| return IRowset_Release(&test_rset.IRowset_iface); |
| } |
| |
| static HRESULT WINAPI chrset_AddRefChapter(IChapteredRowset *iface, HCHAPTER chapter, DBREFCOUNT *refcount) |
| { |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI chrset_ReleaseChapter(IChapteredRowset *iface, HCHAPTER chapter, DBREFCOUNT *refcount) |
| { |
| return S_OK; |
| } |
| |
| static const IChapteredRowsetVtbl chrset_vtbl = { |
| chrset_QI, |
| chrset_AddRef, |
| chrset_Release, |
| chrset_AddRefChapter, |
| chrset_ReleaseChapter |
| }; |
| |
| static void init_test_rset(void) |
| { |
| test_rset.IRowset_iface.lpVtbl = &rset_vtbl; |
| test_rset.IChapteredRowset_iface.lpVtbl = &chrset_vtbl; |
| } |
| |
| static void test_rowpos_initialize(void) |
| { |
| IRowPosition *rowpos; |
| HRESULT hr; |
| |
| hr = CoCreateInstance(&CLSID_OLEDB_ROWPOSITIONLIBRARY, NULL, CLSCTX_INPROC_SERVER, &IID_IRowPosition, (void**)&rowpos); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| init_test_rset(); |
| hr = IRowPosition_Initialize(rowpos, (IUnknown*)&test_rset.IRowset_iface); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| IRowPosition_Release(rowpos); |
| } |
| |
| static HRESULT WINAPI onchange_QI(IRowPositionChange *iface, REFIID riid, void **obj) |
| { |
| if (IsEqualIID(riid, &IID_IUnknown) || |
| IsEqualIID(riid, &IID_IRowPositionChange)) |
| { |
| *obj = iface; |
| return S_OK; |
| } |
| |
| return E_NOINTERFACE; |
| } |
| |
| static ULONG WINAPI onchange_AddRef(IRowPositionChange *iface) |
| { |
| return 2; |
| } |
| |
| static ULONG WINAPI onchange_Release(IRowPositionChange *iface) |
| { |
| return 1; |
| } |
| |
| static HRESULT WINAPI onchange_OnRowPositionChange(IRowPositionChange *iface, DBREASON reason, |
| DBEVENTPHASE phase, BOOL cant_deny) |
| { |
| trace("%d %d %d\n", reason, phase, cant_deny); |
| return S_OK; |
| } |
| |
| static const IRowPositionChangeVtbl onchange_vtbl = { |
| onchange_QI, |
| onchange_AddRef, |
| onchange_Release, |
| onchange_OnRowPositionChange |
| }; |
| |
| static IRowPositionChange onchangesink = { &onchange_vtbl }; |
| |
| static void init_onchange_sink(IRowPosition *rowpos) |
| { |
| IConnectionPointContainer *cpc; |
| IConnectionPoint *cp; |
| DWORD cookie; |
| HRESULT hr; |
| |
| hr = IRowPosition_QueryInterface(rowpos, &IID_IConnectionPointContainer, (void**)&cpc); |
| ok(hr == S_OK, "got %08x\n", hr); |
| hr = IConnectionPointContainer_FindConnectionPoint(cpc, &IID_IRowPositionChange, &cp); |
| ok(hr == S_OK, "got %08x\n", hr); |
| hr = IConnectionPoint_Advise(cp, (IUnknown*)&onchangesink, &cookie); |
| ok(hr == S_OK, "got %08x\n", hr); |
| IConnectionPoint_Release(cp); |
| IConnectionPointContainer_Release(cpc); |
| } |
| |
| static void test_rowpos_clearrowposition(void) |
| { |
| DBPOSITIONFLAGS flags; |
| IRowPosition *rowpos; |
| HCHAPTER chapter; |
| IUnknown *unk; |
| HRESULT hr; |
| HROW row; |
| |
| hr = CoCreateInstance(&CLSID_OLEDB_ROWPOSITIONLIBRARY, NULL, CLSCTX_INPROC_SERVER, &IID_IRowPosition, (void**)&rowpos); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| hr = IRowPosition_ClearRowPosition(rowpos); |
| ok(hr == E_UNEXPECTED, "got %08x\n", hr); |
| |
| hr = IRowPosition_GetRowset(rowpos, &IID_IStream, &unk); |
| ok(hr == E_UNEXPECTED, "got %08x\n", hr); |
| |
| chapter = 1; |
| row = 1; |
| flags = DBPOSITION_OK; |
| hr = IRowPosition_GetRowPosition(rowpos, &chapter, &row, &flags); |
| ok(hr == E_UNEXPECTED, "got %08x\n", hr); |
| ok(chapter == DB_NULL_HCHAPTER, "got %ld\n", chapter); |
| ok(row == DB_NULL_HROW, "got %ld\n", row); |
| ok(flags == DBPOSITION_NOROW, "got %d\n", flags); |
| |
| init_test_rset(); |
| hr = IRowPosition_Initialize(rowpos, (IUnknown*)&test_rset.IRowset_iface); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| chapter = 1; |
| row = 1; |
| flags = DBPOSITION_OK; |
| hr = IRowPosition_GetRowPosition(rowpos, &chapter, &row, &flags); |
| ok(hr == S_OK, "got %08x\n", hr); |
| ok(chapter == DB_NULL_HCHAPTER, "got %ld\n", chapter); |
| ok(row == DB_NULL_HROW, "got %ld\n", row); |
| ok(flags == DBPOSITION_NOROW, "got %d\n", flags); |
| |
| hr = IRowPosition_GetRowset(rowpos, &IID_IRowset, &unk); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| init_onchange_sink(rowpos); |
| hr = IRowPosition_ClearRowPosition(rowpos); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| chapter = 1; |
| row = 1; |
| flags = DBPOSITION_OK; |
| hr = IRowPosition_GetRowPosition(rowpos, &chapter, &row, &flags); |
| ok(hr == S_OK, "got %08x\n", hr); |
| ok(chapter == DB_NULL_HCHAPTER, "got %ld\n", chapter); |
| ok(row == DB_NULL_HROW, "got %ld\n", row); |
| ok(flags == DBPOSITION_NOROW, "got %d\n", flags); |
| |
| IRowPosition_Release(rowpos); |
| } |
| |
| static void test_rowpos_setrowposition(void) |
| { |
| IRowPosition *rowpos; |
| HRESULT hr; |
| |
| hr = CoCreateInstance(&CLSID_OLEDB_ROWPOSITIONLIBRARY, NULL, CLSCTX_INPROC_SERVER, &IID_IRowPosition, (void**)&rowpos); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| init_test_rset(); |
| hr = IRowPosition_Initialize(rowpos, (IUnknown*)&test_rset.IRowset_iface); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| hr = IRowPosition_ClearRowPosition(rowpos); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| init_onchange_sink(rowpos); |
| hr = IRowPosition_SetRowPosition(rowpos, 0x123, 0x456, DBPOSITION_OK); |
| ok(hr == S_OK, "got %08x\n", hr); |
| |
| IRowPosition_Release(rowpos); |
| } |
| |
| START_TEST(database) |
| { |
| OleInitialize(NULL); |
| |
| test_database(); |
| test_errorinfo(); |
| test_initializationstring(); |
| |
| /* row position */ |
| test_rowposition(); |
| test_rowpos_initialize(); |
| test_rowpos_clearrowposition(); |
| test_rowpos_setrowposition(); |
| |
| OleUninitialize(); |
| } |