-/*
+/*-
* ============LICENSE_START=======================================================
* feature-simulators
* ================================================================================
- * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved.
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@Path("/events")
public class DMaaPSimulatorJaxRs {
- private static final String NO_DATA_MSG = "No Data";
+ public static final String NO_TOPIC_MSG = "No topic";
+ public static final String NO_DATA_MSG = "No Data";
+
private static final Map<String, BlockingQueue<String>> queues = new ConcurrentHashMap<>();
private static final Logger logger = LoggerFactory.getLogger(DMaaPSimulatorJaxRs.class);
private static int responseCode = 200;
/**
* Get consumer ID.
- *
+ *
* @param timeout timeout value
* @param topicName the dmaap topic
* @param httpResponse http response object
*/
@GET
@Path("/{topicName}/{consumeGroup}/{consumerId}")
- public String subscribe(@DefaultValue("0") @QueryParam("timeout") int timeout, @PathParam("topicName")
+ public String subscribe(@DefaultValue("0") @QueryParam("timeout") int timeout, @PathParam("topicName")
String topicName, @Context final HttpServletResponse httpResponse) {
int currentRespCode = responseCode;
httpResponse.setStatus(currentRespCode);
else if (timeout > 0) {
return waitForNextMessageFromQueue(timeout, topicName);
}
- return "No topic";
+ return NO_TOPIC_MSG;
}
private String getNextMessageFromQueue(final int timeout, final String topicName) {
BlockingQueue<String> queue = queues.get(topicName);
String response = NO_DATA_MSG;
try {
- response = queue.poll(timeout, TimeUnit.MILLISECONDS);
+ response = poll(queue, timeout);
} catch (InterruptedException e) {
logger.debug("error in DMaaP simulator", e);
Thread.currentThread().interrupt();
private String waitForNextMessageFromQueue(int timeout, String topicName) {
try {
- Thread.sleep(timeout);
+ sleep(timeout);
if (queues.containsKey(topicName)) {
BlockingQueue<String> queue = queues.get(topicName);
String response = queue.poll();
logger.debug("error in DMaaP simulator", e);
Thread.currentThread().interrupt();
}
- return "No topic";
+ return NO_TOPIC_MSG;
}
/**
* Post to a topic.
- *
+ *
* @param topicName name of the topic
* @param body message
* @return empty string
return "Status code set";
}
+ // the following non-static methods may be overridden by junit tests
+
+ protected String poll(BlockingQueue<String> queue, final int timeout) throws InterruptedException {
+ return queue.poll(timeout, TimeUnit.MILLISECONDS);
+ }
+
+ protected void sleep(int timeout) throws InterruptedException {
+ Thread.sleep(timeout);
+ }
+
/**
* Static method to set static response code, synchronized for multiple possible uses.
- *
+ *
* @param incomingResponseCode the response code to set
*/
private static synchronized void setResponseCode(final int incomingResponseCode) {
- responseCode = incomingResponseCode;
+ responseCode = incomingResponseCode;
+ }
+
+ /**
+ * Used only by junit tests to reset the simulator.
+ */
+ protected static void reset() {
+ responseCode = 200;
+ queues.clear();
}
}
--- /dev/null
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP
+ * ================================================================================
+ * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.drools.simulators;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.doThrow;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.LinkedBlockingQueue;
+import javax.servlet.http.HttpServletResponse;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+
+public class DMaaPSimulatorJaxRsTest {
+ private static final String MESSAGE = "hello";
+ private static final String MESSAGE2 = "world";
+ private static final String EXPECTED_EXCEPTION = "expected exception";
+ private static final String TOPIC = "my-topic";
+ private static final int TIMEOUT_MS = 10;
+ private static final int LONG_TIMEOUT_MS = 250;
+
+ @Mock
+ private HttpServletResponse resp;
+
+ private DMaaPSimulatorJaxRs sim;
+
+ /**
+ * Initializes objects and creates the simulator.
+ */
+ @Before
+ public void setUp() {
+ MockitoAnnotations.initMocks(this);
+
+ sim = new DMaaPSimulatorJaxRs();
+ }
+
+ @After
+ public void tearDown() {
+ DMaaPSimulatorJaxRs.reset();
+ }
+
+ @Test
+ public void testSubscribe() {
+ sim.publish(TOPIC, MESSAGE);
+
+ assertEquals(MESSAGE, sim.subscribe(0, TOPIC, resp));
+ }
+
+ @Test
+ public void testSubscribe_FlushEx() throws IOException {
+ doThrow(new IOException(EXPECTED_EXCEPTION)).when(resp).flushBuffer();
+
+ assertEquals("Got an error", sim.subscribe(TIMEOUT_MS, TOPIC, resp));
+ }
+
+ @Test
+ public void testSubscribe_BadStatus_testSetResponseCode() {
+ sim.setStatus(199);
+ assertEquals("You got response code: 199", sim.subscribe(TIMEOUT_MS, TOPIC, resp));
+
+ sim.setStatus(300);
+ assertEquals("You got response code: 300", sim.subscribe(TIMEOUT_MS, TOPIC, resp));
+ }
+
+ @Test
+ public void testSubscribe_UnknownTopic_ZeroTimeout() {
+ assertEquals(DMaaPSimulatorJaxRs.NO_TOPIC_MSG, sim.subscribe(0, TOPIC, resp));
+ }
+
+ @Test
+ public void testSubscribe_UnknownTopic_NonZeroTimeout() {
+ assertEquals(DMaaPSimulatorJaxRs.NO_TOPIC_MSG, sim.subscribe(TIMEOUT_MS, TOPIC, resp));
+ }
+
+ @Test
+ public void testGetNextMessageFromQueue() {
+ sim.publish(TOPIC, MESSAGE);
+ sim.publish(TOPIC, MESSAGE2);
+
+ assertEquals(MESSAGE, sim.subscribe(0, TOPIC, resp));
+ assertEquals(MESSAGE2, sim.subscribe(0, TOPIC, resp));
+
+ // repeat - no message
+ assertEquals(DMaaPSimulatorJaxRs.NO_DATA_MSG, sim.subscribe(0, TOPIC, resp));
+ }
+
+ @Test
+ public void testGetNextMessageFromQueue_Interrupted() throws InterruptedException {
+ sim = new DMaaPSimulatorJaxRs() {
+ @Override
+ protected String poll(BlockingQueue<String> queue, int timeout) throws InterruptedException {
+ throw new InterruptedException(EXPECTED_EXCEPTION);
+ }
+ };
+
+ sim.publish(TOPIC, MESSAGE);
+
+ // put it in the background so we don't interrupt the test thread
+ BlockingQueue<String> queue = new LinkedBlockingQueue<>();
+ backgroundSubscribe(queue);
+
+ assertEquals(DMaaPSimulatorJaxRs.NO_DATA_MSG, queue.take());
+ }
+
+ @Test
+ public void testWaitForNextMessageFromQueue() throws InterruptedException {
+ BlockingQueue<String> queue = new LinkedBlockingQueue<>();
+
+ CountDownLatch latch1 = backgroundSubscribe(queue);
+ CountDownLatch latch2 = backgroundSubscribe(queue);
+
+ // wait for both threads to start
+ latch1.await();
+ latch2.await();
+
+ /*
+ * Must pause to prevent the topic from being created before subscribe() is
+ * invoked.
+ */
+ Thread.sleep(LONG_TIMEOUT_MS / 3);
+
+ // only publish one message
+ sim.publish(TOPIC, MESSAGE);
+
+ // wait for both subscribers to add their messages to the queue
+ List<String> messages = new ArrayList<>();
+ messages.add(queue.take());
+ messages.add(queue.take());
+
+ // sort them so the order is consistent
+ Collections.sort(messages);
+
+ assertEquals("[No Data, hello]", messages.toString());
+ }
+
+ @Test
+ public void testWaitForNextMessageFromQueue_Interrupted() throws InterruptedException {
+ sim = new DMaaPSimulatorJaxRs() {
+ @Override
+ protected void sleep(int timeout) throws InterruptedException {
+ throw new InterruptedException(EXPECTED_EXCEPTION);
+ }
+ };
+
+ // put it in the background so we don't interrupt the test thread
+ BlockingQueue<String> queue = new LinkedBlockingQueue<>();
+ backgroundSubscribe(queue);
+
+ assertEquals(DMaaPSimulatorJaxRs.NO_TOPIC_MSG, queue.take());
+ }
+
+ @Test
+ public void testPublish() {
+ assertEquals("", sim.publish(TOPIC, MESSAGE));
+ assertEquals(MESSAGE, sim.subscribe(0, TOPIC, resp));
+ }
+
+ @Test
+ public void testSetStatus() {
+ assertEquals("Status code set", sim.setStatus(500));
+ assertEquals("You got response code: 500", sim.subscribe(TIMEOUT_MS, TOPIC, resp));
+ }
+
+ /**
+ * Invokes subscribe() in a background thread.
+ *
+ * @param queue where to place the returned result
+ * @return a latch that will be counted down just before the background thread invokes
+ * subscribe()
+ */
+ private CountDownLatch backgroundSubscribe(BlockingQueue<String> queue) {
+ CountDownLatch latch = new CountDownLatch(1);
+
+ new Thread(() -> {
+ latch.countDown();
+ queue.add(sim.subscribe(LONG_TIMEOUT_MS, TOPIC, resp));
+ }).start();
+
+ return latch;
+ }
+}
public class DMaaPSimulatorTest {
- private static final int DMAAPSIM_SERVER_PORT = 6670;
+ private static int DMAAPSIM_SERVER_PORT;
/**
* Setup the simulator.
+ * @throws IOException if a server port cannot be allocated
*/
@BeforeClass
- public static void setUpSimulator() {
+ public static void setUpSimulator() throws IOException {
LoggerUtil.setLevel("ROOT", "INFO");
LoggerUtil.setLevel("org.eclipse.jetty", "WARN");
+ DMAAPSIM_SERVER_PORT = NetworkUtil.allocPort();
try {
final HttpServletServer testServer = HttpServletServerFactoryInstance.getServerFactory().build("dmaapSim",
"localhost", DMAAPSIM_SERVER_PORT, "/", false, true);
Pair<Integer, String> response = dmaapGet("myTopicNoData", timeout);
assertNotNull(response);
assertNotNull(response.first);
- assertEquals("No topic", response.second);
+ assertEquals(DMaaPSimulatorJaxRs.NO_TOPIC_MSG, response.second);
}
@Test
response = dmaapGet(topics[1], 1000);
assertNotNull(response);
assertNotNull(response.first);
- assertEquals("No topic", response.second);
+ assertEquals(DMaaPSimulatorJaxRs.NO_TOPIC_MSG, response.second);
response = dmaapPost(topics[1], data[1][0]);
assertNotNull(response);
response = dmaapGet(topics[0], 1000);
assertNotNull(response);
assertNotNull(response.first);
- assertEquals("No Data", response.second);
+ assertEquals(DMaaPSimulatorJaxRs.NO_DATA_MSG, response.second);
}
@Test
Code Review / policy / drools-pdp.git / commitdiff