01: /**
02: * Licensed to the Apache Software Foundation (ASF) under one
03: * or more contributor license agreements. See the NOTICE file
04: * distributed with this work for additional information
05: * regarding copyright ownership. The ASF licenses this file
06: * to you under the Apache License, Version 2.0 (the
07: * "License"); you may not use this file except in compliance
08: * with the License. You may obtain a copy of the License at
09: *
10: * http://www.apache.org/licenses/LICENSE-2.0
11: *
12: * Unless required by applicable law or agreed to in writing,
13: * software distributed under the License is distributed on an
14: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15: * KIND, either express or implied. See the License for the
16: * specific language governing permissions and limitations
17: * under the License.
18: */package org.apache.cxf.jaxws;
19:
20: import java.io.InputStream;
21: import java.io.OutputStream;
22:
23: import junit.framework.Assert;
24: import org.apache.cxf.helpers.IOUtils;
25: import org.apache.cxf.message.Message;
26: import org.apache.cxf.transport.Conduit;
27: import org.apache.cxf.transport.MessageObserver;
28:
29: public class MessageReplayObserver implements MessageObserver {
30: String responseMessage;
31:
32: public MessageReplayObserver(String responseMessage) {
33: this .responseMessage = responseMessage;
34: }
35:
36: public void onMessage(Message message) {
37: try {
38:
39: InputStream in = message.getContent(InputStream.class);
40: while (in.read() != -1) {
41: // do nothing
42: }
43:
44: Conduit backChannel = message.getDestination()
45: .getBackChannel(message, null, null);
46:
47: backChannel.prepare(message);
48:
49: OutputStream out = message.getContent(OutputStream.class);
50: Assert.assertNotNull(out);
51: InputStream res = getClass().getResourceAsStream(
52: responseMessage);
53: IOUtils.copy(res, out, 2045);
54:
55: res.close();
56: in.close();
57: out.close();
58: backChannel.close(message);
59: } catch (Exception e) {
60: e.printStackTrace();
61: }
62: }
63: }
|