123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157 |
- /*
- * ====================================================================
- * 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.junit.jupiter.api.Assertions.assertEquals;
- import static org.junit.jupiter.api.Assertions.assertThrows;
-
- import java.io.IOException;
- import java.io.InputStream;
- import java.util.List;
- import java.util.regex.Pattern;
-
- import org.apache.poi.POIDataSamples;
- import org.apache.poi.openxml4j.opc.OPCPackage;
- import org.apache.poi.openxml4j.opc.PackagePart;
- import org.apache.poi.ss.usermodel.RichTextString;
- import org.apache.poi.xssf.model.SharedStringsTable;
- import org.apache.poi.xssf.usermodel.XSSFWorkbook;
- import org.junit.jupiter.api.Test;
- import org.xml.sax.SAXException;
-
- /**
- * Tests for {@link org.apache.poi.xssf.eventusermodel.XSSFReader}
- */
- public final class TestReadOnlySharedStringsTable {
- private static final POIDataSamples _ssTests = POIDataSamples.getSpreadSheetInstance();
-
- @Test
- void testParse() throws Exception {
- try (OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("SampleSS.xlsx"))) {
- List<PackagePart> parts = pkg.getPartsByName(Pattern.compile("/xl/sharedStrings.xml"));
- assertEquals(1, parts.size());
-
- try (SharedStringsTable stbl = new SharedStringsTable(parts.get(0))) {
- ReadOnlySharedStringsTable rtbl = new ReadOnlySharedStringsTable(parts.get(0));
- ReadOnlySharedStringsTable rtbl2;
- try (InputStream stream = parts.get(0).getInputStream()) {
- rtbl2 = new ReadOnlySharedStringsTable(stream);
- }
-
- assertEquals(stbl.getCount(), rtbl.getCount());
- assertEquals(stbl.getUniqueCount(), rtbl.getUniqueCount());
- assertEquals(stbl.getUniqueCount(), rtbl2.getUniqueCount());
-
- assertEquals(stbl.getCount(), stbl.getUniqueCount());
- assertEquals(rtbl.getCount(), rtbl.getUniqueCount());
- assertEquals(rtbl.getCount(), rtbl2.getUniqueCount());
- for (int i = 0; i < stbl.getUniqueCount(); i++) {
- RichTextString i1 = stbl.getItemAt(i);
- assertEquals(i1.getString(), rtbl.getItemAt(i).getString());
- assertEquals(i1.getString(), rtbl2.getItemAt(i).getString());
- }
- }
- }
- }
-
- @Test
- void testParseMalformedCountFile() throws Exception {
- try (OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("MalformedSSTCount.xlsx"))) {
- List<PackagePart> parts = pkg.getPartsByName(Pattern.compile("/xl/sharedStrings.xml"));
- assertEquals(1, parts.size());
-
- try (SharedStringsTable stbl = new SharedStringsTable(parts.get(0))) {
- ReadOnlySharedStringsTable rtbl = new ReadOnlySharedStringsTable(parts.get(0));
- ReadOnlySharedStringsTable rtbl2;
- try (InputStream stream = parts.get(0).getInputStream()) {
- rtbl2 = new ReadOnlySharedStringsTable(stream);
- }
-
- assertEquals(stbl.getCount(), rtbl.getCount());
- assertEquals(stbl.getUniqueCount(), rtbl.getUniqueCount());
- assertEquals(stbl.getUniqueCount(), rtbl2.getUniqueCount());
- for (int i = 0; i < stbl.getUniqueCount(); i++) {
- RichTextString i1 = stbl.getItemAt(i);
- assertEquals(i1.getString(), rtbl.getItemAt(i).getString());
- assertEquals(i1.getString(), rtbl2.getItemAt(i).getString());
- }
- }
- }
- }
-
- //51519
- @Test
- void testPhoneticRuns() throws Exception {
- try (OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("51519.xlsx"))) {
- List<PackagePart> parts = pkg.getPartsByName(Pattern.compile("/xl/sharedStrings.xml"));
- assertEquals(1, parts.size());
-
- ReadOnlySharedStringsTable rtbl = new ReadOnlySharedStringsTable(parts.get(0), true);
- assertEquals(49, rtbl.getUniqueCount());
-
- assertEquals("\u30B3\u30E1\u30F3\u30C8", rtbl.getItemAt(0).getString());
- assertEquals("\u65E5\u672C\u30AA\u30E9\u30AF\u30EB \u30CB\u30DB\u30F3", rtbl.getItemAt(3).getString());
-
- //now do not include phonetic runs
- rtbl = new ReadOnlySharedStringsTable(parts.get(0), false);
- assertEquals(49, rtbl.getUniqueCount());
-
- assertEquals("\u30B3\u30E1\u30F3\u30C8", rtbl.getItemAt(0).getString());
- assertEquals("\u65E5\u672C\u30AA\u30E9\u30AF\u30EB", rtbl.getItemAt(3).getString());
- }
- }
-
- @Test
- void testEmptySSTOnPackageObtainedViaWorkbook() throws Exception {
- XSSFWorkbook wb = new XSSFWorkbook(_ssTests.openResourceAsStream("noSharedStringTable.xlsx"));
- OPCPackage pkg = wb.getPackage();
- assertEmptySST(pkg);
- wb.close();
- }
-
- @Test
- void testEmptySSTOnPackageDirect() throws Exception {
- try (OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("noSharedStringTable.xlsx"))) {
- assertEmptySST(pkg);
- }
- }
-
- @Test
- void testNullPointerException() throws Exception {
- try (OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("clusterfuzz-testcase-minimized-XLSX2CSVFuzzer-5025401116950528.xlsx"))) {
- assertEmptySST(pkg);
- }
-
- try (OPCPackage pkg = OPCPackage.open(_ssTests.openResourceAsStream("clusterfuzz-testcase-minimized-XLSX2CSVFuzzer-5025401116950528.xlsx"))) {
- List<PackagePart> parts = pkg.getPartsByName(Pattern.compile("/xl/sharedStrings.xml"));
- assertEquals(1, parts.size());
-
- //noinspection resource
- assertThrows(IOException.class,
- () -> new SharedStringsTable(parts.get(0)));
- }
- }
-
- private void assertEmptySST(OPCPackage pkg) throws IOException, SAXException {
- ReadOnlySharedStringsTable sst = new ReadOnlySharedStringsTable(pkg);
- assertEquals(0, sst.getCount());
- assertEquals(0, sst.getUniqueCount());
- }
- }
|