package model.asn1;

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

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

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

public class IA5StringTest {
    private String stringToTest = "";

    @BeforeEach
    void setup() {
        stringToTest = IntStream.range(0, 127)
                .mapToObj(Character::toString)
                .collect(Collectors.joining());
    }

    @Test
    void testAcceptedString() throws ParseException {
        assertEquals(stringToTest,
                new IA5String(IA5String.TAG, null, stringToTest)
                        .getString());
    }

    @Test
    void testIllegalStrings() {
        assertThrows(ParseException.class,
                () -> new IA5String(IA5String.TAG, null,
                        stringToTest + new String(new byte[]{ -128 }, StandardCharsets.UTF_8)));
        assertThrows(ParseException.class,
                () -> new IA5String(IA5String.TAG, null,
                        stringToTest + new String(new byte[]{ -1 }, StandardCharsets.UTF_8)));
    }

    @Test
    void testEncode() throws ParseException {
        assertArrayEquals(
                new Byte[] { 0x00, 0x01, 0x02 },
                new IA5String(IA5String.TAG, null, new String(new byte[]{ 0, 1, 2}, StandardCharsets.UTF_8))
                        .encodeValueDER());
        assertArrayEquals(
                new Byte[] {
                        0x16, 0x02, // Tag - Length
                        0x68, 0x69 // Value
                }, new IA5String(IA5String.TAG, null, "hi").encodeDER());
        assertArrayEquals(
                new Byte[] {
                        -85, 0x05, // Parent Tag - Length
                        0x16, 0x03, // Inner Tag - Length
                        0x68, 0x69, 0x69 // Value
                }, new IA5String(IA5String.TAG,
                        new Tag(TagClass.CONTEXT_SPECIFIC, true, 11),
                        "hii").encodeDER());
    }

    @Test
    void testParse() throws ParseException {
        assertEquals("123",
                new IA5String(new BytesReader(new Byte[]{ 0x16, 3, '1', '2', '3' }), false)
                        .getString());
        assertEquals("",
                new IA5String(new BytesReader(new Byte[]{ 0x16, 0 }), false)
                        .getString());
    }

    @Test
    void testParseFail() {
        assertThrows(ParseException.class,  () ->
                new IA5String(new BytesReader(new Byte[]{ 0x16, 3, '1', '2' }), false));
        assertThrows(ParseException.class,  () ->
                new IA5String(new BytesReader(new Byte[]{ 0x16, 2, '1',  -128 }), false));
    }
}