1
2
3
4
5
6
7
8
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 }