package com.atlassian.core.util;
   
   import org.apache.commons.io.IOUtils;
   
   import java.io.InputStream;
   
   import junit.framework.TestCase;
   
   public class TestImageInfo extends TestCase
  {
      public void testCheckJPeg() throws Exception
      {
          assertTrue(checkImage("/image.jpg"));
      }
  
      public void testCheckPng() throws Exception
      {
          assertTrue(checkImage("/image.png"));
      }
  
      public void testCheckGif() throws Exception
      {
          assertTrue(checkImage("/image.gif"));
      }
  
      public void testCheckJPegWithInvalidData() throws Exception
      {
          assertFalse(checkImage("/image_invalid.jpg"));
      }
  
      public void testAnInvalidPpmFileDoesNotLeadToEndlessLoops() throws Exception
      {
          assertFalse(checkImage("/image_invalid.ppm"));
      }
  
      public void testPngWithCorruptedMetaData() throws Exception
      {
          // image that was causing OOM in PNGImageReader
          assertTrue(checkImage("/image_corrupted.png"));
      }
  
      public void testPngWithJiraTag() throws Exception
      {
          assertTrue(checkImage("/image_tagged.png", "jira-system-image-type"));
          assertFalse(checkImage("/image_tagged.png", "jira-system-image-typex"));
          assertFalse(checkImage("/image_tagged.png", "jira-system-image-typ"));
          assertFalse(checkImage("/image_without_comment.png", "jira-system-image-type"));
      }
  
      public void testTruncatedPng() throws Exception
      {
          //No IEND chunk, image truncated just after data
          assertFalse(checkImage("/image_truncated.png", true));
      }
  
      private boolean checkImage(String filename)
      {
          return checkImage(filename, null);
      }
  
      private boolean checkImage(String filename, boolean validatePngMetadata)
      {
          return checkImage(filename, null, validatePngMetadata);
      }
  
      private boolean checkImage(String filename, String expectedKeyword)
      {
          return checkImage(filename, expectedKeyword, false);
      }
  
      private boolean checkImage(String filename, String expectedKeyword, boolean validatePngMetadata)
      {
          ImageInfo imageInfo = new ImageInfo();
          if (validatePngMetadata)
          {
              imageInfo.setValidatePngMetadata();
          }
          InputStream stream = this.getClass().getResourceAsStream(filename);
          try
          {
              if(expectedKeyword != null)
              {
                  imageInfo.setExpectedPngTextKeyword(expectedKeyword);
              }
              imageInfo.setInput(stream);
              return imageInfo.check();
          }
          finally
          {
              IOUtils.closeQuietly(stream);
          }
      }
  
  }