2 * Copyright 2017 ZTE Corporation.
\r
4 * Licensed under the Apache License, Version 2.0 (the "License");
\r
5 * you may not use this file except in compliance with the License.
\r
6 * You may obtain a copy of the License at
\r
8 * http://www.apache.org/licenses/LICENSE-2.0
\r
10 * Unless required by applicable law or agreed to in writing, software
\r
11 * distributed under the License is distributed on an "AS IS" BASIS,
\r
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
13 * See the License for the specific language governing permissions and
\r
14 * limitations under the License.
\r
17 package org.openo.holmes.rulemgt.resources;
\r
19 import javax.servlet.http.HttpServletRequest;
\r
20 import javax.ws.rs.WebApplicationException;
\r
21 import org.easymock.EasyMock;
\r
22 import org.eclipse.jetty.server.Request;
\r
23 import org.junit.Before;
\r
24 import org.junit.Rule;
\r
25 import org.junit.Test;
\r
26 import org.junit.rules.ExpectedException;
\r
27 import org.openo.holmes.common.exception.CorrelationException;
\r
28 import org.openo.holmes.rulemgt.bean.request.RuleCreateRequest;
\r
29 import org.openo.holmes.rulemgt.bean.request.RuleDeleteRequest;
\r
30 import org.openo.holmes.rulemgt.bean.request.RuleQueryCondition;
\r
31 import org.openo.holmes.rulemgt.bean.request.RuleUpdateRequest;
\r
32 import org.openo.holmes.rulemgt.bean.response.RuleAddAndUpdateResponse;
\r
33 import org.openo.holmes.rulemgt.bean.response.RuleQueryListResponse;
\r
34 import org.openo.holmes.rulemgt.wrapper.RuleMgtWrapper;
\r
35 import org.powermock.api.easymock.PowerMock;
\r
36 import org.powermock.reflect.Whitebox;
\r
38 public class RuleMgtResourcesTest {
\r
41 public ExpectedException thrown = ExpectedException.none();
\r
43 private HttpServletRequest request = PowerMock.createMock(HttpServletRequest.class);
\r
45 private RuleMgtWrapper ruleMgtWrapper = PowerMock.createMock(RuleMgtWrapper.class);
\r
47 private RuleMgtResources ruleMgtResources = new RuleMgtResources();
\r
49 private Request requestMock = PowerMock.createMock(Request.class);
\r
52 public void setUp() throws Exception {
\r
53 Whitebox.setInternalState(ruleMgtResources, "ruleMgtWrapper", ruleMgtWrapper);
\r
54 PowerMock.resetAll();
\r
58 public void addCorrelationRule_correlation_exception() throws Exception {
\r
59 thrown.expect(WebApplicationException.class);
\r
61 final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest();
\r
62 EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest))
\r
63 .andThrow(new CorrelationException(EasyMock.anyObject(String.class)));
\r
64 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US");
\r
65 EasyMock.expect(request.getHeader("username")).andReturn("admin");
\r
66 PowerMock.replayAll();
\r
67 ruleMgtResources.addCorrelationRule(request, ruleCreateRequest);
\r
68 PowerMock.verifyAll();
\r
72 public void addCorrelationRule_normal() throws Exception {
\r
73 StringBuilder stringBuilder = new StringBuilder("http://localhost");
\r
74 final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest();
\r
75 EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin",
\r
76 ruleCreateRequest)).andReturn(new RuleAddAndUpdateResponse());
\r
77 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US");
\r
78 EasyMock.expect(request.getHeader("username")).andReturn("admin");
\r
79 PowerMock.replayAll();
\r
80 ruleMgtResources.addCorrelationRule(request, ruleCreateRequest);
\r
81 PowerMock.verifyAll();
\r
85 public void updateCorrelationRule_correlation_exception() throws Exception {
\r
86 thrown.expect(WebApplicationException.class);
\r
88 final RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest();
\r
89 EasyMock.expect(ruleMgtWrapper.updateCorrelationRule("admin", ruleUpdateRequest))
\r
90 .andThrow(new CorrelationException(EasyMock.anyObject(String.class)));
\r
91 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US");
\r
92 EasyMock.expect(request.getHeader("username")).andReturn("admin");
\r
93 PowerMock.replayAll();
\r
94 ruleMgtResources.updateCorrelationRule(request, ruleUpdateRequest);
\r
95 PowerMock.verifyAll();
\r
99 public void updateCorrelationRule_normal() throws Exception {
\r
100 final RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest();
\r
101 EasyMock.expect(ruleMgtWrapper.updateCorrelationRule("admin",
\r
102 ruleUpdateRequest)).andReturn(new RuleAddAndUpdateResponse());
\r
103 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US");
\r
104 EasyMock.expect(request.getHeader("username")).andReturn("admin");
\r
105 PowerMock.replayAll();
\r
106 ruleMgtResources.updateCorrelationRule(request, ruleUpdateRequest);
\r
107 PowerMock.verifyAll();
\r
111 public void deleteCorrelationRule_correlation_exception() throws Exception {
\r
112 thrown.expect(WebApplicationException.class);
\r
114 final RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest();
\r
115 ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest);
\r
116 EasyMock.expectLastCall().andThrow(new CorrelationException(EasyMock.anyObject(String.class)));
\r
117 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US");
\r
118 PowerMock.replayAll();
\r
119 ruleMgtResources.deleteCorrelationRule(request, ruleDeleteRequest);
\r
120 PowerMock.verifyAll();
\r
124 public void deleteCorrelationRule_normal() throws Exception {
\r
125 final RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest();
\r
126 ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest);
\r
127 EasyMock.expectLastCall();
\r
128 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US");
\r
129 PowerMock.replayAll();
\r
130 ruleMgtResources.deleteCorrelationRule(request, ruleDeleteRequest);
\r
131 PowerMock.verifyAll();
\r
135 public void getCorrelationRules_data_format_exception() throws Exception {
\r
136 thrown.expect(WebApplicationException.class);
\r
138 final String requestStr = "{\"ruleid\":\"rule_001\",\"rulename\":\"Rule-001\","
\r
139 + "\"enabled\":0,\"creator\":\"admin\"}";
\r
140 EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class)))
\r
141 .andThrow(new CorrelationException(EasyMock.anyObject(String.class)));
\r
142 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2);
\r
143 PowerMock.replayAll();
\r
144 ruleMgtResources.getCorrelationRules(request, requestStr);
\r
145 PowerMock.verifyAll();
\r
149 public void getCorrelationRules_param_translate_exception() {
\r
150 thrown.expect(WebApplicationException.class);
\r
152 String queryRequest = "this is error param";
\r
153 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2);
\r
155 PowerMock.replayAll();
\r
156 ruleMgtResources.getCorrelationRules(request, queryRequest);
\r
157 PowerMock.verifyAll();
\r
162 public void getCorrelationRules_normal_request_string_null() throws Exception {
\r
163 EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class)))
\r
164 .andReturn(new RuleQueryListResponse());
\r
165 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2);
\r
166 PowerMock.replayAll();
\r
167 ruleMgtResources.getCorrelationRules(request, null);
\r
168 PowerMock.verifyAll();
\r
172 public void getCorrelationRules_normal_request_string_enabled_missing() throws Exception {
\r
173 final String requestStr = "{\"ruleid\":\"rule_001\",\"rulename\":\"Rule-001\","
\r
174 + "\"creator\":\"admin\"}";
\r
175 EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class)))
\r
176 .andReturn(new RuleQueryListResponse());
\r
177 EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2);
\r
178 PowerMock.replayAll();
\r
179 ruleMgtResources.getCorrelationRules(request, requestStr);
\r
180 PowerMock.verifyAll();
\r