base64test.java

JXTA&#8482 is a set of open, generalized peer-to-peer (P2P) protocols that allow any networked devi

/*
 * Copyright (c) 2002-2007 Sun Microsystems, Inc.  All rights reserved.
 *  
 *  The Sun Project JXTA(TM) Software License
 *  
 *  Redistribution and use in source and binary forms, with or without 
 *  modification, are permitted provided that the following conditions are met:
 *  
 *  1. Redistributions of source code must retain the above copyright notice,
 *     this list of conditions and the following disclaimer.
 *  
 *  2. Redistributions in binary form must reproduce the above copyright notice, 
 *     this list of conditions and the following disclaimer in the documentation 
 *     and/or other materials provided with the distribution.
 *  
 *  3. The end-user documentation included with the redistribution, if any, must 
 *     include the following acknowledgment: "This product includes software 
 *     developed by Sun Microsystems, Inc. for JXTA(TM) technology." 
 *     Alternately, this acknowledgment may appear in the software itself, if 
 *     and wherever such third-party acknowledgments normally appear.
 *  
 *  4. The names "Sun", "Sun Microsystems, Inc.", "JXTA" and "Project JXTA" must 
 *     not be used to endorse or promote products derived from this software 
 *     without prior written permission. For written permission, please contact 
 *     Project JXTA at http://www.jxta.org.
 *  
 *  5. Products derived from this software may not be called "JXTA", nor may 
 *     "JXTA" appear in their name, without prior written permission of Sun.
 *  
 *  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES,
 *  INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND 
 *  FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SUN 
 *  MICROSYSTEMS OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 
 *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 
 *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, 
 *  OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 
 *  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 
 *  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, 
 *  EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *  
 *  JXTA is a registered trademark of Sun Microsystems, Inc. in the United 
 *  States and other countries.
 *  
 *  Please see the license information page at :
 *  <http://www.jxta.org/project/www/license.html> for instructions on use of 
 *  the license in source files.
 *  
 *  ====================================================================
 *  
 *  This software consists of voluntary contributions made by many individuals 
 *  on behalf of Project JXTA. For more information on Project JXTA, please see 
 *  http://www.jxta.org.
 *  
 *  This license is based on the BSD license adopted by the Apache Foundation. 
 */

package net.jxta.impl.util;


import junit.framework.*;

import java.io.*;
import java.util.*;


/**
 *
 * @author mike
 */
public class BASE64Test extends TestCase {
    
    public BASE64Test(java.lang.String testName) {
        super(testName);
    }
    
    public static void main(java.lang.String[] args) {
        junit.textui.TestRunner.run(suite());
        
        System.err.flush();
        System.out.flush();
    }
    
    public static Test suite() {
        TestSuite suite = new TestSuite(BASE64Test.class);

        return suite;
    }
    
    public void testRoundTrip0() {
        byte testVector1[] = {};
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTrip1() {
        byte testVector1[] = { 0 };
        byte testVector3[] = { 127 };
        byte testVector4[] = { -128 };
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTrip2() {
        byte testVector1[] = { 0, 0 };
        byte testVector2[] = { 0, 1 };
        byte testVector3[] = { 127, 127 };
        byte testVector4[] = { -128, -128 };
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTrip3() {
        byte testVector1[] = { 0, 0, 0 };
        byte testVector2[] = { 0, 1, 2 };
        byte testVector3[] = { 127, 127, 127 };
        byte testVector4[] = { -128, -128, -128 };
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTrip4() {
        byte testVector1[] = { 0, 0, 0, 0 };
        byte testVector2[] = { 0, 1, 2, 3 };
        byte testVector3[] = { 127, 127, 127, 127 };
        byte testVector4[] = { -128, -128, -128, -128 };
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTrip80() {
        byte testVector1[] = new byte[80];

        Arrays.fill(testVector1, (byte) 0);
        byte testVector2[] = new byte[80];

        Arrays.fill(testVector1, (byte) 1);
        byte testVector3[] = new byte[80];

        Arrays.fill(testVector1, (byte) 127);
        byte testVector4[] = new byte[80];

        Arrays.fill(testVector1, (byte) -128);
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTrip1024() {
        byte testVector1[] = new byte[1024];

        Arrays.fill(testVector1, (byte) 0);
        byte testVector2[] = new byte[1024];

        Arrays.fill(testVector2, (byte) 1);
        byte testVector3[] = new byte[1024];

        Arrays.fill(testVector3, (byte) 127);
        byte testVector4[] = new byte[1024];

        Arrays.fill(testVector4, (byte) -128);
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTrip131072() {
        byte testVector1[] = new byte[131072];

        Arrays.fill(testVector1, (byte) 0);
        byte testVector2[] = new byte[131072];

        Arrays.fill(testVector1, (byte) 1);
        byte testVector3[] = new byte[131072];

        Arrays.fill(testVector1, (byte) 127);
        byte testVector4[] = new byte[131072];

        Arrays.fill(testVector1, (byte) -128);
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTest(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTripCR1024() {
        byte testVector1[] = new byte[1024];

        Arrays.fill(testVector1, (byte) 0);
        byte testVector2[] = new byte[1024];

        Arrays.fill(testVector2, (byte) 1);
        byte testVector3[] = new byte[1024];

        Arrays.fill(testVector3, (byte) 127);
        byte testVector4[] = new byte[1024];

        Arrays.fill(testVector4, (byte) -128);
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public void testRoundTripCR131072Random() {
        Random random = new Random();
        byte testVector1[] = new byte[131072];

        random.nextBytes(testVector1);
        byte testVector2[] = new byte[131072];

        random.nextBytes(testVector2);
        byte testVector3[] = new byte[131072];

        random.nextBytes(testVector3);
        byte testVector4[] = new byte[131072];

        random.nextBytes(testVector4);
        
        try {
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector1));
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector2));
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector3));
            assertTrue("vector didn't round trip successfully", roundTripTestCR(testVector4));
        } catch (IOException failed) {
            fail("Exception " + failed);
        }
    }
    
    public boolean roundTripTest(byte[] source) throws IOException {
        
        StringWriter base64Writer = new StringWriter();
        
        OutputStream out = new BASE64OutputStream(base64Writer);
        
        out.write(source);
        out.close();
        
        StringReader base64Reader = new StringReader(base64Writer.toString());
        
        InputStream input = new BASE64InputStream(base64Reader);
        
        DataInput di = new DataInputStream(input);
        
        byte result[] = new byte[source.length];
        
        di.readFully(result);
        
        if (-1 != input.read()) {
            throw new IOException("Not at EOF");
        }
        
        return Arrays.equals(source, result);
    }
    
    /**
     *  A round trip test that also compares whether our encoder works against
     *  an alternate sun private implemenation.
     **/
    public boolean roundTripTestCR(byte[] source) throws IOException {
        
        StringWriter base64Writer = new StringWriter();
        
        OutputStream out = new BASE64OutputStream(base64Writer, 72);
        
        out.write(source);
        out.close();
        
        StringReader base64Reader = new StringReader(base64Writer.toString());
        
        sun.misc.BASE64Encoder sunencoder = new sun.misc.BASE64Encoder();
        String sunencoded = sunencoder.encode(source);
        
        assertEquals("encodings not equal", sunencoded, base64Writer.toString());
        
        sun.misc.BASE64Decoder sundecoder = new sun.misc.BASE64Decoder();
        byte sundecoded[] = sundecoder.decodeBuffer(base64Writer.toString());
        
        assertTrue("decodings not equal", Arrays.equals(source, sundecoded));

        InputStream input = new BASE64InputStream(base64Reader);
        
        DataInput di = new DataInputStream(input);
        
        byte result[] = new byte[source.length];
        
        di.readFully(result);
        
        if (-1 != input.read()) {
            throw new IOException("Not at EOF");
        }
        
        return Arrays.equals(source, result);
    }
}