MockInputStream.java

package net.sf.eclipsefp.haskell.core.project.util;

import java.io.IOException;
import java.io.InputStream;
import org.junit.Assert;

public class MockInputStream extends InputStream {

	private final InputStream fInput;
	private int fTimesClosed = 0;

	public MockInputStream(final InputStream stream) {
		fInput = stream;
	}

	@Override
	public int read() throws IOException {
		return fInput.read();
	}

	@Override
	public void close() throws IOException {
		fTimesClosed ++;
		fInput.close();
	}

	public void verify() {
		if (fTimesClosed != 1) {
      Assert.fail("Stream not closed.");
    }
	}

}