2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2018-2019 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END=========================================================
21 package org.onap.policy.drools.protocol.coders;
23 import com.google.gson.Gson;
24 import com.google.gson.GsonBuilder;
25 import java.io.IOException;
26 import java.nio.file.Files;
27 import java.nio.file.Paths;
28 import java.util.ArrayList;
29 import java.util.List;
30 import java.util.Properties;
31 import org.junit.Assert;
32 import org.junit.Before;
33 import org.junit.Test;
34 import org.kie.api.builder.ReleaseId;
35 import org.onap.policy.common.endpoints.event.comm.TopicEndpoint;
36 import org.onap.policy.common.endpoints.event.comm.TopicSink;
37 import org.onap.policy.common.endpoints.properties.PolicyEndPointProperties;
38 import org.onap.policy.drools.controller.DroolsController;
39 import org.onap.policy.drools.controller.internal.MavenDroolsControllerTest;
40 import org.onap.policy.drools.properties.DroolsProperties;
41 import org.onap.policy.drools.protocol.coders.EventProtocolCoder.CoderFilters;
42 import org.onap.policy.drools.protocol.coders.JsonProtocolFilter.FilterRule;
43 import org.onap.policy.drools.protocol.coders.TopicCoderFilterConfiguration.CustomGsonCoder;
44 import org.onap.policy.drools.util.KieUtils;
45 import org.onap.policy.drools.utils.Triple;
46 import org.slf4j.Logger;
47 import org.slf4j.LoggerFactory;
50 * ProtocolCoder Toolset JUNITs.
52 public class ProtocolCoderToolsetTest {
53 public static final String JUNIT_PROTOCOL_CODER_ARTIFACT_ID = "protocolcoder";
54 public static final String JUNIT_PROTOCOL_CODER_TOPIC = JUNIT_PROTOCOL_CODER_ARTIFACT_ID;
55 public static final String CONTROLLER_ID = "blah";
56 public static final String ARTIFACT_ID_ECHO = "echo";
57 public static final String ARTIFACT_ID_POM_LINE = "<artifactId>" + ARTIFACT_ID_ECHO + "</artifactId>";
59 private static Logger logger = LoggerFactory.getLogger(ProtocolCoderToolset.class);
61 private volatile ReleaseId releaseId;
63 public static final Gson customCoder = new GsonBuilder().create();
68 * @throws IOException throws IO Exception
71 public void setUp() throws IOException {
72 if (releaseId != null) {
76 String pom = new String(Files.readAllBytes(Paths.get(MavenDroolsControllerTest.JUNIT_ECHO_KMODULE_POM_PATH)));
78 if (!pom.contains(ARTIFACT_ID_POM_LINE)) {
79 throw new IllegalArgumentException("unexpected junit test pom");
82 String newPom = pom.replace(ARTIFACT_ID_ECHO, JUNIT_PROTOCOL_CODER_ARTIFACT_ID);
84 String kmodule = new String(Files.readAllBytes(Paths.get(MavenDroolsControllerTest.JUNIT_ECHO_KMODULE_PATH)));
86 String drl = new String(Files.readAllBytes(Paths.get(MavenDroolsControllerTest.JUNIT_ECHO_KMODULE_DRL_PATH)));
88 releaseId = KieUtils.installArtifact(kmodule, newPom, MavenDroolsControllerTest.JUNIT_ECHO_KJAR_DRL_PATH, drl);
92 public void testToolsets() {
94 testGsonToolset(createFilterSet());
98 * Test the Gson toolset.
100 * @param protocolFilter protocol filter
102 public void testGsonToolset(JsonProtocolFilter protocolFilter) {
103 GsonProtocolCoderToolset gsonToolset = new GsonProtocolCoderToolset(
104 EventProtocolParams.builder().topic(JUNIT_PROTOCOL_CODER_TOPIC)
105 .groupId(this.releaseId.getGroupId())
106 .artifactId(this.releaseId.getArtifactId())
107 .eventClass(Triple.class.getCanonicalName())
108 .protocolFilter(protocolFilter)
109 .customGsonCoder(null)
110 .modelClassLoaderHash(12345678), CONTROLLER_ID);
112 Assert.assertNotNull(gsonToolset.getEncoder());
113 Assert.assertNotNull(gsonToolset.getDecoder());
115 testToolset(protocolFilter, gsonToolset);
117 Triple<String, String, String> triple = createTriple();
118 gsonToolset.setCustomCoder(new CustomGsonCoder(this.getClass().getCanonicalName(), "customCoder"));
119 String tripleEncoded = encode(gsonToolset, triple);
120 decode(protocolFilter, gsonToolset, triple, tripleEncoded);
123 private Triple<String, String, String> createTriple() {
124 return new Triple<>("v1", "v2", "v3");
127 private void testToolset(JsonProtocolFilter protocolFilter, ProtocolCoderToolset coderToolset) {
129 validateInitialization(protocolFilter, coderToolset);
131 updateCoderFilterRule(coderToolset);
133 addRemoveCoder(coderToolset);
135 /* restore original filters */
136 coderToolset.addCoder(Triple.class.getCanonicalName(), protocolFilter, 654321);
138 Triple<String, String, String> triple = createTriple();
140 String tripleEncoded = encode(coderToolset, triple);
142 decode(protocolFilter, coderToolset, triple, tripleEncoded);
145 @SuppressWarnings("unchecked")
146 private void decode(JsonProtocolFilter protocolFilter, ProtocolCoderToolset coderToolset,
147 Triple<String, String, String> triple, String tripleEncoded) {
149 Triple<String, String, String> tripleDecoded = null;
151 tripleDecoded = (Triple<String, String, String>) coderToolset.decode(tripleEncoded);
152 } catch (UnsupportedOperationException e) {
154 logger.trace("Junit expected exception - decode does not pass filtering", e);
157 CoderFilters coderFilters = coderToolset.getCoder(Triple.class.getCanonicalName());
158 Assert.assertTrue(coderFilters.getCodedClass() == Triple.class.getCanonicalName());
159 Assert.assertTrue(coderFilters.getFilter() == protocolFilter);
160 Assert.assertTrue(coderFilters.getFilter().getRules("second").size() == 1);
161 Assert.assertTrue(coderFilters.getFilter().getRules("third").size() == 1);
163 coderFilters.getFilter().getRules("second").get(0).setRegex("^v2$");
164 coderFilters.getFilter().getRules("third").get(0).setRegex(".*v3.*");
166 tripleDecoded = (Triple<String, String, String>) coderToolset.decode(tripleEncoded);
168 Assert.assertTrue(tripleDecoded.first().equals(triple.first()));
169 Assert.assertTrue(tripleDecoded.second().equals(triple.second()));
170 Assert.assertTrue(tripleDecoded.third().equals(triple.third()));
172 coderFilters.getFilter().deleteRules("third");
173 Assert.assertTrue(coderFilters.getFilter().getRules("third").isEmpty());
175 tripleDecoded = (Triple<String, String, String>) coderToolset.decode(tripleEncoded);
177 Assert.assertTrue(tripleDecoded.first().equals(triple.first()));
178 Assert.assertTrue(tripleDecoded.second().equals(triple.second()));
179 Assert.assertTrue(tripleDecoded.third().equals(triple.third()));
181 coderFilters.getFilter().addRule("third", ".*v3.*");
184 private String encode(ProtocolCoderToolset coderToolset, Triple<String, String, String> triple) {
185 String tripleEncoded = coderToolset.encode(triple);
186 Assert.assertTrue(!tripleEncoded.isEmpty());
187 return tripleEncoded;
190 private void addRemoveCoder(ProtocolCoderToolset coderToolset) {
191 List<FilterRule> filters = new ArrayList<>();
192 filters.add(new FilterRule("second", ".*"));
194 coderToolset.addCoder(this.getClass().getCanonicalName(), new JsonProtocolFilter(filters), 654321);
195 Assert.assertTrue(coderToolset.getCoders().size() == 2);
197 coderToolset.removeCoders(this.getClass().getCanonicalName());
198 Assert.assertTrue(coderToolset.getCoders().size() == 1);
201 private void updateCoderFilterRule(ProtocolCoderToolset coderToolset) {
202 List<FilterRule> filters = new ArrayList<>();
203 filters.add(new FilterRule("third", ".*"));
204 coderToolset.addCoder(Triple.class.getCanonicalName(), new JsonProtocolFilter(filters), 654321);
206 Assert.assertTrue(coderToolset.getCoders().size() == 1);
208 Assert.assertTrue(coderToolset.getCoder(Triple.class.getCanonicalName()).getModelClassLoaderHash() == 654321);
211 coderToolset.getCoder(Triple.class.getCanonicalName()).getFilter().getRules("third").size() == 1);
214 coderToolset.getCoder(Triple.class.getCanonicalName()).getFilter().getRules("third").size() == 1);
216 Assert.assertTrue(".*".equals(coderToolset.getCoder(Triple.class.getCanonicalName()).getFilter()
217 .getRules("third").get(0).getRegex()));
220 private void validateInitialization(JsonProtocolFilter protocolFilter, ProtocolCoderToolset coderToolset) {
221 Assert.assertTrue(CONTROLLER_ID.equals(coderToolset.getControllerId()));
222 Assert.assertTrue(this.releaseId.getGroupId().equals(coderToolset.getGroupId()));
223 Assert.assertTrue(this.releaseId.getArtifactId().equals(coderToolset.getArtifactId()));
224 Assert.assertNull(coderToolset.getCustomCoder());
226 Assert.assertTrue(coderToolset.getCoders().size() == 1);
228 CoderFilters coderFilters = coderToolset.getCoder(CONTROLLER_ID);
229 Assert.assertTrue(coderFilters == null);
231 coderFilters = coderToolset.getCoder(Triple.class.getCanonicalName());
232 Assert.assertNotNull(coderFilters);
234 Assert.assertEquals(coderFilters.getFilter(), protocolFilter);
237 private void createController() {
238 if (releaseId == null) {
239 throw new IllegalStateException("no prereq artifact installed in maven repository");
242 Properties sinkConfig = new Properties();
243 sinkConfig.put(PolicyEndPointProperties.PROPERTY_NOOP_SINK_TOPICS, JUNIT_PROTOCOL_CODER_TOPIC);
244 final List<? extends TopicSink> noopTopics = TopicEndpoint.manager.addTopicSinks(sinkConfig);
246 Properties droolsControllerConfig = new Properties();
247 droolsControllerConfig.put(DroolsProperties.RULES_GROUPID, releaseId.getGroupId());
248 droolsControllerConfig.put(DroolsProperties.RULES_ARTIFACTID, releaseId.getArtifactId());
249 droolsControllerConfig.put(DroolsProperties.RULES_VERSION, releaseId.getVersion());
250 droolsControllerConfig.put(PolicyEndPointProperties.PROPERTY_NOOP_SINK_TOPICS + "." + JUNIT_PROTOCOL_CODER_TOPIC
251 + PolicyEndPointProperties.PROPERTY_TOPIC_EVENTS_SUFFIX, Triple.class.getCanonicalName());
253 DroolsController.factory.build(droolsControllerConfig, null, noopTopics);
256 private JsonProtocolFilter createFilterSet() {
257 List<FilterRule> filters = new ArrayList<>();
258 filters.add(new FilterRule("first", ".*"));
259 filters.add(new FilterRule("second", "^blah.*"));
260 filters.add(new FilterRule("third", "^hello$"));
262 return new JsonProtocolFilter(filters);