Browse Source
Bug: 559106 Change-Id: Ife06f21a00d258780a2030745738194098c631cf Signed-off-by: Michael Dardis <git@md-5.net> Signed-off-by: Matthias Sohn <matthias.sohn@sap.com>stable-5.8
Michael Dardis
5 years ago
committed by
Matthias Sohn
4 changed files with 156 additions and 2 deletions
@ -0,0 +1,81 @@ |
|||||||
|
/* |
||||||
|
* 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(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} |
@ -0,0 +1,74 @@ |
|||||||
|
/* |
||||||
|
* 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 java.io.IOException; |
||||||
|
import java.io.OutputStream; |
||||||
|
|
||||||
|
/** |
||||||
|
* An output stream that writes all data to two streams. |
||||||
|
* |
||||||
|
* @since 5.7 |
||||||
|
*/ |
||||||
|
public class TeeOutputStream extends OutputStream { |
||||||
|
|
||||||
|
private final OutputStream stream1; |
||||||
|
private final OutputStream stream2; |
||||||
|
|
||||||
|
/** |
||||||
|
* Initialize a tee output stream. |
||||||
|
* |
||||||
|
* @param stream1 first output stream |
||||||
|
* @param stream2 second output stream |
||||||
|
*/ |
||||||
|
public TeeOutputStream(OutputStream stream1, OutputStream stream2) { |
||||||
|
this.stream1 = stream1; |
||||||
|
this.stream2 = stream2; |
||||||
|
} |
||||||
|
|
||||||
|
/** {@inheritDoc} */ |
||||||
|
@Override |
||||||
|
public void write(byte[] buf) throws IOException { |
||||||
|
this.stream1.write(buf); |
||||||
|
this.stream2.write(buf); |
||||||
|
} |
||||||
|
|
||||||
|
/** {@inheritDoc} */ |
||||||
|
@Override |
||||||
|
public void write(byte[] buf, int off, int len) throws IOException { |
||||||
|
this.stream1.write(buf, off, len); |
||||||
|
this.stream2.write(buf, off, len); |
||||||
|
} |
||||||
|
|
||||||
|
/** {@inheritDoc} */ |
||||||
|
@Override |
||||||
|
public void write(int b) throws IOException { |
||||||
|
this.stream1.write(b); |
||||||
|
this.stream2.write(b); |
||||||
|
} |
||||||
|
|
||||||
|
/** {@inheritDoc} */ |
||||||
|
@Override |
||||||
|
public void flush() throws IOException { |
||||||
|
this.stream1.flush(); |
||||||
|
this.stream2.flush(); |
||||||
|
} |
||||||
|
|
||||||
|
/** {@inheritDoc} */ |
||||||
|
@Override |
||||||
|
public void close() throws IOException { |
||||||
|
try { |
||||||
|
this.stream1.close(); |
||||||
|
} finally { |
||||||
|
this.stream2.close(); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue