/*
    * 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.io.input;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertFalse;
  import static org.junit.jupiter.api.Assertions.assertThrowsExactly;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  import java.io.ByteArrayInputStream;
  import java.io.IOException;
  import java.io.InputStream;
  
  import org.apache.commons.io.IOUtils;
  import org.junit.jupiter.api.BeforeEach;
  import org.junit.jupiter.api.Test;
  
  /**
   * Tests {@link CloseShieldInputStream}.
   */
  class CloseShieldInputStreamTest {
  
      private byte[] data;
  
      private InputStream byteArrayInputStream;
  
      private InputStream shielded;
  
      private boolean closed;
  
      @BeforeEach
      public void setUp() {
          data = new byte[] { 'x', 'y', 'z' };
          byteArrayInputStream = new ByteArrayInputStream(data) {
              @Override
              public void close() {
                  closed = true;
              }
          };
          closed = false;
      }
  
      @Test
      void testAvailableAfterClose() throws Exception {
          final InputStream shadow;
          try (InputStream in = CloseShieldInputStream.wrap(byteArrayInputStream)) {
              assertEquals(3, in.available());
              shadow = in;
          }
          assertEquals(0, shadow.available());
      }
  
      @Test
      void testAvailableAfterOpen() throws Exception {
          try (InputStream in = CloseShieldInputStream.wrap(byteArrayInputStream)) {
              assertEquals(3, in.available());
          }
      }
  
      @Test
      void testClose() throws IOException {
          shielded = CloseShieldInputStream.wrap(byteArrayInputStream);
          shielded.close();
          assertFalse(closed, "closed");
          assertEquals(-1, shielded.read(), "read()");
          assertEquals(data[0], byteArrayInputStream.read(), "read()");
      }
  
      @Test
      void testOnClose() throws Exception {
          try (InputStream in = CloseShieldInputStream.builder().setInputStream(byteArrayInputStream).setOnClose(is -> {
              assertFalse(closed);
              closed = true;
              return ClosedInputStream.INSTANCE;
          }).get()) {
              assertEquals(3, in.available());
          }
          assertTrue(closed);
      }
  
      @Test
      void testOnCloseException() throws Exception {
          final String message = "test";
          assertEquals("test", assertThrowsExactly(IOException.class, () -> {
              try (InputStream in = CloseShieldInputStream.builder().setInputStream(byteArrayInputStream).setOnClose(is -> {
                 assertFalse(closed);
                 throw new IOException(message);
             }).get()) {
                 assertEquals("test", assertThrowsExactly(IOException.class, in::close).getMessage());
             }
         }).getMessage());
         assertFalse(closed);
     }
 
     @Test
     void testReadAfterCose() throws Exception {
         final InputStream shadow;
         try (InputStream in = CloseShieldInputStream.wrap(byteArrayInputStream)) {
             assertEquals(3, in.available());
             shadow = in;
         }
         assertEquals(IOUtils.EOF, shadow.read());
     }
 
     @Test
     void testSystemInOnSystemInNo() throws IOException {
         shielded = CloseShieldInputStream.systemIn(byteArrayInputStream);
         shielded.close();
         assertTrue(closed, "closed");
         assertEquals(data[0], shielded.read(), "read()");
         assertEquals(data[1], byteArrayInputStream.read(), "read()");
     }
 
     @Test
     void testSystemInOnSystemInYes() throws IOException {
         shielded = CloseShieldInputStream.systemIn(System.in);
         shielded.close();
         assertFalse(closed, "closed");
         assertEquals(-1, shielded.read(), "read()");
         assertEquals(data[0], byteArrayInputStream.read(), "read()");
     }
 
 }