package ui;

import model.asn1.exceptions.ParseException;
import org.junit.jupiter.api.Test;

import java.nio.charset.StandardCharsets;
import java.util.stream.IntStream;

import static org.junit.jupiter.api.Assertions.*;

public class UtilsTest {
    @Test
    void testByteToByte() {
        final byte[] primitive = new byte[]{1, 1, 4, 5, 1, 4};
        final Byte[] boxed = new Byte[]{1, 1, 4, 5, 1, 4};
        assertArrayEquals(primitive, Utils.byteToByte(boxed));
        assertArrayEquals(boxed, Utils.byteToByte(primitive));
    }

    @Test
    void testBytesToInt() throws ParseException {
        assertEquals(1, Utils.bytesToInt(new Byte[]{1}));
        assertEquals(257, Utils.bytesToInt(new Byte[]{1, 1}));
        assertThrows(ParseException.class, () -> Utils.bytesToInt(new Byte[]{1, -1, -1, -1, -1}));
    }

    @Test
    void testValToByte() {
        assertArrayEquals(new Byte[]{0},
                Utils.valToByte(0));
        assertArrayEquals(new Byte[]{1},
                Utils.valToByte(1));
        assertArrayEquals(new Byte[]{1, 1},
                Utils.valToByte(257));
        assertArrayEquals(new Byte[]{-1},
                Utils.valToByte(-1));
    }

    @Test
    void testParsePEM() throws ParseException {
        assertArrayEquals(IntStream.range(0, 48).mapToObj(i -> (byte) 1).toArray(Byte[]::new),
                Utils.parsePEM(Utils.byteToByte(("-----BEGIN BLABLA-----\n"
                                + "AQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEB\n"
                                + "-----END BLABLA-----").getBytes(StandardCharsets.UTF_8)),
                        "BLABLA"));
        assertThrows(ParseException.class, () -> {
            Utils.parsePEM(Utils.byteToByte(("-----BEGIN BLABLA-----\n"
                            + "AQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEB\n"
                            + "-----END BLABLA-----").getBytes(StandardCharsets.UTF_8)),
                    "LALA");
        });
        assertThrows(ParseException.class, () -> {
            Utils.parsePEM(Utils.byteToByte(("-----BEGIN BLABLA-----\n"
                            + "91023921083910298*@34908302130890123890\n"
                            + "-----END BLABLA-----").getBytes(StandardCharsets.UTF_8)),
                    "BLABLA");
        });
        assertThrows(ParseException.class, () -> {
            Utils.parsePEM(Utils.byteToByte(("-----BEGIN BLABLA-----"
                            + "AQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEB"
                            + "-----END BLABLA-----").getBytes(StandardCharsets.UTF_8)),
                    "BLABLA");
        });
    }

    @Test
    void testToPEM() {
        assertEquals("-----BEGIN ABC-----\n"
                        + "AQ==\n"
                        + "-----END ABC-----",
                Utils.toPEM(new Byte[]{0x1}, "ABC"));
    }
}