Support faulted awaitables in IAwaitableFactory

devlooped · Jan 1, 2021 · 5802db8 · 5802db8
1 parent 047bf95
commit 5802db8
Show file tree

Hide file tree

Showing 7 changed files with 112 additions and 0 deletions.
diff --git a/src/Moq/Async/AwaitableFactory`1.cs b/src/Moq/Async/AwaitableFactory`1.cs
@@ -2,7 +2,9 @@
 // All rights reserved. Licensed under the BSD 3-Clause License; see License.txt.
 
 using System;
+using System.Collections.Generic;
 using System.Diagnostics;
+using System.Linq;
 
 namespace Moq.Async
 {
@@ -23,6 +25,25 @@ object IAwaitableFactory.CreateCompleted(object result)
 			return this.CreateCompleted();
 		}
 
+		public abstract TAwaitable CreateFaulted(Exception exception);
+
+		object IAwaitableFactory.CreateFaulted(Exception exception)
+		{
+			Debug.Assert(exception != null);
+
+			return this.CreateFaulted(exception);
+		}
+
+		public abstract TAwaitable CreateFaulted(IEnumerable<Exception> exceptions);
+
+		object IAwaitableFactory.CreateFaulted(IEnumerable<Exception> exceptions)
+		{
+			Debug.Assert(exceptions != null);
+			Debug.Assert(exceptions.Any());
+
+			return this.CreateFaulted(exceptions);
+		}
+
 		bool IAwaitableFactory.TryGetResult(object awaitable, out object result)
 		{
 			Debug.Assert(awaitable is TAwaitable);

diff --git a/src/Moq/Async/AwaitableFactory`2.cs b/src/Moq/Async/AwaitableFactory`2.cs
@@ -2,7 +2,9 @@
 // All rights reserved. Licensed under the BSD 3-Clause License; see License.txt.
 
 using System;
+using System.Collections.Generic;
 using System.Diagnostics;
+using System.Linq;
 
 namespace Moq.Async
 {
@@ -23,6 +25,25 @@ object IAwaitableFactory.CreateCompleted(object result)
 			return this.CreateCompleted((TResult)result);
 		}
 
+		public abstract TAwaitable CreateFaulted(Exception exception);
+
+		object IAwaitableFactory.CreateFaulted(Exception exception)
+		{
+			Debug.Assert(exception != null);
+
+			return this.CreateFaulted(exception);
+		}
+
+		public abstract TAwaitable CreateFaulted(IEnumerable<Exception> exceptions);
+
+		object IAwaitableFactory.CreateFaulted(IEnumerable<Exception> exceptions)
+		{
+			Debug.Assert(exceptions != null);
+			Debug.Assert(exceptions.Any());
+
+			return this.CreateFaulted(exceptions);
+		}
+
 		public abstract bool TryGetResult(TAwaitable awaitable, out TResult result);
 
 		bool IAwaitableFactory.TryGetResult(object awaitable, out object result)

diff --git a/src/Moq/Async/IAwaitableFactory.cs b/src/Moq/Async/IAwaitableFactory.cs
@@ -2,6 +2,7 @@
 // All rights reserved. Licensed under the BSD 3-Clause License; see License.txt.
 
 using System;
+using System.Collections.Generic;
 
 namespace Moq.Async
 {
@@ -11,6 +12,10 @@ internal interface IAwaitableFactory
 
 		object CreateCompleted(object result = null);
 
+		object CreateFaulted(Exception exception);
+
+		object CreateFaulted(IEnumerable<Exception> exceptions);
+
 		bool TryGetResult(object awaitable, out object result);
 	}
 }
diff --git a/src/Moq/Async/TaskFactory.cs b/src/Moq/Async/TaskFactory.cs
@@ -1,6 +1,9 @@
 // Copyright (c) 2007, Clarius Consulting, Manas Technology Solutions, InSTEDD, and Contributors.
 // All rights reserved. Licensed under the BSD 3-Clause License; see License.txt.
 
+using System;
+using System.Collections.Generic;
+using System.Reflection;
 using System.Threading.Tasks;
 
 namespace Moq.Async
@@ -17,5 +20,19 @@ public override Task CreateCompleted()
 		{
 			return Task.FromResult<object>(default);
 		}
+
+		public override Task CreateFaulted(Exception exception)
+		{
+			var tcs = new TaskCompletionSource<object>();
+			tcs.SetException(exception);
+			return tcs.Task;
+		}
+
+		public override Task CreateFaulted(IEnumerable<Exception> exceptions)
+		{
+			var tcs = new TaskCompletionSource<object>();
+			tcs.SetException(exceptions);
+			return tcs.Task;
+		}
 	}
 }
diff --git a/src/Moq/Async/TaskFactory`1.cs b/src/Moq/Async/TaskFactory`1.cs
@@ -1,6 +1,8 @@
 // Copyright (c) 2007, Clarius Consulting, Manas Technology Solutions, InSTEDD, and Contributors.
 // All rights reserved. Licensed under the BSD 3-Clause License; see License.txt.
 
+using System;
+using System.Collections.Generic;
 using System.Threading.Tasks;
 
 namespace Moq.Async
@@ -12,6 +14,20 @@ public override Task<TResult> CreateCompleted(TResult result)
 			return Task.FromResult(result);
 		}
 
+		public override Task<TResult> CreateFaulted(Exception exception)
+		{
+			var tcs = new TaskCompletionSource<TResult>();
+			tcs.SetException(exception);
+			return tcs.Task;
+		}
+
+		public override Task<TResult> CreateFaulted(IEnumerable<Exception> exceptions)
+		{
+			var tcs = new TaskCompletionSource<TResult>();
+			tcs.SetException(exceptions);
+			return tcs.Task;
+		}
+
 		public override bool TryGetResult(Task<TResult> task, out TResult result)
 		{
 			if (task.Status == TaskStatus.RanToCompletion)

diff --git a/src/Moq/Async/ValueTaskFactory.cs b/src/Moq/Async/ValueTaskFactory.cs
@@ -1,6 +1,8 @@
 // Copyright (c) 2007, Clarius Consulting, Manas Technology Solutions, InSTEDD, and Contributors.
 // All rights reserved. Licensed under the BSD 3-Clause License; see License.txt.
 
+using System;
+using System.Collections.Generic;
 using System.Threading.Tasks;
 
 namespace Moq.Async
@@ -17,5 +19,19 @@ public override ValueTask CreateCompleted()
 		{
 			return default;
 		}
+
+		public override ValueTask CreateFaulted(Exception exception)
+		{
+			var tcs = new TaskCompletionSource<object>();
+			tcs.SetException(exception);
+			return new ValueTask(tcs.Task);
+		}
+
+		public override ValueTask CreateFaulted(IEnumerable<Exception> exceptions)
+		{
+			var tcs = new TaskCompletionSource<object>();
+			tcs.SetException(exceptions);
+			return new ValueTask(tcs.Task);
+		}
 	}
 }
diff --git a/src/Moq/Async/ValueTaskFactory`1.cs b/src/Moq/Async/ValueTaskFactory`1.cs
@@ -1,6 +1,8 @@
 // Copyright (c) 2007, Clarius Consulting, Manas Technology Solutions, InSTEDD, and Contributors.
 // All rights reserved. Licensed under the BSD 3-Clause License; see License.txt.
 
+using System;
+using System.Collections.Generic;
 using System.Threading.Tasks;
 
 namespace Moq.Async
@@ -12,6 +14,20 @@ public override ValueTask<TResult> CreateCompleted(TResult result)
 			return new ValueTask<TResult>(result);
 		}
 
+		public override ValueTask<TResult> CreateFaulted(Exception exception)
+		{
+			var tcs = new TaskCompletionSource<TResult>();
+			tcs.SetException(exception);
+			return new ValueTask<TResult>(tcs.Task);
+		}
+
+		public override ValueTask<TResult> CreateFaulted(IEnumerable<Exception> exceptions)
+		{
+			var tcs = new TaskCompletionSource<TResult>();
+			tcs.SetException(exceptions);
+			return new ValueTask<TResult>(tcs.Task);
+		}
+
 		public override bool TryGetResult(ValueTask<TResult> valueTask, out TResult result)
 		{
 			if (valueTask.IsCompletedSuccessfully)