mirror of
https://github.com/reactos/reactos.git
synced 2025-08-05 16:43:04 +00:00
[ZIPFLDR_APITEST] Add some basic IDataObject tests
CORE-16241
This commit is contained in:
Mark Jansen
parent
534710a6d5
commit
0340c5188c
11 changed files with 915 additions and 0 deletions
266
modules/rostests/apitests/zipfldr/EnumObjects.cpp
Normal file
266
modules/rostests/apitests/zipfldr/EnumObjects.cpp
Normal file
|
|
@ -0,0 +1,266 @@
|
|||
/*
|
||||
* PROJECT: ReactOS api tests
|
||||
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
||||
* PURPOSE: Test for zipfldr
|
||||
* COPYRIGHT: Copyright 2020 Mark Jansen (mark.jansen@reactos.org)
|
||||
*/
|
||||
|
||||
#include "precomp.h"
|
||||
|
||||
static bool g_bOldZipfldr = false;
|
||||
|
||||
#define ok_displayname(pFolder, pidl, Flags, Name) ok_displayname_(__FILE__, __LINE__, pFolder, pidl, Flags, Name)
|
||||
static void ok_displayname_(const char* file, int line, IShellFolder* pFolder, PCUITEMID_CHILD pidl, SHGDNF Flags, LPCWSTR Name)
|
||||
{
|
||||
STRRET NameRet;
|
||||
HRESULT hr;
|
||||
ok_hr_(file, line, (hr = pFolder->GetDisplayNameOf(pidl, Flags, &NameRet)), S_OK);
|
||||
if (!SUCCEEDED(hr))
|
||||
return;
|
||||
|
||||
WCHAR DisplayName[MAX_PATH];
|
||||
ok_hr_(file, line, (hr = StrRetToBufW(&NameRet, pidl, DisplayName, ARRAYSIZE(DisplayName))), S_OK);
|
||||
if (!SUCCEEDED(hr))
|
||||
return;
|
||||
|
||||
ok_wstr_(file, line, DisplayName, Name);
|
||||
}
|
||||
|
||||
struct ZipFiles
|
||||
{
|
||||
LPCWSTR Name;
|
||||
SFGAOF Attributes;
|
||||
};
|
||||
|
||||
static CLIPFORMAT cfHIDA = RegisterClipboardFormatA(CFSTR_SHELLIDLISTA);
|
||||
static CLIPFORMAT cfFileDescriptor = RegisterClipboardFormatW(CFSTR_FILEDESCRIPTORW);
|
||||
|
||||
static void test_EnumObjects_Files(const WCHAR* Filename, IShellFolder* pFolder)
|
||||
{
|
||||
CComPtr<IEnumIDList> spEnum;
|
||||
HRESULT hr = pFolder->EnumObjects(NULL, SHCONTF_NONFOLDERS, &spEnum);
|
||||
ok_hr(hr, S_OK);
|
||||
if (!SUCCEEDED(hr))
|
||||
return;
|
||||
|
||||
SFGAOF BaseAttributes = SFGAO_STREAM | SFGAO_HASPROPSHEET | SFGAO_CANDELETE | SFGAO_CANCOPY | SFGAO_CANMOVE;
|
||||
ZipFiles ExpectedFiles[] = {
|
||||
{ L"cccc.txt", BaseAttributes },
|
||||
{ L"bbbb.txt", BaseAttributes },
|
||||
};
|
||||
|
||||
ULONG totalFetched = 0;
|
||||
do
|
||||
{
|
||||
CComHeapPtr<ITEMID_CHILD> child;
|
||||
ULONG celtFetched = 0;
|
||||
hr = spEnum->Next(1, &child, &celtFetched);
|
||||
if (hr != S_OK)
|
||||
break;
|
||||
ok_int(celtFetched, 1);
|
||||
if (celtFetched != 1)
|
||||
break;
|
||||
|
||||
LPCWSTR ExpectedName = totalFetched < ARRAYSIZE(ExpectedFiles) ? ExpectedFiles[totalFetched].Name : L"<TOO MANY FILES>";
|
||||
SFGAOF ExpectedAttributes = totalFetched < ARRAYSIZE(ExpectedFiles) ? ExpectedFiles[totalFetched].Attributes : 0xdeaddead;
|
||||
|
||||
totalFetched++;
|
||||
|
||||
|
||||
CComPtr<IDataObject> spDataObj;
|
||||
hr = pFolder->GetUIObjectOf(NULL, 1, &child, IID_IDataObject, NULL, reinterpret_cast<LPVOID*>(&spDataObj));
|
||||
ok_hr(hr, S_OK);
|
||||
|
||||
if (!g_bOldZipfldr && !IsFormatAdvertised(spDataObj, cfHIDA, TYMED_HGLOBAL))
|
||||
{
|
||||
trace("Pre-Vista zipfldr\n");
|
||||
// No seconds in filetimes, less functional IStream* implementation
|
||||
g_bOldZipfldr = true;
|
||||
}
|
||||
|
||||
ok_displayname(pFolder, child, SHGDN_NORMAL, ExpectedName);
|
||||
if (g_bOldZipfldr)
|
||||
{
|
||||
ok_displayname(pFolder, child, SHGDN_FORPARSING, ExpectedName);
|
||||
}
|
||||
else
|
||||
{
|
||||
WCHAR Buffer[MAX_PATH];
|
||||
StringCchPrintfW(Buffer, _countof(Buffer), L"%s\\%s", Filename, ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FORPARSING, Buffer);
|
||||
}
|
||||
ok_displayname(pFolder, child, SHGDN_FORPARSING | SHGDN_INFOLDER, ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FOREDITING , ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FOREDITING | SHGDN_INFOLDER, ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FORADDRESSBAR , ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FORADDRESSBAR | SHGDN_INFOLDER, ExpectedName);
|
||||
|
||||
SFGAOF Attributes = 0;
|
||||
hr = pFolder->GetAttributesOf(1, &child, &Attributes);
|
||||
ok_hr(hr, S_OK);
|
||||
ok_hex(Attributes, ExpectedAttributes);
|
||||
|
||||
STGMEDIUM medium = {0};
|
||||
FORMATETC etc = { cfFileDescriptor, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
|
||||
HRESULT hr = spDataObj->GetData(&etc, &medium);
|
||||
ok_hex(hr, S_OK);
|
||||
if (!SUCCEEDED(hr))
|
||||
continue;
|
||||
|
||||
ok_hex(medium.tymed, TYMED_HGLOBAL);
|
||||
PVOID pData = GlobalLock(medium.hGlobal);
|
||||
FILEGROUPDESCRIPTORW* Descriptor = static_cast<FILEGROUPDESCRIPTORW*>(pData);
|
||||
ok_hex(Descriptor->cItems, 1);
|
||||
if (Descriptor->cItems == 1)
|
||||
ok_wstr(Descriptor->fgd[0].cFileName, ExpectedName);
|
||||
GlobalUnlock(medium.hGlobal);
|
||||
ReleaseStgMedium(&medium);
|
||||
} while (true);
|
||||
|
||||
ok_int(totalFetched, ARRAYSIZE(ExpectedFiles));
|
||||
ok_hr(hr, S_FALSE);
|
||||
}
|
||||
|
||||
static void test_EnumObjects_Folders(const WCHAR* Filename, IShellFolder* pFolder)
|
||||
{
|
||||
CComPtr<IEnumIDList> spEnum;
|
||||
HRESULT hr = pFolder->EnumObjects(NULL, SHCONTF_FOLDERS, &spEnum);
|
||||
ok_hr(hr, S_OK);
|
||||
if (!SUCCEEDED(hr))
|
||||
return;
|
||||
|
||||
SFGAOF BaseAttributes = SFGAO_FOLDER | SFGAO_DROPTARGET | SFGAO_HASPROPSHEET | SFGAO_CANDELETE | SFGAO_CANCOPY | SFGAO_CANMOVE;
|
||||
ZipFiles ExpectedFolders[] = {
|
||||
{ L"aaaa", BaseAttributes | (g_bOldZipfldr ? 0 : SFGAO_STORAGE) },
|
||||
{ L"cccc", BaseAttributes | (g_bOldZipfldr ? 0 : SFGAO_STORAGE) },
|
||||
};
|
||||
|
||||
LPCWSTR ExpectedExtraFiles[2][4] =
|
||||
{
|
||||
{
|
||||
L"aaaa\\a_aaaa",
|
||||
L"aaaa\\a_cccc",
|
||||
L"aaaa\\a_bbbb.txt",
|
||||
L"aaaa\\a_cccc.txt",
|
||||
},
|
||||
{
|
||||
L"cccc\\c_cccc",
|
||||
L"cccc\\c_bbbb.txt",
|
||||
L"cccc\\c_cccc.txt",
|
||||
L"cccc\\c_aaaa",
|
||||
},
|
||||
};
|
||||
|
||||
ULONG totalFetched = 0;
|
||||
do
|
||||
{
|
||||
CComHeapPtr<ITEMID_CHILD> child;
|
||||
ULONG celtFetched = 0;
|
||||
hr = spEnum->Next(1, &child, &celtFetched);
|
||||
if (hr != S_OK)
|
||||
break;
|
||||
ok_int(celtFetched, 1);
|
||||
if (celtFetched != 1)
|
||||
break;
|
||||
|
||||
LPCWSTR ExpectedName = totalFetched < ARRAYSIZE(ExpectedFolders) ? ExpectedFolders[totalFetched].Name : L"<TOO MANY FILES>";
|
||||
SFGAOF ExpectedAttributes = totalFetched < ARRAYSIZE(ExpectedFolders) ? ExpectedFolders[totalFetched].Attributes : 0xdeaddead;
|
||||
LPCWSTR* ExtraFiles = totalFetched < ARRAYSIZE(ExpectedExtraFiles) ? ExpectedExtraFiles[totalFetched] : ExpectedExtraFiles[0];
|
||||
|
||||
totalFetched++;
|
||||
|
||||
ok_displayname(pFolder, child, SHGDN_NORMAL, ExpectedName);
|
||||
if (g_bOldZipfldr)
|
||||
{
|
||||
ok_displayname(pFolder, child, SHGDN_FORPARSING, ExpectedName);
|
||||
}
|
||||
else
|
||||
{
|
||||
WCHAR Buffer[MAX_PATH];
|
||||
StringCchPrintfW(Buffer, _countof(Buffer), L"%s\\%s", Filename, ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FORPARSING, Buffer);
|
||||
}
|
||||
ok_displayname(pFolder, child, SHGDN_FORPARSING | SHGDN_INFOLDER, ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FOREDITING , ExpectedName);
|
||||
ok_displayname(pFolder, child, SHGDN_FOREDITING | SHGDN_INFOLDER, ExpectedName);
|
||||
|
||||
SFGAOF Attributes = 0;
|
||||
hr = pFolder->GetAttributesOf(1, &child, &Attributes);
|
||||
ok_hr(hr, S_OK);
|
||||
ok_hex(Attributes, ExpectedAttributes);
|
||||
|
||||
CComPtr<IDataObject> spData;
|
||||
hr = pFolder->GetUIObjectOf(NULL, 1, &child, IID_IDataObject, NULL, reinterpret_cast<LPVOID*>(&spData));
|
||||
ok_hr(hr, S_OK);
|
||||
|
||||
STGMEDIUM medium = {0};
|
||||
FORMATETC etc = { cfFileDescriptor, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
|
||||
HRESULT hr = spData->GetData(&etc, &medium);
|
||||
ok_hex(hr, S_OK);
|
||||
if (!SUCCEEDED(hr))
|
||||
continue;
|
||||
|
||||
ok_hex(medium.tymed, TYMED_HGLOBAL);
|
||||
PVOID pData = GlobalLock(medium.hGlobal);
|
||||
FILEGROUPDESCRIPTORW* Descriptor = static_cast<FILEGROUPDESCRIPTORW*>(pData);
|
||||
ok_hex(Descriptor->cItems, 5);
|
||||
if (Descriptor->cItems == 5)
|
||||
{
|
||||
ok_wstr(Descriptor->fgd[0].cFileName, ExpectedName);
|
||||
ok_wstr(Descriptor->fgd[1].cFileName, ExtraFiles[0]);
|
||||
ok_wstr(Descriptor->fgd[2].cFileName, ExtraFiles[1]);
|
||||
ok_wstr(Descriptor->fgd[3].cFileName, ExtraFiles[2]);
|
||||
ok_wstr(Descriptor->fgd[4].cFileName, ExtraFiles[3]);
|
||||
}
|
||||
GlobalUnlock(medium.hGlobal);
|
||||
ReleaseStgMedium(&medium);
|
||||
} while (true);
|
||||
|
||||
ok_int(totalFetched, ARRAYSIZE(ExpectedFolders));
|
||||
ok_hr(hr, S_FALSE);
|
||||
}
|
||||
|
||||
static void test_EnumObjects(const WCHAR* Filename)
|
||||
{
|
||||
CComPtr<IShellFolder> spFolder;
|
||||
if (!InitializeShellFolder(Filename, spFolder))
|
||||
return;
|
||||
|
||||
// Let's ask the shell how it wants to display the full path to our zip file
|
||||
WCHAR ZipFilename[MAX_PATH] = L"<Failed to query zip path>";
|
||||
{
|
||||
CComPtr<IPersistFolder2> spPersistFolder;
|
||||
HRESULT hr = spFolder->QueryInterface(IID_PPV_ARG(IPersistFolder2, &spPersistFolder));
|
||||
ok_hr(hr, S_OK);
|
||||
if (SUCCEEDED(hr))
|
||||
{
|
||||
CComHeapPtr<ITEMIDLIST_ABSOLUTE> zipPidl;
|
||||
hr = spPersistFolder->GetCurFolder(&zipPidl);
|
||||
ok_hr(hr, S_OK);
|
||||
if (SUCCEEDED(hr))
|
||||
{
|
||||
BOOL bHasPath = SHGetPathFromIDListW(zipPidl, ZipFilename);
|
||||
ok_int(bHasPath, TRUE);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
test_EnumObjects_Files(ZipFilename, spFolder);
|
||||
test_EnumObjects_Folders(ZipFilename, spFolder);
|
||||
}
|
||||
|
||||
START_TEST(EnumObjects)
|
||||
{
|
||||
HRESULT hr = CoInitialize(NULL);
|
||||
|
||||
ok_hr(hr, S_OK);
|
||||
if (!SUCCEEDED(hr))
|
||||
return;
|
||||
|
||||
WCHAR ZipTestFile[MAX_PATH];
|
||||
if (!extract_resource(ZipTestFile, MAKEINTRESOURCEW(IDR_ZIP_TEST_ENUM)))
|
||||
return;
|
||||
test_EnumObjects(ZipTestFile);
|
||||
DeleteFileW(ZipTestFile);
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue