PowerToys/tools/CliShim.UnitTests/CommandLineTests.cpp

// Copyright (c) Microsoft Corporation
// The Microsoft Corporation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#include <string>

#include <CppUnitTest.h>

#include "CommandLine.h"

using namespace Microsoft::VisualStudio::CppUnitTestFramework;

namespace CliShimUnitTests
{
    TEST_CLASS(CommandLineTests)
    {
    public:
        TEST_METHOD(StripArgumentZero_ReturnsForwardedTail)
        {
            struct Case
            {
                const wchar_t* commandLine;
                const wchar_t* expected;
            };

            const Case cases[] = {
                { L"PowerToys.FancyZones.CLI arg", L"arg" },
                { L"PowerToys.FancyZones.CLI a b c", L"a b c" },
                { L"PowerToys.FancyZones.CLI", L"" },
                { L"PowerToys.FileLocksmith.CLI", L"" },

                { LR"("C:\Program Files\PowerToys\bin\PowerToys.FancyZones.CLI.exe" arg)", L"arg" },
                { LR"("C:\Program Files\PowerToys\bin\PowerToys.FancyZones.CLI.exe")", L"" },

                // Quotes only toggle, so a quoted argv[0] does not end at the closing quote: an
                // argument glued to it is part of the program name. CommandLineToArgvW would
                // forward `--help` here; the CRT - and so every target CLI - sees no arguments.
                { LR"("C:\bin\PowerToys.FancyZones.CLI.exe"--help)", L"" },

                // Partially quoted program names - the batch idiom of quoting only the variable,
                // `"%ProgramFiles%"\PowerToys\bin\...`, which cmd.exe passes through verbatim - are
                // resolved exactly as the CRT resolves them, so the target sees the same tail it
                // would have seen had the caller invoked it directly rather than through the shim.
                { LR"("C:\Program Files"\PowerToys\bin\PowerToys.FancyZones.CLI.exe arg)", L"arg" },
                { LR"(C:\Program" Files"\PowerToys\bin\PowerToys.FancyZones.CLI.exe arg)", L"arg" },
                { LR"("C:\Program Files"\PowerToys\bin\PowerToys.FancyZones.CLI.exe)", L"" },

                { LR"("C:\bin\PowerToys.FancyZones.CLI.exe" "a b")", LR"("a b")" },
                { LR"(PowerToys.FancyZones.CLI --path "C:\a b\c.png")", LR"(--path "C:\a b\c.png")" },

                { L"PowerToys.FancyZones.CLI\targ", L"arg" },
                { L"PowerToys.FancyZones.CLI \t arg", L"arg" },

                // Non-shell CreateProcessW callers can prepend whitespace; argv[0] must not leak.
                { L"  PowerToys.FancyZones.CLI arg", L"arg" },
                { L" PowerToys.FancyZones.CLI", L"" },
                { LR"(  "C:\bin\PowerToys.FancyZones.CLI.exe" arg)", L"arg" },

                { L"", L"" },

                // An unterminated argv[0] quote consumes the remaining command line.
                { LR"("C:\Program Files\app)", L"" },
            };

            for (const Case& testCase : cases)
            {
                const std::wstring actual = CommandLine::StripArgumentZero(testCase.commandLine);
                const std::wstring message = std::wstring(L"input: <") + testCase.commandLine + L">";
                Assert::AreEqual(std::wstring(testCase.expected), actual, message.c_str());
            }
        }
    };
}