PowerToys/src/modules/launcher/Plugins/Community.PowerToys.Run.Plugin.ValueGenerator.UnitTests/InputParserTests.cs

// 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.

using System;
using System.Linq;
using System.Text.RegularExpressions;

using Microsoft.VisualStudio.TestTools.UnitTesting;
using Wox.Plugin;

namespace Community.PowerToys.Run.Plugin.ValueGenerator.UnitTests
{
    [TestClass]
    public partial class InputParserTests
    {
        [DataTestMethod]
        [DataRow("md5 abc", typeof(Hashing.HashRequest))]
        [DataRow("sha1 abc", typeof(Hashing.HashRequest))]
        [DataRow("sha256 abc", typeof(Hashing.HashRequest))]
        [DataRow("sha384 abc", typeof(Hashing.HashRequest))]
        [DataRow("sha512 abc", typeof(Hashing.HashRequest))]
        [DataRow("sha1111 abc", null)]
        [DataRow("uuid", typeof(GUID.GUIDRequest))]
        [DataRow("guidv3 ns:DNS abc", typeof(GUID.GUIDRequest))]
        [DataRow("guidv2 ns:DNS abc", null)]
        [DataRow("uUiD5 ns:URL abc", typeof(GUID.GUIDRequest))]
        [DataRow("Guidvv ns:DNS abc", null)]
        [DataRow("guidv4", typeof(GUID.GUIDRequest))]
        [DataRow("guidv7", typeof(GUID.GUIDRequest))]
        [DataRow("GUIDv7", typeof(GUID.GUIDRequest))]
        [DataRow("base64 abc", typeof(Base64.Base64Request))]
        [DataRow("base99 abc", null)]
        [DataRow("base64s abc", null)]
        [DataRow("base64d abc=", typeof(Base64.Base64DecodeRequest))]
        [DataRow("url http://googl.de/u oii d", typeof(Uri.UrlEncodeRequest))]
        [DataRow("urld http://googl.de/u oii d=oo", typeof(Uri.UrlDecodeRequest))]
        [DataRow("esc:data jjdje332j 3 3l2jl32", typeof(Uri.DataEscapeRequest))]
        [DataRow("esc:hex d", typeof(Uri.HexEscapeRequest))]
        [DataRow("esc:hex 4", typeof(Uri.HexEscapeRequest))]
        [DataRow("esc:hex ", typeof(Uri.HexEscapeRequest), true)]
        [DataRow("esc:hex z44", typeof(Uri.HexEscapeRequest), true)]
        [DataRow("uesc:data jjdje332j 3 3l2jl32", typeof(Uri.DataUnescapeRequest))]
        [DataRow("uesc:hex %21", typeof(Uri.HexUnescapeRequest))]
        [DataRow("uesc:hex 4", typeof(Uri.HexUnescapeRequest))]
        [DataRow("uesc:hex ", typeof(Uri.HexUnescapeRequest))]
        [DataRow("uesc:hex z44", typeof(Uri.HexUnescapeRequest))]
        public void ParserTest(string input, Type? expectedRequestType, bool expectException = false)
        {
            var parser = new InputParser();
            var query = new Query(input);

            string? command = null;
            if (query.Terms.Count == 0)
            {
                expectException = true;
            }
            else
            {
                command = query.Terms[0];
            }

            if (command != null && !CommandIsKnown(command))
            {
                expectException = true;
            }

            try
            {
                IComputeRequest request = parser.ParseInput(query);
                if (expectException)
                {
                    Assert.Fail("Parser should have thrown an exception");
                }

                Assert.IsNotNull(request);

                Assert.AreEqual(expectedRequestType, request.GetType());
            }
            catch (AssertFailedException)
            {
                throw;
            }
            catch
            {
                if (!expectException)
                {
                    throw;
                }
            }
        }

        private static bool CommandIsKnown(string command)
        {
            string[] hashes = ["md5", "sha1", "sha256", "sha384", "sha512", "base64", "base64d"];
            if (hashes.Contains(command.ToLowerInvariant()))
            {
                return true;
            }

            if (GetKnownUuidImplementations().IsMatch(command))
            {
                return true;
            }

            string[] uriCommands = ["url", "urld", "esc:hex", "uesc:hex", "esc:data", "uesc:data"];
            return uriCommands.Contains(command.ToLowerInvariant());
        }

        [GeneratedRegex("^(guid|uuid)([13457]{0,1}|v[13457]{1})$", RegexOptions.IgnoreCase | RegexOptions.Compiled, "en-US")]
        private static partial Regex GetKnownUuidImplementations();
    }
}