1 /*******************************************************************************
2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017 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
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
17 * ============LICENSE_END=========================================================
19 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
21 *******************************************************************************/
22 package org.onap.dmaap.mr.client.impl;
25 import java.io.FileOutputStream;
26 import java.io.IOException;
27 import java.util.LinkedList;
28 import java.util.Properties;
30 import junit.framework.TestCase;
32 import org.junit.Test;
33 import org.onap.dmaap.mr.client.MRClientFactory;
34 import org.onap.dmaap.mr.client.impl.MRConstants;
35 import org.onap.dmaap.mr.client.impl.MRConsumerImpl;
36 import org.onap.dmaap.mr.test.clients.ProtocolTypeConstants;
38 public class MRConsumerImplTest extends TestCase {
40 public void testNullFilter() throws IOException {
41 final LinkedList<String> hosts = new LinkedList<String>();
42 hosts.add("localhost:8080");
43 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", -1, -1, null, null, null);
44 final String url = c.createUrlPath(MRConstants.makeConsumerUrl("localhost:8080", "topic", "cg", "cid", "http"),
46 assertEquals("http://localhost:8080/events/" + "topic/cg/cid", url);
50 public void testFilterWithNoTimeoutOrLimit() throws IOException {
51 final LinkedList<String> hosts = new LinkedList<String>();
52 hosts.add("localhost:8080");
53 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", -1, -1, "filter", null, null);
54 final String url = c.createUrlPath(MRConstants.makeConsumerUrl("localhost:8080", "topic", "cg", "cid", "http"),
56 assertEquals("http://localhost:8080/events/" + "topic/cg/cid?filter=filter", url);
60 public void testTimeoutNoLimitNoFilter() throws IOException {
61 final LinkedList<String> hosts = new LinkedList<String>();
62 hosts.add("localhost:8080");
63 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", 30000, -1, null, null, null);
64 final String url = c.createUrlPath(MRConstants.makeConsumerUrl("localhost:8080", "topic", "cg", "cid", "http"),
66 assertEquals("http://localhost:8080/events/" + "topic/cg/cid?timeout=30000", url);
70 public void testNoTimeoutWithLimitNoFilter() throws IOException {
71 final LinkedList<String> hosts = new LinkedList<String>();
72 hosts.add("localhost:8080");
73 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", -1, 100, null, null, null);
74 final String url = c.createUrlPath(MRConstants.makeConsumerUrl("localhost:8080", "topic", "cg", "cid", "http"),
76 assertEquals("http://localhost:8080/events/" + "topic/cg/cid?limit=100", url);
80 public void testWithTimeoutWithLimitWithFilter() throws IOException {
81 final LinkedList<String> hosts = new LinkedList<String>();
82 hosts.add("localhost:8080");
83 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", 1000, 400, "f", null, null);
84 final String url = c.createUrlPath(MRConstants.makeConsumerUrl("localhost:8080", "topic", "cg", "cid", "http"),
86 assertEquals("http://localhost:8080/events/" + "topic/cg/cid?timeout=1000&limit=400&filter=f", url);
90 public void testFilterEncoding() throws IOException {
91 final LinkedList<String> hosts = new LinkedList<String>();
92 hosts.add("localhost:8080");
93 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", -1, -1, "{ \"foo\"=\"bar\"bar\" }",
95 final String url = c.createUrlPath(MRConstants.makeConsumerUrl("localhost:8080", "topic", "cg", "cid", "http"),
97 assertEquals("http://localhost:8080/events/" + "topic/cg/cid?filter=%7B+%22foo%22%3D%22bar%22bar%22+%7D", url);
101 public void testFetchWithReturnConsumerResponse() throws IOException {
102 final LinkedList<String> hosts = new LinkedList<String>();
103 hosts.add("localhost:8080");
104 Properties properties = new Properties();
106 MRSimplerBatchConsumerTest.class.getClassLoader().getResourceAsStream("dme2/consumer.properties"));
108 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", -1, -1, "{ \"foo\"=\"bar\"bar\" }",
110 assertNotNull(c.fetchWithReturnConsumerResponse());
111 c.setProtocolFlag(ProtocolTypeConstants.AAF_AUTH.getValue());
112 assertNotNull(c.fetchWithReturnConsumerResponse());
113 c.setProtocolFlag(ProtocolTypeConstants.HTTPNOAUTH.getValue());
114 assertNotNull(c.fetchWithReturnConsumerResponse());
115 c.setProtocolFlag(ProtocolTypeConstants.AUTH_KEY.getValue());
116 assertNotNull(c.fetchWithReturnConsumerResponse());
121 public void testFetch() throws Exception {
122 final LinkedList<String> hosts = new LinkedList<String>();
123 hosts.add("localhost:8080");
126 Properties properties = new Properties();
128 MRSimplerBatchPublisherTest.class.getClassLoader().getResourceAsStream("dme2/consumer.properties"));
130 String routeFilePath = "dme2/preferredRoute.txt";
132 File file = new File(MRSimplerBatchPublisherTest.class.getClassLoader().getResource(routeFilePath).getFile());
133 properties.put("routeFilePath",
134 MRSimplerBatchPublisherTest.class.getClassLoader().getResource(routeFilePath).getFile());
136 File outFile = new File(file.getParent() + "/consumer_tmp.properties");
137 properties.store(new FileOutputStream(outFile), "");
139 MRClientFactory.prop=properties;
140 final MRConsumerImpl c = new MRConsumerImpl(hosts, "topic", "cg", "cid", -1, -1, "{ \"foo\"=\"bar\"bar\" }",
142 c.setProps(properties);
145 } catch (Exception e) {
148 c.setProtocolFlag(ProtocolTypeConstants.AAF_AUTH.getValue());
149 assertNotNull(c.fetchWithReturnConsumerResponse());
150 c.setProtocolFlag(ProtocolTypeConstants.HTTPNOAUTH.getValue());
151 assertNotNull(c.fetchWithReturnConsumerResponse());
152 c.setProtocolFlag(ProtocolTypeConstants.AUTH_KEY.getValue());
153 assertNotNull(c.fetchWithReturnConsumerResponse());