From d1eeda291c02a9f74e50106c91d393e36a80f09c Mon Sep 17 00:00:00 2001
From: ramkumar <ramkumar@13f79535-47bb-0310-9956-ffa450edef68>
Date: Tue, 10 Feb 2009 10:31:58 +0000
Subject: Fixes for MTOM TestCase

git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@742927 13f79535-47bb-0310-9956-ffa450edef68
---
 .../itests/mtom/FileTransferMTOMTestCase.java      | 98 ++++++++++++++++++----
 1 file changed, 82 insertions(+), 16 deletions(-)

(limited to 'branches/sca-java-1.x/modules/binding-ws-axis2/src/test/java/org/apache/tuscany/sca/binding/ws/axis2/itests/mtom/FileTransferMTOMTestCase.java')

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();
+        }
     }
-
 }
-- 
cgit v1.2.3