package com.oreilly.mock;

import junit.framework.Assert;

import java.io.PrintWriter;
import java.io.StringWriter;

public class MockPrintWriter extends PrintWriter {
    private String expectedOutput;

    public MockPrintWriter() {
        super(new StringWriter());
    }

    public void setExpectedOutput(String expectedOutput) {
        this.expectedOutput = expectedOutput;
    }

    public void verify() {
        String actualOutput = super.out.toString();
        Assert.assertEquals("PrintWriter content",
                this.expectedOutput, actualOutput);
    }
}