StubManagerExtensions hotfix

MCKanpolat · Feb 28, 2018 · 8a6f768 · 8a6f768
1 parent f591447
commit 8a6f768
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/StubMiddleware.Core/Extensions/StubManagerExtensions.cs b/src/StubMiddleware.Core/Extensions/StubManagerExtensions.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Linq;
 using System.Reflection;
 
 namespace StubGenerator.Core
@@ -42,7 +43,7 @@ public static object InvokeCreateNew(this IStubManager stubManager, string typeN
             var type = LoadType(typeName);
 
 
-            MethodInfo method = typeof(IStubManager).GetMethod("CreateNew");
+            MethodInfo method = typeof(IStubManager).GetMethods().FirstOrDefault(w => w.Name == "CreateNew" && w.GetParameters().Count() == 2);
             MethodInfo genericMethod = method.MakeGenericMethod(type);
             return genericMethod.Invoke(stubManager, new object[] { subItemSize, null });
         }
@@ -65,7 +66,7 @@ public static object InvokeCreateListOfSize(this IStubManager stubManager, strin
             }
 
             var type = LoadType(typeName);
-            MethodInfo method = typeof(IStubManager).GetMethod("CreateListOfSize");
+            MethodInfo method = typeof(IStubManager).GetMethods().FirstOrDefault(w => w.Name == "CreateListOfSize" && w.GetParameters().Count() == 3);
             MethodInfo genericMethod = method.MakeGenericMethod(type);
             return genericMethod.Invoke(stubManager, parameters: new object[] { size, subItemSize, null });
         }