mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-26 18:58:29 +08:00
321 lines
12 KiB
C++
321 lines
12 KiB
C++
#include "stdafx.h"
|
|
#include "CppUnitTest.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_METHOD(GeneralReplaceTest){
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->put_searchTerm(L"foo") == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_searchTerm(L"notfound") == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_searchTerm(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->put_searchTerm(L"foo") == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->get_flags(&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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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->put_flags(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->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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(VerifyReplaceFirstWildcardUseRegEx)
|
|
{
|
|
CComPtr<IPowerRenameRegEx> renameRegEx;
|
|
Assert::IsTrue(CPowerRenameRegEx::s_CreateInstance(&renameRegEx) == S_OK);
|
|
DWORD flags = UseRegularExpressions;
|
|
Assert::IsTrue(renameRegEx->put_flags(flags) == S_OK);
|
|
|
|
SearchReplaceExpected sreTable[] = {
|
|
{ L".*", L"Foo", L"AAAAAA", L"FooAAAA" },
|
|
};
|
|
|
|
for (int i = 0; i < ARRAYSIZE(sreTable); i++)
|
|
{
|
|
PWSTR result = nullptr;
|
|
Assert::IsTrue(renameRegEx->put_searchTerm(sreTable[i].search) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(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(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->put_flags(flags) == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_searchTerm(L"FOO") == S_OK);
|
|
Assert::IsTrue(renameRegEx->put_replaceTerm(L"BAR") == 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();
|
|
}
|
|
}
|
|
;
|
|
} |