opendaylight/adsal/hosttracker/shell/src/test/java/org/opendaylight/controller/hosttracker/shell/HostTrackerShellTest.java

   1 package org.opendaylight.controller.hosttracker.shell;
   2
   3 import java.io.ByteArrayOutputStream;
   4 import java.io.PrintStream;
   5 import java.util.ArrayList;
   6 import java.util.Arrays;
   7 import java.util.List;
   8
   9 import org.junit.Assert;
  10 import org.junit.Test;
  11 import static org.mockito.Mockito.mock;
  12 import static org.mockito.Mockito.when;
  13 import org.opendaylight.controller.hosttracker.IHostTrackerShell;
  14
  15 public class HostTrackerShellTest {
  16
  17     private final long COMMAND_TIMEOUT = 1000;
  18     private IHostTrackerShell hostTracker;
  19
  20     @Test
  21     public void testDumpPendingARPReqList() throws Exception {
  22         DumpPendingARPReqList dumpPendTest = new DumpPendingARPReqList();
  23         hostTracker = mock(IHostTrackerShell.class);
  24         List<String> failedList = new ArrayList<String>(Arrays.asList("a", "b", "c"));
  25         when(hostTracker.dumpPendingArpReqList()).thenReturn(failedList);
  26         dumpPendTest.setHostTracker(hostTracker);
  27         ByteArrayOutputStream baos = new ByteArrayOutputStream();
  28         System.setOut(new PrintStream(baos));
  29         dumpPendTest.doExecute();
  30         //Assert.assertTrue(true);
  31         Assert.assertEquals("[a, b, c]", baos.toString());
  32     }
  33
  34     @Test
  35     public void testDumpFailedARPReqList() throws Exception {
  36         DumpFailedARPReqList dumpFailTest = new DumpFailedARPReqList();
  37         hostTracker = mock(IHostTrackerShell.class);
  38         List<String> failedList = new ArrayList<String>(Arrays.asList("a", "b", "c"));
  39         when(hostTracker.dumpFailedArpReqList()).thenReturn(failedList);
  40         dumpFailTest.setHostTracker(hostTracker);
  41         ByteArrayOutputStream baos = new ByteArrayOutputStream();
  42         System.setOut(new PrintStream(baos));
  43         dumpFailTest.doExecute();
  44         //Assert.assertTrue(true);
  45         Assert.assertEquals("[a, b, c]", baos.toString());
  46     }
  47 }