View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *    https://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  package org.apache.commons.io.input;
18  
19  import static org.junit.jupiter.api.Assertions.assertThrows;
20  import static org.junit.jupiter.api.Assertions.assertTrue;
21  
22  import java.io.IOException;
23  import java.io.InputStream;
24  import java.nio.channels.FileChannel;
25  import java.nio.file.StandardOpenOption;
26  
27  import org.junit.jupiter.api.BeforeEach;
28  import org.junit.jupiter.api.Test;
29  
30  /**
31   * Tests functionality of {@link BufferedFileChannelInputStream}.
32   *
33   * This class was ported and adapted from Apache Spark commit 933dc6cb7b3de1d8ccaf73d124d6eb95b947ed19 where it was called
34   * {@code BufferedFileChannelInputStreamSuite}.
35   */
36  class BufferedFileChannelInputStreamTest extends AbstractInputStreamTest {
37  
38      @SuppressWarnings("resource")
39      @BeforeEach
40      public void setUpInputStreams() throws IOException {
41          // @formatter:off
42          inputStreams = new InputStream[] {
43              new BufferedFileChannelInputStream(InputPath), // default
44              new BufferedFileChannelInputStream(InputPath, 123), // small, unaligned buffer size
45              BufferedFileChannelInputStream.builder().setPath(InputPath).get(), // default
46              BufferedFileChannelInputStream.builder().setPath(InputPath).setBufferSize(123).get(), // small, unaligned buffer size
47              BufferedFileChannelInputStream.builder().setURI(InputPath.toUri()).setBufferSize(1024).get(), // URI and buffer size
48              BufferedFileChannelInputStream.builder().setPath(InputPath).setOpenOptions(StandardOpenOption.READ).get(), // open options
49              BufferedFileChannelInputStream.builder().setFileChannel(FileChannel.open(InputPath)).get(), // FileChannel
50          };
51          //@formatter:on
52      }
53  
54      @Override
55      @Test
56      void testAvailableAfterOpen() throws Exception {
57          for (final InputStream inputStream : inputStreams) {
58              assertTrue(inputStream.available() > 0);
59          }
60      }
61  
62      @Test
63      void testBuilderGet() {
64          // java.lang.IllegalStateException: origin == null
65          assertThrows(IllegalStateException.class, () -> BufferedFileChannelInputStream.builder().get());
66      }
67  
68      @Test
69      void testReadAfterClose() throws Exception {
70          for (final InputStream inputStream : inputStreams) {
71              inputStream.close();
72              assertThrows(IOException.class, inputStream::read);
73          }
74      }
75  
76  }