package com.atlassian.plugins.rest.common.expand.resolver;
   
   import com.atlassian.plugins.rest.common.expand.EntityCrawler;
   import com.atlassian.plugins.rest.common.expand.EntityExpander;
   import com.atlassian.plugins.rest.common.expand.ExpandContext;
   import com.atlassian.plugins.rest.common.expand.Expander;
   
   import static org.junit.Assert.*;
   
  import org.junit.Before;
  import org.junit.Test;
  
  import static org.mockito.Mockito.mock;
  
  /**
   * Testing {@link AbstractAnnotationEntityExpanderResolver}
   */
  public class AbstractAnnotationEntityExpanderResolverTest {
      private AbstractAnnotationEntityExpanderResolver resolver;
      private EntityExpander entityExpander;
  
      @Before
      public void setUp() throws Exception {
          entityExpander = mock(EntityExpander.class);
  
          resolver = new AbstractAnnotationEntityExpanderResolver() {
              protected EntityExpander<?> getEntityExpander(Expander e) {
                  if (e.value().equals(AnnotatedClassEntityExpander.class)) {
                      return entityExpander;
                  }
                  fail("Should not have been called with a different expander");
                  return null; // just for compilation
              }
          };
      }
  
      @Test
      public void testHasExpanderWithNullClass() {
          try {
              final Class clazz = null;
              resolver.hasExpander(clazz);
              fail();
          } catch (NullPointerException e) {
              // expected
          }
      }
  
      @Test
      public void testHasExpanderWithAnnotatedClass() {
          assertTrue(resolver.hasExpander(AnnotatedClass.class));
      }
  
      @Test
      public void testHasExpanderWithNonAnnotatedClass() {
          assertFalse(resolver.hasExpander(Object.class));
      }
  
      @Test
      public void testGetExpanderWithAnnotatedClass() {
          assertEquals(entityExpander, resolver.getExpander(AnnotatedClass.class));
      }
  
      @Test
      public void testGetExpanderWithNonAnnotatedClass() {
          assertNull(resolver.getExpander(Object.class));
      }
  
      @Expander(AnnotatedClassEntityExpander.class)
      private static class AnnotatedClass {
      }
  
      private static class AnnotatedClassEntityExpander implements EntityExpander<AnnotatedClass> {
          public AnnotatedClass expand(ExpandContext expandContext, EntityExpanderResolver expanderResolver, EntityCrawler entityCrawler) {
              return null;
          }
      }
  }