1   package org.codehaus.xfire.attachments;
2   
3   import javax.activation.DataHandler;
4   import javax.activation.DataSource;
5   import javax.activation.FileDataSource;
6   
7   import org.codehaus.xfire.test.AbstractXFireTest;
8   import org.codehaus.xfire.type.mtom.MTOMDataContentHandlerFactory;
9   
10  /***
11   * @author <a href="mailto:dan@envoisolutiosn.com">Dan Diephouse</a>
12   */
13  public class DataHandlerTest
14      extends AbstractXFireTest
15  {
16      public void testText()
17          throws Exception
18      {
19          DataSource ds = new FileDataSource(
20              getTestFile("src/test/org/codehaus/xfire/attachments/test.txt"));
21  
22          DataHandler handler = new DataHandler(ds);
23  
24          Object content = handler.getContent();
25          assertNotNull(content);
26          assertTrue(content instanceof String);
27          assertEquals("bleh", content);
28      }
29      
30      public void testImages()
31          throws Exception
32      {
33          DataSource ds = new FileDataSource(
34              getTestFile("src/test/org/codehaus/xfire/attachments/xfire_logo.jpg"));
35      
36          MTOMDataContentHandlerFactory factory = new MTOMDataContentHandlerFactory();
37  
38          Object content = factory.createDataContentHandler("image/jpeg").getContent(ds);
39          assertNotNull(content);
40          assertTrue(content instanceof java.awt.Image);
41      }
42  }