Add msb register code

[holmes/rule-management.git] / rulemgt / src / test / java / org / openo / holmes / rulemgt / bolt / enginebolt / EngineServiceTest.java

diff --git a/rulemgt/src/test/java/org/openo/holmes/rulemgt/bolt/enginebolt/EngineServiceTest.java b/rulemgt/src/test/java/org/openo/holmes/rulemgt/bolt/enginebolt/EngineServiceTest.java
index 596a408..3b1c9bd 100644 (file)
--- a/rulemgt/src/test/java/org/openo/holmes/rulemgt/bolt/enginebolt/EngineServiceTest.java
+++ b/rulemgt/src/test/java/org/openo/holmes/rulemgt/bolt/enginebolt/EngineServiceTest.java
@@ -19,18 +19,22 @@ package org.openo.holmes.rulemgt.bolt.enginebolt;
 \r
 \r
 import org.apache.http.HttpResponse;\r
+import org.apache.http.client.methods.CloseableHttpResponse;\r
+import org.apache.http.client.methods.HttpPut;\r
+import org.apache.http.impl.client.CloseableHttpClient;\r
+import org.apache.http.impl.client.HttpClients;\r
 import org.easymock.EasyMock;\r
 import org.junit.Before;\r
 import org.junit.Rule;\r
 import org.junit.Test;\r
 import org.junit.rules.ExpectedException;\r
-import org.openo.holmes.rulemgt.RuleAppConfig;\r
 import org.openo.holmes.rulemgt.bean.request.CorrelationCheckRule4Engine;\r
 import org.openo.holmes.rulemgt.bean.request.CorrelationDeployRule4Engine;\r
 import org.powermock.api.easymock.PowerMock;\r
+import org.powermock.core.classloader.annotations.PrepareForTest;\r
 import org.powermock.modules.junit4.rule.PowerMockRule;\r
-import org.powermock.reflect.Whitebox;\r
 \r
+@PrepareForTest({HttpClients.class, CloseableHttpClient.class})\r
 public class EngineServiceTest {\r
 \r
     @Rule\r
@@ -39,14 +43,16 @@ public class EngineServiceTest {
     public PowerMockRule powerMockRule = new PowerMockRule();\r
     private EngineService engineService;\r
     private HttpResponse httpResponseMock;\r
-    private RuleAppConfig ruleAppConfig = new RuleAppConfig();\r
+    private CloseableHttpClient closeableHttpClient;\r
     private CorrelationDeployRule4Engine correlationDeployRule4Engine;\r
+    private CloseableHttpResponse closeableHttpResponseMock;\r
 \r
     @Before\r
     public void setUp() {\r
         engineService = new EngineService();\r
+        closeableHttpClient = PowerMock.createMock(CloseableHttpClient.class);\r
         httpResponseMock = PowerMock.createMock(HttpResponse.class);\r
-        Whitebox.setInternalState(engineService, "ruleAppConfig", ruleAppConfig);\r
+        closeableHttpResponseMock = PowerMock.createMock(CloseableHttpResponse.class);\r
         correlationDeployRule4Engine = new CorrelationDeployRule4Engine();\r
         correlationDeployRule4Engine.setContent("{\"package\":\"test\"}");\r
         correlationDeployRule4Engine.setEngineId("engine_id");\r