1   /*
2    * $Id: UdpMessageAdapterTestCase.java 10787 2008-02-12 18:51:50Z dfeist $
3    * --------------------------------------------------------------------------------------
4    * Copyright (c) MuleSource, Inc.  All rights reserved.  http://www.mulesource.com
5    *
6    * The software in this package is published under the terms of the CPAL v1.0
7    * license, a copy of which has been included with this distribution in the
8    * LICENSE.txt file.
9    */
10  
11  package org.mule.transport.udp;
12  
13  import org.mule.DefaultMuleMessage;
14  import org.mule.api.MessagingException;
15  import org.mule.api.transport.MessageAdapter;
16  import org.mule.transport.AbstractMessageAdapterTestCase;
17  import org.mule.transport.udp.UdpMessageAdapter;
18  
19  import java.net.DatagramPacket;
20  
21  public class UdpMessageAdapterTestCase extends AbstractMessageAdapterTestCase
22  {
23  
24      public Object getValidMessage() throws Exception
25      {
26          return new DatagramPacket("Hello".getBytes(), 5);
27      }
28  
29      public MessageAdapter createAdapter(Object payload) throws MessagingException
30      {
31          return new UdpMessageAdapter(payload);
32      }
33  
34      public void testMessageRetrieval() throws Exception
35      {
36          Object message = getValidMessage();
37          MessageAdapter adapter = createAdapter(message);
38          DefaultMuleMessage muleMessage = new DefaultMuleMessage(adapter);
39          assertEquals(new String(((DatagramPacket)message).getData()), muleMessage.getPayloadAsString());
40          byte[] bytes = muleMessage.getPayloadAsBytes();
41          assertNotNull(bytes);
42  
43          String stringMessage = muleMessage.getPayloadAsString();
44          assertNotNull(stringMessage);
45  
46          assertNotNull(adapter.getPayload());
47  
48          try
49          {
50              adapter = createAdapter(getInvalidMessage());
51              fail("Message adapter should throw exception if an invalid messgae is set");
52          }
53          catch (Exception e)
54          {
55              // expected
56          }
57      }
58  
59  }