import java.io.*;
import junit.framework.*;

public class Base64EncodeOutputStreamTest extends TestCase
{
    public Base64EncodeOutputStreamTest(String name)
    {
        super(name);
    }

    private byte[] testData;
    private byte[] expectedResult;

    private void scenario() throws Exception
    {
        InputStream in = new ByteArrayInputStream(testData);
        
        ByteArrayOutputStream byteArrayOut = new ByteArrayOutputStream();
        OutputStream out = new Base64EncodeOutputStream(byteArrayOut);
            
        int c = in.read();
        while(c != -1) {
            out.write(c);
            c = in.read();
        }
        in.close();
        out.close();

        assertTrue(java.util.Arrays.equals(expectedResult, byteArrayOut));
    }

    public void testEmpty() throws Exception
    {
        testData = new byte[] {};
        expectedResult = new byte[] {
        };
        scenario();
    }

    public void test1() throws Exception
    {
        testData = new byte[] { 0 };
        expectedResult = new byte[] {
            'A', 'A', '=', '=',
        };
        scenario();
    }

    public void test2() throws Exception
    {
        testData = new byte[] { 0, 0 };
        expectedResult = new byte[] {
            'A', 'A', 'A', '=',
        };
        scenario();
    }

    public void test10() throws Exception
    {
        testData = new byte[] {
            0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
        };
        expectedResult = new byte[] {
            'A', 'A', 'E', 'C', 'A', 'w', 'Q', 'F', 'B', 'g', 
            'c', 'I', 'C', 'Q', '=', '=',
        };
        scenario();
    }

    public void test20() throws Exception
    {
        testData = new byte[] {
            0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
            10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
        };
        expectedResult = new byte[] {
            'A', 'A', 'E', 'C', 'A', 'w', 'Q', 'F', 'B', 'g', 
            'c', 'I', 'C', 'Q', 'o', 'L', 'D', 'A', '0', 'O',
            'D', 'x', 'A', 'R', 'E', 'h', 'M', '=',
        };
        scenario();
    }

    public void test76LineBreak76() throws Exception
    {
        // 76個のデータ
        testData = new byte[] {
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0,
        };
        
        //76文字を超えたときに改行
        expectedResult = new byte[] {
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', '\n', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', '=', '=', 
        };
        scenario();
    }

    public void test76LineBreak57() throws Exception
    {
        // 57個のデータ
        testData = new byte[] {
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0,
        };
        
        //76文字を超えないぎりぎり
        expectedResult = new byte[] {
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A',
        };
        scenario();
    }

    public void test76LineBreak58() throws Exception
    {
        // 58個のデータ
        testData = new byte[] {
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 
        };
        
        //76文字を超えた
        expectedResult = new byte[] {
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', '\n', 'A', 'A', '=',
            '=',
        };
        scenario();
    }

    public void test76LineBreak56() throws Exception
    {
        // 56個のデータ
        testData = new byte[] {
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
            0, 0, 0, 0, 0, 0,
        };
        
        //76文字を超えた
        expectedResult = new byte[] {
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A',  'A', 
            'A', 'A', 'A', 'A', 'A', '=',
        };
        scenario();
    }

    public static void main (String[] args)
    {
        Test test;
        if (args.length == 0) {
            test = suite();
        }
        else {
            TestSuite suite = new TestSuite();
            for (int i = 0; i < args.length; ++i) {
                suite.addTest(testNamed(args[i]));
            }
            test = suite;
        }
        junit.textui.TestRunner.run(test);
    }

    public static Test testNamed(String name)
    {
        return new Base64EncodeOutputStreamTest(name);
    }

    public static Test suite()
    {
        return new TestSuite(Base64EncodeOutputStreamTest.class);
    }
}