Allow mocking synchronous methods

mingyaulee · Sep 12, 2024 · 3eef452 · 3eef452
1 parent e2f9e24
commit 3eef452
Show file tree

Hide file tree

Showing 6 changed files with 126 additions and 10 deletions.
diff --git a/src/WebExtensions.Net/Mock/Configurators/ApiConfigurator.cs b/src/WebExtensions.Net/Mock/Configurators/ApiConfigurator.cs
@@ -2,7 +2,6 @@
 using System.Collections.Generic;
 using System.Linq.Expressions;
 using WebExtensions.Net.Mock.Handlers;
-using WebExtensions.Net.Mock.Resolvers;
 
 namespace WebExtensions.Net.Mock.Configurators
 {

diff --git a/src/WebExtensions.Net/Mock/Configurators/BaseConfigurator.cs b/src/WebExtensions.Net/Mock/Configurators/BaseConfigurator.cs
@@ -2,7 +2,6 @@
 using System.Collections.Generic;
 using System.Linq;
 using System.Linq.Expressions;
-using System.Reflection;
 using System.Threading.Tasks;
 using WebExtensions.Net.Mock.Handlers;
 
@@ -35,6 +34,63 @@ protected BaseConfigurator(IWebExtensionsApi webExtensionsApi, IList<IMockHandle
         public MockPropertyConfigurator<T> Property<T>(Expression<Func<TObject, T>> expression)
             => new(mockHandlers, CreateHandler(expression));
 
+        #region Synchronous method without return value
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Action configurator.</returns>
+        public MockActionConfigurator Method(Expression<Func<TObject, Action>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T">The action argument type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Action configurator.</returns>
+        public MockActionConfigurator<T> Method<T>(Expression<Func<TObject, Func<T, Action>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The action argument 1 type.</typeparam>
+        /// <typeparam name="T2">The action argument 2 type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Action configurator.</returns>
+        public MockActionConfigurator<T1, T2> Method<T1, T2>(Expression<Func<TObject, Action<T1, T2>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The action argument 1 type.</typeparam>
+        /// <typeparam name="T2">The action argument 2 type.</typeparam>
+        /// <typeparam name="T3">The action argument 3 type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Action configurator.</returns>
+        public MockActionConfigurator<T1, T2, T3> Method<T1, T2, T3>(Expression<Func<TObject, Action<T1, T2, T3>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The action argument 1 type.</typeparam>
+        /// <typeparam name="T2">The action argument 2 type.</typeparam>
+        /// <typeparam name="T3">The action argument 3 type.</typeparam>
+        /// <typeparam name="T4">The action argument 4 type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Action configurator.</returns>
+        public MockActionConfigurator<T1, T2, T3, T4> Method<T1, T2, T3, T4>(Expression<Func<TObject, Action<T1, T2, T3, T4>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The action argument 1 type.</typeparam>
+        /// <typeparam name="T2">The action argument 2 type.</typeparam>
+        /// <typeparam name="T3">The action argument 3 type.</typeparam>
+        /// <typeparam name="T4">The action argument 4 type.</typeparam>
+        /// <typeparam name="T5">The action argument 5 type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Action configurator.</returns>
+        public MockActionConfigurator<T1, T2, T3, T4, T5> Method<T1, T2, T3, T4, T5>(Expression<Func<TObject, Action<T1, T2, T3, T4, T5>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        #endregion Synchronous method without return value
+
+        #region Asynchronous method without return value
+
         /// <summary>Configures the mock handler for the method.</summary>
         /// <param name="expression">The expression returning the method to mock.</param>
         /// <returns>Action configurator.</returns>
@@ -86,6 +142,71 @@ public MockActionConfigurator<T1, T2, T3, T4> Method<T1, T2, T3, T4>(Expression<
         public MockActionConfigurator<T1, T2, T3, T4, T5> Method<T1, T2, T3, T4, T5>(Expression<Func<TObject, Func<T1, T2, T3, T4, T5, ValueTask>>> expression)
             => new(mockHandlers, CreateHandler(expression));
 
+        #endregion Asynchronous method without return value
+
+        #region Synchronous method with return value
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="TResult">The function return type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Function configurator.</returns>
+        public MockFunctionConfigurator<TResult> Method<TResult>(Expression<Func<TObject, Func<TResult>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T">The function argument type.</typeparam>
+        /// <typeparam name="TResult">The function return type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Function configurator.</returns>
+        public MockFunctionConfigurator<T, TResult> Method<T, TResult>(Expression<Func<TObject, Func<T, TResult>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The function argument 1 type.</typeparam>
+        /// <typeparam name="T2">The function argument 2 type.</typeparam>
+        /// <typeparam name="TResult">The function return type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Function configurator.</returns>
+        public MockFunctionConfigurator<T1, T2, TResult> Method<T1, T2, TResult>(Expression<Func<TObject, Func<T1, T2, TResult>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The function argument 1 type.</typeparam>
+        /// <typeparam name="T2">The function argument 2 type.</typeparam>
+        /// <typeparam name="T3">The function argument 3 type.</typeparam>
+        /// <typeparam name="TResult">The function return type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Function configurator.</returns>
+        public MockFunctionConfigurator<T1, T2, T3, TResult> Method<T1, T2, T3, TResult>(Expression<Func<TObject, Func<T1, T2, T3, TResult>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The function argument 1 type.</typeparam>
+        /// <typeparam name="T2">The function argument 2 type.</typeparam>
+        /// <typeparam name="T3">The function argument 3 type.</typeparam>
+        /// <typeparam name="T4">The function argument 4 type.</typeparam>
+        /// <typeparam name="TResult">The function return type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Function configurator.</returns>
+        public MockFunctionConfigurator<T1, T2, T3, T4, TResult> Method<T1, T2, T3, T4, TResult>(Expression<Func<TObject, Func<T1, T2, T3, T4, TResult>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        /// <summary>Configures the mock handler for the method.</summary>
+        /// <typeparam name="T1">The function argument 1 type.</typeparam>
+        /// <typeparam name="T2">The function argument 2 type.</typeparam>
+        /// <typeparam name="T3">The function argument 3 type.</typeparam>
+        /// <typeparam name="T4">The function argument 4 type.</typeparam>
+        /// <typeparam name="T5">The function argument 5 type.</typeparam>
+        /// <typeparam name="TResult">The function return type.</typeparam>
+        /// <param name="expression">The expression returning the method to mock.</param>
+        /// <returns>Function configurator.</returns>
+        public MockFunctionConfigurator<T1, T2, T3, T4, T5, TResult> Method<T1, T2, T3, T4, T5, TResult>(Expression<Func<TObject, Func<T1, T2, T3, T4, T5, TResult>>> expression)
+            => new(mockHandlers, CreateHandler(expression));
+
+        #endregion Synchronous method with return value
+
+        #region Asynchronous method with return value
+
         /// <summary>Configures the mock handler for the method.</summary>
         /// <typeparam name="TResult">The function return type.</typeparam>
         /// <param name="expression">The expression returning the method to mock.</param>
@@ -143,6 +264,8 @@ public MockFunctionConfigurator<T1, T2, T3, T4, TResult> Method<T1, T2, T3, T4,
         public MockFunctionConfigurator<T1, T2, T3, T4, T5, TResult> Method<T1, T2, T3, T4, T5, TResult>(Expression<Func<TObject, Func<T1, T2, T3, T4, T5, ValueTask<TResult>>>> expression)
             => new(mockHandlers, CreateHandler(expression));
 
+        #endregion Asynchronous method with return value
+
         /// <summary>
         /// Creates the mock handler based on the lamdba expression.
         /// </summary>

diff --git a/src/WebExtensions.Net/Mock/Configurators/IMockConfigurator.cs b/src/WebExtensions.Net/Mock/Configurators/IMockConfigurator.cs
@@ -1,6 +1,4 @@
-using System;
-
-namespace WebExtensions.Net.Mock.Configurators
+namespace WebExtensions.Net.Mock.Configurators
 {
     /// <summary>
     /// Mock configurator interface.

diff --git a/src/WebExtensions.Net/Mock/Configurators/MockConfigurator.cs b/src/WebExtensions.Net/Mock/Configurators/MockConfigurator.cs
@@ -1,5 +1,4 @@
-using System;
-using System.Collections.Generic;
+using System.Collections.Generic;
 using System.Linq;
 using WebExtensions.Net.Mock.Handlers;
 using WebExtensions.Net.Mock.Resolvers;

diff --git a/src/WebExtensions.Net/Mock/MockResolvers.cs b/src/WebExtensions.Net/Mock/MockResolvers.cs
@@ -1,8 +1,6 @@
 using System;
 using System.Collections.Generic;
-using System.Linq;
 using WebExtensions.Net.Mock.Configurators;
-using WebExtensions.Net.Mock.Handlers;
 using WebExtensions.Net.Mock.Resolvers;
 
 namespace WebExtensions.Net.Mock

diff --git a/src/WebExtensions.Net/Mock/Resolvers/DefaultMockResolver.cs b/src/WebExtensions.Net/Mock/Resolvers/DefaultMockResolver.cs
@@ -2,7 +2,6 @@
 using System.Linq;
 using WebExtensions.Net.Mock.Configurators;
 using WebExtensions.Net.Mock.Handlers;
-using WebExtensions.Net.Storage;
 
 namespace WebExtensions.Net.Mock.Resolvers
 {