fix legacy mock issue

zczc0128 · Oct 4, 2018 · 2243da9 · 2243da9
1 parent 50f0e6e
commit 2243da9
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/...om/ctrip/framework/apollo/adminservice/controller/ControllerIntegrationExceptionTest.java b/...om/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/";
   }