TestRootValues.java

package com.fasterxml.jackson.core.json;

import java.io.*;

import org.junit.jupiter.api.Test;

import com.fasterxml.jackson.core.*;

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

class TestRootValues
        extends com.fasterxml.jackson.core.JUnit5TestBase
{
    static class Issue516InputStream extends InputStream
    {
        private final byte[][] reads;
        private int currentRead;

        public Issue516InputStream(byte[][] reads) {
            this.reads = reads;
            this.currentRead = 0;
        }

        @Override
        public int read() throws IOException {
            throw new UnsupportedOperationException();
        }

        @Override
        public int read(byte[] b, int off, int len) throws IOException {
            if (currentRead >= reads.length) {
                return -1;
            }
            byte[] bytes = reads[currentRead++];
            if (len < bytes.length) {
                throw new IllegalArgumentException();
            }
            System.arraycopy(bytes, 0, b, off, bytes.length);
            return bytes.length;
        }
    }

    static class Issue516Reader extends Reader
    {
        private final char[][] reads;
        private int currentRead;

        public Issue516Reader(char[][] reads) {
            this.reads = reads;
            this.currentRead = 0;
        }

        @Override
        public void close() { }

        @Override
        public int read() throws IOException {
            throw new UnsupportedOperationException();
        }

        @Override
        public int read(char[] b, int off, int len) throws IOException {
            if (currentRead >= reads.length) {
                return -1;
            }
            char[] bytes = reads[currentRead++];
            if (len < bytes.length) {
                throw new IllegalArgumentException();
            }
            System.arraycopy(bytes, 0, b, off, bytes.length);
            return bytes.length;
        }
    }

    /*
    /**********************************************************
    /* Test methods, reads
    /**********************************************************
     */

    private final JsonFactory JSON_F = sharedStreamFactory();

    @Test
    void simpleNumbers() throws Exception {
        // DataInput cannot detect EOF so:
        _testSimpleNumbers(MODE_INPUT_STREAM);
        _testSimpleNumbers(MODE_INPUT_STREAM_THROTTLED);
        _testSimpleNumbers(MODE_READER);
    }

    private void _testSimpleNumbers(int mode) throws Exception
    {
        final String DOC = "1 2\t3\r4\n5\r\n6\r\n   7";
        JsonParser p = createParser(mode, DOC);
        for (int i = 1; i <= 7; ++i) {
            assertToken(JsonToken.VALUE_NUMBER_INT, p.nextToken());
            assertEquals(i, p.getIntValue());
        }
        assertNull(p.nextToken());
        p.close();
    }

    @Test
    void brokenNumber() throws Exception
    {
        _testBrokenNumber(MODE_INPUT_STREAM);
        _testBrokenNumber(MODE_INPUT_STREAM_THROTTLED);
        _testBrokenNumber(MODE_READER);
        // I think DataInput _SHOULD_ be able to detect, fail, but for now...
//        _testBrokenNumber(MODE_DATA_INPUT);
    }

    private void _testBrokenNumber(int mode) throws Exception
    {
        final String DOC = "14:89:FD:D3:E7:8C";
        JsonParser p = createParser(mode, DOC);
        // Should fail, right away
        try {
            p.nextToken();
            fail("Ought to fail! Instead, got token: "+p.currentToken());
        } catch (JsonParseException e) {
            verifyException(e, "unexpected character");
        }
        p.close();
    }

    @Test
    void simpleBooleans() throws Exception {
        // can't do DataInput so
        _testSimpleBooleans(MODE_INPUT_STREAM);
        _testSimpleBooleans(MODE_INPUT_STREAM_THROTTLED);
        _testSimpleBooleans(MODE_READER);
    }

    private void _testSimpleBooleans(int mode) throws Exception
    {
        final String DOC = "true false\ttrue\rfalse\ntrue\r\nfalse\r\n   true";
        JsonParser p = createParser(mode, DOC);
        boolean exp = true;
        for (int i = 1; i <= 7; ++i) {
            assertToken(exp ? JsonToken.VALUE_TRUE : JsonToken.VALUE_FALSE, p.nextToken());
            exp = !exp;
        }
        assertNull(p.nextToken());
        p.close();
    }

    @Test
    void invalidToken() throws Exception
    {
        _testInvalidToken(MODE_INPUT_STREAM, '\u00c4');
        _testInvalidToken(MODE_INPUT_STREAM_THROTTLED, '\u00c4');
        _testInvalidToken(MODE_READER, '\u00c4');
        _testInvalidToken(MODE_DATA_INPUT, '\u00c4');

        _testInvalidToken(MODE_INPUT_STREAM, '\u3456');
        _testInvalidToken(MODE_INPUT_STREAM_THROTTLED, '\u3456');
        _testInvalidToken(MODE_READER, '\u3456');
        _testInvalidToken(MODE_DATA_INPUT, '\u3456');
    }

    private void _testInvalidToken(int mode, char weirdChar) throws Exception
    {
        final String DOC = " A\u3456C ";
        JsonParser p = createParser(mode, DOC);
        // Should fail, right away
        try {
            p.nextToken();
            fail("Ought to fail! Instead, got token: "+p.currentToken());
        } catch (JsonParseException e) {
            verifyException(e, "Unrecognized token");
        }
        p.close();
    }

    /*
    /**********************************************************
    /* Test methods, writes
    /**********************************************************
     */

    @Test
    void simpleWrites() throws Exception
    {
        _testSimpleWrites(false);
        _testSimpleWrites(true);
    }

    private void _testSimpleWrites(boolean useStream) throws Exception
    {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        StringWriter w = new StringWriter();
        JsonGenerator gen;

        if (useStream) {
            gen = JSON_F.createGenerator(out, JsonEncoding.UTF8);
        } else {
            gen = JSON_F.createGenerator(w);
        }
        gen.writeNumber(123);
        gen.writeString("abc");
        gen.writeBoolean(true);

        gen.close();
        out.close();
        w.close();

        // and verify
        String json = useStream ? out.toString("UTF-8") : w.toString();
        assertEquals("123 \"abc\" true", json);
    }

    /*
    /**********************************************************
    /* Test methods, other
    /**********************************************************
     */

    // [core#516]: Off-by-one read problem
    @Test
    void rootOffsetIssue516Bytes() throws Exception
    {
        // InputStream that forces _parseNumber2 to be invoked.
        final InputStream in = new Issue516InputStream(new byte[][] {
            "1234".getBytes("UTF-8"),
            "5 true".getBytes("UTF-8")
        });

        JsonParser parser = JSON_F.createParser(in);
        assertEquals(12345, parser.nextIntValue(0));

        // Fails with com.fasterxml.jackson.core.JsonParseException: Unrecognized token 'rue': was expecting ('true', 'false' or 'null')
        assertTrue(parser.nextBooleanValue());

        parser.close();
        in.close();
    }

    // [core#516]: Off-by-one read problem
    @Test
    void rootOffsetIssue516Chars() throws Exception
    {
        // InputStream that forces _parseNumber2 to be invoked.
        final Reader in = new Issue516Reader(new char[][] {
            "1234".toCharArray(), "5 true".toCharArray()
        });

        JsonParser parser = JSON_F.createParser(in);
        assertEquals(12345, parser.nextIntValue(0));

        // Fails with com.fasterxml.jackson.core.JsonParseException: Unrecognized token 'rue': was expecting ('true', 'false' or 'null')
        assertTrue(parser.nextBooleanValue());

        parser.close();
        in.close();
    }
}