176 lines
4.9 KiB
Java
176 lines
4.9 KiB
Java
|
|
/* ====================================================================
|
|
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.poi.poifs.filesystem;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.fail;
|
|
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
import java.util.Arrays;
|
|
|
|
import org.junit.Test;
|
|
|
|
/**
|
|
* Class to test DocumentOutputStream functionality
|
|
*/
|
|
public final class TestDocumentOutputStream {
|
|
|
|
/**
|
|
* test write(int) behavior
|
|
*/
|
|
@Test
|
|
public void testWrite1() throws IOException {
|
|
ByteArrayOutputStream stream = new ByteArrayOutputStream();
|
|
DocumentOutputStream dstream = new DocumentOutputStream(stream, 25);
|
|
|
|
for (int j = 0; j < 25; j++)
|
|
{
|
|
dstream.write(j);
|
|
}
|
|
try
|
|
{
|
|
dstream.write(0);
|
|
fail("Should have caught IOException");
|
|
}
|
|
catch (IOException ignored)
|
|
{
|
|
}
|
|
byte[] output = stream.toByteArray();
|
|
|
|
assertEquals(25, output.length);
|
|
for (int j = 0; j < 25; j++)
|
|
{
|
|
assertEquals(( byte ) j, output[ j ]);
|
|
}
|
|
dstream.close();
|
|
stream.close();
|
|
}
|
|
|
|
/**
|
|
* test write(byte[]) behavior
|
|
*/
|
|
@Test
|
|
public void testWrite2() throws IOException {
|
|
ByteArrayOutputStream stream = new ByteArrayOutputStream();
|
|
DocumentOutputStream dstream = new DocumentOutputStream(stream, 25);
|
|
|
|
for (int j = 0; j < 6; j++)
|
|
{
|
|
byte[] array = new byte[ 4 ];
|
|
|
|
Arrays.fill(array, ( byte ) j);
|
|
dstream.write(array);
|
|
}
|
|
try
|
|
{
|
|
byte[] array = new byte[ 4 ];
|
|
|
|
Arrays.fill(array, ( byte ) 7);
|
|
dstream.write(array);
|
|
fail("Should have caught IOException");
|
|
}
|
|
catch (IOException ignored)
|
|
{
|
|
}
|
|
byte[] output = stream.toByteArray();
|
|
|
|
assertEquals(24, output.length);
|
|
for (int j = 0; j < 6; j++)
|
|
{
|
|
for (int k = 0; k < 4; k++)
|
|
{
|
|
assertEquals(String.valueOf((j * 4) + k), ( byte ) j,
|
|
output[ (j * 4) + k ]);
|
|
}
|
|
}
|
|
dstream.close();
|
|
stream.close();
|
|
}
|
|
|
|
/**
|
|
* test write(byte[], int, int) behavior
|
|
*/
|
|
@Test
|
|
public void testWrite3() throws IOException {
|
|
ByteArrayOutputStream stream = new ByteArrayOutputStream();
|
|
DocumentOutputStream dstream = new DocumentOutputStream(stream, 25);
|
|
byte[] array = new byte[ 50 ];
|
|
|
|
for (int j = 0; j < 50; j++)
|
|
{
|
|
array[ j ] = ( byte ) j;
|
|
}
|
|
dstream.write(array, 1, 25);
|
|
try
|
|
{
|
|
dstream.write(array, 0, 1);
|
|
fail("Should have caught IOException");
|
|
}
|
|
catch (IOException ignored)
|
|
{
|
|
}
|
|
byte[] output = stream.toByteArray();
|
|
|
|
assertEquals(25, output.length);
|
|
for (int j = 0; j < 25; j++)
|
|
{
|
|
assertEquals(( byte ) (j + 1), output[ j ]);
|
|
}
|
|
dstream.close();
|
|
stream.close();
|
|
}
|
|
|
|
/**
|
|
* test writeFiller()
|
|
*/
|
|
@Test
|
|
public void testWriteFiller() throws IOException {
|
|
ByteArrayOutputStream stream = new ByteArrayOutputStream();
|
|
DocumentOutputStream dstream = new DocumentOutputStream(stream, 25);
|
|
|
|
for (int j = 0; j < 25; j++)
|
|
{
|
|
dstream.write(j);
|
|
}
|
|
try
|
|
{
|
|
dstream.write(0);
|
|
fail("Should have caught IOException");
|
|
}
|
|
catch (IOException ignored)
|
|
{
|
|
}
|
|
dstream.writeFiller(100, ( byte ) 0xff);
|
|
byte[] output = stream.toByteArray();
|
|
|
|
assertEquals(100, output.length);
|
|
for (int j = 0; j < 25; j++)
|
|
{
|
|
assertEquals(( byte ) j, output[ j ]);
|
|
}
|
|
for (int j = 25; j < 100; j++)
|
|
{
|
|
assertEquals(String.valueOf(j), ( byte ) 0xff, output[ j ]);
|
|
}
|
|
dstream.close();
|
|
stream.close();
|
|
}
|
|
}
|