using System.Threading.Tasks;
using Shouldly;
using Xunit;

namespace Volo.Abp.DistributedLocking;

public class LocalDistributedLock_Tests : AbpDistributedLockingAbstractionsTestBase
{
    private readonly IAbpDistributedLock _distributedLock;

    public LocalDistributedLock_Tests()
    {
        _distributedLock = GetRequiredService<IAbpDistributedLock>();
    }

    [Fact]
    public void Should_Be_Instance_Of_LocalAbpDistributedLock()
    {
        _distributedLock.ShouldBeOfType<LocalAbpDistributedLock>();
    }

    [Fact]
    public async Task Should_Lock_With_TryAcquire()
    {
        await using (var handle = await _distributedLock.TryAcquireAsync("lock1"))
        {
            handle.ShouldNotBeNull();
        }
    }

    [Fact]
    public async Task Should_Not_Acquire_If_Already_Locked()
    {
        await using (var handle = await _distributedLock.TryAcquireAsync("lock1"))
        {
            handle.ShouldNotBeNull();

            await Task.Run(async () =>
            {
                await using (var handle2 = await _distributedLock.TryAcquireAsync("lock1"))
                {
                    handle2.ShouldBeNull();
                }
            });
        }

        await Task.Run(async () =>
        {
            await using (var handle = await _distributedLock.TryAcquireAsync("lock1"))
            {
                handle.ShouldNotBeNull();
            }
        });
    }

    [Fact]
    public async Task Should_Obtain_Multiple_Locks()
    {
        await using (var handle = await _distributedLock.TryAcquireAsync("lock1"))
        {
            handle.ShouldNotBeNull();

            await Task.Run(async () =>
            {
                await using (var handle2 = await _distributedLock.TryAcquireAsync("lock2"))
                {
                    handle2.ShouldNotBeNull();
                }
            });
        }
    }
}