feat: verify @MockTokenServices on test class

Author: nigel.zheng

src/main/java/com/github/ahunigel/test/security/oauth2/MockTokenServices.java

@@ -0,0 +1,12 @@
+package com.github.ahunigel.test.security.oauth2;
+
+import org.springframework.boot.test.mock.mockito.MockBean;
+import org.springframework.boot.test.mock.mockito.MockReset;
+import org.springframework.security.oauth2.provider.token.ResourceServerTokenServices;
+
+/**
+ * Created by Nigel Zheng on 8/7/2018.
+ */
+@MockBean(value = {ResourceServerTokenServices.class}, reset = MockReset.NONE)
+public @interface MockTokenServices {
+}

src/main/java/com/github/ahunigel/test/security/oauth2/WithToken.java

@@ -1,7 +1,5 @@
 package com.github.ahunigel.test.security.oauth2;
 
-import org.springframework.boot.test.mock.mockito.MockBean;
-import org.springframework.boot.test.mock.mockito.MockReset;
 import org.springframework.security.oauth2.provider.token.ResourceServerTokenServices;
 import org.springframework.security.web.authentication.preauth.PreAuthenticatedAuthenticationToken;
 
@@ -12,10 +10,11 @@
  * Created by Nigel Zheng on 2018/8/7.
  * <p>
  * Emulate an OAuth2 token request, would extract an {@link PreAuthenticatedAuthenticationToken}
+ * <p>
+ * require annotation {@link MockTokenServices} on test class level or {@link ResourceServerTokenServices} @MockBean exists
  *
  * @author nigel
  */
-@MockBean(value = {ResourceServerTokenServices.class}, reset = MockReset.NONE)
 @Retention(RetentionPolicy.RUNTIME)
 public @interface WithToken {
 

src/main/java/com/github/ahunigel/test/security/oauth2/WithTokenTestExecutionListener.java

@@ -46,6 +46,7 @@ public class WithTokenTestExecutionListener extends AbstractTestExecutionListene
   public void beforeTestClass(TestContext testContext) throws Exception {
     Annotation annotation = AnnotatedElementUtils.findMergedAnnotation(testContext.getTestClass(), WithToken.class);
     if (annotation != null) {
+      verifyTokenServicesMocked(testContext.getTestClass());
       addAuthHeader(testContext.getTestClass(), testContext, (WithToken) annotation);
     }
   }
@@ -54,6 +55,7 @@ public void beforeTestClass(TestContext testContext) throws Exception {
   public void beforeTestMethod(TestContext testContext) throws Exception {
     Annotation annotation = AnnotatedElementUtils.findMergedAnnotation(testContext.getTestMethod(), WithToken.class);
     if (annotation != null) {
+      verifyTokenServicesMocked(testContext.getTestClass());
       addAuthHeader(testContext.getTestMethod(), testContext, (WithToken) annotation);
     }
   }
@@ -74,6 +76,11 @@ public void afterTestClass(TestContext testContext) throws Exception {
     }
   }
 
+  private void verifyTokenServicesMocked(Class<?> testClass) {
+    MockTokenServices annotation = AnnotatedElementUtils.findMergedAnnotation(testClass, MockTokenServices.class);
+    Assert.state(annotation != null, "Missing @MockTokenServices on class level");
+  }
+
   private void addAuthHeader(AnnotatedElement annotated, TestContext testContext, WithToken withToken) {
     Assert.state(withToken != null, "No @WithToken exists!!!");
     MockMvc mockMvc = testContext.getApplicationContext().getBean(MockMvc.class);