01: package com.mockrunner.test.web;
02:
03: import java.io.IOException;
04: import java.io.StringWriter;
05:
06: import com.mockrunner.mock.web.MockJspWriter;
07:
08: import junit.framework.TestCase;
09:
10: public class MockJspWriterTest extends TestCase {
11: public void testWithDefaultWriter() throws Exception {
12: MockJspWriter writer = new MockJspWriter();
13: writer.print("test1");
14: writer.clear();
15: writer.print("test2");
16: writer.clearBuffer();
17: writer.print("test3");
18: assertEquals("test3", writer.getOutputAsString());
19: }
20:
21: public void testWithProvidedWriter() throws Exception {
22: StringWriter providedWriter = new StringWriter();
23: MockJspWriter writer = new MockJspWriter(providedWriter);
24: writer.print("test1");
25: try {
26: writer.clear();
27: fail();
28: } catch (IOException exc) {
29: //should throw exception
30: }
31: writer.print("test2");
32: writer.clearBuffer();
33: writer.print("test3");
34: assertEquals("", writer.getOutputAsString());
35: writer.flush();
36: assertEquals("test1test2test3", providedWriter.toString());
37: }
38: }
|