SPT-Server-Build/UnitTests/Tests/Utils/HashUtilTests.cs

using System.Collections.Concurrent;
using System.Diagnostics;
using SPTarkov.Server.Core.Utils;

namespace UnitTests.Tests.Utils;

[TestClass]
public class HashUtilTests
{
    private HashUtil _hashUtil;

    [TestInitialize]
    public void Initialize()
    {
        _hashUtil = DI.GetService<HashUtil>();
    }

    [TestMethod]
    public void GenerateTest()
    {
        // Generate 100 MongoId's
        for (var i = 0; i < 100; i++)
        {
            // Invalid mongoId character
            var result = _hashUtil.Generate();

            // Invalid mongoId length
            var test = _hashUtil.IsValidMongoId(result);

            Assert.AreEqual(true, test, $"IsValidMongoId() `{result}` is not a valid MongoId.");
        }
    }

    [TestMethod]
    [DataRow(
        "677ddb67406e9918a0264bbz",
        false,
        "677ddb67406e9918a0264bbz contains invalid char `z`, but result was true"
    )]
    [DataRow(
        "677ddb67406e9918a0264bbcc",
        false,
        "677ddb67406e9918a0264bbcc is 25 characters, but result was true"
    )]
    [DataRow(
        "677ddb67406e9918a0264bbc",
        true,
        "IsValidMongoId() `677ddb67406e9918a0264bbc` is a valid mongoId, but result was false"
    )]
    public void IsValidMongoIdTest(string mongoId, bool passes, string failMessage)
    {
        var result = _hashUtil.IsValidMongoId(mongoId);
        Assert.AreEqual(passes, result, failMessage);
    }

    [TestMethod]
    [DataRow(
        "123456789",
        "25F9E794323B453885F5181F1B624D0B",
        "Not valid output, expected '25F9E794323B453885F5181F1B624D0B'"
    )]
    public void GenerateValidMd5Test(string input, string expectedOutput, string failMessage)
    {
        var result = _hashUtil.GenerateHashForData(HashingAlgorithm.MD5, input);
        Assert.AreEqual(expectedOutput, result, failMessage);
    }

    [TestMethod]
    public void MultiThreadedMongoIDGenerationTest()
    {
        var concurrentBag = new ConcurrentBag<string>();
        var random = new Random();
        var stopwatch = new Stopwatch();
        stopwatch.Start();

        Parallel.For(
            0,
            1000,
            i =>
            {
                Thread.Sleep(random.Next(0, 10));
                var mongoId = _hashUtil.Generate();
                concurrentBag.Add(mongoId);
            }
        );

        stopwatch.Stop();
        Console.WriteLine($"Elapsed time: {stopwatch.ElapsedMilliseconds} ms");
        var uniqueCount = concurrentBag.Distinct().Count();
        var totalCount = concurrentBag.Count;
        Assert.AreEqual(
            totalCount,
            uniqueCount,
            $"Expected all generated MongoId's to be unique, but found {totalCount - uniqueCount} duplicates."
        );
    }
}