fix legacy mock issue

nobodyiam · nobodyiam · commit 2243da9f0236 · 2018-10-04T09:56:59.000+08:00
diff --git a/apollo-adminservice/src/test/java/com/ctrip/framework/apollo/adminservice/controller/ControllerIntegrationExceptionTest.java b/apollo-adminservice/src/test/java/com/ctrip/framework/apollo/adminservice/controller/ControllerIntegrationExceptionTest.java
@@ -7,6 +7,7 @@
 import com.ctrip.framework.apollo.common.dto.AppDTO;
 import com.ctrip.framework.apollo.common.entity.App;
 
+import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
@@ -31,6 +32,8 @@ public class ControllerIntegrationExceptionTest extends AbstractControllerTest {
   @Mock
   AdminService adminService;
 
+  private Object realAdminService;
+
   @Autowired
   AppService appService;
 
@@ -39,9 +42,17 @@ public class ControllerIntegrationExceptionTest extends AbstractControllerTest {
   @Before
   public void setUp() {
     MockitoAnnotations.initMocks(this);
+
+    realAdminService = ReflectionTestUtils.getField(appController, "adminService");
+
     ReflectionTestUtils.setField(appController, "adminService", adminService);
   }
 
+  @After
+  public void tearDown() throws Exception {
+    ReflectionTestUtils.setField(appController, "adminService", realAdminService);
+  }
+
   private String getBaseAppUrl() {
     return "http://localhost:" + port + "/apps/";
   }
