explorerframe: Implement GetRootItems.
This commit is contained in:
parent
b431995cc4
commit
2761b0ded6
@ -683,8 +683,35 @@ static HRESULT WINAPI NSTC2_fnGetRootItems(INameSpaceTreeControl2* iface,
|
||||
IShellItemArray **ppsiaRootItems)
|
||||
{
|
||||
NSTC2Impl *This = (NSTC2Impl*)iface;
|
||||
FIXME("stub, %p (%p)\n", This, ppsiaRootItems);
|
||||
return E_NOTIMPL;
|
||||
IShellFolder *psf;
|
||||
LPITEMIDLIST *array;
|
||||
nstc_root *root;
|
||||
UINT count, i;
|
||||
HRESULT hr;
|
||||
TRACE("%p (%p)\n", This, ppsiaRootItems);
|
||||
|
||||
count = list_count(&This->roots);
|
||||
|
||||
if(!count)
|
||||
return E_INVALIDARG;
|
||||
|
||||
array = HeapAlloc(GetProcessHeap(), 0, sizeof(LPITEMIDLIST*)*count);
|
||||
|
||||
i = 0;
|
||||
LIST_FOR_EACH_ENTRY(root, &This->roots, nstc_root, entry)
|
||||
SHGetIDListFromObject((IUnknown*)root->psi, &array[i++]);
|
||||
|
||||
SHGetDesktopFolder(&psf);
|
||||
hr = SHCreateShellItemArray(NULL, psf, count, (PCUITEMID_CHILD_ARRAY)array,
|
||||
ppsiaRootItems);
|
||||
IShellFolder_Release(psf);
|
||||
|
||||
for(i = 0; i < count; i++)
|
||||
ILFree(array[i]);
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, array);
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
static HRESULT WINAPI NSTC2_fnSetItemState(INameSpaceTreeControl2* iface,
|
||||
|
@ -32,6 +32,7 @@ static HWND hwnd;
|
||||
|
||||
static HRESULT (WINAPI *pSHCreateShellItem)(LPCITEMIDLIST,IShellFolder*,LPCITEMIDLIST,IShellItem**);
|
||||
static HRESULT (WINAPI *pSHGetIDListFromObject)(IUnknown*, PIDLIST_ABSOLUTE*);
|
||||
static HRESULT (WINAPI *pSHCreateItemFromParsingName)(PCWSTR,IBindCtx*,REFIID,void**);
|
||||
|
||||
static void init_function_pointers(void)
|
||||
{
|
||||
@ -40,6 +41,7 @@ static void init_function_pointers(void)
|
||||
hmod = GetModuleHandleA("shell32.dll");
|
||||
pSHCreateShellItem = (void*)GetProcAddress(hmod, "SHCreateShellItem");
|
||||
pSHGetIDListFromObject = (void*)GetProcAddress(hmod, "SHGetIDListFromObject");
|
||||
pSHCreateItemFromParsingName = (void*)GetProcAddress(hmod, "SHCreateItemFromParsingName");
|
||||
}
|
||||
|
||||
/*******************************************************
|
||||
@ -344,6 +346,63 @@ static void process_msgs(void)
|
||||
}
|
||||
}
|
||||
|
||||
/** Some functions from shell32/tests/shlfolder.c */
|
||||
/* creates a file with the specified name for tests */
|
||||
static void CreateTestFile(const CHAR *name)
|
||||
{
|
||||
HANDLE file;
|
||||
DWORD written;
|
||||
|
||||
file = CreateFileA(name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);
|
||||
if (file != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
WriteFile(file, name, strlen(name), &written, NULL);
|
||||
WriteFile(file, "\n", strlen("\n"), &written, NULL);
|
||||
CloseHandle(file);
|
||||
}
|
||||
}
|
||||
/* initializes the tests */
|
||||
static void CreateFilesFolders(void)
|
||||
{
|
||||
CreateDirectoryA(".\\testdir", NULL);
|
||||
CreateTestFile (".\\testdir\\test1.txt ");
|
||||
CreateTestFile (".\\testdir\\test2.txt ");
|
||||
CreateTestFile (".\\testdir\\test3.txt ");
|
||||
CreateDirectoryA(".\\testdir\\testdir2 ", NULL);
|
||||
CreateDirectoryA(".\\testdir\\testdir2\\subdir", NULL);
|
||||
}
|
||||
|
||||
/* cleans after tests */
|
||||
static void Cleanup(void)
|
||||
{
|
||||
DeleteFileA(".\\testdir\\test1.txt");
|
||||
DeleteFileA(".\\testdir\\test2.txt");
|
||||
DeleteFileA(".\\testdir\\test3.txt");
|
||||
RemoveDirectoryA(".\\testdir\\testdir2\\subdir");
|
||||
RemoveDirectoryA(".\\testdir\\testdir2");
|
||||
RemoveDirectoryA(".\\testdir");
|
||||
}
|
||||
|
||||
/* Based on PathAddBackslashW from dlls/shlwapi/path.c */
|
||||
static LPWSTR myPathAddBackslashW( LPWSTR lpszPath )
|
||||
{
|
||||
size_t iLen;
|
||||
|
||||
if (!lpszPath || (iLen = lstrlenW(lpszPath)) >= MAX_PATH)
|
||||
return NULL;
|
||||
|
||||
if (iLen)
|
||||
{
|
||||
lpszPath += iLen;
|
||||
if (lpszPath[-1] != '\\')
|
||||
{
|
||||
*lpszPath++ = '\\';
|
||||
*lpszPath = '\0';
|
||||
}
|
||||
}
|
||||
return lpszPath;
|
||||
}
|
||||
|
||||
/* Returns FALSE if the NamespaceTreeControl failed to be instantiated. */
|
||||
static BOOL test_initialization(void)
|
||||
{
|
||||
@ -495,21 +554,66 @@ static BOOL test_initialization(void)
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static void verify_root_order_(INameSpaceTreeControl *pnstc, IShellItem **roots,
|
||||
const char *file, int line)
|
||||
{
|
||||
HRESULT hr;
|
||||
IShellItemArray *psia;
|
||||
|
||||
hr = INameSpaceTreeControl_GetRootItems(pnstc, &psia);
|
||||
ok_(file,line) (hr == S_OK, "GetRootItems: got (0x%08x)\n", hr);
|
||||
if(SUCCEEDED(hr))
|
||||
{
|
||||
DWORD i, expected, count = -1;
|
||||
hr = IShellItemArray_GetCount(psia, &count);
|
||||
ok_(file,line) (hr == S_OK, "Got (0x%08x)\n", hr);
|
||||
|
||||
for(expected = 0; roots[expected] != NULL; expected++);
|
||||
ok_(file,line) (count == expected, "Got %d roots, expected %d\n", count, expected);
|
||||
|
||||
for(i = 0; i < count && roots[i] != NULL; i++)
|
||||
{
|
||||
IShellItem *psi;
|
||||
hr = IShellItemArray_GetItemAt(psia, i, &psi);
|
||||
ok_(file,line) (hr == S_OK, "GetItemAt %i: got 0x%08x\n", i, hr);
|
||||
if(SUCCEEDED(hr))
|
||||
{
|
||||
int cmp;
|
||||
hr = IShellItem_Compare(psi, roots[i], SICHINT_DISPLAY, &cmp);
|
||||
ok_(file,line) (hr == S_OK, "Compare %i: got 0x%08x\n", i, hr);
|
||||
IShellItem_Release(psi);
|
||||
}
|
||||
}
|
||||
IShellItem_Release(psia);
|
||||
}
|
||||
}
|
||||
#define verify_root_order(pnstc, psi_a) \
|
||||
verify_root_order_(pnstc, psi_a, __FILE__, __LINE__)
|
||||
|
||||
static void test_basics(void)
|
||||
{
|
||||
INameSpaceTreeControl *pnstc;
|
||||
INameSpaceTreeControl2 *pnstc2;
|
||||
IShellItemArray *psia;
|
||||
IShellFolder *psfdesktop;
|
||||
IShellItem *psidesktop, *psidesktop2;
|
||||
IShellItem *psitestdir, *psitestdir2;
|
||||
IOleWindow *pow;
|
||||
LPITEMIDLIST pidl_desktop;
|
||||
HRESULT hr;
|
||||
UINT i, res;
|
||||
RECT rc;
|
||||
IShellItem *roots[10];
|
||||
WCHAR curdirW[MAX_PATH];
|
||||
WCHAR buf[MAX_PATH];
|
||||
static const WCHAR testdirW[] = {'t','e','s','t','d','i','r',0};
|
||||
static const WCHAR testdir2W[] =
|
||||
{'t','e','s','t','d','i','r','\\','t','e','s','t','d','i','r','2',0};
|
||||
|
||||
/* These should exist on platforms supporting the NSTC */
|
||||
ok(pSHCreateShellItem != NULL, "No SHCreateShellItem.\n");
|
||||
ok(pSHGetIDListFromObject != NULL, "No SHCreateShellItem.\n");
|
||||
ok(pSHCreateItemFromParsingName != NULL, "No SHCreateItemFromParsingName\n");
|
||||
|
||||
/* Create ShellItems for testing. */
|
||||
SHGetDesktopFolder(&psfdesktop);
|
||||
@ -536,6 +640,23 @@ static void test_basics(void)
|
||||
return;
|
||||
}
|
||||
|
||||
CreateFilesFolders();
|
||||
GetCurrentDirectoryW(MAX_PATH, curdirW);
|
||||
ok(lstrlenW(curdirW), "Got 0 length string.\n");
|
||||
|
||||
lstrcpyW(buf, curdirW);
|
||||
myPathAddBackslashW(buf);
|
||||
lstrcatW(buf, testdirW);
|
||||
hr = pSHCreateItemFromParsingName(buf, NULL, &IID_IShellItem, (void**)&psitestdir);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
if(FAILED(hr)) goto cleanup;
|
||||
lstrcpyW(buf, curdirW);
|
||||
myPathAddBackslashW(buf);
|
||||
lstrcatW(buf, testdir2W);
|
||||
hr = pSHCreateItemFromParsingName(buf, NULL, &IID_IShellItem, (void**)&psitestdir2);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
if(FAILED(hr)) goto cleanup;
|
||||
|
||||
hr = CoCreateInstance(&CLSID_NamespaceTreeControl, NULL, CLSCTX_INPROC_SERVER,
|
||||
&IID_INameSpaceTreeControl, (void**)&pnstc);
|
||||
ok(hr == S_OK, "Failed to initialize control (0x%08x)\n", hr);
|
||||
@ -820,8 +941,89 @@ static void test_basics(void)
|
||||
hr = INameSpaceTreeControl_RemoveAllRoots(pnstc);
|
||||
ok(hr == S_OK, "Got (0x%08x)\n", hr);
|
||||
|
||||
/* GetRootItems */
|
||||
if(0)
|
||||
{
|
||||
/* Crashes on native. */
|
||||
hr = INameSpaceTreeControl_GetRootItems(pnstc, NULL);
|
||||
}
|
||||
|
||||
hr = INameSpaceTreeControl_GetRootItems(pnstc, &psia);
|
||||
ok(hr == E_INVALIDARG, "Got (0x%08x)\n", hr);
|
||||
|
||||
hr = INameSpaceTreeControl_AppendRoot(pnstc, psidesktop, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
hr = INameSpaceTreeControl_AppendRoot(pnstc, psidesktop2, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
hr = INameSpaceTreeControl_AppendRoot(pnstc, psitestdir, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
hr = INameSpaceTreeControl_AppendRoot(pnstc, psitestdir2, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
|
||||
roots[0] = psidesktop;
|
||||
roots[1] = psidesktop2;
|
||||
roots[2] = psitestdir;
|
||||
roots[3] = psitestdir2;
|
||||
roots[4] = NULL;
|
||||
verify_root_order(pnstc, roots);
|
||||
|
||||
hr = INameSpaceTreeControl_InsertRoot(pnstc, 0, psitestdir2, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
|
||||
roots[0] = psitestdir2;
|
||||
roots[1] = psidesktop;
|
||||
roots[2] = psidesktop2;
|
||||
roots[3] = psitestdir;
|
||||
roots[4] = psitestdir2;
|
||||
roots[5] = NULL;
|
||||
verify_root_order(pnstc, roots);
|
||||
|
||||
hr = INameSpaceTreeControl_InsertRoot(pnstc, 5, psidesktop, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
|
||||
roots[5] = psidesktop;
|
||||
roots[6] = NULL;
|
||||
verify_root_order(pnstc, roots);
|
||||
|
||||
hr = INameSpaceTreeControl_InsertRoot(pnstc, 3, psitestdir2, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
|
||||
roots[3] = psitestdir2;
|
||||
roots[4] = psitestdir;
|
||||
roots[5] = psitestdir2;
|
||||
roots[6] = psidesktop;
|
||||
roots[7] = NULL;
|
||||
verify_root_order(pnstc, roots);
|
||||
|
||||
hr = INameSpaceTreeControl_AppendRoot(pnstc, psitestdir2, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
|
||||
roots[7] = psitestdir2;
|
||||
roots[8] = NULL;
|
||||
verify_root_order(pnstc, roots);
|
||||
|
||||
hr = INameSpaceTreeControl_InsertRoot(pnstc, -1, psidesktop, 0, 0, NULL);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
|
||||
roots[0] = psidesktop;
|
||||
roots[1] = psitestdir2;
|
||||
roots[2] = psidesktop;
|
||||
roots[3] = psidesktop2;
|
||||
roots[4] = psitestdir2;
|
||||
roots[5] = psitestdir;
|
||||
roots[6] = psitestdir2;
|
||||
roots[7] = psidesktop;
|
||||
roots[8] = psitestdir2;
|
||||
roots[9] = NULL;
|
||||
verify_root_order(pnstc, roots);
|
||||
|
||||
hr = INameSpaceTreeControl_RemoveAllRoots(pnstc);
|
||||
ok(hr == S_OK, "Got (0x%08x)\n", hr);
|
||||
|
||||
IShellItem_Release(psidesktop);
|
||||
IShellItem_Release(psidesktop2);
|
||||
IShellItem_Release(psitestdir);
|
||||
IShellItem_Release(psitestdir2);
|
||||
|
||||
hr = INameSpaceTreeControl_QueryInterface(pnstc, &IID_IOleWindow, (void**)&pow);
|
||||
ok(hr == S_OK, "Got 0x%08x\n", hr);
|
||||
@ -836,6 +1038,9 @@ static void test_basics(void)
|
||||
|
||||
res = INameSpaceTreeControl_Release(pnstc);
|
||||
ok(!res, "res was %d!\n", res);
|
||||
|
||||
cleanup:
|
||||
Cleanup();
|
||||
}
|
||||
|
||||
static void test_events(void)
|
||||
|
Loading…
x
Reference in New Issue
Block a user