dlls/shell32/tests/shlfolder.c - wine - Git at Google

 /*
  * Unit test of the IShellFolder functions.
  *
  * Copyright 2004 Vitaliy Margolen
  *
  * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */

 #include <stdarg.h>
 #include <stdio.h>

 #define COBJMACROS

 #include "windef.h"
 #include "winbase.h"
 #include "wtypes.h"
 #include "shellapi.h"


 #include "shlguid.h"
 #include "shlobj.h"
 #include "shobjidl.h"


 #include "wine/unicode.h"
 #include "wine/test.h"


 static IMalloc *ppM;

 /* 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);
     CreateDirectoryA(".\\testdir\\test.txt", 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\\test.txt");
     RemoveDirectoryA(".\\testdir\\testdir2\\subdir");
     RemoveDirectoryA(".\\testdir\\testdir2");
     RemoveDirectoryA(".\\testdir");
 }


 /* perform test */
 static void test_EnumObjects(IShellFolder *iFolder)
 {
     IEnumIDList *iEnumList;
     ITEMIDLIST *newPIDL, *(idlArr [5]);
     ULONG NumPIDLs;
     int i=0, j;
     HRESULT hr;

     static const WORD iResults [5][5] =
     {
 	{ 0,-1,-1,-1,-1},
 	{ 1, 0,-1,-1,-1},
 	{ 1, 1, 0,-1,-1},
 	{ 1, 1, 1, 0,-1},
 	{ 1, 1, 1, 1, 0}
     };

     /* Just test SFGAO_FILESYSTEM | SFGAO_FOLDER | SFGAO_FILESYSANCESTOR for now */
     static const ULONG attrs[5] =
     {
         SFGAO_FILESYSTEM | SFGAO_FOLDER | SFGAO_FILESYSANCESTOR,
         SFGAO_FILESYSTEM | SFGAO_FOLDER | SFGAO_FILESYSANCESTOR,
         SFGAO_FILESYSTEM,
         SFGAO_FILESYSTEM,
         SFGAO_FILESYSTEM,
     };

     hr = IShellFolder_EnumObjects(iFolder, NULL, SHCONTF_FOLDERS | SHCONTF_NONFOLDERS | SHCONTF_INCLUDEHIDDEN, &iEnumList);
     ok(hr == S_OK, "EnumObjects failed %08lx\n", hr);

     while (IEnumIDList_Next(iEnumList, 1, &newPIDL, &NumPIDLs) == S_OK)
     {
         idlArr[i++] = newPIDL;
     }

     hr = IEnumIDList_Release(iEnumList);
     ok(hr == S_OK, "IEnumIDList_Release failed %08lx\n", hr);

     /* Sort them first in case of wrong order from system */
     for (i=0;i<5;i++) for (j=0;j<5;j++)
         if ((SHORT)IShellFolder_CompareIDs(iFolder, 0, idlArr[i], idlArr[j]) < 0)
 	{
             newPIDL = idlArr[i];
             idlArr[i] = idlArr[j];
             idlArr[j] = newPIDL;
         }

     for (i=0;i<5;i++) for (j=0;j<5;j++)
     {
         hr = IShellFolder_CompareIDs(iFolder, 0, idlArr[i], idlArr[j]);
         ok(hr == iResults[i][j], "Got %lx expected [%d]-[%d]=%x\n", hr, i, j, iResults[i][j]);
     }


     for (i = 0; i < 5; i++)
     {
         SFGAOF flags;
         flags = SFGAO_FILESYSTEM | SFGAO_FOLDER | SFGAO_FILESYSANCESTOR;
         hr = IShellFolder_GetAttributesOf(iFolder, 1, (LPCITEMIDLIST*)(idlArr + i), &flags);
         flags &= SFGAO_FILESYSTEM | SFGAO_FOLDER | SFGAO_FILESYSANCESTOR;
         ok(hr == S_OK, "GetAttributesOf returns %08lx\n", hr);
         ok(flags == attrs[i], "GetAttributesOf gets attrs %08lx, expects %08lx\n", flags, attrs[i]);
     }

     for (i=0;i<5;i++)
         IMalloc_Free(ppM, idlArr[i]);
 }

 static void test_BindToObject(void)
 {
     HRESULT hr;
     UINT cChars;
     IShellFolder *psfDesktop, *psfChild, *psfMyComputer, *psfSystemDir;
     SHITEMID emptyitem = { 0, { 0 } };
     LPITEMIDLIST pidlMyComputer, pidlSystemDir, pidlEmpty = (LPITEMIDLIST)&emptyitem;
     WCHAR wszSystemDir[MAX_PATH];
     WCHAR wszMyComputer[] = {
         ':',':','{','2','0','D','0','4','F','E','0','-','3','A','E','A','-','1','0','6','9','-',
         'A','2','D','8','-','0','8','0','0','2','B','3','0','3','0','9','D','}',0 };

     /* The following tests shows that BindToObject should fail with E_INVALIDARG if called
      * with an empty pidl. This is tested for Desktop, MyComputer and the FS ShellFolder
      */
     hr = SHGetDesktopFolder(&psfDesktop);
     ok (SUCCEEDED(hr), "SHGetDesktopFolder failed! hr = %08lx\n", hr);
     if (FAILED(hr)) return;

     hr = IShellFolder_BindToObject(psfDesktop, pidlEmpty, NULL, &IID_IShellFolder, (LPVOID*)&psfChild);
     ok (hr == E_INVALIDARG, "Desktop's BindToObject should fail, when called with empty pidl! hr = %08lx\n", hr);

     hr = IShellFolder_ParseDisplayName(psfDesktop, NULL, NULL, wszMyComputer, NULL, &pidlMyComputer, NULL);
     ok (SUCCEEDED(hr), "Desktop's ParseDisplayName failed to parse MyComputer's CLSID! hr = %08lx\n", hr);
     if (FAILED(hr)) {
         IShellFolder_Release(psfDesktop);
         return;
     }

     hr = IShellFolder_BindToObject(psfDesktop, pidlMyComputer, NULL, &IID_IShellFolder, (LPVOID*)&psfMyComputer);
     ok (SUCCEEDED(hr), "Desktop failed to bind to MyComputer object! hr = %08lx\n", hr);
     IShellFolder_Release(psfDesktop);
     ILFree(pidlMyComputer);
     if (FAILED(hr)) return;

     hr = IShellFolder_BindToObject(psfMyComputer, pidlEmpty, NULL, &IID_IShellFolder, (LPVOID*)&psfChild);
     ok (hr == E_INVALIDARG, "MyComputers's BindToObject should fail, when called with empty pidl! hr = %08lx\n", hr);

     cChars = GetSystemDirectoryW(wszSystemDir, MAX_PATH);
     ok (cChars > 0 && cChars < MAX_PATH, "GetSystemDirectoryW failed! LastError: %08lx\n", GetLastError());
     if (cChars == 0 || cChars >= MAX_PATH) {
         IShellFolder_Release(psfMyComputer);
         return;
     }

     hr = IShellFolder_ParseDisplayName(psfMyComputer, NULL, NULL, wszSystemDir, NULL, &pidlSystemDir, NULL);
     ok (SUCCEEDED(hr), "MyComputers's ParseDisplayName failed to parse the SystemDirectory! hr = %08lx\n", hr);
     if (FAILED(hr)) {
         IShellFolder_Release(psfMyComputer);
         return;
     }

     hr = IShellFolder_BindToObject(psfMyComputer, pidlSystemDir, NULL, &IID_IShellFolder, (LPVOID*)&psfSystemDir);
     ok (SUCCEEDED(hr), "MyComputer failed to bind to a FileSystem ShellFolder! hr = %08lx\n", hr);
     IShellFolder_Release(psfMyComputer);
     ILFree(pidlSystemDir);
     if (FAILED(hr)) return;

     hr = IShellFolder_BindToObject(psfSystemDir, pidlEmpty, NULL, &IID_IShellFolder, (LPVOID*)&psfChild);
     ok (hr == E_INVALIDARG,
         "FileSystem ShellFolder's BindToObject should fail, when called with empty pidl! hr = %08lx\n", hr);

     IShellFolder_Release(psfSystemDir);
 }

 START_TEST(shlfolder)
 {
     ITEMIDLIST *newPIDL;
     IShellFolder *IDesktopFolder, *testIShellFolder;
     char  cCurrDirA [MAX_PATH] = {0};
     WCHAR cCurrDirW [MAX_PATH];
     static const WCHAR cTestDirW[] = {'\\','t','e','s','t','d','i','r',0};
     HRESULT hr;

     GetCurrentDirectoryA(MAX_PATH, cCurrDirA);
     MultiByteToWideChar(CP_ACP, 0, cCurrDirA, -1, cCurrDirW, MAX_PATH);
     strcatW(cCurrDirW, cTestDirW);

     OleInitialize(NULL);

     hr = SHGetMalloc(&ppM);
     ok(hr == S_OK, "SHGetMalloc failed %08lx\n", hr);

     CreateFilesFolders();

     hr = SHGetDesktopFolder(&IDesktopFolder);
     ok(hr == S_OK, "SHGetDesktopfolder failed %08lx\n", hr);

     hr = IShellFolder_ParseDisplayName(IDesktopFolder, NULL, NULL, cCurrDirW, NULL, &newPIDL, 0);
     ok(hr == S_OK, "ParseDisplayName failed %08lx\n", hr);

     hr = IShellFolder_BindToObject(IDesktopFolder, newPIDL, NULL, (REFIID)&IID_IShellFolder, (LPVOID *)&testIShellFolder);
     ok(hr == S_OK, "BindToObject failed %08lx\n", hr);

     test_EnumObjects(testIShellFolder);
     test_BindToObject();

     hr = IShellFolder_Release(testIShellFolder);
     ok(hr == S_OK, "IShellFolder_Release failed %08lx\n", hr);

     IMalloc_Free(ppM, newPIDL);

     Cleanup();
 }
	/*
	* Unit test of the IShellFolder functions.
	*
	* Copyright 2004 Vitaliy Margolen
	*
	* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	*/

	#include <stdarg.h>
	#include <stdio.h>

	#define COBJMACROS

	#include "windef.h"
	#include "winbase.h"
	#include "wtypes.h"
	#include "shellapi.h"


	#include "shlguid.h"
	#include "shlobj.h"
	#include "shobjidl.h"


	#include "wine/unicode.h"
	#include "wine/test.h"


	static IMalloc *ppM;

	/* 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);
	CreateDirectoryA(".\\testdir\\test.txt", 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\\test.txt");
	RemoveDirectoryA(".\\testdir\\testdir2\\subdir");
	RemoveDirectoryA(".\\testdir\\testdir2");
	RemoveDirectoryA(".\\testdir");
	}


	/* perform test */
	static void test_EnumObjects(IShellFolder *iFolder)
	{
	IEnumIDList *iEnumList;
	ITEMIDLIST newPIDL, (idlArr [5]);
	ULONG NumPIDLs;
	int i=0, j;
	HRESULT hr;

	static const WORD iResults [5][5] =
	{
	{ 0,-1,-1,-1,-1},
	{ 1, 0,-1,-1,-1},
	{ 1, 1, 0,-1,-1},
	{ 1, 1, 1, 0,-1},
	{ 1, 1, 1, 1, 0}
	};

	/* Just test SFGAO_FILESYSTEM \| SFGAO_FOLDER \| SFGAO_FILESYSANCESTOR for now */
	static const ULONG attrs[5] =
	{
	SFGAO_FILESYSTEM \| SFGAO_FOLDER \| SFGAO_FILESYSANCESTOR,
	SFGAO_FILESYSTEM \| SFGAO_FOLDER \| SFGAO_FILESYSANCESTOR,
	SFGAO_FILESYSTEM,
	SFGAO_FILESYSTEM,
	SFGAO_FILESYSTEM,
	};

	hr = IShellFolder_EnumObjects(iFolder, NULL, SHCONTF_FOLDERS \| SHCONTF_NONFOLDERS \| SHCONTF_INCLUDEHIDDEN, &iEnumList);
	ok(hr == S_OK, "EnumObjects failed %08lx\n", hr);

	while (IEnumIDList_Next(iEnumList, 1, &newPIDL, &NumPIDLs) == S_OK)
	{
	idlArr[i++] = newPIDL;
	}

	hr = IEnumIDList_Release(iEnumList);
	ok(hr == S_OK, "IEnumIDList_Release failed %08lx\n", hr);

	/* Sort them first in case of wrong order from system */
	for (i=0;i<5;i++) for (j=0;j<5;j++)
	if ((SHORT)IShellFolder_CompareIDs(iFolder, 0, idlArr[i], idlArr[j]) < 0)
	{
	newPIDL = idlArr[i];
	idlArr[i] = idlArr[j];
	idlArr[j] = newPIDL;
	}

	for (i=0;i<5;i++) for (j=0;j<5;j++)
	{
	hr = IShellFolder_CompareIDs(iFolder, 0, idlArr[i], idlArr[j]);
	ok(hr == iResults[i][j], "Got %lx expected [%d]-[%d]=%x\n", hr, i, j, iResults[i][j]);
	}


	for (i = 0; i < 5; i++)
	{
	SFGAOF flags;
	flags = SFGAO_FILESYSTEM \| SFGAO_FOLDER \| SFGAO_FILESYSANCESTOR;
	hr = IShellFolder_GetAttributesOf(iFolder, 1, (LPCITEMIDLIST*)(idlArr + i), &flags);
	flags &= SFGAO_FILESYSTEM \| SFGAO_FOLDER \| SFGAO_FILESYSANCESTOR;
	ok(hr == S_OK, "GetAttributesOf returns %08lx\n", hr);
	ok(flags == attrs[i], "GetAttributesOf gets attrs %08lx, expects %08lx\n", flags, attrs[i]);
	}

	for (i=0;i<5;i++)
	IMalloc_Free(ppM, idlArr[i]);
	}

	static void test_BindToObject(void)
	{
	HRESULT hr;
	UINT cChars;
	IShellFolder psfDesktop, psfChild, psfMyComputer, psfSystemDir;
	SHITEMID emptyitem = { 0, { 0 } };
	LPITEMIDLIST pidlMyComputer, pidlSystemDir, pidlEmpty = (LPITEMIDLIST)&emptyitem;
	WCHAR wszSystemDir[MAX_PATH];
	WCHAR wszMyComputer[] = {
	':',':','{','2','0','D','0','4','F','E','0','-','3','A','E','A','-','1','0','6','9','-',
	'A','2','D','8','-','0','8','0','0','2','B','3','0','3','0','9','D','}',0 };

	/* The following tests shows that BindToObject should fail with E_INVALIDARG if called
	* with an empty pidl. This is tested for Desktop, MyComputer and the FS ShellFolder
	*/
	hr = SHGetDesktopFolder(&psfDesktop);
	ok (SUCCEEDED(hr), "SHGetDesktopFolder failed! hr = %08lx\n", hr);
	if (FAILED(hr)) return;

	hr = IShellFolder_BindToObject(psfDesktop, pidlEmpty, NULL, &IID_IShellFolder, (LPVOID*)&psfChild);
	ok (hr == E_INVALIDARG, "Desktop's BindToObject should fail, when called with empty pidl! hr = %08lx\n", hr);

	hr = IShellFolder_ParseDisplayName(psfDesktop, NULL, NULL, wszMyComputer, NULL, &pidlMyComputer, NULL);
	ok (SUCCEEDED(hr), "Desktop's ParseDisplayName failed to parse MyComputer's CLSID! hr = %08lx\n", hr);
	if (FAILED(hr)) {
	IShellFolder_Release(psfDesktop);
	return;
	}

	hr = IShellFolder_BindToObject(psfDesktop, pidlMyComputer, NULL, &IID_IShellFolder, (LPVOID*)&psfMyComputer);
	ok (SUCCEEDED(hr), "Desktop failed to bind to MyComputer object! hr = %08lx\n", hr);
	IShellFolder_Release(psfDesktop);
	ILFree(pidlMyComputer);
	if (FAILED(hr)) return;

	hr = IShellFolder_BindToObject(psfMyComputer, pidlEmpty, NULL, &IID_IShellFolder, (LPVOID*)&psfChild);
	ok (hr == E_INVALIDARG, "MyComputers's BindToObject should fail, when called with empty pidl! hr = %08lx\n", hr);

	cChars = GetSystemDirectoryW(wszSystemDir, MAX_PATH);
	ok (cChars > 0 && cChars < MAX_PATH, "GetSystemDirectoryW failed! LastError: %08lx\n", GetLastError());
	if (cChars == 0 \|\| cChars >= MAX_PATH) {
	IShellFolder_Release(psfMyComputer);
	return;
	}

	hr = IShellFolder_ParseDisplayName(psfMyComputer, NULL, NULL, wszSystemDir, NULL, &pidlSystemDir, NULL);
	ok (SUCCEEDED(hr), "MyComputers's ParseDisplayName failed to parse the SystemDirectory! hr = %08lx\n", hr);
	if (FAILED(hr)) {
	IShellFolder_Release(psfMyComputer);
	return;
	}

	hr = IShellFolder_BindToObject(psfMyComputer, pidlSystemDir, NULL, &IID_IShellFolder, (LPVOID*)&psfSystemDir);
	ok (SUCCEEDED(hr), "MyComputer failed to bind to a FileSystem ShellFolder! hr = %08lx\n", hr);
	IShellFolder_Release(psfMyComputer);
	ILFree(pidlSystemDir);
	if (FAILED(hr)) return;

	hr = IShellFolder_BindToObject(psfSystemDir, pidlEmpty, NULL, &IID_IShellFolder, (LPVOID*)&psfChild);
	ok (hr == E_INVALIDARG,
	"FileSystem ShellFolder's BindToObject should fail, when called with empty pidl! hr = %08lx\n", hr);

	IShellFolder_Release(psfSystemDir);
	}

	START_TEST(shlfolder)
	{
	ITEMIDLIST *newPIDL;
	IShellFolder IDesktopFolder, testIShellFolder;
	char cCurrDirA [MAX_PATH] = {0};
	WCHAR cCurrDirW [MAX_PATH];
	static const WCHAR cTestDirW[] = {'\\','t','e','s','t','d','i','r',0};
	HRESULT hr;

	GetCurrentDirectoryA(MAX_PATH, cCurrDirA);
	MultiByteToWideChar(CP_ACP, 0, cCurrDirA, -1, cCurrDirW, MAX_PATH);
	strcatW(cCurrDirW, cTestDirW);

	OleInitialize(NULL);

	hr = SHGetMalloc(&ppM);
	ok(hr == S_OK, "SHGetMalloc failed %08lx\n", hr);

	CreateFilesFolders();

	hr = SHGetDesktopFolder(&IDesktopFolder);
	ok(hr == S_OK, "SHGetDesktopfolder failed %08lx\n", hr);

	hr = IShellFolder_ParseDisplayName(IDesktopFolder, NULL, NULL, cCurrDirW, NULL, &newPIDL, 0);
	ok(hr == S_OK, "ParseDisplayName failed %08lx\n", hr);

	hr = IShellFolder_BindToObject(IDesktopFolder, newPIDL, NULL, (REFIID)&IID_IShellFolder, (LPVOID *)&testIShellFolder);
	ok(hr == S_OK, "BindToObject failed %08lx\n", hr);

	test_EnumObjects(testIShellFolder);
	test_BindToObject();

	hr = IShellFolder_Release(testIShellFolder);
	ok(hr == S_OK, "IShellFolder_Release failed %08lx\n", hr);

	IMalloc_Free(ppM, newPIDL);

	Cleanup();
	}