1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package org.springframework.ws.transport;
18
19 import java.io.IOException;
20 import java.io.OutputStream;
21 import java.util.HashMap;
22 import java.util.Map;
23
24 import org.springframework.util.Assert;
25
26 public class MockTransportOutputStream extends TransportOutputStream {
27
28 private Map headers = new HashMap();
29
30 private OutputStream outputStream;
31
32 public MockTransportOutputStream(OutputStream outputStream) {
33 Assert.notNull(outputStream, "outputStream must not be null");
34 this.outputStream = outputStream;
35 }
36
37 protected OutputStream createOutputStream() throws IOException {
38 return outputStream;
39 }
40
41 public Map getHeaders() {
42 return headers;
43 }
44
45 public void addHeader(String name, String value) throws IOException {
46 headers.put(name, value);
47 }
48 }