Java 类org.apache.commons.io.output.ClosedOutputStream 实例源码

@Test
@SuppressWarnings("unchecked")
public void testPumpOutClosed() {
  final byte[] eout = "Jackdaws love my big sphinx of quartz.\n".getBytes();

  final ByteArrayInputStream in = new ByteArrayInputStream(eout);
  final OutputStream out = new ClosedOutputStream();

  final EventListener<IOException> el = context.mock(EventListener.class);
  context.checking(new Expectations() {
    {
      oneOf(el).receive(with(aNonNull(InputOutputStreamPump.class)),
                        with(any(IOException.class)));
    }
  });

  final InputOutputStreamPump p = new InputOutputStreamPump(in, out, el);
  p.run();
}

@Test
@SuppressWarnings("unchecked")
public void testPumpBothClosed() {
  final InputStream in = new ClosedInputStream();
  final OutputStream out = new ClosedOutputStream();

  final EventListener<IOException> el = context.mock(EventListener.class);
  context.checking(new Expectations() {
    {
      never(el).receive(with(any(Object.class)),
                        with(any(IOException.class)));
    }
  });

  final InputOutputStreamPump p = new InputOutputStreamPump(in, out, el);
  p.run();
}

@Test
@SuppressWarnings("unchecked")
public void testPumpOutClosed() {
  final byte[] eout = "Jackdaws love my big sphinx of quartz.\n".getBytes();

  final ByteArrayInputStream in = new ByteArrayInputStream(eout);
  final OutputStream out = new ClosedOutputStream();

  final EventListener<IOException> el = context.mock(EventListener.class);
  context.checking(new Expectations() {
    {
      oneOf(el).receive(with(aNonNull(InputOutputStreamPump.class)),
                        with(any(IOException.class)));
    }
  });

  final InputOutputStreamPump p = new InputOutputStreamPump(in, out, el);
  p.run();
}

@Test
@SuppressWarnings("unchecked")
public void testPumpBothClosed() {
  final InputStream in = new ClosedInputStream();
  final OutputStream out = new ClosedOutputStream();

  final EventListener<IOException> el = context.mock(EventListener.class);
  context.checking(new Expectations() {
    {
      never(el).receive(with(any(Object.class)),
                        with(any(IOException.class)));
    }
  });

  final InputOutputStreamPump p = new InputOutputStreamPump(in, out, el);
  p.run();
}

@Test(expected=UnsupportedOperationException.class)
public void testSetOutputStreamRunning() {
  final InputOutputStreamPump p = new IOSP();
  p.run();

  final OutputStream out = new ClosedOutputStream();
  p.setOutputStream(out);
}

@Override
public void close() throws IOException {
    if (delegate != null) {
        delegate.close();
    }
    //forget the reference to original outputstream to potentially free some memory
    delegate = ClosedOutputStream.CLOSED_OUTPUT_STREAM;
}

@Test(expected=UnsupportedOperationException.class)
public void testSetOutputStreamRunning() {
  final InputOutputStreamPump p = new IOSP();
  p.run();

  final OutputStream out = new ClosedOutputStream();
  p.setOutputStream(out);
}

@Test
public void testSetOutputStreamNotRunning() {
  final InputOutputStreamPump p = new InputOutputStreamPump();
  final OutputStream out = new ClosedOutputStream();
  p.setOutputStream(out);
}

public boolean isClosed() {
    return delegate == ClosedOutputStream.CLOSED_OUTPUT_STREAM;
}

@Test
public void testSetOutputStreamNotRunning() {
  final InputOutputStreamPump p = new InputOutputStreamPump();
  final OutputStream out = new ClosedOutputStream();
  p.setOutputStream(out);
}