exception testing

2026-05-16 03:50:11 +00:00 · 2020-05-10 22:44:40 +09:00
parent af6dbd8868
commit 93dd82e3d4
17 changed files with 926 additions and 3963 deletions
--- a/src/UniTask.NetCoreTests/Linq/Concat.cs
+++ b/src/UniTask.NetCoreTests/Linq/Concat.cs
@@ -27,6 +27,19 @@ namespace NetCoreTests.Linq
            xs.Should().BeEquivalentTo(ys);
        }

+        [Fact]
+        public async Task AppendThrow()
+        {
+            var xs = UniTaskTestException.ThrowImmediate().Append(99).ToArrayAsync();
+            await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+
+            var ys = UniTaskTestException.ThrowAfter().Append(99).ToArrayAsync();
+            await Assert.ThrowsAsync<UniTaskTestException>(async () => await ys);
+
+            var zs = UniTaskTestException.ThrowInMoveNext().Append(99).ToArrayAsync();
+            await Assert.ThrowsAsync<UniTaskTestException>(async () => await zs);
+        }
+
        [Theory]
        [InlineData(0, 0)]
        [InlineData(0, 1)]
@@ -40,6 +53,19 @@ namespace NetCoreTests.Linq
            xs.Should().BeEquivalentTo(ys);
        }

+        [Fact]
+        public async Task PrependThrow()
+        {
+            var xs = UniTaskTestException.ThrowImmediate().Prepend(99).ToArrayAsync();
+            await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+
+            var ys = UniTaskTestException.ThrowAfter().Prepend(99).ToArrayAsync();
+            await Assert.ThrowsAsync<UniTaskTestException>(async () => await ys);
+
+            var zs = UniTaskTestException.ThrowInMoveNext().Prepend(99).ToArrayAsync();
+            await Assert.ThrowsAsync<UniTaskTestException>(async () => await zs);
+        }
+
        public static IEnumerable<object[]> array1 = new object[][]
        {
            new object[] { (0, 0), (0, 0) }, // empty + empty
@@ -61,5 +87,30 @@ namespace NetCoreTests.Linq
            var ys = l.Concat(r).ToArray();
            xs.Should().BeEquivalentTo(ys);
        }
+
+        [Fact]
+        public async Task ConcatThrow()
+        {
+            {
+                var xs = UniTaskTestException.ThrowImmediate().Concat(UniTaskAsyncEnumerable.Range(1, 10)).ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+
+                var ys = UniTaskTestException.ThrowAfter().Concat(UniTaskAsyncEnumerable.Range(1, 10)).ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await ys);
+
+                var zs = UniTaskTestException.ThrowInMoveNext().Concat(UniTaskAsyncEnumerable.Range(1, 10)).ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await zs);
+            }
+            {
+                var xs = UniTaskAsyncEnumerable.Range(1, 10).Concat(UniTaskTestException.ThrowImmediate()).ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+
+                var ys = UniTaskAsyncEnumerable.Range(1, 10).Concat(UniTaskTestException.ThrowAfter()).ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await ys);
+
+                var zs = UniTaskAsyncEnumerable.Range(1, 10).Concat(UniTaskTestException.ThrowInMoveNext()).ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await zs);
+            }
+        }
    }
 }
--- a/src/UniTask.NetCoreTests/Linq/Filtering.cs
+++ b/src/UniTask.NetCoreTests/Linq/Filtering.cs
@@ -0,0 +1,118 @@
+using Cysharp.Threading.Tasks;
+using Cysharp.Threading.Tasks.Linq;
+using FluentAssertions;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reactive.Concurrency;
+using System.Reactive.Linq;
+using System.Threading.Tasks;
+using Xunit;
+
+
+namespace NetCoreTests.Linq
+{
+    public class Filtering
+    {
+        [Fact]
+        public async Task Where()
+        {
+            var range = Enumerable.Range(1, 10);
+            var src = range.ToUniTaskAsyncEnumerable();
+
+            {
+                var a = await src.Where(x => x % 2 == 0).ToArrayAsync();
+                var expected = range.Where(x => x % 2 == 0).ToArray();
+                a.Should().BeEquivalentTo(expected);
+            }
+            {
+                var a = await src.Where((x, i) => (x + i) % 2 == 0).ToArrayAsync();
+                var expected = range.Where((x, i) => (x + i) % 2 == 0).ToArray();
+                a.Should().BeEquivalentTo(expected);
+            }
+            {
+                var a = await src.WhereAwait(x => UniTask.Run(() => x % 2 == 0)).ToArrayAsync();
+                var b = await src.WhereAwait(x => UniTask.FromResult(x % 2 == 0)).ToArrayAsync();
+                var expected = range.Where(x => x % 2 == 0).ToArray();
+                a.Should().BeEquivalentTo(expected);
+                b.Should().BeEquivalentTo(expected);
+            }
+            {
+                var a = await src.WhereAwait((x, i) => UniTask.Run(() => (x + i) % 2 == 0)).ToArrayAsync();
+                var b = await src.WhereAwait((x, i) => UniTask.FromResult((x + i) % 2 == 0)).ToArrayAsync();
+                var expected = range.Where((x, i) => (x + i) % 2 == 0).ToArray();
+                a.Should().BeEquivalentTo(expected);
+                b.Should().BeEquivalentTo(expected);
+            }
+        }
+
+
+        [Fact]
+        public async Task WhereException()
+        {
+            foreach (var item in UniTaskTestException.Throws())
+            {
+                {
+                    var xs = item.Where(x => x % 2 == 0).ToArrayAsync();
+                    await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+                }
+                {
+                    var xs = item.Where((x, i) => x % 2 == 0).ToArrayAsync();
+                    await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+                }
+                {
+                    var xs = item.WhereAwait(x => UniTask.FromResult(x % 2 == 0)).ToArrayAsync();
+                    await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+                }
+                {
+                    var xs = item.WhereAwait((x, i) => UniTask.FromResult(x % 2 == 0)).ToArrayAsync();
+                    await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+                }
+            }
+        }
+
+        [Fact]
+        public async Task OfType()
+        {
+            var data = new object[] { 0, null, 10, 30, null, "foo", 99 };
+
+            var a = await data.ToUniTaskAsyncEnumerable().OfType<int>().ToArrayAsync();
+            var b = data.OfType<int>().ToArray();
+
+            a.Should().BeEquivalentTo(b);
+        }
+
+
+        [Fact]
+        public async Task OfTypeException()
+        {
+            foreach (var item in UniTaskTestException.Throws())
+            {
+                var xs = item.Select(x => (object)x).OfType<int>().ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+            }
+        }
+
+        [Fact]
+        public async Task Cast()
+        {
+            var data = new object[] { 0, 10, 30, 99 };
+
+            var a = await data.ToUniTaskAsyncEnumerable().Cast<int>().ToArrayAsync();
+            var b = data.Cast<int>().ToArray();
+
+            a.Should().BeEquivalentTo(b);
+        }
+
+
+        [Fact]
+        public async Task CastException()
+        {
+            foreach (var item in UniTaskTestException.Throws())
+            {
+                var xs = item.Select(x => (object)x).Cast<int>().ToArrayAsync();
+                await Assert.ThrowsAsync<UniTaskTestException>(async () => await xs);
+            }
+        }
+    }
+}
--- a/src/UniTask.NetCoreTests/Linq/_Exception.cs
+++ b/src/UniTask.NetCoreTests/Linq/_Exception.cs
@@ -0,0 +1,126 @@
+using Cysharp.Threading.Tasks;
+using Cysharp.Threading.Tasks.Linq;
+using System;
+using System.Collections.Generic;
+using System.Runtime.ExceptionServices;
+using System.Threading;
+
+namespace NetCoreTests.Linq
+{
+    public class UniTaskTestException : Exception
+    {
+        public static IUniTaskAsyncEnumerable<int> ThrowImmediate()
+        {
+            return UniTaskAsyncEnumerable.Throw<int>(new UniTaskTestException());
+        }
+        public static IUniTaskAsyncEnumerable<int> ThrowAfter()
+        {
+            return new ThrowAfter<int>(new UniTaskTestException());
+        }
+        public static IUniTaskAsyncEnumerable<int> ThrowInMoveNext()
+        {
+            return new ThrowIn<int>(new UniTaskTestException());
+        }
+
+
+        public static IEnumerable<IUniTaskAsyncEnumerable<int>> Throws()
+        {
+            yield return ThrowImmediate();
+            yield return ThrowAfter();
+            yield return ThrowInMoveNext();
+            yield return UniTaskAsyncEnumerable.Range(1, 3).Concat(ThrowImmediate());
+            yield return UniTaskAsyncEnumerable.Range(1, 3).Concat(ThrowAfter());
+            yield return UniTaskAsyncEnumerable.Range(1, 3).Concat(ThrowInMoveNext());
+        }
+    }
+
+    internal class ThrowIn<TValue> : IUniTaskAsyncEnumerable<TValue>
+    {
+        readonly Exception exception;
+
+        public ThrowIn(Exception exception)
+        {
+            this.exception = exception;
+        }
+
+        public IUniTaskAsyncEnumerator<TValue> GetAsyncEnumerator(CancellationToken cancellationToken = default)
+        {
+            return new Enumerator(exception, cancellationToken);
+        }
+
+        class Enumerator : IUniTaskAsyncEnumerator<TValue>
+        {
+            readonly Exception exception;
+            CancellationToken cancellationToken;
+
+            public Enumerator(Exception exception, CancellationToken cancellationToken)
+            {
+                this.exception = exception;
+                this.cancellationToken = cancellationToken;
+            }
+
+            public TValue Current => default;
+
+            public UniTask<bool> MoveNextAsync()
+            {
+                ExceptionDispatchInfo.Capture(exception).Throw();
+                return new UniTask<bool>(false);
+            }
+
+            public UniTask DisposeAsync()
+            {
+                return default;
+            }
+        }
+    }
+
+    internal class ThrowAfter<TValue> : IUniTaskAsyncEnumerable<TValue>
+    {
+        readonly Exception exception;
+
+        public ThrowAfter(Exception exception)
+        {
+            this.exception = exception;
+        }
+
+        public IUniTaskAsyncEnumerator<TValue> GetAsyncEnumerator(CancellationToken cancellationToken = default)
+        {
+            return new Enumerator(exception, cancellationToken);
+        }
+
+        class Enumerator : IUniTaskAsyncEnumerator<TValue>
+        {
+            readonly Exception exception;
+            CancellationToken cancellationToken;
+
+            public Enumerator(Exception exception, CancellationToken cancellationToken)
+            {
+                this.exception = exception;
+                this.cancellationToken = cancellationToken;
+            }
+
+            public TValue Current => default;
+
+            public UniTask<bool> MoveNextAsync()
+            {
+                cancellationToken.ThrowIfCancellationRequested();
+
+                var tcs = new UniTaskCompletionSource<bool>();
+
+                var awaiter = UniTask.Yield().GetAwaiter();
+                awaiter.UnsafeOnCompleted(() =>
+                {
+                    Thread.Sleep(1);
+                    tcs.TrySetException(exception);
+                });
+
+                return tcs.Task;
+            }
+
+            public UniTask DisposeAsync()
+            {
+                return default;
+            }
+        }
+    }
+}