tests/DotNetCommon.Test/Extensions/TaskExtensions/TaskTimeoutTests.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

namespace DotNetCommon.Test.Extensions.TaskExtensions
{
    using System;
    using System.Threading;
    using System.Threading.Tasks;
    using DotNetCommon.Data;
    using DotNetCommon.Extensions;
    using NUnit.Framework;
    using Shouldly;

    [TestFixture]
    internal sealed class TaskTimeoutTests
    {
        public class Person { public int Id { get; set; } }
        public class PersonDto { public int Id { get; set; } }
        [Test]
        public async Task TestMapperAsync()
        {
            var task = Task.FromResult<Person>(new Person { Id = 1 });
            var task2 = await task.MapperAsync<Person, PersonDto>();
            task2.ShouldNotBeNull();
            task2.Id.ShouldBe(1);
        }

        [Test]
        public async Task TestUnWrapAsync()
        {
            var task = Task.FromResult(Result.Ok(1));
            var num = await task.UnWrapAsync();
            num.ShouldBe(1);
        }

        [Test]
        public void When_executing_task_with_result_that_cancels_before_timeout()
        {
            using (var cts = new CancellationTokenSource(1.Seconds()))
            {
                var task = GetSomethingAfter(5.Seconds(), cts.Token);

                Should.Throw<TaskCanceledException>(async () => await task.TimeoutAfter(3.Seconds()));
            }
        }

        [Test]
        public async Task When_executing_task_with_result_that_does_not_timeout()
        {
            var task = GetSomethingAfter(2.Seconds());

            var result = await task.TimeoutAfter(3.Seconds());
            result.ShouldBe(42);
        }

        [Test]
        public void When_executing_task_with_result_that_does_timeout()
        {
            var task = GetSomethingAfter(2.Seconds());

            Should.Throw<TimeoutException>(async () => await task.TimeoutAfter(1.Seconds()))
                .Message.ShouldBe("Task timed out after: 00:00:01");
        }