#12 Add support for async methods

Miista · May 2, 2024 · eea75ec · eea75ec
1 parent 2504e79
commit eea75ec
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 2 deletions.
diff --git a/src/Pose/Helpers/StubHelper.cs b/src/Pose/Helpers/StubHelper.cs
@@ -59,6 +59,14 @@ public static MethodInfo DeVirtualizeMethod(Type thisType, MethodInfo virtualMet
             var bindingFlags = BindingFlags.Instance | (virtualMethod.IsPublic ? BindingFlags.Public : BindingFlags.NonPublic);
             var types = virtualMethod.GetParameters().Select(p => p.ParameterType).ToArray();
 
+            if (thisType.IsNestedPrivate && thisType.GetInterfaces().Any(i => i.Name == "IAsyncStateMachine"))
+            {
+                var nestedType = thisType;
+                var targetMethod = nestedType.GetInterfaceMap(nestedType.GetInterfaces()[0].GetMethod("MoveNext").DeclaringType).TargetMethods[0];
+
+                return targetMethod;
+            }
+
             return thisType.GetMethod(virtualMethod.Name, bindingFlags, null, types, null);
         }
 

diff --git a/src/Pose/Shim.cs b/src/Pose/Shim.cs
@@ -63,6 +63,7 @@ public static Shim Replace<T>(Expression<Func<T>> expression, bool setter = fals
 
         private static Shim ReplaceImpl<T>(Expression<T> expression, bool setter)
         {
+            // TODO: Figure out if method is an async method. Do that by finding the attribute on the method which designates the state machine
             var methodBase = ShimHelper.GetMethodFromExpression(expression.Body, setter, out var instance);
             return new Shim(methodBase, instance) { _setter = setter };
         }

diff --git a/src/Sandbox/Program.cs b/src/Sandbox/Program.cs
@@ -1,11 +1,18 @@
 // See https://aka.ms/new-console-template for more information
 
 using System;
+using System.Threading.Tasks;
 
 namespace Pose.Sandbox
 {
     public class Program
     {
+        public static async Task<int> GetIntAsync()
+        {
+            Console.WriteLine("Here");
+            return await Task.FromResult(1);
+        }
+
         public static void Main(string[] args)
         {
 #if NET48
@@ -19,11 +26,14 @@ public static void Main(string[] args)
 #elif NETCOREAPP2_0
             Console.WriteLine("2.0");
             var dateTimeShim = Shim.Replace(() => DateTime.Now).With(() => new DateTime(2004, 1, 1));
+            var asyncShim = Shim.Replace(() => GetIntAsync()).With(() => Task.FromResult(10));
             PoseContext.Isolate(
                 () =>
                 {
-                    Console.WriteLine(DateTime.Now);
-                }, dateTimeShim);
+                    var result = GetIntAsync().GetAwaiter().GetResult();
+                    Console.WriteLine($"Result: {result}");
+                    //Console.WriteLine(DateTime.Now);
+                }, dateTimeShim, asyncShim);
 #elif NET6_0
             Console.WriteLine("6.0");
             var dateTimeShim = Shim.Replace(() => DateTime.Now).With(() => new DateTime(2004, 1, 1));