aboutsummaryrefslogtreecommitdiffstats
path: root/services/services-engine
diff options
context:
space:
mode:
authorhuaxing <huaxing.jin@est.tech>2020-06-10 14:58:34 +0800
committerhuaxing <huaxing.jin@est.tech>2020-06-12 09:59:42 +0800
commit718a1611a1e580ed7874088e1a0e013416334989 (patch)
tree605f35ec8940bb986deaf7b96a76b97709ba7b76 /services/services-engine
parent950622063ecc737ef7106ed9c380a077441b76aa (diff)
Improve robustness of unit testing
Issue-ID: POLICY-2630 Signed-off-by: huaxing <huaxing.jin@est.tech> Change-Id: I6475f9272c1a770836af537c13b23e486b66ac3e
Diffstat (limited to 'services/services-engine')
-rw-r--r--services/services-engine/src/main/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListener.java4
-rw-r--r--services/services-engine/src/test/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListenerTest.java43
2 files changed, 25 insertions, 22 deletions
diff --git a/services/services-engine/src/main/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListener.java b/services/services-engine/src/main/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListener.java
index 2e8218a99..57c547b8c 100644
--- a/services/services-engine/src/main/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListener.java
+++ b/services/services-engine/src/main/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListener.java
@@ -77,7 +77,7 @@ public class EngDepMessageListener implements MessageListener<Message>, Runnable
// The message listener thread and stopping flag
private Thread messageListenerThread;
- private boolean stopOrderedFlag = false;
+ private volatile boolean stopOrderedFlag = false;
// The message queue is used to hold messages prior to forwarding to Apex
private final BlockingQueue<MessageBlock<Message>> messageQueue = new LinkedBlockingDeque<>();
@@ -149,7 +149,7 @@ public class EngDepMessageListener implements MessageListener<Message>, Runnable
@Override
public void run() {
// Take messages off the queue and forward them to the Apex engine
- while (messageListenerThread.isAlive() && !stopOrderedFlag) {
+ while (!stopOrderedFlag) {
pollAndHandleMessage();
}
}
diff --git a/services/services-engine/src/test/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListenerTest.java b/services/services-engine/src/test/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListenerTest.java
index f755a669c..9f5329e55 100644
--- a/services/services-engine/src/test/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListenerTest.java
+++ b/services/services-engine/src/test/java/org/onap/policy/apex/service/engine/engdep/EngDepMessageListenerTest.java
@@ -5,36 +5,37 @@
* 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.
- *
+ *
* SPDX-License-Identifier: Apache-2.0
* ============LICENSE_END=========================================================
*/
package org.onap.policy.apex.service.engine.engdep;
+import static org.awaitility.Awaitility.await;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.List;
-
+import java.util.concurrent.BlockingQueue;
import org.java_websocket.WebSocket;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
+import org.mockito.internal.util.reflection.Whitebox;
import org.onap.policy.apex.core.infrastructure.messaging.impl.ws.messageblock.MessageBlock;
-import org.onap.policy.apex.core.infrastructure.threading.ThreadUtilities;
import org.onap.policy.apex.core.protocols.Message;
import org.onap.policy.apex.core.protocols.engdep.messages.GetEngineInfo;
import org.onap.policy.apex.core.protocols.engdep.messages.GetEngineServiceInfo;
@@ -57,7 +58,7 @@ public class EngDepMessageListenerTest {
/**
* Set up mocking of the engine service facade.
- *
+ *
* @throws ApexException on engine service facade setup errors
*/
@Before
@@ -72,6 +73,8 @@ public class EngDepMessageListenerTest {
public void testMessageListener() throws ApexException {
DummyEngineService dummyEngineService = new DummyEngineService();
EngDepMessageListener listener = new EngDepMessageListener(dummyEngineService);
+ BlockingQueue<?> messageQueue
+ = (BlockingQueue<?>) Whitebox.getInternalState(listener, "messageQueue");
listener.startProcessorThread();
try {
@@ -84,49 +87,49 @@ public class EngDepMessageListenerTest {
List<Message> messageList = new ArrayList<>();
messageList.add(new StartEngine(new AxArtifactKey("Start:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals("Start:0.0.1", dummyEngineService.getStartEngineKey().getId());
messageList.clear();
messageList.add(new StopEngine(new AxArtifactKey("Stop:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals("Stop:0.0.1", dummyEngineService.getStopEngineKey().getId());
messageList.clear();
messageList.add(new StartPeriodicEvents(new AxArtifactKey("StartPeriodic:0.0.1"), "12345"));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals(12345, dummyEngineService.getPeriodicPeriod());
messageList.clear();
messageList.add(new StopPeriodicEvents(new AxArtifactKey("StopPeriodic:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals(0, dummyEngineService.getPeriodicPeriod());
messageList.clear();
messageList.add(new GetEngineInfo(new AxArtifactKey("EngineInfo:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals("EngineInfo:0.0.1", dummyEngineService.getRuntimeInfoKey().getId());
messageList.clear();
messageList.add(new GetEngineStatus(new AxArtifactKey("EngineStatus:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals("EngineStatus:0.0.1", dummyEngineService.getStatusKey().getId());
messageList.clear();
messageList.add(new GetEngineServiceInfo(new AxArtifactKey("EngineServiceInfo:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals(1, dummyEngineService.getModelKeyGetCalled());
messageList.clear();
messageList.add(new UpdateModel(new AxArtifactKey("UpdateModel:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals("UpdateModel:0.0.1", dummyEngineService.getUpdateModelKey().getId());
try {
@@ -134,35 +137,35 @@ public class EngDepMessageListenerTest {
messageList.add(new Response(new AxArtifactKey("UpdateModel:0.0.1"), false,
new GetEngineInfo(new AxArtifactKey("EngineInfo:0.0.1"))));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
assertEquals("UpdateModel:0.0.1", dummyEngineService.getUpdateModelKey().getId());
messageList.clear();
Message badMessage0 = new DummyMessage(null, null);
messageList.add(badMessage0);
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
messageList.clear();
Message badMessage1 = new DummyMessage(new DummyAction(null), null);
messageList.add(badMessage1);
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
messageList.clear();
Message badMessage2 = new DummyMessage(new DummyAction("throw exception"), null);
messageList.add(badMessage2);
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
+ messageList.clear();
Mockito.doReturn(false).when(webSocketMock).isOpen();
messageList.add(new StartEngine(new AxArtifactKey("Start:0.0.1")));
listener.onMessage(new MessageBlock<>(messageList, webSocketMock));
- ThreadUtilities.sleep(50);
+ await().until(messageQueue::isEmpty);
} catch (Exception e) {
fail("test should not throw exceptions on bad messages");
}
listener.stopProcessorThreads();
- ThreadUtilities.sleep(50);
}
}