package model.asn1;

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

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

public class ObjectIdentifierTest {
    @Test
    void testConstructor() {
        assertArrayEquals(new Integer[]{ 1, 3, 6, 1, 4},
                new ObjectIdentifier(ObjectIdentifier.TAG, null, new Integer[]{ 1, 3, 6, 1, 4 }).getInts());
        assertArrayEquals(new Integer[]{ 1, 2, 3, 4, 5},
                new ObjectIdentifier(ObjectIdentifier.TAG, null, new Integer[]{ 1, 2, 3, 4, 5 }).getInts());
    }

    @Test
    void testToString() {
        assertEquals("1.3.6.1.4",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, new Integer[]{ 1, 3, 6, 1, 4 }).toString());
        assertEquals("1.2.3.4.5.6",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, new Integer[]{ 1, 2, 3, 4, 5, 6 }).toString());
        assertEquals("CN",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_CN).toString());
        assertEquals("SN",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_SN).toString());
        assertEquals("C",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_C).toString());
        assertEquals("L",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_L).toString());
        assertEquals("O",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_O).toString());
        assertEquals("OU",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_OU).toString());
        assertEquals("DC",
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_DC).toString());
    }

    @Test
    void testEncode() {
        assertArrayEquals(new Byte[]{ 0x55, 0x04, 0x0A },
                new ObjectIdentifier(ObjectIdentifier.TAG, null, ObjectIdentifier.OID_O).encodeValueDER());
        assertArrayEquals(new Byte[]{ 0x67, -127, 0x0C, 0x01, 0x02, 0x01 },
                new ObjectIdentifier(ObjectIdentifier.TAG, null,
                        new Integer[]{ 2, 23, 140, 1, 2, 1 }).encodeValueDER());
        assertArrayEquals(new Byte[]{ 0x2B, 0x06, 0x01, 0x04, 0x01, -126, -33, 0x13, 0x01, 0x01, 0x01 },
                new ObjectIdentifier(ObjectIdentifier.TAG, null,
                        new Integer[]{ 1, 3, 6, 1, 4, 1, 44947, 1, 1, 1 }).encodeValueDER());
        assertArrayEquals(new Byte[]{ 0x2A, -122, 0x48, -50, 0x3D, 0x02, 0x01 },
                new ObjectIdentifier(ObjectIdentifier.TAG, null,
                        new Integer[]{ 1, 2, 840, 10045, 2, 1 }).encodeValueDER());
        assertArrayEquals(new Byte[]{ 0x2A, -122, 0x48, -122, -9, 0x0D, 0x01, 0x01, 0x0B },
                new ObjectIdentifier(ObjectIdentifier.TAG, null,
                        ObjectIdentifier.OID_SHA256_WITH_RSA_ENCRYPTION).encodeValueDER());

    }

    @Test
    void testParse() throws ParseException {
        assertArrayEquals(ObjectIdentifier.OID_SHA256_WITH_RSA_ENCRYPTION,
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x6, 0x9, 0x2A, -122, 0x48, -122, -9, 0x0D,
                        0x01, 0x01, 0x0B }),false).getInts());
        assertArrayEquals(new Integer[]{ 1, 2, 840, 10045, 2, 1 },
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x6, 7, 0x2A, -122, 0x48, -50, 0x3D, 0x02, 0x01 }),
                        false).getInts());
        assertArrayEquals(new Integer[]{ 0, 2, 840, 10045, 2, 1 },
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x6, 7, 2, -122, 0x48, -50, 0x3D, 0x02, 0x01 }),
                        false).getInts());
        assertArrayEquals(new Integer[]{ 2, 2, 840, 10045, 2, 1 },
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x6, 7, 82, -122, 0x48, -50, 0x3D, 0x02, 0x01 }),
                        false).getInts());
        assertArrayEquals(new Integer[]{ 2, 42, 840, 10045, 2, 1 },
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x6, 7, 122, -122, 0x48, -50, 0x3D, 0x02, 0x01 }),
                        false).getInts());
        assertArrayEquals(new Integer[]{ 1, 2, 840, 113549, 1, 9, 14 },
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x06, 0x09, 0x2A, -122, 0x48, -122, -9, 0x0D, 0x01,
                        0x09, 0x0E }), false).getInts());
    }

    @Test
    void testParseFail() {
        assertThrows(ParseException.class, () ->
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x6, 0x0 }), false));
        assertThrows(ParseException.class, () ->
                new ObjectIdentifier(new BytesReader(new Byte[]{ 0x6, 0x9, 0x2A, -122, 0x48, -122, -9, 0x0D,
                        0x01, 0x01, -117 }), false));
    }
}