Merge pull request #7 from atsmalec/issues/6_enable-disable-injection

Fix #6 - Allows enabling/disabling of injection, also allows to query status of injection.

Author: exabrial

README.md

@@ -129,6 +129,24 @@ class MyControllerTest {
 
 Ref: https://docs.oracle.com/javaee/7/api/javax/annotation/PostConstruct.html
 
+## Enabling/Disabling Injection Behavior
+
+The InjectExtension provides methods to turn on and off the behavior as well as query the status of injection.
+
+```
+InjectExtension.enable();
+InjectExtension.bypass();
+InjectExtension.status(); // returns true if enabled
+```
+
+It's recommended that if you're using the above APIs that you reset the Injector status between tests.
+
+```
+@AfterEach
+void afterEach() {
+    InjectExtension.enable();
+}
+```
 
 ## License
 
@@ -154,7 +172,7 @@ Maven Coordinates:
 <dependency>
  <groupId>com.github.exabrial</groupId>
  <artifactId>mockito-object-injection</artifactId>
- <version>2.1.0</version>
+ <version>2.2.0</version>
  <scope>test</scope>
 </dependency>
 ```

pom.xml

@@ -3,7 +3,7 @@
 	<modelVersion>4.0.0</modelVersion>
 	<groupId>com.github.exabrial</groupId>
 	<artifactId>mockito-object-injection</artifactId>
-	<version>2.1.1-SNAPSHOT</version>
+	<version>2.2.0-SNAPSHOT</version>
 	<packaging>jar</packaging>
 	<description>Mockito Object Injection</description>
 	<name>${project.artifactId}</name>

src/main/java/com/github/exabrial/junit5/injectmap/InjectExtension.java

@@ -21,6 +21,20 @@
 import javassist.util.proxy.ProxyFactory;
 
 public class InjectExtension implements BeforeTestExecutionCallback {
+	private static boolean isEnabled = true;
+
+	public static void enable() {
+		isEnabled = true;
+	}
+
+	public static void bypass() {
+		isEnabled = false;
+	}
+
+	public static boolean status() {
+		return isEnabled;
+	}
+
 	@Override
 	public void beforeTestExecution(ExtensionContext context) throws Exception {
 		Object testInstance = context.getTestInstance().get();
@@ -80,20 +94,22 @@ private Map<String, List<Field>> createFieldMap(Class<? extends Object> targetCl
 	}
 
 	private MethodHandler createMethodHandler(final Map<String, Field> injectMap, final Object injectionTarget,
-			final Map<String, List<Field>> fieldMap, final Object testInstance, final Method postConstructMethod) {
+											  final Map<String, List<Field>> fieldMap, final Object testInstance, final Method postConstructMethod) {
 		return (proxy, invokedMethod, proceedMethod, args) -> {
 			invokedMethod.setAccessible(true);
-			for (String fieldName : injectMap.keySet()) {
-				for (Field targetField : fieldMap.get(fieldName)) {
-					Field sourceField = injectMap.get(fieldName);
-					sourceField.setAccessible(true);
-					targetField.setAccessible(true);
-					targetField.set(injectionTarget, sourceField.get(testInstance));
+			if (InjectExtension.isEnabled) {
+				for (String fieldName : injectMap.keySet()) {
+					for (Field targetField : fieldMap.get(fieldName)) {
+						Field sourceField = injectMap.get(fieldName);
+						sourceField.setAccessible(true);
+						targetField.setAccessible(true);
+						targetField.set(injectionTarget, sourceField.get(testInstance));
+					}
+				}
+				if (postConstructMethod != null) {
+					postConstructMethod.setAccessible(true);
+					postConstructMethod.invoke(injectionTarget);
 				}
-			}
-			if (postConstructMethod != null) {
-				postConstructMethod.setAccessible(true);
-				postConstructMethod.invoke(injectionTarget);
 			}
 			try {
 				return invokedMethod.invoke(injectionTarget, args);