jpexs-decompiler/libsrc/ffdec_lib/test/com/jpexs/helpers/JpegFixerTest.java

/*
 *  Copyright (C) 2010-2023 JPEXS, All rights reserved.
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3.0 of the License, or (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.
 */
package com.jpexs.helpers;

import static com.jpexs.helpers.JpegFixer.EOI;
import static com.jpexs.helpers.JpegFixer.SOI;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import org.testng.Assert;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

/**
 *
 * @author JPEXS
 */
public class JpegFixerTest {

    private JpegFixer fixer = new JpegFixer();

    @DataProvider(name = "bytes")
    public static Object[][] provideSamples() {
        return new Object[][]{
            {
                new byte[]{(byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI, (byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x21,
                    (byte) 0xFF, (byte) EOI},
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x21,
                    (byte) 0xFF, (byte) EOI},},
            {
                new byte[]{(byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI},
                new byte[]{(byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI},},
            {
                new byte[]{(byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x23
                },
                new byte[]{(byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x23
                },},
            {
                new byte[]{(byte) 0xFF, (byte) EOI},
                new byte[]{(byte) 0xFF, (byte) EOI}},
            {
                new byte[]{(byte) 0xFF},
                new byte[]{(byte) 0xFF}
            },
            {
                new byte[]{(byte) 0x26},
                new byte[]{(byte) 0x26}
            },
            {
                new byte[]{(byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x27},
                new byte[]{(byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x27}
            },
            {
                new byte[]{},
                new byte[]{}},
            {
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x29,
                     (byte) 0xFF, (byte) EOI},
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x29,
                     (byte) 0xFF, (byte) EOI}
            },
            {
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x2A,
                     (byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x3A,
                     (byte) 0xFF, (byte) EOI},
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x2A,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x3A,
                     (byte) 0xFF, (byte) EOI}
            },
            {
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x2B,
                     (byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x3B,
                     (byte) 0xFF, (byte) EOI},
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x2B,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x3B,
                     (byte) 0xFF, (byte) EOI},},
            {
                new byte[]{(byte) 0xFF, (byte) SOI, (byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x2C,
                     (byte) 0xFF, (byte) EOI},
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x03, 0x2C,
                     (byte) 0xFF, (byte) EOI}
            },
            {
                new byte[]{(byte) 0xFF, (byte) SOI, (byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x06, (byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI,
                     (byte) 0xFF, (byte) EOI},
                new byte[]{(byte) 0xFF, (byte) SOI,
                    (byte) 0xFF, (byte) JpegMarker.APP0, 0x00, 0x06, (byte) 0xFF, (byte) EOI, (byte) 0xFF, (byte) SOI,
                     (byte) 0xFF, (byte) EOI}
            }
        };
    }

    @Test(dataProvider = "bytes")
    public void testFixJpeg(byte[] inputData, byte[] expectedOutput) throws IOException {
        JpegFixer fixer = new JpegFixer();
        ByteArrayInputStream bais = new ByteArrayInputStream(inputData);
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        fixer.fixJpeg(bais, baos);
        byte[] actualOutput = baos.toByteArray();
        try {
            Assert.assertEquals(Helper.byteArrToString(actualOutput), Helper.byteArrToString(expectedOutput), "Bytes do not match");
        } catch (AssertionError er) {
            System.out.println("Bytes do not match:");
            System.out.println("INPUT: " + Helper.byteArrToString(inputData));
            System.out.println("ACTUAL OUTPUT: " + Helper.byteArrToString(actualOutput));
            System.out.println("EXPECTED OUTPUT: " + Helper.byteArrToString(expectedOutput));
            Assert.fail();
        }
    }
}