package com.drew.imaging.png;

import com.drew.lang.KeyValuePair;
import com.drew.lang.annotations.NotNull;
import com.drew.metadata.Metadata;
import com.drew.metadata.png.PngDirectory;
import org.junit.Test;

import java.io.FileInputStream;
import java.io.IOException;
import java.util.Collection;
import java.util.List;
import java.util.TimeZone;

import static org.junit.Assert.*;

/**
 * @author Drew Noakes https://drewnoakes.com
 */
public class PngMetadataReaderTest
{
    @NotNull
    private static Metadata processFile(@NotNull String filePath) throws PngProcessingException, IOException
    {
        FileInputStream inputStream = null;
        try {
            inputStream = new FileInputStream(filePath);
            return PngMetadataReader.readMetadata(inputStream);
        } finally {
            if (inputStream != null) {
                inputStream.close();
            }
        }
    }

    @Test
    public void testGimpGreyscaleWithManyChunks() throws Exception
    {
        TimeZone timeZone = TimeZone.getDefault();

        try {
            TimeZone.setDefault(TimeZone.getTimeZone("GMT"));

            Metadata metadata = processFile("Tests/Data/gimp-8x12-greyscale-alpha-time-background.png");
            Collection<PngDirectory> directories = metadata.getDirectoriesOfType(PngDirectory.class);

            assertNotNull(directories);
            assertEquals(5, directories.size());

            PngDirectory[] dirs = new PngDirectory[directories.size()];
            directories.toArray(dirs);

            assertEquals(PngChunkType.IHDR, dirs[0].getPngChunkType());
            assertEquals(8, dirs[0].getInt(PngDirectory.TAG_IMAGE_WIDTH));
            assertEquals(12, dirs[0].getInt(PngDirectory.TAG_IMAGE_HEIGHT));
            assertEquals(8, dirs[0].getInt(PngDirectory.TAG_BITS_PER_SAMPLE));
            assertEquals(4, dirs[0].getInt(PngDirectory.TAG_COLOR_TYPE));
            assertEquals(0, dirs[0].getInt(PngDirectory.TAG_COMPRESSION_TYPE));
            assertEquals(0, dirs[0].getInt(PngDirectory.TAG_FILTER_METHOD));
            assertEquals(0, dirs[0].getInt(PngDirectory.TAG_INTERLACE_METHOD));

            assertEquals(PngChunkType.gAMA, dirs[1].getPngChunkType());
            assertEquals(0.45455, dirs[1].getDouble(PngDirectory.TAG_GAMMA), 0.00001);

            assertEquals(PngChunkType.bKGD, dirs[2].getPngChunkType());
            assertArrayEquals(new byte[]{0, 52}, dirs[2].getByteArray(PngDirectory.TAG_BACKGROUND_COLOR));

            //noinspection ConstantConditions
            assertEquals(PngChunkType.tIME, dirs[3].getPngChunkType());
            assertEquals("Tue Jan 01 04:08:30 GMT 2013", dirs[3].getDate(PngDirectory.TAG_LAST_MODIFICATION_TIME).toString());

            assertEquals(PngChunkType.iTXt, dirs[4].getPngChunkType());
            @SuppressWarnings("unchecked")
            List<KeyValuePair> pairs = (List<KeyValuePair>)dirs[4].getObject(PngDirectory.TAG_TEXTUAL_DATA);
            assertNotNull(pairs);
            assertEquals(1, pairs.size());
            assertEquals("Comment", pairs.get(0).getKey());
            assertEquals("Created with GIMP", pairs.get(0).getValue());
        } finally {
            TimeZone.setDefault(timeZone);
        }
    }
}

@@ -1388,17 +1388,17 @@
 tEquals(
-5
+6
 , direct
@@ -2526,16 +2526,335 @@
 ditions%0A
+            assertEquals(PngChunkType.pHYs, dirs%5B3%5D.getPngChunkType());%0A            assertEquals(1, dirs%5B3%5D.getInt(PngDirectory.TAG_UNIT_SPECIFIER));%0A            assertEquals(2835, dirs%5B3%5D.getInt(PngDirectory.TAG_PIXELS_PER_UNIT_X));%0A            assertEquals(2835, dirs%5B3%5D.getInt(PngDirectory.TAG_PIXELS_PER_UNIT_Y));%0A%0A
         
@@ -2890,25 +2890,25 @@
 .tIME, dirs%5B
-3
+4
 %5D.getPngChun
@@ -2975,25 +2975,25 @@
 2013%22, dirs%5B
-3
+4
 %5D.getDate(Pn
@@ -3090,25 +3090,25 @@
 .iTXt, dirs%5B
-4
+5
 %5D.getPngChun
@@ -3220,25 +3220,25 @@
 ePair%3E)dirs%5B
-4
+5
 %5D.getObject(

Fix broken unit test.