1 files changed, 145 insertions, 0 deletions
diff --git a/utils-test/src/test/java/org/onap/policy/common/utils/time/PseudoScheduledFutureTest.java b/utils-test/src/test/java/org/onap/policy/common/utils/time/PseudoScheduledFutureTest.java
new file mode 100644
index 00000000..e23bbd29
--- /dev/null
+++ b/utils-test/src/test/java/org/onap/policy/common/utils/time/PseudoScheduledFutureTest.java
@@ -0,0 +1,145 @@
+/*-
+ * ============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.common.utils.time;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertSame;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import java.util.concurrent.Delayed;
+import java.util.concurrent.TimeUnit;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+
+public class PseudoScheduledFutureTest {
+    private static final long DELAY_MS = 1000L;
+
+    private int count;
+
+    @Mock
+    private WorkItem work;
+
+    private PseudoScheduledFuture<Integer> future;
+
+    /**
+     * Sets up objects, including {@link #future}.
+     */
+    @Before
+    public void setUp() {
+        MockitoAnnotations.initMocks(this);
+
+        when(work.getDelay()).thenReturn(DELAY_MS);
+
+        count = 0;
+        future = new PseudoScheduledFuture<>(() -> ++count, true);
+        future.setWorkItem(work);
+    }
+
+    @Test
+    public void testRun() {
+        // verify with a periodic task - should execute twice
+        count = 0;
+        future.run();
+        future.run();
+        assertEquals(2, count);
+
+        // verify with an aperiodic task - should only execute once
+        future = new PseudoScheduledFuture<>(() -> ++count, false);
+        count = 0;
+        future.run();
+        future.run();
+        assertEquals(1, count);
+    }
+
+    @Test
+    public void testPseudoScheduledFutureRunnableTBoolean() throws Exception {
+        final Integer result = 100;
+        future = new PseudoScheduledFuture<>(() -> ++count, result, true);
+        assertTrue(future.isPeriodic());
+        future.run();
+        future.run();
+        assertEquals(2, count);
+
+        // verify with aperiodic constructor
+        future = new PseudoScheduledFuture<>(() -> ++count, result, false);
+        count = 0;
+        assertFalse(future.isPeriodic());
+        future.run();
+        future.run();
+        assertEquals(1, count);
+        assertEquals(result, future.get());
+    }
+
+    @Test
+    public void testPseudoScheduledFutureCallableOfTBoolean() throws Exception {
+        assertTrue(future.isPeriodic());
+        future.run();
+        future.run();
+        assertEquals(2, count);
+
+        // verify with aperiodic constructor
+        future = new PseudoScheduledFuture<>(() -> ++count, false);
+        count = 0;
+        assertFalse(future.isPeriodic());
+        future.run();
+        assertEquals(1, future.get().intValue());
+        future.run();
+        assertEquals(1, count);
+    }
+
+    @Test
+    public void testGetDelay() {
+        assertEquals(DELAY_MS, future.getDelay(TimeUnit.MILLISECONDS));
+        assertEquals(TimeUnit.MILLISECONDS.toSeconds(DELAY_MS), future.getDelay(TimeUnit.SECONDS));
+    }
+
+    @Test
+    public void testCompareTo() {
+        Delayed delayed = mock(Delayed.class);
+        when(delayed.getDelay(TimeUnit.MILLISECONDS)).thenReturn(DELAY_MS + 1);
+
+        assertTrue(future.compareTo(delayed) < 0);
+    }
+
+    @Test
+    public void testIsPeriodic() {
+        assertTrue(future.isPeriodic());
+        assertFalse(new PseudoScheduledFuture<>(() -> ++count, false).isPeriodic());
+    }
+
+    @Test
+    public void testGetWorkItem() {
+        assertSame(work, future.getWorkItem());
+    }
+
+    @Test
+    public void testSetWorkItem() {
+        work = mock(WorkItem.class);
+        future.setWorkItem(work);
+        assertSame(work, future.getWorkItem());
+    }
+
+}