Fixes for MTOM TestCase

git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@742927 13f79535-47bb-0310-9956-ffa450edef68

1 files changed, 82 insertions, 16 deletions

diff --git a/branches/sca-java-1.x/modules/binding-ws-axis2/src/test/java/org/apache/tuscany/sca/binding/ws/axis2/itests/mtom/FileTransferMTOMTestCase.java b/branches/sca-java-1.x/modules/binding-ws-axis2/src/test/java/org/apache/tuscany/sca/binding/ws/axis2/itests/mtom/FileTransferMTOMTestCase.java
index 69b3b60ea7..a342fa2414 100644
--- a/branches/sca-java-1.x/modules/binding-ws-axis2/src/test/java/org/apache/tuscany/sca/binding/ws/axis2/itests/mtom/FileTransferMTOMTestCase.java
+++ b/branches/sca-java-1.x/modules/binding-ws-axis2/src/test/java/org/apache/tuscany/sca/binding/ws/axis2/itests/mtom/FileTransferMTOMTestCase.java
@@ -21,34 +21,100 @@ package org.apache.tuscany.sca.binding.ws.axis2.itests.mtom;
 

 import javax.activation.DataHandler;

 import javax.activation.FileDataSource;

+import java.awt.Image;

+

+import javax.xml.namespace.QName;

+import javax.xml.transform.Source;

+import org.apache.axiom.om.OMAbstractFactory;

+import org.apache.axiom.om.OMElement;

+import org.apache.axiom.om.impl.llom.OMElementImpl;

+import org.apache.axiom.om.OMText;

+import javax.xml.transform.dom.DOMSource;

+import org.apache.axiom.om.OMFactory;

+import java.awt.image.BufferedImage;

 import junit.framework.TestCase;

-import org.apache.tuscany.sca.binding.ws.axis2.itests.mtom.FileTransferService;

 import org.apache.tuscany.sca.host.embedded.SCADomain;

+import org.apache.tuscany.sca.binding.ws.axis2.itests.mtom.FileTransferServiceClient;

+import org.apache.tuscany.sca.databinding.xml.String2Node;

+import org.junit.AfterClass;

+import org.junit.BeforeClass;

+import org.junit.Test;

 

 public class FileTransferMTOMTestCase extends TestCase {

 

     private SCADomain domain;

+    

+    /**

+     * Runs once before the tests

+     */

+    @BeforeClass

+    protected void setUp() throws Exception {

+    	domain = SCADomain.newInstance("org/apache/tuscany/sca/binding/ws/axis2/itests/mtom/filetransferservice.composite");

+    }

+   

+    /**

+     * Runs once after the tests

+     */

+    @AfterClass

+    protected void tearDown() throws Exception {

+    	domain.close();

+    }

 

-    public void testFileTransfer() throws Exception {

-        try {

+    @Test

+    public void testImageFileTransfer() throws Exception {

+        try {        	

+            Image image = new BufferedImage(80, 24, BufferedImage.TYPE_INT_RGB);

+            

+            FileTransferServiceClient filetransfer = domain.getService(FileTransferServiceClient.class, "FileTransferClientComponent");

+            assertEquals("File uploaded Sucessfully", filetransfer.uploadImageFileForward(image));            

+        } catch (Exception ex){

+            ex.printStackTrace();

+        }

+    }

+    

+    @Test

+    public void testSourceFileTransfer() throws Exception {

+        try {        	 

+        	String xml = "<a>A<b>B</b><c>C</c></a>";

+            Source source = new DOMSource(new String2Node().transform(xml, null));

+            

+            FileTransferServiceClient filetransfer = domain.getService(FileTransferServiceClient.class, "FileTransferClientComponent");

+            assertEquals("File uploaded Sucessfully", filetransfer.uploadSourceFileForward(source));            

+        } catch (Exception ex){

+            ex.printStackTrace();

+        }

+    }

+    

+    @Test

+    public void testDataHandlerFileTransfer() throws Exception {

+        try {            

             // For testing purpose lets try uploading FileTransferClient.java file.

             DataHandler dataHandler = new DataHandler(new FileDataSource("./LICENSE"));

-            FileTransferService filetransfer = domain.getService(FileTransferService.class, "FileTransferClientComponent");

-            //dataHandler.writeTo(System.out);

-            assertEquals("File uploaded Sucessfully", filetransfer.uploadFile(dataHandler));

+            

+            FileTransferServiceClient filetransfer = domain.getService(FileTransferServiceClient.class, "FileTransferClientComponent");

+            assertEquals("File uploaded Sucessfully", filetransfer.uploadDataHandlerFileForward(dataHandler));            

         } catch (Exception ex){

             ex.printStackTrace();

         }

     }

-

-    @Override

-    protected void setUp() throws Exception {

-        domain = SCADomain.newInstance("org/apache/tuscany/sca/binding/ws/axis2/itests/mtom/filetransferservice.composite");

-    }

-   

-    @Override

-    protected void tearDown() throws Exception {

-        domain.close();

+    

+    @Test

+    public void testOMElementFileTransfer() throws Exception {

+        try {        	

+        	OMFactory factory = OMAbstractFactory.getOMFactory();                        

+            OMElement imageElement = factory.createOMElement(new QName("image"));   

+            

+            // For testing purpose lets try uploading FileTransferClient.java file.

+            DataHandler dataHandler = new DataHandler(new FileDataSource("./LICENSE"));

+            

+            OMText textData = factory.createOMText(dataHandler, true);

+            imageElement.addChild(textData);

+                       

+            FileTransferServiceClient filetransfer = domain.getService(FileTransferServiceClient.class, "FileTransferClientComponent");

+            assertEquals("File uploaded Sucessfully", filetransfer.uploadOMElementFileForward(imageElement));

+            

+        } catch (Exception ex){

+            ex.printStackTrace();

+        }

     }

-

 }