1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.mina.filter.codec.textline;
21
22 import java.net.SocketAddress;
23 import java.nio.charset.Charset;
24 import java.nio.charset.CharsetEncoder;
25 import java.util.LinkedList;
26 import java.util.Queue;
27
28 import junit.framework.Assert;
29 import junit.framework.TestCase;
30 import org.apache.mina.common.ByteBuffer;
31 import org.apache.mina.common.IoFilterChain;
32 import org.apache.mina.common.IoHandler;
33 import org.apache.mina.common.IoService;
34 import org.apache.mina.common.IoServiceConfig;
35 import org.apache.mina.common.IoSession;
36 import org.apache.mina.common.IoSessionConfig;
37 import org.apache.mina.common.TransportType;
38 import org.apache.mina.common.support.BaseIoSession;
39 import org.apache.mina.filter.codec.ProtocolDecoderOutput;
40
41
42
43
44
45
46
47 public class TextLineDecoderTest extends TestCase {
48 public static void main(String[] args) {
49 junit.textui.TestRunner.run(TextLineDecoderTest.class);
50 }
51
52 public void testNormalDecode() throws Exception {
53 TextLineDecoder decoder = new TextLineDecoder(Charset.forName("UTF-8"),
54 LineDelimiter.WINDOWS);
55
56 CharsetEncoder encoder = Charset.forName("UTF-8").newEncoder();
57 IoSession session = new DummySession();
58 TestDecoderOutput out = new TestDecoderOutput();
59 ByteBuffer in = ByteBuffer.allocate(16);
60
61
62 in.putString("ABC\r\n", encoder);
63 in.flip();
64 decoder.decode(session, in, out);
65 Assert.assertEquals(1, out.getMessageQueue().size());
66 Assert.assertEquals("ABC", out.getMessageQueue().poll());
67
68
69 in.clear();
70 in.putString("DEF", encoder);
71 in.flip();
72 decoder.decode(session, in, out);
73 Assert.assertEquals(0, out.getMessageQueue().size());
74 in.clear();
75 in.putString("GHI\r\n", encoder);
76 in.flip();
77 decoder.decode(session, in, out);
78 Assert.assertEquals(1, out.getMessageQueue().size());
79 Assert.assertEquals("DEFGHI", out.getMessageQueue().poll());
80
81
82 in.clear();
83 in.putString("JKL\r\nMNO\r\n", encoder);
84 in.flip();
85 decoder.decode(session, in, out);
86 Assert.assertEquals(2, out.getMessageQueue().size());
87 Assert.assertEquals("JKL", out.getMessageQueue().poll());
88 Assert.assertEquals("MNO", out.getMessageQueue().poll());
89
90
91 decoder = new TextLineDecoder(Charset.forName("UTF-8"),
92 new LineDelimiter("\n\n\n"));
93 in.clear();
94 in.putString("PQR\n", encoder);
95 in.flip();
96 decoder.decode(session, in, out);
97 Assert.assertEquals(0, out.getMessageQueue().size());
98 in.clear();
99 in.putString("\n", encoder);
100 in.flip();
101 decoder.decode(session, in, out);
102 Assert.assertEquals(0, out.getMessageQueue().size());
103 in.clear();
104 in.putString("\n", encoder);
105 in.flip();
106 decoder.decode(session, in, out);
107 Assert.assertEquals(1, out.getMessageQueue().size());
108 Assert.assertEquals("PQR", out.getMessageQueue().poll());
109
110
111 decoder = new TextLineDecoder(Charset.forName("UTF-8"),
112 new LineDelimiter("\n\n\n"));
113 in.clear();
114 in.putString("PQR\n", encoder);
115 in.flip();
116 decoder.decode(session, in, out);
117 Assert.assertEquals(0, out.getMessageQueue().size());
118 in.clear();
119 in.putString("\n", encoder);
120 in.flip();
121 decoder.decode(session, in, out);
122 Assert.assertEquals(0, out.getMessageQueue().size());
123 in.clear();
124 in.putString("\nSTU\n\n\n", encoder);
125 in.flip();
126 decoder.decode(session, in, out);
127 Assert.assertEquals(2, out.getMessageQueue().size());
128 Assert.assertEquals("PQR", out.getMessageQueue().poll());
129 Assert.assertEquals("STU", out.getMessageQueue().poll());
130
131
132 decoder = new TextLineDecoder(Charset.forName("UTF-8"),
133 new LineDelimiter("\n\n\n"));
134 in.clear();
135 in.putString("PQR\n", encoder);
136 in.flip();
137 decoder.decode(session, in, out);
138 Assert.assertEquals(0, out.getMessageQueue().size());
139 in.clear();
140 in.putString("X\n", encoder);
141 in.flip();
142 decoder.decode(session, in, out);
143 Assert.assertEquals(0, out.getMessageQueue().size());
144 in.clear();
145 in.putString("\n\nSTU\n\n\n", encoder);
146 in.flip();
147 decoder.decode(session, in, out);
148 Assert.assertEquals(2, out.getMessageQueue().size());
149 Assert.assertEquals("PQR\nX", out.getMessageQueue().poll());
150 Assert.assertEquals("STU", out.getMessageQueue().poll());
151 }
152
153 public void testAutoDecode() throws Exception {
154 TextLineDecoder decoder = new TextLineDecoder(Charset.forName("UTF-8"),
155 LineDelimiter.AUTO);
156
157 CharsetEncoder encoder = Charset.forName("UTF-8").newEncoder();
158 IoSession session = new DummySession();
159 TestDecoderOutput out = new TestDecoderOutput();
160 ByteBuffer in = ByteBuffer.allocate(16);
161
162
163 in.putString("ABC\r\n", encoder);
164 in.flip();
165 decoder.decode(session, in, out);
166 Assert.assertEquals(1, out.getMessageQueue().size());
167 Assert.assertEquals("ABC", out.getMessageQueue().poll());
168
169
170 in.clear();
171 in.putString("DEF", encoder);
172 in.flip();
173 decoder.decode(session, in, out);
174 Assert.assertEquals(0, out.getMessageQueue().size());
175 in.clear();
176 in.putString("GHI\r\n", encoder);
177 in.flip();
178 decoder.decode(session, in, out);
179 Assert.assertEquals(1, out.getMessageQueue().size());
180 Assert.assertEquals("DEFGHI", out.getMessageQueue().poll());
181
182
183 in.clear();
184 in.putString("JKL\r\nMNO\r\n", encoder);
185 in.flip();
186 decoder.decode(session, in, out);
187 Assert.assertEquals(2, out.getMessageQueue().size());
188 Assert.assertEquals("JKL", out.getMessageQueue().poll());
189 Assert.assertEquals("MNO", out.getMessageQueue().poll());
190
191
192 in.clear();
193 in.putString("\n\n\n", encoder);
194 in.flip();
195 decoder.decode(session, in, out);
196 Assert.assertEquals(3, out.getMessageQueue().size());
197 Assert.assertEquals("", out.getMessageQueue().poll());
198 Assert.assertEquals("", out.getMessageQueue().poll());
199 Assert.assertEquals("", out.getMessageQueue().poll());
200
201
202 in.clear();
203 in.putString("PQR\r", encoder);
204 in.flip();
205 decoder.decode(session, in, out);
206 Assert.assertEquals(0, out.getMessageQueue().size());
207 in.clear();
208 in.putString("\r", encoder);
209 in.flip();
210 decoder.decode(session, in, out);
211 Assert.assertEquals(0, out.getMessageQueue().size());
212 in.clear();
213 in.putString("\n", encoder);
214 in.flip();
215 decoder.decode(session, in, out);
216 Assert.assertEquals(1, out.getMessageQueue().size());
217 Assert.assertEquals("PQR", out.getMessageQueue().poll());
218
219
220 in.clear();
221 in.putString("PQR\r", encoder);
222 in.flip();
223 decoder.decode(session, in, out);
224 Assert.assertEquals(0, out.getMessageQueue().size());
225 in.clear();
226 in.putString("\r", encoder);
227 in.flip();
228 decoder.decode(session, in, out);
229 Assert.assertEquals(0, out.getMessageQueue().size());
230 in.clear();
231 in.putString("\nSTU\r\r\n", encoder);
232 in.flip();
233 decoder.decode(session, in, out);
234 Assert.assertEquals(2, out.getMessageQueue().size());
235 Assert.assertEquals("PQR", out.getMessageQueue().poll());
236 Assert.assertEquals("STU", out.getMessageQueue().poll());
237
238
239 in.clear();
240 in.putString("PQR\r", encoder);
241 in.flip();
242 decoder.decode(session, in, out);
243 Assert.assertEquals(0, out.getMessageQueue().size());
244 in.clear();
245 in.putString("X\r", encoder);
246 in.flip();
247 decoder.decode(session, in, out);
248 Assert.assertEquals(0, out.getMessageQueue().size());
249 in.clear();
250 in.putString("\r\nSTU\r\r\n", encoder);
251 in.flip();
252 decoder.decode(session, in, out);
253 Assert.assertEquals(2, out.getMessageQueue().size());
254 Assert.assertEquals("PQR\rX", out.getMessageQueue().poll());
255 Assert.assertEquals("STU", out.getMessageQueue().poll());
256 }
257
258 private static class DummySession extends BaseIoSession {
259 @Override
260 protected void updateTrafficMask() {
261 }
262
263 public IoService getService() {
264 return null;
265 }
266
267 public IoServiceConfig getServiceConfig() {
268 return null;
269 }
270
271 public IoHandler getHandler() {
272 return null;
273 }
274
275 public IoFilterChain getFilterChain() {
276 return null;
277 }
278
279 public TransportType getTransportType() {
280 return null;
281 }
282
283 public SocketAddress getRemoteAddress() {
284 return null;
285 }
286
287 public SocketAddress getLocalAddress() {
288 return null;
289 }
290
291 public int getScheduledWriteRequests() {
292 return 0;
293 }
294
295 public IoSessionConfig getConfig() {
296 return null;
297 }
298
299 public SocketAddress getServiceAddress() {
300 return null;
301 }
302
303 public int getScheduledWriteBytes() {
304 return 0;
305 }
306 }
307
308 private static class TestDecoderOutput implements ProtocolDecoderOutput {
309 private Queue<Object> messageQueue = new LinkedList<Object>();
310
311 public void write(Object message) {
312 messageQueue.add(message);
313 }
314
315 public Queue<Object> getMessageQueue() {
316 return messageQueue;
317 }
318
319 public void flush() {
320 }
321 }
322 }