/*
    * Copyright (C) 2020, Michael Dardis. and others
    *
    * This program and the accompanying materials are made available under the
    * terms of the Eclipse Distribution License v. 1.0 which is available at
    * https://www.eclipse.org/org/documents/edl-v10.php.
    *
    * SPDX-License-Identifier: BSD-3-Clause
    */
  package org.eclipse.jgit.util.io;
  
  import static org.junit.Assert.assertArrayEquals;
  import static org.junit.Assert.assertFalse;
  import static org.junit.Assert.assertTrue;
  
  import java.io.ByteArrayOutputStream;
  import java.io.IOException;
  
  import org.eclipse.jgit.lib.Constants;
  import org.junit.Test;
  
  public class TeeOutputStreamTest {
  
  	@Test
  	public void test() throws IOException {
  		byte[] data = Constants.encode("Hello World");
  
  		TestOutput first = new TestOutput();
  		TestOutput second = new TestOutput();
  		try (TeeOutputStream tee = new TeeOutputStream(first, second)) {
  			tee.write(data);
  			assertArrayEquals("Stream output must match", first.toByteArray(),
  					second.toByteArray());
  
  			tee.write(1);
  			assertArrayEquals("Stream output must match", first.toByteArray(),
  					second.toByteArray());
  
  			tee.write(data, 1, 4); // Test partial write methods
  			assertArrayEquals("Stream output must match", first.toByteArray(),
  					second.toByteArray());
  		}
  		assertTrue("First stream should be closed", first.closed);
  		assertTrue("Second stream should be closed", second.closed);
  	}
  
  	@Test
  	public void testCloseException() {
  		TestOutput first = new TestOutput() {
  			@Override
  			public void close() throws IOException {
  				throw new IOException();
  			}
  
  		};
  		TestOutput second = new TestOutput();
  
  		@SuppressWarnings("resource")
  		TeeOutputStream tee = new TeeOutputStream(first, second);
  		try {
  			tee.close();
  		} catch (IOException ex) {
  			// Expected from first closed
  		}
  		assertFalse("First stream should not be closed", first.closed);
  		assertTrue("Second stream should still be closed if first close failed",
  				second.closed);
  	}
  
  	private static class TestOutput extends ByteArrayOutputStream {
  
  		private boolean closed;
  
  		@Override
  		public void close() throws IOException {
  			closed = true;
  			super.close();
  		}
  	}
  
  }