Du kan inte välja fler än 25 ämnen Ämnen måste starta med en bokstav eller siffra, kan innehålla bindestreck ('-') och vara max 35 tecken långa.

TestSXSSFWorkbook.java 30KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738
  1. /*
  2. * ====================================================================
  3. * Licensed to the Apache Software Foundation (ASF) under one or more
  4. * contributor license agreements. See the NOTICE file distributed with
  5. * this work for additional information regarding copyright ownership.
  6. * The ASF licenses this file to You under the Apache License, Version 2.0
  7. * (the "License"); you may not use this file except in compliance with
  8. * the License. You may obtain a copy of the License at
  9. *
  10. * http://www.apache.org/licenses/LICENSE-2.0
  11. *
  12. * Unless required by applicable law or agreed to in writing, software
  13. * distributed under the License is distributed on an "AS IS" BASIS,
  14. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. * See the License for the specific language governing permissions and
  16. * limitations under the License.
  17. * ====================================================================
  18. */
  19. package org.apache.poi.xssf.streaming;
  20. import static org.apache.poi.POITestCase.assertEndsWith;
  21. import static org.apache.poi.POITestCase.assertStartsWith;
  22. import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
  23. import static org.junit.jupiter.api.Assertions.assertEquals;
  24. import static org.junit.jupiter.api.Assertions.assertFalse;
  25. import static org.junit.jupiter.api.Assertions.assertNotNull;
  26. import static org.junit.jupiter.api.Assertions.assertSame;
  27. import static org.junit.jupiter.api.Assertions.assertThrows;
  28. import static org.junit.jupiter.api.Assertions.assertTrue;
  29. import java.io.File;
  30. import java.io.FileInputStream;
  31. import java.io.FileOutputStream;
  32. import java.io.IOException;
  33. import java.io.OutputStream;
  34. import java.time.LocalDate;
  35. import java.util.Arrays;
  36. import org.apache.commons.io.output.NullOutputStream;
  37. import org.apache.commons.io.output.UnsynchronizedByteArrayOutputStream;
  38. import org.apache.poi.POIDataSamples;
  39. import org.apache.poi.common.usermodel.HyperlinkType;
  40. import org.apache.poi.openxml4j.opc.OPCPackage;
  41. import org.apache.poi.openxml4j.opc.PackageAccess;
  42. import org.apache.poi.ss.tests.usermodel.BaseTestXWorkbook;
  43. import org.apache.poi.ss.usermodel.Cell;
  44. import org.apache.poi.ss.usermodel.CellStyle;
  45. import org.apache.poi.ss.usermodel.CellType;
  46. import org.apache.poi.ss.usermodel.DataFormatter;
  47. import org.apache.poi.ss.usermodel.Row;
  48. import org.apache.poi.ss.usermodel.Sheet;
  49. import org.apache.poi.ss.usermodel.Workbook;
  50. import org.apache.poi.ss.usermodel.WorkbookFactory;
  51. import org.apache.poi.ss.util.CellRangeAddress;
  52. import org.apache.poi.ss.util.CellReference;
  53. import org.apache.poi.xssf.SXSSFITestDataProvider;
  54. import org.apache.poi.xssf.XSSFTestDataSamples;
  55. import org.apache.poi.xssf.model.SharedStringsTable;
  56. import org.apache.poi.xssf.usermodel.DefaultIndexedColorMap;
  57. import org.apache.poi.xssf.usermodel.XSSFCell;
  58. import org.apache.poi.xssf.usermodel.XSSFColor;
  59. import org.apache.poi.xssf.usermodel.XSSFFont;
  60. import org.apache.poi.xssf.usermodel.XSSFHyperlink;
  61. import org.apache.poi.xssf.usermodel.XSSFRichTextString;
  62. import org.apache.poi.xssf.usermodel.XSSFSheet;
  63. import org.apache.poi.xssf.usermodel.XSSFWorkbook;
  64. import org.junit.jupiter.api.AfterEach;
  65. import org.junit.jupiter.api.Disabled;
  66. import org.junit.jupiter.api.Test;
  67. import org.junit.jupiter.params.ParameterizedTest;
  68. import org.junit.jupiter.params.provider.ValueSource;
  69. public final class TestSXSSFWorkbook extends BaseTestXWorkbook {
  70. public TestSXSSFWorkbook() {
  71. super(SXSSFITestDataProvider.instance);
  72. }
  73. @AfterEach
  74. void tearDown(){
  75. ((SXSSFITestDataProvider)_testDataProvider).cleanup();
  76. }
  77. /**
  78. * cloning of sheets is not supported in SXSSF
  79. */
  80. @Override
  81. @Test
  82. public void cloneSheet() throws IOException {
  83. RuntimeException e = assertThrows(RuntimeException.class, super::cloneSheet);
  84. assertEquals("Not Implemented", e.getMessage());
  85. }
  86. /**
  87. * cloning of sheets is not supported in SXSSF
  88. */
  89. @Override
  90. @Test
  91. public void sheetClone() {
  92. RuntimeException e = assertThrows(RuntimeException.class, super::sheetClone);
  93. assertEquals("Not Implemented", e.getMessage());
  94. }
  95. /**
  96. * Skip this test, as SXSSF doesn't update formulas on sheet name
  97. * changes.
  98. */
  99. @Override
  100. @Disabled("SXSSF doesn't update formulas on sheet name changes, as most cells probably aren't in memory at the time")
  101. protected void setSheetName() {
  102. }
  103. @Test
  104. void existingWorkbook() throws IOException {
  105. try (XSSFWorkbook xssfWb1 = new XSSFWorkbook()) {
  106. xssfWb1.createSheet("S1");
  107. try (SXSSFWorkbook wb1 = new SXSSFWorkbook(xssfWb1);
  108. XSSFWorkbook xssfWb2 = SXSSFITestDataProvider.instance.writeOutAndReadBack(wb1)) {
  109. assertTrue(wb1.dispose());
  110. try (SXSSFWorkbook wb2 = new SXSSFWorkbook(xssfWb2)) {
  111. assertEquals(1, wb2.getNumberOfSheets());
  112. Sheet sheet = wb2.getSheetAt(0);
  113. assertNotNull(sheet);
  114. assertEquals("S1", sheet.getSheetName());
  115. assertTrue(wb2.dispose());
  116. }
  117. }
  118. }
  119. }
  120. @Test
  121. void useSharedStringsTable() throws Exception {
  122. try (SXSSFWorkbook wb = new SXSSFWorkbook(null, 10, false, true)) {
  123. SharedStringsTable sss = wb.getSharedStringSource();
  124. assertNotNull(sss);
  125. Row row = wb.createSheet("S1").createRow(0);
  126. row.createCell(0).setCellValue("A");
  127. row.createCell(1).setCellValue("B");
  128. row.createCell(2).setCellValue("A");
  129. try (XSSFWorkbook xssfWorkbook = SXSSFITestDataProvider.instance.writeOutAndReadBack(wb)) {
  130. sss = wb.getSharedStringSource();
  131. assertEquals(2, sss.getUniqueCount());
  132. assertTrue(wb.dispose());
  133. Sheet sheet1 = xssfWorkbook.getSheetAt(0);
  134. assertEquals("S1", sheet1.getSheetName());
  135. assertEquals(1, sheet1.getPhysicalNumberOfRows());
  136. row = sheet1.getRow(0);
  137. assertNotNull(row);
  138. Cell cell = row.getCell(0);
  139. assertNotNull(cell);
  140. assertEquals("A", cell.getStringCellValue());
  141. cell = row.getCell(1);
  142. assertNotNull(cell);
  143. assertEquals("B", cell.getStringCellValue());
  144. cell = row.getCell(2);
  145. assertNotNull(cell);
  146. assertEquals("A", cell.getStringCellValue());
  147. }
  148. }
  149. }
  150. @Test
  151. void useSharedStringsTableWithRichText() throws Exception {
  152. testUseSharedStringsTableWithRichText(false);
  153. }
  154. @Test
  155. void useSharedStringsTableWithRichTextAndCompression() throws Exception {
  156. testUseSharedStringsTableWithRichText(true);
  157. }
  158. @Test
  159. void addToExistingWorkbook() throws IOException {
  160. try (XSSFWorkbook xssfWb1 = new XSSFWorkbook()) {
  161. xssfWb1.createSheet("S1");
  162. Sheet sheet = xssfWb1.createSheet("S2");
  163. Row row = sheet.createRow(1);
  164. Cell cell = row.createCell(1);
  165. cell.setCellValue("value 2_1_1");
  166. try (SXSSFWorkbook wb1 = new SXSSFWorkbook(xssfWb1);
  167. XSSFWorkbook xssfWb2 = SXSSFITestDataProvider.instance.writeOutAndReadBack(wb1)) {
  168. assertTrue(wb1.dispose());
  169. try (SXSSFWorkbook wb2 = new SXSSFWorkbook(xssfWb2)) {
  170. // Add a row to the existing empty sheet
  171. Sheet sheet1 = wb2.getSheetAt(0);
  172. Row row1_1 = sheet1.createRow(1);
  173. Cell cell1_1_1 = row1_1.createCell(1);
  174. cell1_1_1.setCellValue("value 1_1_1");
  175. // Add a row to the existing non-empty sheet
  176. Sheet sheet2 = wb2.getSheetAt(1);
  177. Row row2_2 = sheet2.createRow(2);
  178. Cell cell2_2_1 = row2_2.createCell(1);
  179. cell2_2_1.setCellValue("value 2_2_1");
  180. // Add a sheet with one row
  181. Sheet sheet3 = wb2.createSheet("S3");
  182. Row row3_1 = sheet3.createRow(1);
  183. Cell cell3_1_1 = row3_1.createCell(1);
  184. cell3_1_1.setCellValue("value 3_1_1");
  185. try (XSSFWorkbook xssfWb3 = SXSSFITestDataProvider.instance.writeOutAndReadBack(wb2)) {
  186. assertEquals(3, xssfWb3.getNumberOfSheets());
  187. // Verify sheet 1
  188. sheet1 = xssfWb3.getSheetAt(0);
  189. assertEquals("S1", sheet1.getSheetName());
  190. assertEquals(1, sheet1.getPhysicalNumberOfRows());
  191. row1_1 = sheet1.getRow(1);
  192. assertNotNull(row1_1);
  193. cell1_1_1 = row1_1.getCell(1);
  194. assertNotNull(cell1_1_1);
  195. assertEquals("value 1_1_1", cell1_1_1.getStringCellValue());
  196. // Verify sheet 2
  197. sheet2 = xssfWb3.getSheetAt(1);
  198. assertEquals("S2", sheet2.getSheetName());
  199. assertEquals(2, sheet2.getPhysicalNumberOfRows());
  200. Row row2_1 = sheet2.getRow(1);
  201. assertNotNull(row2_1);
  202. Cell cell2_1_1 = row2_1.getCell(1);
  203. assertNotNull(cell2_1_1);
  204. assertEquals("value 2_1_1", cell2_1_1.getStringCellValue());
  205. row2_2 = sheet2.getRow(2);
  206. assertNotNull(row2_2);
  207. cell2_2_1 = row2_2.getCell(1);
  208. assertNotNull(cell2_2_1);
  209. assertEquals("value 2_2_1", cell2_2_1.getStringCellValue());
  210. // Verify sheet 3
  211. sheet3 = xssfWb3.getSheetAt(2);
  212. assertEquals("S3", sheet3.getSheetName());
  213. assertEquals(1, sheet3.getPhysicalNumberOfRows());
  214. row3_1 = sheet3.getRow(1);
  215. assertNotNull(row3_1);
  216. cell3_1_1 = row3_1.getCell(1);
  217. assertNotNull(cell3_1_1);
  218. assertEquals("value 3_1_1", cell3_1_1.getStringCellValue());
  219. }
  220. }
  221. }
  222. }
  223. }
  224. @Test
  225. void sheetdataWriter() throws IOException{
  226. try (SXSSFWorkbook wb = new SXSSFWorkbook()) {
  227. SXSSFSheet sh = wb.createSheet();
  228. SheetDataWriter wr = sh.getSheetDataWriter();
  229. assertSame(wr.getClass(), SheetDataWriter.class);
  230. File tmp = wr.getTempFile();
  231. assertStartsWith(tmp.getName(), "poi-sxssf-sheet");
  232. assertEndsWith(tmp.getName(), ".xml");
  233. assertTrue(wb.dispose());
  234. }
  235. try (SXSSFWorkbook wb = new SXSSFWorkbook()) {
  236. wb.setCompressTempFiles(true);
  237. SXSSFSheet sh = wb.createSheet();
  238. SheetDataWriter wr = sh.getSheetDataWriter();
  239. assertSame(wr.getClass(), GZIPSheetDataWriter.class);
  240. File tmp = wr.getTempFile();
  241. assertStartsWith(tmp.getName(), "poi-sxssf-sheet-xml");
  242. assertEndsWith(tmp.getName(), ".gz");
  243. assertTrue(wb.dispose());
  244. }
  245. //Test escaping of Unicode control characters
  246. try (SXSSFWorkbook wb = new SXSSFWorkbook()) {
  247. wb.createSheet("S1").createRow(0).createCell(0).setCellValue("value\u0019");
  248. try (XSSFWorkbook xssfWorkbook = SXSSFITestDataProvider.instance.writeOutAndReadBack(wb)) {
  249. Cell cell = xssfWorkbook.getSheet("S1").getRow(0).getCell(0);
  250. assertEquals("value?", cell.getStringCellValue());
  251. assertTrue(wb.dispose());
  252. }
  253. }
  254. }
  255. @Test
  256. void gzipSheetdataWriter() throws IOException {
  257. try (SXSSFWorkbook wb = new SXSSFWorkbook()) {
  258. wb.setCompressTempFiles(true);
  259. final int rowNum = 1000;
  260. final int sheetNum = 5;
  261. populateData(wb);
  262. try (XSSFWorkbook xwb = SXSSFITestDataProvider.instance.writeOutAndReadBack(wb)) {
  263. for (int i = 0; i < sheetNum; i++) {
  264. Sheet sh = xwb.getSheetAt(i);
  265. assertEquals("sheet" + i, sh.getSheetName());
  266. for (int j = 0; j < rowNum; j++) {
  267. Row row = sh.getRow(j);
  268. assertNotNull(row, "row[" + j + "]");
  269. Cell cell1 = row.getCell(0);
  270. assertEquals(new CellReference(cell1).formatAsString(), cell1.getStringCellValue());
  271. Cell cell2 = row.getCell(1);
  272. assertEquals(i, (int) cell2.getNumericCellValue());
  273. Cell cell3 = row.getCell(2);
  274. assertEquals(j, (int) cell3.getNumericCellValue());
  275. }
  276. }
  277. assertTrue(wb.dispose());
  278. }
  279. }
  280. }
  281. private static void populateData(Workbook wb) {
  282. for(int i = 0; i < 5; i++){
  283. Sheet sh = wb.createSheet("sheet" + i);
  284. for(int j = 0; j < 1000; j++){
  285. Row row = sh.createRow(j);
  286. Cell cell1 = row.createCell(0);
  287. cell1.setCellValue(new CellReference(cell1).formatAsString());
  288. Cell cell2 = row.createCell(1);
  289. cell2.setCellValue(i);
  290. Cell cell3 = row.createCell(2);
  291. cell3.setCellValue(j);
  292. }
  293. }
  294. }
  295. @ParameterizedTest
  296. @ValueSource(booleans = {false, true})
  297. void workbookDispose(boolean compressTempFiles) throws IOException {
  298. try (SXSSFWorkbook wb = new SXSSFWorkbook()) {
  299. // compressTempFiles == false: the underlying writer is SheetDataWriter
  300. // compressTempFiles == true: the underlying writer is GZIPSheetDataWriter
  301. wb.setCompressTempFiles(compressTempFiles);
  302. populateData(wb);
  303. for (Sheet sheet : wb) {
  304. SXSSFSheet sxSheet = (SXSSFSheet) sheet;
  305. assertTrue(sxSheet.getSheetDataWriter().getTempFile().exists());
  306. }
  307. assertTrue(wb.dispose());
  308. for (Sheet sheet : wb) {
  309. SXSSFSheet sxSheet = (SXSSFSheet) sheet;
  310. assertFalse(sxSheet.getSheetDataWriter().getTempFile().exists());
  311. }
  312. }
  313. }
  314. @Test
  315. void bug53515() throws Exception {
  316. try (Workbook wb1 = new SXSSFWorkbook(10)) {
  317. populateWorkbook(wb1);
  318. assertDoesNotThrow(() -> wb1.write(NullOutputStream.INSTANCE));
  319. assertDoesNotThrow(() -> wb1.write(NullOutputStream.INSTANCE));
  320. try (Workbook wb2 = new XSSFWorkbook()) {
  321. populateWorkbook(wb2);
  322. assertDoesNotThrow(() -> wb2.write(NullOutputStream.INSTANCE));
  323. assertDoesNotThrow(() -> wb2.write(NullOutputStream.INSTANCE));
  324. }
  325. }
  326. }
  327. @Disabled("Crashes the JVM because of documented JVM behavior with concurrent writing/reading of zip-files, "
  328. + "see http://www.oracle.com/technetwork/java/javase/documentation/overview-156328.html")
  329. @Test
  330. void bug53515a() throws Exception {
  331. File out = new File("Test.xlsx");
  332. assertTrue(!out.exists() || out.delete());
  333. for (int i = 0; i < 2; i++) {
  334. final SXSSFWorkbook wb;
  335. if (out.exists()) {
  336. wb = new SXSSFWorkbook(
  337. (XSSFWorkbook) WorkbookFactory.create(out));
  338. } else {
  339. wb = new SXSSFWorkbook(10);
  340. }
  341. try {
  342. FileOutputStream outSteam = new FileOutputStream(out);
  343. if (i == 0) {
  344. populateWorkbook(wb);
  345. } else {
  346. /*
  347. Code explicitly invokes garbage collection. Except for specific use in benchmarking,
  348. this is very dubious.
  349. In the past, situations where people have explicitly invoked the garbage collector in
  350. routines such as close or finalize methods has led to huge performance black holes.
  351. Garbage collection can be expensive. Any situation that forces hundreds or thousands
  352. of garbage collections will bring the machine to a crawl.
  353. */
  354. //System.gc();
  355. //System.gc();
  356. //System.gc();
  357. }
  358. wb.write(outSteam);
  359. // assertTrue(wb.dispose());
  360. outSteam.close();
  361. } finally {
  362. assertTrue(wb.dispose());
  363. }
  364. wb.close();
  365. }
  366. assertTrue(out.exists());
  367. assertTrue(out.delete());
  368. }
  369. private static void populateWorkbook(Workbook wb) {
  370. Sheet sh = wb.createSheet();
  371. for (int rownum = 0; rownum < 100; rownum++) {
  372. Row row = sh.createRow(rownum);
  373. for (int cellnum = 0; cellnum < 10; cellnum++) {
  374. Cell cell = row.createCell(cellnum);
  375. String address = new CellReference(cell).formatAsString();
  376. cell.setCellValue(address);
  377. }
  378. }
  379. }
  380. @Test
  381. void closeDoesNotModifyWorkbook() throws IOException {
  382. final String filename = "SampleSS.xlsx";
  383. final File file = POIDataSamples.getSpreadSheetInstance().getFile(filename);
  384. // Some tests commented out because close() modifies the file
  385. // See bug 58779
  386. // String
  387. //wb = new SXSSFWorkbook(new XSSFWorkbook(file.getPath()));
  388. //assertCloseDoesNotModifyFile(filename, wb);
  389. // File
  390. //wb = new SXSSFWorkbook(new XSSFWorkbook(file));
  391. //assertCloseDoesNotModifyFile(filename, wb);
  392. // InputStream
  393. try (FileInputStream fis = new FileInputStream(file);
  394. XSSFWorkbook xwb = new XSSFWorkbook(fis);
  395. SXSSFWorkbook wb = new SXSSFWorkbook(xwb)) {
  396. assertCloseDoesNotModifyFile(filename, wb);
  397. }
  398. // OPCPackage
  399. //wb = new SXSSFWorkbook(new XSSFWorkbook(OPCPackage.open(file)));
  400. //assertCloseDoesNotModifyFile(filename, wb);
  401. }
  402. /**
  403. * Bug #59743
  404. *
  405. * this is only triggered on other files apart of sheet[1,2,...].xml
  406. * as those are either copied uncompressed or with the use of GZIPInputStream
  407. * so we use shared strings
  408. */
  409. @Test
  410. void testZipBombNotTriggeredOnUselessContent() throws IOException {
  411. try (SXSSFWorkbook swb = new SXSSFWorkbook(null, 1, true, true)) {
  412. SXSSFSheet s = swb.createSheet();
  413. char[] useless = new char[32767];
  414. Arrays.fill(useless, ' ');
  415. for (int row = 0; row < 10; row++) {
  416. Row r = s.createRow(row);
  417. for (int col = 0; col < 10; col++) {
  418. char[] prefix = Integer.toHexString(row * 10 + col).toCharArray();
  419. Arrays.fill(useless, 0, 10, ' ');
  420. System.arraycopy(prefix, 0, useless, 0, prefix.length);
  421. String ul = new String(useless);
  422. r.createCell(col, CellType.STRING).setCellValue(ul);
  423. }
  424. }
  425. assertDoesNotThrow(() -> swb.write(NullOutputStream.INSTANCE));
  426. swb.dispose();
  427. }
  428. }
  429. /**
  430. * To avoid accident changes to the template, you should be able
  431. * to create a SXSSFWorkbook from a read-only XSSF one, then
  432. * change + save that (only). See bug #60010
  433. * TODO Fix this to work!
  434. */
  435. @Test
  436. @Disabled
  437. void createFromReadOnlyWorkbook() throws Exception {
  438. String sheetName = "Test SXSSF";
  439. File input = XSSFTestDataSamples.getSampleFile("sample.xlsx");
  440. try (OPCPackage pkg = OPCPackage.open(input, PackageAccess.READ)) {
  441. UnsynchronizedByteArrayOutputStream bos = UnsynchronizedByteArrayOutputStream.builder().get();
  442. try (XSSFWorkbook xssf = new XSSFWorkbook(pkg)) {
  443. try (SXSSFWorkbook wb = new SXSSFWorkbook(xssf, 2)) {
  444. Sheet s = wb.createSheet(sheetName);
  445. for (int i = 0; i < 10; i++) {
  446. Row r = s.createRow(i);
  447. r.createCell(0).setCellValue(true);
  448. r.createCell(1).setCellValue(2.4);
  449. r.createCell(2).setCellValue("Test Row " + i);
  450. }
  451. assertEquals(10, s.getLastRowNum());
  452. wb.write(bos);
  453. wb.dispose();
  454. }
  455. }
  456. try (XSSFWorkbook xssf = new XSSFWorkbook(bos.toInputStream())) {
  457. Sheet s = xssf.getSheet(sheetName);
  458. assertEquals(10, s.getLastRowNum());
  459. assertTrue(s.getRow(0).getCell(0).getBooleanCellValue());
  460. assertEquals("Test Row 9", s.getRow(9).getCell(2).getStringCellValue());
  461. }
  462. }
  463. }
  464. @Test
  465. void test56557() throws IOException {
  466. try (XSSFWorkbook wb = XSSFTestDataSamples.openSampleWorkbook("56557.xlsx");
  467. // Using streaming XSSFWorkbook makes the output file invalid
  468. Workbook wb2 = new SXSSFWorkbook(wb);
  469. // Should not throw POIXMLException: java.io.IOException: Unable to parse xml bean when reading back
  470. Workbook wbBack = XSSFTestDataSamples.writeOutAndReadBack(wb2)
  471. ) {
  472. assertNotNull(wbBack);
  473. }
  474. }
  475. @Test
  476. void addHyperlink() throws Exception {
  477. try (
  478. SXSSFWorkbook wb = new SXSSFWorkbook();
  479. UnsynchronizedByteArrayOutputStream bos = UnsynchronizedByteArrayOutputStream.builder().get()
  480. ) {
  481. SXSSFSheet sheet = wb.createSheet("s1");
  482. SXSSFRow row = sheet.createRow(0);
  483. SXSSFCell cell = row.createCell(0);
  484. cell.setCellValue("Example Website");
  485. XSSFHyperlink hyperlink = (XSSFHyperlink)wb.getCreationHelper().createHyperlink(HyperlinkType.URL);
  486. hyperlink.setAddress("http://example.com");
  487. hyperlink.setCellReference("A1");
  488. sheet.addHyperlink(hyperlink);
  489. wb.write(bos);
  490. try (XSSFWorkbook xssfWorkbook = new XSSFWorkbook(bos.toInputStream())) {
  491. XSSFSheet xssfSheet = xssfWorkbook.getSheet(sheet.getSheetName());
  492. XSSFCell xssfCell = xssfSheet.getRow(0).getCell(0);
  493. assertEquals("Example Website", xssfCell.getStringCellValue());
  494. XSSFHyperlink xssfHyperlink = xssfCell.getHyperlink();
  495. assertEquals(hyperlink.getAddress(), xssfHyperlink.getAddress());
  496. }
  497. }
  498. }
  499. @Test
  500. void addDimension() throws IOException {
  501. try (
  502. SXSSFWorkbook wb = new SXSSFWorkbook();
  503. UnsynchronizedByteArrayOutputStream bos = UnsynchronizedByteArrayOutputStream.builder().get()
  504. ) {
  505. SXSSFSheet sheet = wb.createSheet();
  506. sheet.createRow(2).createCell(3).setCellValue("top left");
  507. sheet.createRow(6).createCell(5).setCellValue("bottom right");
  508. assertEquals(2, sheet.getFirstRowNum());
  509. assertEquals(6, sheet.getLastRowNum());
  510. wb.write(bos);
  511. try (XSSFWorkbook xssfWorkbook = new XSSFWorkbook(bos.toInputStream())) {
  512. XSSFSheet xssfSheet = xssfWorkbook.getSheetAt(0);
  513. assertEquals(CellRangeAddress.valueOf("D3:F7"), xssfSheet.getDimension());
  514. }
  515. }
  516. }
  517. @Test
  518. void addDimension1() throws IOException {
  519. try (
  520. SXSSFWorkbook wb = new SXSSFWorkbook(1);
  521. UnsynchronizedByteArrayOutputStream bos = UnsynchronizedByteArrayOutputStream.builder().get()
  522. ) {
  523. SXSSFSheet sheet = wb.createSheet();
  524. sheet.createRow(2).createCell(3).setCellValue("top left");
  525. sheet.createRow(6).createCell(5).setCellValue("bottom right");
  526. assertEquals(2, sheet.getFirstRowNum());
  527. assertEquals(6, sheet.getLastRowNum());
  528. wb.write(bos);
  529. try (XSSFWorkbook xssfWorkbook = new XSSFWorkbook(bos.toInputStream())) {
  530. XSSFSheet xssfSheet = xssfWorkbook.getSheetAt(0);
  531. assertEquals(CellRangeAddress.valueOf("D3:F7"), xssfSheet.getDimension());
  532. }
  533. }
  534. }
  535. @Test
  536. void addDimensionXSSFtoSXSSF() throws IOException {
  537. try (XSSFWorkbook wb = new XSSFWorkbook()) {
  538. XSSFSheet sheet = wb.createSheet();
  539. sheet.createRow(2).createCell(3).setCellValue("top left");
  540. sheet.createRow(6).createCell(5).setCellValue("bottom right");
  541. assertEquals(2, sheet.getFirstRowNum());
  542. assertEquals(6, sheet.getLastRowNum());
  543. try (
  544. SXSSFWorkbook sxssfWorkbook = new SXSSFWorkbook(wb);
  545. UnsynchronizedByteArrayOutputStream bos = UnsynchronizedByteArrayOutputStream.builder().get()
  546. ) {
  547. sxssfWorkbook.write(bos);
  548. try (XSSFWorkbook xssfWorkbook = new XSSFWorkbook(bos.toInputStream())) {
  549. XSSFSheet xssfSheet = xssfWorkbook.getSheetAt(0);
  550. assertEquals(CellRangeAddress.valueOf("D3:F7"), xssfSheet.getDimension());
  551. }
  552. }
  553. }
  554. }
  555. @Test
  556. void addDimensionDisabled() throws IOException {
  557. try (
  558. SXSSFWorkbook wb = new SXSSFWorkbook();
  559. UnsynchronizedByteArrayOutputStream bos = UnsynchronizedByteArrayOutputStream.builder().get()
  560. ) {
  561. wb.setShouldCalculateSheetDimensions(false);
  562. SXSSFSheet sheet = wb.createSheet();
  563. sheet.createRow(2).createCell(3).setCellValue("top left");
  564. sheet.createRow(6).createCell(5).setCellValue("bottom right");
  565. assertEquals(2, sheet.getFirstRowNum());
  566. assertEquals(6, sheet.getLastRowNum());
  567. wb.write(bos);
  568. try (XSSFWorkbook xssfWorkbook = new XSSFWorkbook(bos.toInputStream())) {
  569. XSSFSheet xssfSheet = xssfWorkbook.getSheetAt(0);
  570. assertEquals(CellRangeAddress.valueOf("A1:A1"), xssfSheet.getDimension());
  571. }
  572. }
  573. }
  574. @Test
  575. void getStylesSource() throws IOException {
  576. try (SXSSFWorkbook workbook = new SXSSFWorkbook()) {
  577. assertNotNull(workbook.getXSSFWorkbook().getStylesSource(), "style source available");
  578. assertNotNull(workbook.getXSSFWorkbook().getStylesSource().getIndexedColors(), "indexed colors available");
  579. }
  580. }
  581. @Test
  582. void dateStyle() throws IOException {
  583. try (
  584. SXSSFWorkbook workbook = new SXSSFWorkbook();
  585. UnsynchronizedByteArrayOutputStream bos = UnsynchronizedByteArrayOutputStream.builder().get()
  586. ) {
  587. SXSSFSheet sheet = workbook.createSheet();
  588. SXSSFRow row = sheet.createRow(0);
  589. SXSSFCreationHelper createHelper = (SXSSFCreationHelper) workbook.getCreationHelper();
  590. CellStyle cellStyle = workbook.createCellStyle();
  591. cellStyle.setDataFormat(createHelper.createDataFormat().getFormat("dd.MM.yyyy"));
  592. SXSSFCell cell = row.createCell(0);
  593. cell.setCellValue(LocalDate.parse("2023-03-07"));
  594. cell.setCellStyle(cellStyle);
  595. workbook.write(bos);
  596. try (XSSFWorkbook xssfWorkbook = new XSSFWorkbook(bos.toInputStream())) {
  597. XSSFSheet xssfSheet = xssfWorkbook.getSheetAt(0);
  598. DataFormatter dataFormatter = new DataFormatter();
  599. final String result = dataFormatter.formatCellValue(xssfSheet.getRow(0).getCell(0));
  600. assertEquals("07.03.2023", result);
  601. }
  602. }
  603. }
  604. @Override
  605. @Disabled("not implemented")
  606. protected void changeSheetNameWithSharedFormulas() {
  607. }
  608. private void testUseSharedStringsTableWithRichText(boolean compressTempFiles) throws Exception {
  609. try (SXSSFWorkbook wb = new SXSSFWorkbook(null, 10, compressTempFiles, true)) {
  610. SharedStringsTable sss = wb.getSharedStringSource();
  611. assertNotNull(sss);
  612. XSSFFont redFont = new XSSFFont();
  613. redFont.setColor(new XSSFColor(new java.awt.Color(241,76,93), new DefaultIndexedColorMap()));
  614. Row row = wb.createSheet("S1").createRow(0);
  615. row.createCell(0).setCellValue("A");
  616. row.createCell(1).setCellValue("B");
  617. XSSFRichTextString rts = new XSSFRichTextString("A");
  618. rts.applyFont(redFont);
  619. row.createCell(2).setCellValue(rts);
  620. try (XSSFWorkbook xssfWorkbook = SXSSFITestDataProvider.instance.writeOutAndReadBack(wb)) {
  621. sss = wb.getSharedStringSource();
  622. assertEquals(3, sss.getUniqueCount());
  623. assertTrue(wb.dispose());
  624. Sheet sheet1 = xssfWorkbook.getSheetAt(0);
  625. assertEquals("S1", sheet1.getSheetName());
  626. assertEquals(1, sheet1.getPhysicalNumberOfRows());
  627. row = sheet1.getRow(0);
  628. assertNotNull(row);
  629. Cell cell = row.getCell(0);
  630. assertNotNull(cell);
  631. assertEquals("A", cell.getStringCellValue());
  632. cell = row.getCell(1);
  633. assertNotNull(cell);
  634. assertEquals("B", cell.getStringCellValue());
  635. cell = row.getCell(2);
  636. assertNotNull(cell);
  637. assertEquals("A", cell.getStringCellValue());
  638. XSSFRichTextString outputRichTextString = (XSSFRichTextString) cell.getRichStringCellValue();
  639. XSSFFont outputFont = outputRichTextString.getFontAtIndex(0);
  640. assertEquals(redFont, outputFont);
  641. }
  642. }
  643. }
  644. @Test
  645. void writeBrokenFile() throws IOException {
  646. try (final Workbook wb = _testDataProvider.openSampleWorkbook("clusterfuzz-testcase-minimized-POIXSSFFuzzer-5185049589579776.xlsx")) {
  647. try (OutputStream out = NullOutputStream.INSTANCE) {
  648. assertThrows(IllegalArgumentException.class,
  649. () -> wb.write(out));
  650. }
  651. }
  652. }
  653. }