271 lines
8.8 KiB
Java
271 lines
8.8 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.xssf.eventusermodel;
|
|
|
|
import static org.apache.poi.POITestCase.assertContains;
|
|
|
|
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);
|
|
assertContains(text, "Line 1");
|
|
assertContains(text, "Line 2");
|
|
assertContains(text, "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) {
|
|
assertContains(e.getMessage(), "57699");
|
|
return;
|
|
}
|
|
|
|
XSSFReader.SheetIterator it = (XSSFReader.SheetIterator) r.getSheetsData();
|
|
|
|
String text = getShapesString(it);
|
|
assertContains(text, "Line 1");
|
|
assertContains(text, "Line 2");
|
|
assertContains(text, "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());
|
|
|
|
pkg.close();
|
|
}
|
|
|
|
/**
|
|
* NPE when sheet has no relationship id in the workbook
|
|
* 60825
|
|
*/
|
|
public void testSheetWithNoRelationshipId() throws Exception {
|
|
OPCPackage pkg = XSSFTestDataSamples.openSamplePackage("60825.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();
|
|
assertNotNull(iter.next());
|
|
assertFalse(iter.hasNext());
|
|
|
|
pkg.close();
|
|
}
|
|
}
|