1
2
3
4
5
6
7
8
9
10 package org.mule.module.ibeans.spi.support;
11
12 import org.mule.message.DefaultExceptionPayload;
13 import org.mule.module.ibeans.spi.MuleResponseMessage;
14
15 import org.ibeans.impl.test.MockMessageCallback;
16
17
18
19
20 public class HttpStatusCodeCallback implements MockMessageCallback<MuleResponseMessage>
21 {
22 private int status;
23
24 public HttpStatusCodeCallback(int status)
25 {
26 this.status = status;
27 }
28
29 public void onMessage(MuleResponseMessage response)
30 {
31
32 response.setStatusCode(String.valueOf(status));
33 if(status >= 400)
34 {
35 response.getMessage().setExceptionPayload(new DefaultExceptionPayload(new Exception("Mock Http Error")));
36 }
37 }
38 }