17dd252788645e940eada959bdde927426e2531c9Paul Duffin/* 27dd252788645e940eada959bdde927426e2531c9Paul Duffin * Copyright (C) 2012 The Guava Authors 37dd252788645e940eada959bdde927426e2531c9Paul Duffin * 47dd252788645e940eada959bdde927426e2531c9Paul Duffin * Licensed under the Apache License, Version 2.0 (the "License"); 57dd252788645e940eada959bdde927426e2531c9Paul Duffin * you may not use this file except in compliance with the License. 67dd252788645e940eada959bdde927426e2531c9Paul Duffin * You may obtain a copy of the License at 77dd252788645e940eada959bdde927426e2531c9Paul Duffin * 87dd252788645e940eada959bdde927426e2531c9Paul Duffin * http://www.apache.org/licenses/LICENSE-2.0 97dd252788645e940eada959bdde927426e2531c9Paul Duffin * 107dd252788645e940eada959bdde927426e2531c9Paul Duffin * Unless required by applicable law or agreed to in writing, software 117dd252788645e940eada959bdde927426e2531c9Paul Duffin * distributed under the License is distributed on an "AS IS" BASIS, 127dd252788645e940eada959bdde927426e2531c9Paul Duffin * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 137dd252788645e940eada959bdde927426e2531c9Paul Duffin * See the License for the specific language governing permissions and 147dd252788645e940eada959bdde927426e2531c9Paul Duffin * limitations under the License. 157dd252788645e940eada959bdde927426e2531c9Paul Duffin */ 167dd252788645e940eada959bdde927426e2531c9Paul Duffin 177dd252788645e940eada959bdde927426e2531c9Paul Duffinpackage com.google.common.io; 187dd252788645e940eada959bdde927426e2531c9Paul Duffin 197dd252788645e940eada959bdde927426e2531c9Paul Duffinimport static com.google.common.base.Preconditions.checkNotNull; 207dd252788645e940eada959bdde927426e2531c9Paul Duffinimport static com.google.common.io.TestOption.CLOSE_THROWS; 217dd252788645e940eada959bdde927426e2531c9Paul Duffinimport static com.google.common.io.TestOption.OPEN_THROWS; 227dd252788645e940eada959bdde927426e2531c9Paul Duffinimport static com.google.common.io.TestOption.WRITE_THROWS; 237dd252788645e940eada959bdde927426e2531c9Paul Duffin 247dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.ImmutableSet; 257dd252788645e940eada959bdde927426e2531c9Paul Duffin 267dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.io.FilterOutputStream; 277dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.io.IOException; 287dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.io.OutputStream; 297dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.Arrays; 307dd252788645e940eada959bdde927426e2531c9Paul Duffin 317dd252788645e940eada959bdde927426e2531c9Paul Duffin/** 327dd252788645e940eada959bdde927426e2531c9Paul Duffin * @author Colin Decker 337dd252788645e940eada959bdde927426e2531c9Paul Duffin */ 347dd252788645e940eada959bdde927426e2531c9Paul Duffinpublic class TestOutputStream extends FilterOutputStream { 357dd252788645e940eada959bdde927426e2531c9Paul Duffin 367dd252788645e940eada959bdde927426e2531c9Paul Duffin private final ImmutableSet<TestOption> options; 377dd252788645e940eada959bdde927426e2531c9Paul Duffin private boolean closed; 387dd252788645e940eada959bdde927426e2531c9Paul Duffin 397dd252788645e940eada959bdde927426e2531c9Paul Duffin public TestOutputStream(OutputStream out, TestOption... options) throws IOException { 407dd252788645e940eada959bdde927426e2531c9Paul Duffin this(out, Arrays.asList(options)); 417dd252788645e940eada959bdde927426e2531c9Paul Duffin } 427dd252788645e940eada959bdde927426e2531c9Paul Duffin 437dd252788645e940eada959bdde927426e2531c9Paul Duffin public TestOutputStream(OutputStream out, Iterable<TestOption> options) throws IOException { 447dd252788645e940eada959bdde927426e2531c9Paul Duffin super(checkNotNull(out)); 457dd252788645e940eada959bdde927426e2531c9Paul Duffin this.options = ImmutableSet.copyOf(options); 467dd252788645e940eada959bdde927426e2531c9Paul Duffin throwIf(OPEN_THROWS); 477dd252788645e940eada959bdde927426e2531c9Paul Duffin } 487dd252788645e940eada959bdde927426e2531c9Paul Duffin 497dd252788645e940eada959bdde927426e2531c9Paul Duffin public boolean closed() { 507dd252788645e940eada959bdde927426e2531c9Paul Duffin return closed; 517dd252788645e940eada959bdde927426e2531c9Paul Duffin } 527dd252788645e940eada959bdde927426e2531c9Paul Duffin 537dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 547dd252788645e940eada959bdde927426e2531c9Paul Duffin public void write(byte[] b, int off, int len) throws IOException { 557dd252788645e940eada959bdde927426e2531c9Paul Duffin throwIf(closed); 567dd252788645e940eada959bdde927426e2531c9Paul Duffin throwIf(WRITE_THROWS); 577dd252788645e940eada959bdde927426e2531c9Paul Duffin super.write(b, off, len); 587dd252788645e940eada959bdde927426e2531c9Paul Duffin } 597dd252788645e940eada959bdde927426e2531c9Paul Duffin 607dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 617dd252788645e940eada959bdde927426e2531c9Paul Duffin public void write(int b) throws IOException { 627dd252788645e940eada959bdde927426e2531c9Paul Duffin throwIf(closed); 637dd252788645e940eada959bdde927426e2531c9Paul Duffin throwIf(WRITE_THROWS); 647dd252788645e940eada959bdde927426e2531c9Paul Duffin super.write(b); 657dd252788645e940eada959bdde927426e2531c9Paul Duffin } 667dd252788645e940eada959bdde927426e2531c9Paul Duffin 677dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 687dd252788645e940eada959bdde927426e2531c9Paul Duffin public void close() throws IOException { 697dd252788645e940eada959bdde927426e2531c9Paul Duffin closed = true; 707dd252788645e940eada959bdde927426e2531c9Paul Duffin super.close(); 717dd252788645e940eada959bdde927426e2531c9Paul Duffin throwIf(CLOSE_THROWS); 727dd252788645e940eada959bdde927426e2531c9Paul Duffin } 737dd252788645e940eada959bdde927426e2531c9Paul Duffin 747dd252788645e940eada959bdde927426e2531c9Paul Duffin private void throwIf(TestOption option) throws IOException { 757dd252788645e940eada959bdde927426e2531c9Paul Duffin throwIf(options.contains(option)); 767dd252788645e940eada959bdde927426e2531c9Paul Duffin } 777dd252788645e940eada959bdde927426e2531c9Paul Duffin 787dd252788645e940eada959bdde927426e2531c9Paul Duffin private static void throwIf(boolean condition) throws IOException { 797dd252788645e940eada959bdde927426e2531c9Paul Duffin if (condition) { 807dd252788645e940eada959bdde927426e2531c9Paul Duffin throw new IOException(); 817dd252788645e940eada959bdde927426e2531c9Paul Duffin } 827dd252788645e940eada959bdde927426e2531c9Paul Duffin } 837dd252788645e940eada959bdde927426e2531c9Paul Duffin} 84