#10 - removed workaround for older Moq versions

astaerk · Mar 6, 2022 · dda8e2f · dda8e2f
1 parent 5df72e3
commit dda8e2f
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/Tests.AutoRegistration/AutoRegistrationFixture.cs b/Tests.AutoRegistration/AutoRegistrationFixture.cs
@@ -36,18 +36,12 @@ public void SetUp()
             _containerMock = new Mock<IUnityContainer>();
             _registered = new List<RegisterEvent>();
             var setup = _containerMock
-                .Setup(c => c.RegisterType(It.IsAny<Type>(), It.IsAny<Type>(), It.IsAny<string>(), It.IsAny<ITypeLifetimeManager>()));
-            var callback = new RegistrationCallback((from, to, name, lifetime, ips) =>
+                .Setup(c => c.RegisterType(It.IsAny<Type>(), It.IsAny<Type>(), It.IsAny<string>(), It.IsAny<ITypeLifetimeManager>()))
+                .Callback<Type, Type, string, ITypeLifetimeManager, InjectionMember[]>((from, to, name, lifetime, ips) =>
                 {
                     _registered.Add(new RegisterEvent(from, to, name, lifetime));
                     _realContainer.RegisterType(from, to, name, lifetime);
                 });
-
-            // Using reflection, because current version of Moq doesn't support callbacks with more than 4 arguments
-            setup
-                .GetType()
-                .GetMethod("SetCallbackWithArguments", BindingFlags.NonPublic | BindingFlags.Instance)
-                .Invoke(setup, new object[] {callback});
 
             _container = _containerMock.Object;
         }