1
2
3
4
5
6
7
8
9
10
11 package org.mule.module.cxf.support;
12
13 import org.mule.tck.AbstractMuleTestCase;
14
15 import java.util.ArrayList;
16 import java.util.Arrays;
17 import java.util.List;
18
19 import javax.xml.namespace.QName;
20
21 import org.apache.cxf.message.MessageContentsList;
22 import org.apache.cxf.service.model.MessagePartInfo;
23
24 public class OutputPayloadInterceptorTestCase extends AbstractMuleTestCase
25 {
26
27 private OutputPayloadInterceptor interceptor;
28
29 @Override
30 protected void doSetUp() throws Exception
31 {
32 super.doSetUp();
33 interceptor = new OutputPayloadInterceptor();
34 }
35
36 public void testCleanUpPayload_NotAnArray() throws Exception
37 {
38 final List<?> someList = new ArrayList<Object>();
39 assertSame(someList, interceptor.cleanUpPayload(someList));
40
41 final String someString = "Some String";
42 assertSame(someString, interceptor.cleanUpPayload(someString));
43 }
44
45 public void testCleanUpPayload_NonObjectArray() throws Exception
46 {
47 final String someString = "Some String";
48 assertSame(someString, interceptor.cleanUpPayload(new String[]{someString, null}));
49
50 final String[] arrayOf2Strings = {someString, "someOther String"};
51 assertSame(arrayOf2Strings, interceptor.cleanUpPayload(arrayOf2Strings));
52
53 final String[] arrayOf2Strings1Null = {someString, "someOther String", null};
54 assertTrue(Arrays.equals(new String[]{arrayOf2Strings1Null[0], arrayOf2Strings1Null[1]},
55 (Object[]) interceptor.cleanUpPayload(arrayOf2Strings1Null)));
56 }
57
58 public void testCleanUpPayload_ArrayOfObjects()
59 {
60 final Integer someInteger = Integer.valueOf(1);
61 assertSame(someInteger, interceptor.cleanUpPayload(new Object[]{someInteger, null}));
62
63 final Object[] arrayOf2Objects = {someInteger, new Object()};
64 assertSame(arrayOf2Objects, interceptor.cleanUpPayload(arrayOf2Objects));
65
66 final Object[] arrayOf2Objects1Null = {someInteger, null, new Object(),
67 MessageContentsList.REMOVED_MARKER};
68 assertTrue(Arrays.equals(new Object[]{arrayOf2Objects1Null[0], arrayOf2Objects1Null[2]},
69 (Object[]) interceptor.cleanUpPayload(arrayOf2Objects1Null)));
70 }
71
72 public void testAddsMissingPartContentOnHandleMessage() {
73 MessageContentsList messageContentsList = new MessageContentsList();
74 Object content1 = new Object();
75 messageContentsList.add(content1);
76
77 List<MessagePartInfo> parts = new ArrayList<MessagePartInfo>();
78
79 MessagePartInfo part1 = new MessagePartInfo(new QName("part1"), null);
80 part1.setIndex(1);
81 parts.add(part1);
82
83 interceptor.ensurePartIndexMatchListIndex(messageContentsList, parts);
84
85 assertEquals(2, messageContentsList.size());
86 assertEquals(null, messageContentsList.get(0));
87 assertEquals(content1, messageContentsList.get(1));
88 }
89
90 public void testAddsMissingPartContentWithUnsortedPartListOnHandleMessage() {
91 MessageContentsList messageContentsList = new MessageContentsList();
92 Object content1 = new Object();
93 messageContentsList.add(content1);
94 Object content2 = new Object();
95 messageContentsList.add(content2);
96
97 List<MessagePartInfo> parts = new ArrayList<MessagePartInfo>();
98
99 MessagePartInfo part2 = new MessagePartInfo(new QName("part2"), null);
100 part2.setIndex(3);
101 parts.add(part2);
102
103 MessagePartInfo part1 = new MessagePartInfo(new QName("part1"), null);
104 part1.setIndex(1);
105 parts.add(part1);
106
107 interceptor.ensurePartIndexMatchListIndex(messageContentsList, parts);
108
109 assertEquals(4, messageContentsList.size());
110 assertEquals(null, messageContentsList.get(0));
111 assertEquals(content1, messageContentsList.get(1));
112 assertEquals(null, messageContentsList.get(2));
113 assertEquals(content2, messageContentsList.get(3));
114 }
115 }