| /* |
| * 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 |
| * |
| * http://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.io.input; |
| |
| import static org.junit.jupiter.api.Assertions.assertEquals; |
| import static org.junit.jupiter.api.Assertions.assertTrue; |
| |
| import java.io.CharArrayReader; |
| import java.io.IOException; |
| import java.io.StringReader; |
| import java.nio.charset.Charset; |
| import java.nio.charset.StandardCharsets; |
| import java.util.Random; |
| |
| import org.junit.jupiter.api.Test; |
| |
| public class ReaderInputStreamTest { |
| private static final String TEST_STRING = "\u00e0 peine arriv\u00e9s nous entr\u00e2mes dans sa chambre"; |
| private static final String LARGE_TEST_STRING; |
| |
| static { |
| final StringBuilder buffer = new StringBuilder(); |
| for (int i=0; i<100; i++) { |
| buffer.append(TEST_STRING); |
| } |
| LARGE_TEST_STRING = buffer.toString(); |
| } |
| |
| private final Random random = new Random(); |
| |
| private void testWithSingleByteRead(final String testString, final String charsetName) throws IOException { |
| final byte[] bytes = testString.getBytes(charsetName); |
| final ReaderInputStream in = new ReaderInputStream(new StringReader(testString), charsetName); |
| for (final byte b : bytes) { |
| final int read = in.read(); |
| assertTrue(read >= 0); |
| assertTrue(read <= 255); |
| assertEquals(b, (byte)read); |
| } |
| assertEquals(-1, in.read()); |
| in.close(); |
| } |
| |
| private void testWithBufferedRead(final String testString, final String charsetName) throws IOException { |
| final byte[] expected = testString.getBytes(charsetName); |
| final ReaderInputStream in = new ReaderInputStream(new StringReader(testString), charsetName); |
| final byte[] buffer = new byte[128]; |
| int offset = 0; |
| while (true) { |
| int bufferOffset = random.nextInt(64); |
| final int bufferLength = random.nextInt(64); |
| int read = in.read(buffer, bufferOffset, bufferLength); |
| if (read == -1) { |
| assertEquals(offset, expected.length); |
| break; |
| } |
| assertTrue(read <= bufferLength); |
| while (read > 0) { |
| assertTrue(offset < expected.length); |
| assertEquals(expected[offset], buffer[bufferOffset]); |
| offset++; |
| bufferOffset++; |
| read--; |
| } |
| } |
| in.close(); |
| } |
| |
| @Test |
| public void testUTF8WithSingleByteRead() throws IOException { |
| testWithSingleByteRead(TEST_STRING, "UTF-8"); |
| } |
| |
| @Test |
| public void testLargeUTF8WithSingleByteRead() throws IOException { |
| testWithSingleByteRead(LARGE_TEST_STRING, "UTF-8"); |
| } |
| |
| @Test |
| public void testUTF8WithBufferedRead() throws IOException { |
| testWithBufferedRead(TEST_STRING, "UTF-8"); |
| } |
| |
| @Test |
| public void testLargeUTF8WithBufferedRead() throws IOException { |
| testWithBufferedRead(LARGE_TEST_STRING, "UTF-8"); |
| } |
| |
| @Test |
| public void testUTF16WithSingleByteRead() throws IOException { |
| testWithSingleByteRead(TEST_STRING, "UTF-16"); |
| } |
| |
| @SuppressWarnings("deprecation") |
| @Test |
| public void testReadZero() throws Exception { |
| final String inStr = "test"; |
| final ReaderInputStream r = new ReaderInputStream(new StringReader(inStr)); |
| final byte[] bytes = new byte[30]; |
| assertEquals(0, r.read(bytes, 0, 0)); |
| assertEquals(inStr.length(), r.read(bytes, 0, inStr.length()+1)); |
| // Should always return 0 for length == 0 |
| assertEquals(0, r.read(bytes, 0, 0)); |
| r.close(); |
| } |
| |
| @SuppressWarnings("deprecation") |
| @Test |
| public void testReadZeroEmptyString() throws Exception { |
| final ReaderInputStream r = new ReaderInputStream(new StringReader("")); |
| final byte[] bytes = new byte[30]; |
| // Should always return 0 for length == 0 |
| assertEquals(0, r.read(bytes, 0, 0)); |
| assertEquals(-1, r.read(bytes, 0, 1)); |
| assertEquals(0, r.read(bytes, 0, 0)); |
| assertEquals(-1, r.read(bytes, 0, 1)); |
| r.close(); |
| } |
| |
| /* |
| * Tests https://issues.apache.org/jira/browse/IO-277 |
| */ |
| @Test |
| public void testCharsetMismatchInfiniteLoop() throws IOException { |
| // Input is UTF-8 bytes: 0xE0 0xB2 0xA0 |
| final char[] inputChars = new char[] { (char) 0xE0, (char) 0xB2, (char) 0xA0 }; |
| // Charset charset = Charset.forName("UTF-8"); // works |
| final Charset charset = StandardCharsets.US_ASCII; // infinite loop |
| try (ReaderInputStream stream = new ReaderInputStream(new CharArrayReader(inputChars), charset)) { |
| while (stream.read() != -1) { |
| } |
| } |
| } |
| } |