src/test/java/org/apache/commons/net/io/DotTerminatedMessageReaderTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.commons.net.io;

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

import java.io.IOException;
import java.io.StringReader;

import org.junit.jupiter.api.Test;

class DotTerminatedMessageReaderTest {

    private static final String CRLF = "\r\n";
    private static final String DOT = ".";
    private static final String EOM = CRLF + DOT + CRLF;
    private DotTerminatedMessageReader reader;
    private final StringBuilder str = new StringBuilder();
    private final char[] buf = new char[64];

    @Test
    void testDoubleCrBeforeDot() throws IOException {
        final String test = "Hello World!\r" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello World!\r" + CRLF, str.toString());
    }

    @Test
    void testEmbeddedDot1() throws IOException {
        final String test = "Hello . World!" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello . World!" + CRLF, str.toString());
    }

    @Test
    void testEmbeddedDot2() throws IOException {
        final String test = "Hello .. World!" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello .. World!" + CRLF, str.toString());
    }

    @Test
    void testEmbeddedDot3() throws IOException {
        final String test = "Hello World." + CRLF + "more" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello World." + CRLF + "more" + CRLF, str.toString());
    }

    @Test
    void testEmbeddedDot4() throws IOException {
        final String test = "Hello World\r.\nmore" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello World\r.\nmore" + CRLF, str.toString());
    }

    @Test
    void testEmbeddedNewlines() throws IOException {
        final String test = "Hello" + CRLF + "World\nA\rB" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello" + CRLF + "World\nA\rB" + CRLF, str.toString());
    }

    @Test
    void testLeadingDot() throws IOException {
        final String test = "Hello World!" + CRLF + "..text" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello World!" + CRLF + ".text" + CRLF, str.toString());
    }

    @Test
    void testReadLine1() throws Exception {
        final String test = "Hello World" + CRLF + "more" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        String line;
        while ((line = reader.readLine()) != null) {
            str.append(line);
            str.append("#");
        }

        assertEquals("Hello World#more#", str.toString());

    }

    @Test
    void testReadLine2() throws Exception {
        final String test = "Hello World\r.\nmore" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        String line;
        while ((line = reader.readLine()) != null) {
            str.append(line);
            str.append("#");
        }

        assertEquals("Hello World\r.\nmore#", str.toString());

    }

    @Test
    void testReadSimpleStringCrLfLineEnding() throws IOException {
        final String test = "Hello World!" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello World!" + CRLF, str.toString());
    }

    @Test
    void testReadSimpleStringLfLineEnding() throws IOException {
        final String test = "Hello World!" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello World!" + CRLF, str.toString());
    }

    @Test
    void testSingleDotWithTrailingText() throws IOException {
        final String test = "Hello World!" + CRLF + ".text" + EOM;
        reader = new DotTerminatedMessageReader(new StringReader(test));

        int read = 0;
        while ((read = reader.read(buf)) != -1) {
            str.append(buf, 0, read);
        }

        assertEquals("Hello World!" + CRLF + ".text" + CRLF, str.toString());
    }

}