base64test.java

来自「Xfire文件 用于开发web service 的一个开源工具 很好用的」· Java 代码 · 共 84 行

package org.codehaus.xfire.aegis.type.basic;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;

import javax.xml.stream.XMLStreamReader;

import org.codehaus.xfire.MessageContext;
import org.codehaus.xfire.aegis.MessageWriter;
import org.codehaus.xfire.aegis.stax.ElementReader;
import org.codehaus.xfire.aegis.stax.ElementWriter;
import org.codehaus.xfire.aegis.type.DefaultTypeMappingRegistry;
import org.codehaus.xfire.aegis.type.TypeMapping;
import org.codehaus.xfire.aegis.type.TypeMappingRegistry;
import org.codehaus.xfire.soap.SoapConstants;
import org.codehaus.xfire.test.AbstractXFireTest;
import org.codehaus.xfire.util.STAXUtils;

public class Base64Test
    extends AbstractXFireTest
{
    TypeMapping mapping;
    
    public void setUp() throws Exception
    {
        super.setUp();
        
        addNamespace("t", "urn:test");
        addNamespace("xsd", SoapConstants.XSD);
        addNamespace("xsi", SoapConstants.XSI_NS);
        
        TypeMappingRegistry reg = new DefaultTypeMappingRegistry(true);
        mapping = reg.createTypeMapping(true);
    }

    public void testBean()
        throws Exception
    {
        Base64Type type = new Base64Type();
        
        byte[] data = new byte[10];
        
        File file = File.createTempFile("base64", "xml");
        FileOutputStream fos = new FileOutputStream(file);
        ElementWriter writer = new ElementWriter(fos, "root", "urn:test");
        MessageWriter b64writer = writer.getElementWriter("base64", "urn:test");
        type.writeObject(data, 
                         b64writer, 
                         new MessageContext());
        b64writer.close();

        b64writer = writer.getElementWriter("base64", "urn:test");
        type.writeObject(data, 
                         b64writer, 
                         new MessageContext());
        b64writer.close();
        
        writer.close();
        writer.flush();
        fos.close();
        
        FileInputStream fis = new FileInputStream(file);
        XMLStreamReader reader = STAXUtils.createXMLStreamReader(fis, null,null);
        reader.next();
        reader.next();
        assertEquals("base64", reader.getLocalName());
        
        byte[] data2 = (byte[]) type.readObject(new ElementReader(reader), new MessageContext());
        assertEquals(data.length, data2.length);
        
        assertEquals("base64", reader.getLocalName());
        assertEquals(XMLStreamReader.START_ELEMENT, reader.getEventType());
        data2 = (byte[]) type.readObject(new ElementReader(reader), new MessageContext());
        assertEquals(data.length, data2.length);
        
        assertEquals(XMLStreamReader.END_ELEMENT, reader.getEventType());
        
        fis.close();
        file.deleteOnExit();
    }
    
}