1 files changed, 17 insertions, 0 deletions

diff --git a/rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java b/rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java
index e25c5b3..16a547b 100644
--- a/rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java
+++ b/rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java
@@ -19,6 +19,7 @@ package org.openo.holmes.rulemgt.resources;
 import javax.servlet.http.HttpServletRequest;

 import javax.ws.rs.WebApplicationException;

 import org.easymock.EasyMock;

+import org.eclipse.jetty.server.Request;

 import org.junit.Before;

 import org.junit.Rule;

 import org.junit.Test;

@@ -45,6 +46,8 @@ public class RuleMgtResourcesTest {
 

     private RuleMgtResources ruleMgtResources = new RuleMgtResources();

 

+    private Request requestMock = PowerMock.createMock(Request.class);

+

     @Before

     public void setUp() throws Exception {

         Whitebox.setInternalState(ruleMgtResources, "ruleMgtWrapper", ruleMgtWrapper);

@@ -67,6 +70,7 @@ public class RuleMgtResourcesTest {
 

     @Test

     public void addCorrelationRule_normal() throws Exception {

+        StringBuilder stringBuilder = new StringBuilder("http://localhost");

         final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest();

         EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin",

                 ruleCreateRequest)).andReturn(new RuleAddAndUpdateResponse());

@@ -142,6 +146,19 @@ public class RuleMgtResourcesTest {
     }

 

     @Test

+    public void getCorrelationRules_param_translate_exception() {

+        thrown.expect(WebApplicationException.class);

+

+        String queryRequest = "this is error param";

+        EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2);

+

+        PowerMock.replayAll();

+        ruleMgtResources.getCorrelationRules(request, queryRequest);

+        PowerMock.verifyAll();

+

+    }

+

+    @Test

     public void getCorrelationRules_normal_request_string_null() throws Exception {

         EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class)))

                 .andReturn(new RuleQueryListResponse());