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<String, String> headers = new HashMap<String, String>();
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 @Override
38 protected OutputStream createOutputStream() throws IOException {
39 return outputStream;
40 }
41
42 public Map<String, String> getHeaders() {
43 return headers;
44 }
45
46 @Override
47 public void addHeader(String name, String value) throws IOException {
48 headers.put(name, value);
49 }
50 }