summaryrefslogtreecommitdiffstats
path: root/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/KeepaliveReaderWrapperTest.java
diff options
context:
space:
mode:
Diffstat (limited to 'infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/KeepaliveReaderWrapperTest.java')
-rw-r--r--infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/KeepaliveReaderWrapperTest.java108
1 files changed, 108 insertions, 0 deletions
diff --git a/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/KeepaliveReaderWrapperTest.java b/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/KeepaliveReaderWrapperTest.java
new file mode 100644
index 000000000..ca2872139
--- /dev/null
+++ b/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/KeepaliveReaderWrapperTest.java
@@ -0,0 +1,108 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * 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.
+ */
+
+package io.fd.honeycomb.translate.util.read;
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import io.fd.honeycomb.translate.read.ReadContext;
+import io.fd.honeycomb.translate.read.Reader;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.Executors;
+import java.util.concurrent.Future;
+import java.util.concurrent.ScheduledExecutorService;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.opendaylight.yangtools.concepts.Builder;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+public class KeepaliveReaderWrapperTest {
+
+ @Mock
+ private ReadContext ctx;
+ @Mock
+ private Reader<DataObject, Builder<DataObject>> delegate;
+ @Mock
+ private Builder<DataObject> builder;
+
+ private InstanceIdentifier<DataObject> id = InstanceIdentifier.create(DataObject.class);
+ private ScheduledExecutorService exec;
+
+ @Before
+ public void setUp() throws Exception {
+ MockitoAnnotations.initMocks(this);
+ exec = Executors.newScheduledThreadPool(1);
+ when(delegate.read(any(InstanceIdentifier.class), any(ReadContext.class))).thenThrow(TestingException.class);
+ }
+
+ @After
+ public void tearDown() throws Exception {
+ exec.shutdownNow();
+ }
+
+ @Test(timeout = 10000)
+ public void testKeepalive() throws Exception {
+ final CapturingFailListener listener = new CapturingFailListener();
+
+ final KeepaliveReaderWrapper<DataObject, Builder<DataObject>> keepaliveWrapper =
+ new KeepaliveReaderWrapper<>(delegate, exec, TestingException.class, 1, listener);
+
+ keepaliveWrapper.readCurrentAttributes(id, builder, ctx);
+ verify(delegate).readCurrentAttributes(id, builder, ctx);
+
+ keepaliveWrapper.getBuilder(id);
+ verify(delegate).getBuilder(id);
+
+
+ assertTrue(listener.getTriggerFuture().get());
+ verify(delegate).read(any(InstanceIdentifier.class), any(ReadContext.class));
+
+ keepaliveWrapper.close();
+ }
+
+ @Test(timeout = 10000)
+ public void testKeepaliveCancel() throws Exception {
+ final CapturingFailListener listener = new CapturingFailListener();
+
+ final KeepaliveReaderWrapper<DataObject, Builder<DataObject>> keepaliveWrapper =
+ new KeepaliveReaderWrapper<>(delegate, exec, TestingException.class, 100000, listener);
+ keepaliveWrapper.close();
+ assertFalse(listener.getTriggerFuture().isDone());
+ }
+
+ private static final class TestingException extends RuntimeException {}
+
+ private static class CapturingFailListener implements KeepaliveReaderWrapper.KeepaliveFailureListener {
+ private CompletableFuture<Boolean> booleanCompletableFuture = new CompletableFuture<>();
+
+ @Override
+ public synchronized void onKeepaliveFailure() {
+ booleanCompletableFuture.complete(true);
+ }
+
+ synchronized Future<Boolean> getTriggerFuture() {
+ return booleanCompletableFuture;
+ }
+ }
+} \ No newline at end of file