package com.atlassian.plugin.loaders.classloading;
   
   import java.io.File;
   import java.util.Arrays;
   
   import org.junit.Test;
   
   import static org.junit.Assert.assertEquals;
   import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.verify;
  import static org.mockito.Mockito.when;
  
  public class TestForwardingScanner
  {
      @Test
      public void allMethodsForwardToDelegate()
      {
          DeploymentUnit unitA = new DeploymentUnit(new File("unitA"));
          DeploymentUnit unitB = new DeploymentUnit(new File("unitB"));
          Scanner mockScanner = mock(Scanner.class);
          when(mockScanner.scan()).thenReturn(Arrays.asList(unitA));
          when(mockScanner.getDeploymentUnits()).thenReturn(Arrays.asList(unitA, unitB));
  
          Scanner forwardingScanner = new ForwardingScanner(mockScanner);
          assertEquals(forwardingScanner.scan(), Arrays.asList(unitA));
          verify(mockScanner).scan();
          assertEquals(forwardingScanner.getDeploymentUnits(), Arrays.asList(unitA, unitB));
          verify(mockScanner).getDeploymentUnits();
          forwardingScanner.reset();
          verify(mockScanner).reset();
          forwardingScanner.remove(unitA);
          verify(mockScanner).remove(unitA);
      }
  }