package com.atlassian.plugins.rest.common.expand;
   
   import com.atlassian.plugins.rest.common.expand.parameter.DefaultExpandParameter;
   import com.atlassian.plugins.rest.common.expand.parameter.ExpandParameter;
   import com.atlassian.plugins.rest.common.expand.resolver.EntityExpanderResolver;
   import com.google.common.collect.Lists;
   import com.google.common.collect.Sets;
   
   import static org.hamcrest.Matchers.contains;
  import static org.junit.Assert.*;
  
  import org.junit.Before;
  import org.junit.Test;
  
  import javax.xml.bind.annotation.XmlAttribute;
  import javax.xml.bind.annotation.XmlElement;
  import java.lang.reflect.Field;
  import java.util.Arrays;
  import java.util.Collections;
  import java.util.List;
  import java.util.Set;
  
  /**
   * Testing {@link EntityCrawler}
   */
  public class EntityCrawlerTest {
      private EntityCrawler entityCrawler;
  
      @Before
      public void setUp() {
          entityCrawler = new EntityCrawler(Collections.singleton(new SomeExpandsProvider()));
      }
  
      @Test
      public void testGetExpandableWithNullField() {
          assertNull(entityCrawler.getExpandable(null));
      }
  
      @Test
      public void testGetExpandableWithNoExpandableAnnotation() throws Exception {
          final Field field = SomeClass.class.getField("field1");
          assertNull(entityCrawler.getExpandable(field));
      }
  
      @Test
      public void testGetExpandableWithValuedExpandable() throws Exception {
          final Field field = SomeClass.class.getField("field2");
          assertEquals("field2Value", entityCrawler.getExpandable(field).value());
      }
  
      @Test
      public void testGetExpandableWithNamedXmlElement() throws Exception {
          final Field field = SomeClass.class.getField("field3");
          assertEquals("field3Value", entityCrawler.getExpandable(field).value());
      }
  
      @Test
      public void testGetExpandableWithUnNamedXmlElement() throws Exception {
          final Field field = SomeClass.class.getField("field4");
          assertEquals("field4", entityCrawler.getExpandable(field).value());
      }
  
      @Test
      public void testGetExpandableWithNoXmlElement() throws Exception {
          final Field field = SomeClass.class.getField("field5");
          assertEquals("field5", entityCrawler.getExpandable(field).value());
      }
  
      @Test
      public void testGetPossibleExpands() throws Exception {
          SomeDerivedClass object = new SomeDerivedClass();
          entityCrawler.crawl(object, new DefaultExpandParameter(Collections.emptyList()), null);
          List<String> newExpands = Arrays.asList(object.expand.split(","));
  
          assertThat(newExpands, contains("field2Value","field3Value","field4","field5","someAdditionalExpand"));
      }
  
      @Test
      public void testCrawlGetsFieldsFromSuperClass() throws Exception {
          final Set<String> expectedFields = Sets.newHashSet("field2Value", "field3Value", "field4", "field5");
  
          entityCrawler.crawl(new SomeDerivedClass(), new DefaultExpandParameter(Lists.newArrayList("*")),
                  new EntityExpanderResolver() {
                      public boolean hasExpander(Class<?> type) {
                          return true;
                      }
  
                      public <T> EntityExpander<T> getExpander(Class<? extends T> type) {
                          return new EntityExpander<T>() {
                              public T expand(ExpandContext<T> tExpandContext, EntityExpanderResolver expanderResolver, EntityCrawler entityCrawler) {
                                  assertTrue(expectedFields.remove(tExpandContext.getExpandable().value()));
                                  return null;
                              }
                          };
                      }
                  });
  
          assertTrue(expectedFields.isEmpty());
      }
 
     private static class SomeClass {
         public Object field1 = new Object();
 
         @Expandable("field2Value")
         public Object field2 = new Object();
 
         @Expandable
         @XmlElement(name = "field3Value")
         public Object field3 = new Object();
 
         @Expandable
         @XmlElement
         public Object field4 = new Object();
 
         @Expandable
         public Object field5 = new Object();
 
         @Expandable
         public Object ignoredField;
     }
 
     private static class SomeDerivedClass extends SomeClass {
         @XmlAttribute
         private String expand;
     }
 
     private static class SomeExpandsProvider implements AdditionalExpandsProvider<SomeDerivedClass> {
         @Override
         public List<String> getAdditionalExpands(SomeDerivedClass entity) {
             return Collections.singletonList("someAdditionalExpand");
         }
 
         @Override
         public Class<SomeDerivedClass> getSupportedType() {
             return SomeDerivedClass.class;
         }
     }
 }