123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128 |
- /* ====================================================================
- 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 java.io.ByteArrayInputStream;
- import java.io.IOException;
- import java.io.InputStream;
- import java.io.PushbackInputStream;
- import java.util.Arrays;
-
- import org.apache.poi.hssf.HSSFTestDataSamples;
-
- import junit.framework.TestCase;
-
- /**
- * Class to test that POIFS complains when given an Office 2003 XML
- * of Office Open XML (OOXML, 2007+) document
- */
- public class TestOfficeXMLException extends TestCase {
-
- private static final InputStream openSampleStream(String sampleFileName) {
- return HSSFTestDataSamples.openSampleFileStream(sampleFileName);
- }
- public void testOOXMLException() throws IOException
- {
- InputStream in = openSampleStream("sample.xlsx");
-
- try {
- new POIFSFileSystem(in).close();
- fail("expected exception was not thrown");
- } catch(OfficeXmlFileException e) {
- // expected during successful test
- assertTrue(e.getMessage().indexOf("The supplied data appears to be in the Office 2007+ XML") > -1);
- assertTrue(e.getMessage().indexOf("You are calling the part of POI that deals with OLE2 Office Documents") > -1);
- }
- }
- public void test2003XMLException() throws IOException
- {
- InputStream in = openSampleStream("SampleSS.xml");
-
- try {
- new POIFSFileSystem(in).close();
- fail("expected exception was not thrown");
- } catch(NotOLE2FileException e) {
- // expected during successful test
- assertTrue(e.getMessage().indexOf("The supplied data appears to be a raw XML file") > -1);
- assertTrue(e.getMessage().indexOf("Formats such as Office 2003 XML") > -1);
- }
- }
-
- public void testDetectAsPOIFS() throws IOException {
- // ooxml file isn't
- confirmIsPOIFS("SampleSS.xlsx", false);
-
- // 2003 xml file isn't
- confirmIsPOIFS("SampleSS.xml", false);
-
- // xls file is
- confirmIsPOIFS("SampleSS.xls", true);
-
- // text file isn't
- confirmIsPOIFS("SampleSS.txt", false);
- }
- private void confirmIsPOIFS(String sampleFileName, boolean expectedResult) throws IOException {
- InputStream in = new PushbackInputStream(openSampleStream(sampleFileName), 10);
- try {
- boolean actualResult;
- try {
- actualResult = POIFSFileSystem.hasPOIFSHeader(in);
- } catch (IOException e) {
- throw new RuntimeException(e);
- }
- assertEquals(expectedResult, actualResult);
- } finally {
- in.close();
- }
- }
-
- public void testFileCorruption() throws Exception {
-
- // create test InputStream
- byte[] testData = { (byte)1, (byte)2, (byte)3 };
- InputStream testInput = new ByteArrayInputStream(testData);
-
- // detect header
- InputStream in = new PushbackInputStream(testInput, 10);
- assertFalse(POIFSFileSystem.hasPOIFSHeader(in));
-
- // check if InputStream is still intact
- byte[] test = new byte[3];
- in.read(test);
- assertTrue(Arrays.equals(testData, test));
- assertEquals(-1, in.read());
- }
-
-
- public void testFileCorruptionOPOIFS() throws Exception {
-
- // create test InputStream
- byte[] testData = { (byte)1, (byte)2, (byte)3 };
- InputStream testInput = new ByteArrayInputStream(testData);
-
- // detect header
- InputStream in = new PushbackInputStream(testInput, 10);
- assertFalse(OPOIFSFileSystem.hasPOIFSHeader(in));
-
- // check if InputStream is still intact
- byte[] test = new byte[3];
- in.read(test);
- assertTrue(Arrays.equals(testData, test));
- assertEquals(-1, in.read());
- }
- }
|