package com.atlassian.plugins.rest.module.jersey;
   
   import static org.junit.Assert.assertEquals;
   import static org.mockito.Matchers.anyObject;
   import static org.mockito.Matchers.eq;
   import static org.mockito.Mockito.when;
   
   import java.lang.annotation.Annotation;
   import java.nio.charset.Charset;
  
  import javax.validation.constraints.NotNull;
  import javax.ws.rs.core.MediaType;
  import javax.ws.rs.ext.ContextResolver;
  import javax.ws.rs.ext.MessageBodyWriter;
  import javax.ws.rs.ext.Providers;
  import javax.xml.bind.annotation.XmlRootElement;
  import javax.xml.parsers.SAXParserFactory;
  
  import org.junit.Test;
  import org.junit.runner.RunWith;
  import org.mockito.Mock;
  import org.mockito.runners.MockitoJUnitRunner;
  
  import com.atlassian.plugins.rest.module.ResourceConfigManager;
  import com.sun.jersey.core.impl.provider.entity.XMLRootElementProvider.App;
  import com.sun.jersey.core.spi.factory.MessageBodyFactory;
  import com.sun.jersey.spi.inject.Injectable;
  
  @RunWith(value = MockitoJUnitRunner.class)
  public class JerseyEntityHandlerTest {
  
      @Mock
      MessageBodyFactory messageBodyFactory;
      @Mock
      ResourceConfigManager resourceConfigManager;
      @Mock
      Injectable<SAXParserFactory> injectableSAXParserFactory;
      @Mock
      Providers providers;
      @Mock
      ContextResolver resolver;
  
      /**
       * @throws Exception
       * @see REST-172
       */
      @Test
      public void testMarshall() throws Exception {
          MessageBodyWriter writer = new App(injectableSAXParserFactory, providers);
  
          when(messageBodyFactory.getMessageBodyWriter(eq(AnEntity.class), eq(AnEntity.class),
                  (Annotation[]) anyObject(), eq(MediaType.APPLICATION_XML_TYPE)))
                  .thenReturn(writer);
  
          JerseyEntityHandler handler = new JerseyEntityHandler(messageBodyFactory, resourceConfigManager);
  
          final String text = "Some \n\ntext";
  
          AnEntity entity = new AnEntity();
          entity.setText(text);
  
          String marshalledEntity = handler.marshall(entity, MediaType.APPLICATION_XML_TYPE, Charset.forName("UTF-8"));
  
          assertEquals("<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?><anEntity><text>"
                          + text + "</text></anEntity>",
                  marshalledEntity);
      }
  
  
      @XmlRootElement
      public static class AnEntity {
          @NotNull
          private String text;
  
          public String getText() {
              return text;
          }
  
          public void setText(String text) {
              this.text = text;
          }
      }
  }