Introduce mockAutowired() in ReflectionTestUtils

spring-test/spring-test.gradle

@@ -59,6 +59,7 @@ dependencies {
 	optional("org.jetbrains.kotlin:kotlin-reflect:${kotlinVersion}")
 	optional("org.jetbrains.kotlin:kotlin-stdlib:${kotlinVersion}")
 	optional("io.projectreactor:reactor-test")
+	optional("org.mockito:mockito-core:1.10.19")
 	testCompile(project(":spring-context-support"))
 	testCompile(project(":spring-oxm"))
 	testCompile("javax.cache:cache-api:1.0.0")

spring-test/src/main/java/org/springframework/test/util/ReflectionTestUtils.java

@@ -18,10 +18,14 @@
 
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
+import java.util.HashMap;
+import java.util.Map;
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 
+import org.mockito.Mockito;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.lang.Nullable;
 import org.springframework.util.Assert;
 import org.springframework.util.ClassUtils;
@@ -452,4 +456,34 @@ private static String safeToString(@Nullable Object target) {
 		}
 	}
 
+	public static Map<Class<?>, Object> mockAutowired(Object target) throws IllegalAccessException {
+		Map<Class<?>, Object> beans = new HashMap<>();
+		for(Field field : target.getClass().getDeclaredFields()) {
+			if (field.getDeclaredAnnotation(Autowired.class) == null) {
+				continue;
+			}
+
+			Object value = Mockito.mock(field.getType());
+			beans.put(field.getType(), value);
+			field.setAccessible(true);
+			field.set(target, value);
+		}
+
+		return beans;
+	}
+
+	@SuppressWarnings("unchecked")
+	public static <T> T getBean(Map<Class<?>, Object> beans, Class<T> type) {
+		Object v = beans.get(type);
+		if (v == null) {
+			return null;
+		}
+
+		Class<?> dataType = v.getClass();
+		if (dataType.equals(type) || type.isAssignableFrom(v.getClass())) {
+			return (T) v;
+		}
+		return null;
+	}
+
 }

spring-test/src/test/java/org/springframework/test/util/ReflectionTestUtilsTests.java

@@ -24,12 +24,15 @@
 
 import org.springframework.aop.framework.ProxyFactory;
 import org.springframework.aop.support.AopUtils;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.test.util.subpackage.Component;
 import org.springframework.test.util.subpackage.LegacyEntity;
 import org.springframework.test.util.subpackage.Person;
 import org.springframework.test.util.subpackage.PersonEntity;
 import org.springframework.test.util.subpackage.StaticFields;
 
+import java.util.Map;
+
 import static org.hamcrest.CoreMatchers.*;
 import static org.junit.Assert.*;
 import static org.springframework.test.util.ReflectionTestUtils.*;
@@ -421,5 +424,28 @@ public void invokeSetterMethodOnLegacyEntityWithSideEffectsInToString() {
 		invokeSetterMethod(entity, "collaborator", testCollaborator);
 		assertTrue(entity.toString().contains(testCollaborator));
 	}
+	@Test
+	public void invokeMockAutowiredWillAutoMockAllAutowiredFields() throws IllegalAccessException {
+		MockAutowiredStub stub = new MockAutowiredStub();
+		Map<Class<?>, Object> beans = mockAutowired(stub);
+		assertNotNull(beans);
+
+		assertNotNull(getBean(beans, LegacyEntity.class));
+		assertNotNull(getBean(beans, Component.class));
+	}
+	private static class MockAutowiredStub {
+		@Autowired
+		private LegacyEntity legacyEntity;
 
+		@Autowired
+		private Component component;
+
+		public LegacyEntity getLegacyEntity() {
+			return legacyEntity;
+		}
+
+		public Component getComponent() {
+			return component;
+		}
+	}
 }