Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Core/DelegateExtensionsTests.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using System;
using Lucene.Net.Index;
using NUnit.Framework;
using Umbraco.Cms.Core;
using Umbraco.Extensions;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Core;

[TestFixture]
public class DelegateExtensionsTests
{
    [Test]
    public void Only_Executes_Specific_Count()
    {
        const int maxTries = 5;
        var totalTries = 0;
        DelegateExtensions.RetryUntilSuccessOrMaxAttempts(
            currentTry =>
            {
                totalTries = currentTry;
                return Attempt<IndexWriter>.Fail();
            },
            5,
            TimeSpan.FromMilliseconds(10));

        Assert.AreEqual(maxTries, totalTries);
    }

    [Test]
    public void Quits_On_Success_Count()
    {
        var totalTries = 0;
        DelegateExtensions.RetryUntilSuccessOrMaxAttempts(
            currentTry =>
            {
                totalTries = currentTry;
                return totalTries == 2 ? Attempt<string>.Succeed() : Attempt<string>.Fail();
            },
            5,
            TimeSpan.FromMilliseconds(10));

        Assert.AreEqual(2, totalTries);
    }
}