Add junit for more APPC client-lib classes

[appc.git] / appc-client / client-lib / src / test / java / org / onap / appc / client / impl / core / TaskQueueManagerTest.java

diff --git a/appc-client/client-lib/src/test/java/org/onap/appc/client/impl/core/TaskQueueManagerTest.java b/appc-client/client-lib/src/test/java/org/onap/appc/client/impl/core/TaskQueueManagerTest.java
new file mode 100644 (file)
index 0000000..72dac68
--- /dev/null
+++ b/appc-client/client-lib/src/test/java/org/onap/appc/client/impl/core/TaskQueueManagerTest.java
@@ -0,0 +1,75 @@
+/*
+ * ============LICENSE_START=======================================================
+ * ONAP : APPC
+ * ================================================================================
+ * Copyright 2018 AT&T
+ * =================================================================================
+ * 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.appc.client.impl.core;
+
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+import java.util.Properties;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
+
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+public class TaskQueueManagerTest {
+
+    private TaskQueueManager qm;
+    private CountDownLatch latch;
+    private static Properties props = new Properties();
+
+    @BeforeClass
+    public static void setUpBeforeClass() throws Exception {
+        props.setProperty("client.pool.size", "3");
+    }
+
+    @Before
+    public void setUp() throws Exception {
+        qm = new TaskQueueManager(props);
+    }
+
+    @Test
+    public void testSubmit() throws InterruptedException {
+        int count = 10;
+        latch = new CountDownLatch(count);
+        for (int i = 0; i < count; i++) {
+            qm.submit(i + "", new RunTask());
+        }
+        assertTrue(latch.await(3, TimeUnit.SECONDS));
+    }
+
+    @Test
+    public void testStopQueueManager() throws InterruptedException {
+        latch = new CountDownLatch(1);
+        qm.stopQueueManager();
+        qm.submit("1", new RunTask());
+        assertFalse(latch.await(1, TimeUnit.SECONDS));
+    }
+
+    private class RunTask implements Runnable {
+        @Override
+        public void run() {
+            latch.countDown();
+        }
+    }
+}