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

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

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

namespace DotNetCommon.Test.Extensions.TaskExtensions
{
    using System;
    using System.Collections.Generic;
    using System.IO;
    using System.Linq;
    using System.Threading.Tasks;
    using NUnit.Framework;
    using Shouldly;
    using DotNetCommon.Extensions;

    [TestFixture]
    public class TaskExceptionsTests
    {
        [Test]
        public void ShouldHandleExpectedExceptionWhenNoException()
        {
            var exceptionsQueue = new Queue<Exception>();

            Action action = () =>
            {
                var t = Task.Factory.StartNew(() =>
                {
                    try { throw new InvalidOperationException(); } catch (Exception) { /* Ignore */ }

                }).HandleException<InvalidOperationException>(e => { exceptionsQueue.Enqueue(e); });

                t.Wait();
            };

            action.ShouldNotThrow();

            exceptionsQueue.ShouldBeEmpty();
        }

        [Test]
        public void ShouldHandleExpectedExceptionNonPresentException()
        {
            var exceptionsQueue = new Queue<Exception>();

            Action action = () =>
            {
                var t = Task.Factory.StartNew(() => { throw new DivideByZeroException(); })
                    .HandleException<InvalidOperationException>(
                        e => { exceptionsQueue.Enqueue(e); });

                t.Wait();
            };

            action.ShouldThrow<AggregateException>()
                .Flatten().InnerExceptions
                .Count(e => e.GetType() == typeof(DivideByZeroException) && e.Message == "Attempted to divide by zero.")
                .ShouldBe(1);

            exceptionsQueue.ShouldBeEmpty();
        }

        [Test]
        public void ShouldHandleExpectedExceptionSingleException()
        {