Index: pdfbox/src/test/java/org/apache/pdfbox/TestAll.java =================================================================== --- pdfbox/src/test/java/org/apache/pdfbox/TestAll.java (revision 1476591) +++ pdfbox/src/test/java/org/apache/pdfbox/TestAll.java (working copy) @@ -26,6 +26,7 @@ import org.apache.pdfbox.filter.TestFilters; import org.apache.pdfbox.io.TestIOUtils; import org.apache.pdfbox.io.TestRandomAccessBuffer; +import org.apache.pdfbox.io.TestRandomAccessFileOutputStream; import org.apache.pdfbox.io.ccitt.TestCCITTFaxG31DDecodeInputStream; import org.apache.pdfbox.io.ccitt.TestPackedBitArray; import org.apache.pdfbox.pdmodel.TestFDF; @@ -94,6 +95,7 @@ suite.addTestSuite( TestIOUtils.class ); suite.addTestSuite( TestRandomAccessBuffer.class ); + suite.addTestSuite( TestRandomAccessFileOutputStream.class ); suite.addTestSuite( TestImageIOUtils.class ); suite.addTestSuite( TestPackedBitArray.class ); suite.addTestSuite( TestCCITTFaxG31DDecodeInputStream.class ); Index: pdfbox/src/test/java/org/apache/pdfbox/io/TestRandomAccessFileOutputStream.java =================================================================== --- pdfbox/src/test/java/org/apache/pdfbox/io/TestRandomAccessFileOutputStream.java (revision 0) +++ pdfbox/src/test/java/org/apache/pdfbox/io/TestRandomAccessFileOutputStream.java (working copy) @@ -0,0 +1,214 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.pdfbox.io; + +import java.io.File; +import java.io.IOException; + +import org.apache.pdfbox.cos.COSInteger; +import org.apache.pdfbox.cos.COSObject; +import org.apache.pdfbox.cos.COSString; + +import junit.framework.TestCase; + +public class TestRandomAccessFileOutputStream extends TestCase +{ + private File testResultsDir = new File("target/test-output"); + + @Override + protected void setUp() throws Exception + { + super.setUp(); + testResultsDir.mkdirs(); + } + + public void testWrite() throws IOException + { + RandomAccessFileOutputStream out; + byte buffer[]; + + File file = new File(testResultsDir, "raf-outputstream.bin"); + + file.delete(); + + RandomAccessFile raFile = new RandomAccessFile(file, "rw"); + + // Test single byte writes + buffer = createDataSequence(16, 10); + out = new RandomAccessFileOutputStream(raFile); + for (byte b : buffer) + { + out.write(b); + } + assertEquals(0, out.getPosition()); + assertEquals(16, out.getLength()); + assertEquals(16, out.getLengthWritten()); + assertEquals(null, out.getExpectedLength()); + assertEquals(16, raFile.length()); + assertEquals(16, raFile.getPosition()); + out.close(); + + raFile.seek(0); + + // Test no write + out = new RandomAccessFileOutputStream(raFile); + assertEquals(16, out.getPosition()); + assertEquals(0, out.getLength()); + assertEquals(0, out.getLengthWritten()); + assertEquals(null, out.getExpectedLength()); + assertEquals(16, raFile.length()); + assertEquals(0, raFile.getPosition()); + out.close(); + + raFile.seek(0); + + // Test buffer writes + buffer = createDataSequence(8, 30); + out = new RandomAccessFileOutputStream(raFile); + out.write(buffer); + assertEquals(16, out.getPosition()); + assertEquals(8, out.getLength()); + assertEquals(8, out.getLengthWritten()); + assertEquals(null, out.getExpectedLength()); + assertEquals(24, raFile.length()); + assertEquals(24, raFile.getPosition()); + out.close(); + + raFile.seek(0); + + // Test partial buffer writes + buffer = createDataSequence(16, 50); + out = new RandomAccessFileOutputStream(raFile); + out.write(buffer, 8, 4); + out.write(buffer, 4, 2); + assertEquals(24, out.getPosition()); + assertEquals(6, out.getLength()); + assertEquals(6, out.getLengthWritten()); + assertEquals(null, out.getExpectedLength()); + assertEquals(30, raFile.length()); + assertEquals(30, raFile.getPosition()); + out.close(); + + // Verify written data + buffer = new byte[(int) raFile.length()]; + raFile.seek(0); + assertEquals(buffer.length, raFile.read(buffer, 0, buffer.length)); + assertEquals(10, buffer[0]); + assertEquals(11, buffer[1]); + assertEquals(25, buffer[15]); + + assertEquals(30, buffer[16]); + assertEquals(31, buffer[17]); + assertEquals(37, buffer[23]); + + assertEquals(58, buffer[24]); + assertEquals(59, buffer[25]); + assertEquals(60, buffer[26]); + assertEquals(61, buffer[27]); + assertEquals(54, buffer[28]); + assertEquals(55, buffer[29]); + + // Cleanup + raFile.close(); + file.delete(); + } + + public void testExpectedLength() throws IOException + { + RandomAccessFileOutputStream out; + + File file = new File(testResultsDir, "raf-outputstream2.bin"); + + file.delete(); + + RandomAccessFile raFile = new RandomAccessFile(file, "rw"); + + byte buffer[] = createDataSequence(16, 10); + + // Test COSInteger + out = new RandomAccessFileOutputStream(raFile); + out.setExpectedLength(COSInteger.get(24)); + out.write(buffer); + assertEquals(0, out.getPosition()); + assertEquals(24, out.getLength()); + assertEquals(16, out.getLengthWritten()); + assertEquals(COSInteger.get(24), out.getExpectedLength()); + assertEquals(16, raFile.length()); + assertEquals(16, raFile.getPosition()); + out.close(); + + raFile.seek(0); + + // Test COSInteger -1 + out = new RandomAccessFileOutputStream(raFile); + out.setExpectedLength(COSInteger.get(-1)); + out.write(buffer); + assertEquals(16, out.getPosition()); + assertEquals(16, out.getLength()); + assertEquals(16, out.getLengthWritten()); + assertEquals(COSInteger.get(-1), out.getExpectedLength()); + assertEquals(32, raFile.length()); + assertEquals(32, raFile.getPosition()); + out.close(); + + raFile.seek(0); + + // Test COSObject + out = new RandomAccessFileOutputStream(raFile); + COSObject expLength = new COSObject(COSInteger.get(24)); + out.setExpectedLength(expLength); + out.write(buffer); + assertEquals(32, out.getPosition()); + assertEquals(24, out.getLength()); + assertEquals(16, out.getLengthWritten()); + assertSame(expLength, out.getExpectedLength()); + assertEquals(48, raFile.length()); + assertEquals(48, raFile.getPosition()); + out.close(); + + raFile.seek(0); + + // Test COSString + out = new RandomAccessFileOutputStream(raFile); + out.setExpectedLength(new COSString("24")); + out.write(buffer); + assertEquals(48, out.getPosition()); + assertEquals(16, out.getLength()); + assertEquals(16, out.getLengthWritten()); + assertEquals(new COSString("24"), out.getExpectedLength()); + assertEquals(64, raFile.length()); + assertEquals(64, raFile.getPosition()); + out.close(); + + raFile.seek(0); + + // Cleanup + raFile.close(); + file.delete(); + } + + protected byte[] createDataSequence(int length, int firstByteValue) + { + byte buffer[] = new byte[length]; + for (int i = 0; i < buffer.length; i++) + { + buffer[i] = (byte) (firstByteValue + i); + } + + return buffer; + } +} Property changes on: pdfbox/src/test/java/org/apache/pdfbox/io/TestRandomAccessFileOutputStream.java ___________________________________________________________________ Added: svn:eol-style ## -0,0 +1 ## +native \ No newline at end of property