1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
|
/*-------------------------------------------------------------------------
*
* Copyright (c) 2004-2011, PostgreSQL Global Development Group
*
*
*-------------------------------------------------------------------------
*/
package org.postgresql.test.util;
import java.io.InputStream;
import java.io.IOException;
public class BrokenInputStream extends InputStream {
private InputStream _is;
private long _numRead;
private long _breakOn;
public BrokenInputStream(InputStream is, long breakOn) {
_is = is;
_breakOn = breakOn;
_numRead = 0;
}
public int read() throws IOException {
if (_breakOn > _numRead++)
{
throw new IOException("I was told to break on " + _breakOn);
}
return _is.read();
}
}
|