123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247 |
- /* ====================================================================
- 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.xssf.eventusermodel;
-
- import java.io.InputStream;
- import java.util.Iterator;
- import java.util.List;
-
- import org.apache.poi.POIDataSamples;
- import org.apache.poi.POIXMLException;
- import org.apache.poi.openxml4j.opc.OPCPackage;
- import org.apache.poi.util.IOUtils;
- import org.apache.poi.xssf.XSSFTestDataSamples;
- import org.apache.poi.xssf.model.CommentsTable;
- import org.apache.poi.xssf.model.StylesTable;
- import org.apache.poi.xssf.usermodel.XSSFRichTextString;
- import org.apache.poi.xssf.usermodel.XSSFShape;
- import org.apache.poi.xssf.usermodel.XSSFSimpleShape;
-
- import junit.framework.TestCase;
-
- /**
- * Tests for {@link XSSFReader}
- */
- public final class TestXSSFReader extends TestCase {
- private static POIDataSamples _ssTests = POIDataSamples.getSpreadSheetInstance();
-
- public void testGetBits() throws Exception {
- OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("SampleSS.xlsx"));
-
- XSSFReader r = new XSSFReader(pkg);
-
- assertNotNull(r.getWorkbookData());
- assertNotNull(r.getSharedStringsData());
- assertNotNull(r.getStylesData());
-
- assertNotNull(r.getSharedStringsTable());
- assertNotNull(r.getStylesTable());
- }
-
- public void testStyles() throws Exception {
- OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("SampleSS.xlsx"));
-
- XSSFReader r = new XSSFReader(pkg);
-
- assertEquals(3, r.getStylesTable().getFonts().size());
- assertEquals(0, r.getStylesTable().getNumDataFormats());
-
- // The Styles Table should have the themes associated with it too
- assertNotNull(r.getStylesTable().getTheme());
-
- // Check we get valid data for the two
- assertNotNull(r.getStylesData());
- assertNotNull(r.getThemesData());
- }
-
- public void testStrings() throws Exception {
- OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("SampleSS.xlsx"));
-
- XSSFReader r = new XSSFReader(pkg);
-
- assertEquals(11, r.getSharedStringsTable().getItems().size());
- assertEquals("Test spreadsheet", new XSSFRichTextString(r.getSharedStringsTable().getEntryAt(0)).toString());
- }
-
- public void testSheets() throws Exception {
- OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("SampleSS.xlsx"));
-
- XSSFReader r = new XSSFReader(pkg);
- byte[] data = new byte[4096];
-
- // By r:id
- assertNotNull(r.getSheet("rId2"));
- int read = IOUtils.readFully(r.getSheet("rId2"), data);
- assertEquals(974, read);
-
- // All
- Iterator<InputStream> it = r.getSheetsData();
-
- int count = 0;
- while(it.hasNext()) {
- count++;
- InputStream inp = it.next();
- assertNotNull(inp);
- read = IOUtils.readFully(inp, data);
- inp.close();
-
- assertTrue(read > 400);
- assertTrue(read < 1500);
- }
- assertEquals(3, count);
- }
-
- /**
- * Check that the sheet iterator returns sheets in the logical order
- * (as they are defined in the workbook.xml)
- */
- public void testOrderOfSheets() throws Exception {
- OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("reordered_sheets.xlsx"));
-
- XSSFReader r = new XSSFReader(pkg);
-
- String[] sheetNames = {"Sheet4", "Sheet2", "Sheet3", "Sheet1"};
- XSSFReader.SheetIterator it = (XSSFReader.SheetIterator)r.getSheetsData();
-
- int count = 0;
- while(it.hasNext()) {
- InputStream inp = it.next();
- assertNotNull(inp);
- inp.close();
-
- assertEquals(sheetNames[count], it.getSheetName());
- count++;
- }
- assertEquals(4, count);
- }
-
- public void testComments() throws Exception {
- OPCPackage pkg = XSSFTestDataSamples.openSamplePackage("comments.xlsx");
- XSSFReader r = new XSSFReader(pkg);
- XSSFReader.SheetIterator it = (XSSFReader.SheetIterator)r.getSheetsData();
-
- int count = 0;
- while(it.hasNext()) {
- count++;
- InputStream inp = it.next();
- inp.close();
-
- if(count == 1) {
- assertNotNull(it.getSheetComments());
- CommentsTable ct = it.getSheetComments();
- assertEquals(1, ct.getNumberOfAuthors());
- assertEquals(3, ct.getNumberOfComments());
- } else {
- assertNull(it.getSheetComments());
- }
- }
- assertEquals(3, count);
- }
-
- /**
- * Iterating over a workbook with chart sheets in it, using the
- * XSSFReader method
- * @throws Exception
- */
- public void test50119() throws Exception {
- OPCPackage pkg = XSSFTestDataSamples.openSamplePackage("WithChartSheet.xlsx");
- XSSFReader r = new XSSFReader(pkg);
- XSSFReader.SheetIterator it = (XSSFReader.SheetIterator)r.getSheetsData();
-
- while(it.hasNext())
- {
- InputStream stream = it.next();
- stream.close();
- }
- }
-
- /**
- * Test text extraction from text box using getShapes()
- *
- * @throws Exception
- */
- public void testShapes() throws Exception {
- OPCPackage pkg = XSSFTestDataSamples.openSamplePackage("WithTextBox.xlsx");
- XSSFReader r = new XSSFReader(pkg);
- XSSFReader.SheetIterator it = (XSSFReader.SheetIterator) r.getSheetsData();
-
- String text = getShapesString(it);
- assertTrue(text.contains("Line 1"));
- assertTrue(text.contains("Line 2"));
- assertTrue(text.contains("Line 3"));
- }
-
- private String getShapesString(XSSFReader.SheetIterator it) {
- StringBuilder sb = new StringBuilder();
- while (it.hasNext()) {
- it.next();
- List<XSSFShape> shapes = it.getShapes();
- if (shapes != null) {
- for (XSSFShape shape : shapes) {
- if (shape instanceof XSSFSimpleShape) {
- String t = ((XSSFSimpleShape) shape).getText();
- sb.append(t).append('\n');
- }
- }
- }
- }
- return sb.toString();
- }
-
- public void testBug57914() throws Exception {
- OPCPackage pkg = XSSFTestDataSamples.openSamplePackage("57914.xlsx");
- final XSSFReader r;
-
- // for now expect this to fail, when we fix 57699, this one should fail so we know we should adjust
- // this test as well
- try {
- r = new XSSFReader(pkg);
- fail("This will fail until bug 57699 is fixed");
- } catch (POIXMLException e) {
- assertTrue("Had " + e, e.getMessage().contains("57699"));
- return;
- }
-
- XSSFReader.SheetIterator it = (XSSFReader.SheetIterator) r.getSheetsData();
-
- String text = getShapesString(it);
- assertTrue(text.contains("Line 1"));
- assertTrue(text.contains("Line 2"));
- assertTrue(text.contains("Line 3"));
- }
-
- /**
- * NPE from XSSFReader$SheetIterator.<init> on XLSX files generated by
- * the openpyxl library
- */
- public void test58747() throws Exception {
- OPCPackage pkg = XSSFTestDataSamples.openSamplePackage("58747.xlsx");
- ReadOnlySharedStringsTable strings = new ReadOnlySharedStringsTable(pkg);
- assertNotNull(strings);
- XSSFReader reader = new XSSFReader(pkg);
- StylesTable styles = reader.getStylesTable();
- assertNotNull(styles);
-
- XSSFReader.SheetIterator iter = (XSSFReader.SheetIterator) reader.getSheetsData();
- assertEquals(true, iter.hasNext());
- iter.next();
-
- assertEquals(false, iter.hasNext());
- assertEquals("Orders", iter.getSheetName());
- }
- }
|