1   /*
2    * $Id: VMAttachmentsTestCase.java 7976 2007-08-21 14:26:13Z dirk.olmes $
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  package org.mule.providers.vm.functional;
11  
12  import org.mule.extras.client.MuleClient;
13  import org.mule.impl.MuleMessage;
14  import org.mule.tck.FunctionalTestCase;
15  import org.mule.umo.UMOMessage;
16  
17  import java.io.File;
18  
19  import javax.activation.DataHandler;
20  import javax.activation.FileDataSource;
21  
22  public class VMAttachmentsTestCase extends FunctionalTestCase
23  {
24      protected String getConfigResources()
25      {
26          return "vm-attachments-test.xml";
27      }
28  
29      public void testAttachments() throws Exception
30      {
31           MuleMessage m = new MuleMessage("Mmm... attachments!");
32          FileDataSource ds = new FileDataSource(new File("transports/vm/src/test/resources/" + getConfigResources()).getAbsoluteFile());
33          m.addAttachment("test-attachment", new DataHandler(ds));
34          MuleClient client = new MuleClient();
35          UMOMessage msg = client.send("vm-in", m);
36          assertNotNull(msg);
37          if(msg.getExceptionPayload()!=null)
38          {
39              fail(msg.getExceptionPayload().getException().getCause().toString());
40          }
41          assertEquals(1, msg.getAttachmentNames().size());
42          assertNotNull(msg.getAttachment("mule"));
43          assertTrue(msg.getAttachment("mule").getContentType().startsWith("image/gif"));
44  
45      }
46  }