mirror of
https://github.com/microsoft/PowerToys.git
synced 2026-04-05 18:57:19 +02:00
* Move retrieveing file attibutes to PowerRenameRegex Move file attributes unit tests to PowerRenameRegexTests Add file time field to MockPowerRenameItem * Add file attributes unittests to PowerRenameManagerTests * Change variable name * Rearrange function arguments * Check if file attributes are used only once * Change variable name LocalTime -> fileTime, date -> time * Set fileTime as a member of PowerRenameRegEx rather than passing as an argument * Change function name isFileAttributesUsed() -> isFileTimeUsed() Check before resetting fileTime * Fix small bugs * Fix typos * Refactor for readability, move free calls to reachable places * Fix search for area empty bug searchTerm being empty is not an invalid argument rather it must return OK without any operation Tests must check if Replace() returns S_OK becuase later it checks its result * Check return values of method calls in PowerRenameManager Remove received argments checks from some methods because argument being null or empty string doesnt mean it is invalid or method fails * Fix formatting. Remove overlooked comment. Fix error message. * Change HRESULT declarations according to coding style * Fix unhandled case. Refactor.
517 lines
20 KiB
C++
517 lines
20 KiB
C++
#include "pch.h"
|
|
#include "CppUnitTest.h"
|
|
#include "powerrename/lib/Settings.h"
|
|
#include <PowerRenameInterfaces.h>
|
|
#include <PowerRenameRegEx.h>
|
|
#include "MockPowerRenameRegExEvents.h"
|
|
|
|
using namespace Microsoft::VisualStudio::CppUnitTestFramework;
|
|
|
|
namespace PowerRenameRegExTests
|
|
{
|
|
struct SearchReplaceExpected
|
|
{
|
|
PCWSTR search;
|
|
PCWSTR replace;
|
|
PCWSTR test;
|
|
PCWSTR expected;
|
|
};
|
|
|
|
TEST_CLASS(SimpleTests){
|
|
public:
|
|
TEST_CLASS_INITIALIZE(ClassInitialize)
|
|
{
|
|
CSettingsInstance().SetUseBoostLib(false);
|
|
}
|
|
|
|
TEST_METHOD(GeneralReplaceTest)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(L"foo") == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(L"big") == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(L"foobar", &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, L"bigbar") == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
|
|
TEST_METHOD(ReplaceNoMatch)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(L"notfound") == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(L"big") == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(L"foobar", &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, L"foobar") == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
|
|
TEST_METHOD(ReplaceNoSearchOrReplaceTerm)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->Replace(L"foobar", &result) == S_OK);
|
|
Assert::IsTrue(result == nullptr);
|
|
CoTaskMemFree(result);
|
|
}
|
|
|
|
TEST_METHOD(ReplaceNoReplaceTerm)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(L"foo") == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(L"foobar", &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, L"bar") == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
|
|
TEST_METHOD(ReplaceEmptyStringReplaceTerm)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(L"foo") == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(L"") == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(L"foobar", &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, L"bar") == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
|
|
TEST_METHOD(VerifyDefaultFlags)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = 0;
|
|
Assert::IsTrue(renameRegEx->GetFlags(&flags) == S_OK);
|
|
Assert::IsTrue(flags == MatchAllOccurences);
|
|
}
|
|
|
|
TEST_METHOD(VerifyCaseSensitiveSearch)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = CaseSensitive;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L"Foo", L"Foo", L"FooBar", L"FooBar" },
|
|
{ L"Foo", L"boo", L"FooBar", L"booBar" },
|
|
{ L"Foo", L"boo", L"foobar", L"foobar" },
|
|
{ L"123", L"654", L"123456", L"654456" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceFirstOnly)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = 0;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L"B", L"BB", L"ABA", L"ABBA" },
|
|
{ L"B", L"A", L"ABBBA", L"AABBA" },
|
|
{ L"B", L"BBB", L"ABABAB", L"ABBBABAB" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceAll)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L"B", L"BB", L"ABA", L"ABBA" },
|
|
{ L"B", L"A", L"ABBBA", L"AAAAA" },
|
|
{ L"B", L"BBB", L"ABABAB", L"ABBBABBBABBB" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceAllCaseInsensitive)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | CaseSensitive;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L"B", L"BB", L"ABA", L"ABBA" },
|
|
{ L"B", L"A", L"ABBBA", L"AAAAA" },
|
|
{ L"B", L"BBB", L"ABABAB", L"ABBBABBBABBB" },
|
|
{ L"b", L"BBB", L"AbABAb", L"ABBBABABBB" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceFirstOnlyUseRegEx)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = UseRegularExpressions;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L"B", L"BB", L"ABA", L"ABBA" },
|
|
{ L"B", L"A", L"ABBBA", L"AABBA" },
|
|
{ L"B", L"BBB", L"ABABAB", L"ABBBABAB" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceAllUseRegEx)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L"B", L"BB", L"ABA", L"ABBA" },
|
|
{ L"B", L"A", L"ABBBA", L"AAAAA" },
|
|
{ L"B", L"BBB", L"ABABAB", L"ABBBABBBABBB" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceAllUseRegExCaseSensitive)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions | CaseSensitive;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L"B", L"BB", L"ABA", L"ABBA" },
|
|
{ L"B", L"A", L"ABBBA", L"AAAAA" },
|
|
{ L"b", L"BBB", L"AbABAb", L"ABBBABABBB" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyMatchAllWildcardUseRegEx)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L".*", L"Foo", L"AAAAAA", L"Foo" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
void VerifyReplaceFirstWildcard(SearchReplaceExpected sreTable[], int tableSize, DWORD flags)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
for (int i = 0; i < tableSize; i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::AreEqual(sreTable[i].expected, result);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceFirstWildCardUseRegex)
|
|
{
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L".*", L"Foo", L"AAAAAA", L"Foo" },
|
|
};
|
|
VerifyReplaceFirstWildcard(sreTable, ARRAYSIZE(sreTable), UseRegularExpressions);
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceFirstWildCardUseRegexMatchAllOccurrences)
|
|
{
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L".*", L"Foo", L"AAAAAA", L"Foo" },
|
|
};
|
|
VerifyReplaceFirstWildcard(sreTable, ARRAYSIZE(sreTable), UseRegularExpressions | MatchAllOccurences);
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceFirstWildCardMatchAllOccurrences)
|
|
{
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L".*", L"Foo", L"AAAAAA", L"AAAAAA" },
|
|
{ L".*", L"Foo", L".*", L"Foo" },
|
|
{ L".*", L"Foo", L".*Bar.*", L"FooBarFoo" },
|
|
};
|
|
VerifyReplaceFirstWildcard(sreTable, ARRAYSIZE(sreTable), MatchAllOccurences);
|
|
}
|
|
|
|
TEST_METHOD(VerifyReplaceFirstWildNoFlags)
|
|
{
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L".*", L"Foo", L"AAAAAA", L"AAAAAA" },
|
|
{ L".*", L"Foo", L".*", L"Foo" },
|
|
};
|
|
VerifyReplaceFirstWildcard(sreTable, ARRAYSIZE(sreTable), 0);
|
|
}
|
|
|
|
TEST_METHOD(VerifyHandleCapturingGroups)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions | CaseSensitive;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L"(foo)(bar)", L"$1_$002_$223_$001021_$00001", L"foobar", L"foo_$002_bar23_$001021_$00001" },
|
|
{ L"(foo)(bar)", L"_$1$2_$123$040", L"foobar", L"_foobar_foo23$040" },
|
|
{ L"(foo)(bar)", L"$$$1", L"foobar", L"$foo" },
|
|
{ L"(foo)(bar)", L"$$1", L"foobar", L"$1" },
|
|
{ L"(foo)(bar)", L"$12", L"foobar", L"foo2" },
|
|
{ L"(foo)(bar)", L"$10", L"foobar", L"foo0" },
|
|
{ L"(foo)(bar)", L"$01", L"foobar", L"$01" },
|
|
{ L"(foo)(bar)", L"$$$11", L"foobar", L"$foo1" },
|
|
{ L"(foo)(bar)", L"$$$$113a", L"foobar", L"$$113a" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD (VerifyFileAttributesNoPadding)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions ;
|
|
SYSTEMTIME fileTime = SYSTEMTIME{ 2020, 7, 3, 22, 15, 6, 42, 453 };
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L"foo", L"bar$YY-$M-$D-$h-$m-$s-$f", L"foo", L"bar20-7-22-15-6-42-4" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutFileTime(fileTime) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD (VerifyFileAttributesPadding)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
SYSTEMTIME fileTime = SYSTEMTIME{ 2020, 7, 3, 22, 15, 6, 42, 453 };
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L"foo", L"bar$YYYY-$MM-$DD-$hh-$mm-$ss-$fff", L"foo", L"bar2020-07-22-15-06-42-453" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutFileTime(fileTime) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD (VerifyFileAttributesMonthandDayNames)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
|
|
std::locale::global(std::locale(""));
|
|
SYSTEMTIME fileTime = { 2020, 1, 3, 1, 15, 6, 42, 453 };
|
|
wchar_t localeName[LOCALE_NAME_MAX_LENGTH];
|
|
wchar_t result[MAX_PATH] = L"bar";
|
|
wchar_t formattedDate[MAX_PATH];
|
|
if (GetUserDefaultLocaleName(localeName, LOCALE_NAME_MAX_LENGTH) == 0)
|
|
StringCchCopy(localeName, LOCALE_NAME_MAX_LENGTH, L"en_US");
|
|
|
|
GetDateFormatEx(localeName, NULL, &fileTime, L"MMM", formattedDate, MAX_PATH, NULL);
|
|
formattedDate[0] = towupper(formattedDate[0]);
|
|
StringCchPrintf(result, MAX_PATH, TEXT("%s%s"), result, formattedDate);
|
|
|
|
GetDateFormatEx(localeName, NULL, &fileTime, L"MMMM", formattedDate, MAX_PATH, NULL);
|
|
formattedDate[0] = towupper(formattedDate[0]);
|
|
StringCchPrintf(result, MAX_PATH, TEXT("%s-%s"), result, formattedDate);
|
|
|
|
GetDateFormatEx(localeName, NULL, &fileTime, L"ddd", formattedDate, MAX_PATH, NULL);
|
|
formattedDate[0] = towupper(formattedDate[0]);
|
|
StringCchPrintf(result, MAX_PATH, TEXT("%s-%s"), result, formattedDate);
|
|
|
|
GetDateFormatEx(localeName, NULL, &fileTime, L"dddd", formattedDate, MAX_PATH, NULL);
|
|
formattedDate[0] = towupper(formattedDate[0]);
|
|
StringCchPrintf(result, MAX_PATH, TEXT("%s-%s"), result, formattedDate);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L"foo", L"bar$MMM-$MMMM-$DDD-$DDDD", L"foo", result },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutFileTime(fileTime) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == S_OK);
|
|
Assert::IsTrue(wcscmp(result, sreTable[i].expected) == 0);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyLookbehindFails)
|
|
{
|
|
// Standard Library Regex Engine does not support lookbehind, thus test should fail.
|
|
SearchReplaceExpected sreTable[] = {
|
|
//search, replace, test, result
|
|
{ L"(?<=E12).*", L"Foo", L"AAAAAA", nullptr },
|
|
{ L"(?<!E12).*", L"Foo", L"AAAAAA", nullptr },
|
|
};
|
|
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutFlags(UseRegularExpressions) == S_OK);
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(sreTable[i].replace) == S_OK);
|
|
Assert::IsTrue(renameRegEx->Replace(sreTable[i].test, &result) == E_FAIL);
|
|
Assert::AreEqual(sreTable[i].expected, result);
|
|
CoTaskMemFree(result);
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(VerifyEventsFire)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
CMockPowerRenameRegExEvents* mockEvents = new CMockPowerRenameRegExEvents();
|
|
CComPtr<IPowerRenameRegExEvents> regExEvents;
|
|
Assert::IsTrue(mockEvents->QueryInterface(IID_PPV_ARGS(®ExEvents)) == S_OK);
|
|
DWORD cookie = 0;
|
|
Assert::IsTrue(renameRegEx->Advise(regExEvents, &cookie) == S_OK);
|
|
DWORD flags = MatchAllOccurences | UseRegularExpressions | CaseSensitive;
|
|
Assert::IsTrue(renameRegEx->PutFlags(flags) == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutSearchTerm(L"FOO") == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutReplaceTerm(L"BAR") == S_OK);
|
|
Assert::IsTrue(renameRegEx->PutFileTime(SYSTEMTIME{ 0 }) == S_OK);
|
|
Assert::IsTrue(renameRegEx->ResetFileTime() == S_OK);
|
|
Assert::IsTrue(lstrcmpi(L"FOO", mockEvents->m_searchTerm) == 0);
|
|
Assert::IsTrue(lstrcmpi(L"BAR", mockEvents->m_replaceTerm) == 0);
|
|
Assert::IsTrue(flags == mockEvents->m_flags);
|
|
Assert::IsTrue(renameRegEx->UnAdvise(cookie) == S_OK);
|
|
mockEvents->Release();
|
|
}
|
|
}
|
|
;
|
|
}
|