gerrit.fd Code Review - hc2vpp.git/blob

   1 package io.fd.honeycomb.lisp.translate.read.dump.executor;
   2
   3
   4 import static org.junit.Assert.assertEquals;
   5 import static org.junit.Assert.assertNotNull;
   6 import static org.junit.Assert.assertTrue;
   7 import static org.junit.Assert.fail;
   8 import static org.mockito.Mockito.any;
   9 import static org.mockito.Mockito.times;
  10 import static org.mockito.Mockito.verify;
  11
  12 import com.google.common.collect.ImmutableList;
  13 import io.fd.honeycomb.translate.util.read.cache.EntityDumpExecutor;
  14 import io.fd.honeycomb.translate.util.read.cache.exceptions.execution.DumpExecutionFailedException;
  15 import io.fd.honeycomb.translate.util.read.cache.exceptions.execution.i.DumpCallFailedException;
  16 import io.fd.honeycomb.translate.util.read.cache.exceptions.execution.i.DumpTimeoutException;
  17 import io.fd.honeycomb.vpp.test.read.JvppDumpExecutorTest;
  18 import java.nio.charset.StandardCharsets;
  19 import java.util.concurrent.TimeoutException;
  20 import org.junit.Before;
  21 import org.junit.Test;
  22 import org.mockito.ArgumentCaptor;
  23 import org.mockito.Captor;
  24 import io.fd.vpp.jvpp.core.dto.LispLocatorSetDetails;
  25 import io.fd.vpp.jvpp.core.dto.LispLocatorSetDetailsReplyDump;
  26 import io.fd.vpp.jvpp.core.dto.LispLocatorSetDump;
  27
  28
  29 public class LocatorSetsDumpExecutorTest extends JvppDumpExecutorTest<LocatorSetsDumpExecutor> {
  30
  31     public static final byte[] LOCATOR_SET_NAME_BYTES = "loc-set".getBytes(StandardCharsets.UTF_8);
  32
  33     private LispLocatorSetDetailsReplyDump validDump;
  34
  35     @Captor
  36     private ArgumentCaptor<LispLocatorSetDump> requestCaptor;
  37
  38     @Before
  39     public void init() {
  40         validDump = new LispLocatorSetDetailsReplyDump();
  41         LispLocatorSetDetails detail = new LispLocatorSetDetails();
  42         detail.lsIndex = 2;
  43         detail.lsName = LOCATOR_SET_NAME_BYTES;
  44         detail.context = 4;
  45
  46         validDump.lispLocatorSetDetails = ImmutableList.of(detail);
  47     }
  48
  49     @Test
  50     public void testExecuteDumpTimeout() throws Exception {
  51         doThrowTimeoutExceptionWhen().lispLocatorSetDump(any());
  52         try {
  53             getExecutor().executeDump(EntityDumpExecutor.NO_PARAMS);
  54         } catch (Exception e) {
  55             assertTrue(e instanceof DumpTimeoutException);
  56             assertTrue(e.getCause() instanceof TimeoutException);
  57             return;
  58         }
  59         fail("Test should have thrown exception");
  60     }
  61
  62     @Test(expected = DumpCallFailedException.class)
  63     public void testExecuteDumpHalted() throws DumpExecutionFailedException {
  64         doThrowFailExceptionWhen().lispLocatorSetDump(any());
  65         getExecutor().executeDump(EntityDumpExecutor.NO_PARAMS);
  66     }
  67
  68     @Test
  69     public void testExecuteDump() throws DumpExecutionFailedException {
  70         doReturnResponseWhen(validDump).lispLocatorSetDump(any());
  71
  72         final LispLocatorSetDetailsReplyDump replyDump = getExecutor().executeDump(EntityDumpExecutor.NO_PARAMS);
  73         verify(api, times(1)).lispLocatorSetDump(requestCaptor.capture());
  74
  75         final LispLocatorSetDump request = requestCaptor.getValue();
  76
  77         assertNotNull(request);
  78         assertEquals(1, request.filter);
  79
  80         assertNotNull(replyDump);
  81         assertNotNull(replyDump.lispLocatorSetDetails);
  82         assertEquals(1, replyDump.lispLocatorSetDetails.size());
  83         final LispLocatorSetDetails detail = replyDump.lispLocatorSetDetails.get(0);
  84
  85         assertNotNull(detail);
  86         assertEquals(4, detail.context);
  87         assertEquals(2, detail.lsIndex);
  88         assertEquals(LOCATOR_SET_NAME_BYTES, detail.lsName);
  89     }
  90
  91     @Override
  92     protected LocatorSetsDumpExecutor initExecutor() {
  93         return new LocatorSetsDumpExecutor(api);
  94     }
  95 }