You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

BaseTestWorkbook.java 38KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931
  1. /* ====================================================================
  2. Licensed to the Apache Software Foundation (ASF) under one or more
  3. contributor license agreements. See the NOTICE file distributed with
  4. this work for additional information regarding copyright ownership.
  5. The ASF licenses this file to You under the Apache License, Version 2.0
  6. (the "License"); you may not use this file except in compliance with
  7. the License. You may obtain a copy of the License at
  8. http://www.apache.org/licenses/LICENSE-2.0
  9. Unless required by applicable law or agreed to in writing, software
  10. distributed under the License is distributed on an "AS IS" BASIS,
  11. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. See the License for the specific language governing permissions and
  13. limitations under the License.
  14. ==================================================================== */
  15. package org.apache.poi.ss.usermodel;
  16. import static org.apache.commons.io.output.NullOutputStream.NULL_OUTPUT_STREAM;
  17. import static org.junit.jupiter.api.Assertions.assertArrayEquals;
  18. import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
  19. import static org.junit.jupiter.api.Assertions.assertEquals;
  20. import static org.junit.jupiter.api.Assertions.assertFalse;
  21. import static org.junit.jupiter.api.Assertions.assertNotNull;
  22. import static org.junit.jupiter.api.Assertions.assertNull;
  23. import static org.junit.jupiter.api.Assertions.assertSame;
  24. import static org.junit.jupiter.api.Assertions.assertThrows;
  25. import static org.junit.jupiter.api.Assertions.assertTrue;
  26. import java.io.IOException;
  27. import java.util.ConcurrentModificationException;
  28. import java.util.Iterator;
  29. import java.util.Spliterator;
  30. import org.apache.poi.hssf.HSSFTestDataSamples;
  31. import org.apache.poi.ss.ITestDataProvider;
  32. import org.apache.poi.ss.usermodel.ClientAnchor.AnchorType;
  33. import org.apache.poi.ss.util.CellAddress;
  34. import org.apache.poi.ss.util.CellRangeAddress;
  35. import org.junit.jupiter.api.Test;
  36. public abstract class BaseTestWorkbook {
  37. protected final ITestDataProvider _testDataProvider;
  38. protected BaseTestWorkbook(ITestDataProvider testDataProvider) {
  39. _testDataProvider = testDataProvider;
  40. }
  41. @Test
  42. void sheetIterator_forEach() throws IOException {
  43. try (Workbook wb = _testDataProvider.createWorkbook()) {
  44. wb.createSheet("Sheet0");
  45. wb.createSheet("Sheet1");
  46. wb.createSheet("Sheet2");
  47. int i = 0;
  48. for (Sheet sh : wb) {
  49. assertEquals("Sheet" + i, sh.getSheetName());
  50. i++;
  51. }
  52. }
  53. }
  54. /**
  55. * Expected ConcurrentModificationException:
  56. * should not be able to advance an iterator when the
  57. * underlying data has been reordered
  58. */
  59. @Test
  60. void sheetIterator_sheetsReordered() throws IOException {
  61. try (Workbook wb = _testDataProvider.createWorkbook()) {
  62. wb.createSheet("Sheet0");
  63. wb.createSheet("Sheet1");
  64. wb.createSheet("Sheet2");
  65. Iterator<Sheet> it = wb.sheetIterator();
  66. Spliterator<Sheet> split = wb.spliterator();
  67. it.next();
  68. split.tryAdvance(sheet -> {});
  69. wb.setSheetOrder("Sheet2", 1);
  70. // Iterator order should be fixed when iterator is created
  71. assertThrows(ConcurrentModificationException.class, it::next);
  72. // Spliterator order should be fixed when spliterator is created
  73. assertThrows(ConcurrentModificationException.class, () -> split.tryAdvance(sheet -> {}));
  74. }
  75. }
  76. /**
  77. * Expected ConcurrentModificationException:
  78. * should not be able to advance an iterator when the
  79. * underlying data has been reordered
  80. */
  81. @Test
  82. void sheetIterator_sheetRemoved() throws IOException {
  83. try (Workbook wb = _testDataProvider.createWorkbook()) {
  84. wb.createSheet("Sheet0");
  85. wb.createSheet("Sheet1");
  86. wb.createSheet("Sheet2");
  87. Iterator<Sheet> it = wb.sheetIterator();
  88. Spliterator<Sheet> split = wb.spliterator();
  89. it.next();
  90. split.tryAdvance(sheet -> {});
  91. wb.removeSheetAt(1);
  92. // Iterator order should be fixed when iterator is created
  93. assertThrows(ConcurrentModificationException.class, it::next);
  94. // Spliterator order should be fixed when spliterator is created
  95. assertThrows(ConcurrentModificationException.class, () -> split.tryAdvance(sheet -> {}));
  96. }
  97. }
  98. /**
  99. * Expected UnsupportedOperationException:
  100. * should not be able to remove sheets from the sheet iterator
  101. */
  102. @Test
  103. void sheetIterator_remove() throws IOException {
  104. try (Workbook wb = _testDataProvider.createWorkbook()) {
  105. wb.createSheet("Sheet0");
  106. Iterator<Sheet> it = wb.sheetIterator();
  107. it.next(); //Sheet0
  108. assertThrows(UnsupportedOperationException.class, it::remove);
  109. }
  110. }
  111. @Test
  112. void createSheet() throws IOException {
  113. try (Workbook wb = _testDataProvider.createWorkbook()) {
  114. assertEquals(0, wb.getNumberOfSheets());
  115. //getting a sheet by invalid index or non-existing name
  116. assertNull(wb.getSheet("Sheet1"));
  117. IllegalArgumentException ex = assertThrows(
  118. IllegalArgumentException.class,
  119. () -> wb.getSheetAt(0),
  120. "should have thrown exception due to invalid sheet index"
  121. );
  122. // expected during successful test no negative index in the range message
  123. assertFalse(ex.getMessage().contains("-1"));
  124. assertThrows(IllegalArgumentException.class, () -> wb.getSheetAt(0));
  125. Sheet sheet0 = wb.createSheet();
  126. Sheet sheet1 = wb.createSheet();
  127. assertEquals("Sheet0", sheet0.getSheetName());
  128. assertEquals("Sheet1", sheet1.getSheetName());
  129. assertEquals(2, wb.getNumberOfSheets());
  130. //fetching sheets by name is case-insensitive
  131. Sheet originalSheet = wb.createSheet("Sheet3");
  132. Sheet fetchedSheet = wb.getSheet("sheet3");
  133. assertNotNull(fetchedSheet, "Identified bug 44892");
  134. assertEquals("Sheet3", fetchedSheet.getSheetName());
  135. assertEquals(3, wb.getNumberOfSheets());
  136. assertSame(originalSheet, fetchedSheet);
  137. ex = assertThrows(
  138. IllegalArgumentException.class,
  139. () -> wb.createSheet("sHeeT3"),
  140. "should have thrown exception due to duplicate sheet name"
  141. );
  142. // expected during successful test
  143. assertEquals("The workbook already contains a sheet named 'sHeeT3'", ex.getMessage());
  144. //names cannot be blank or contain any of /\*?[]
  145. String[] invalidNames = {"", "Sheet/", "Sheet\\",
  146. "Sheet?", "Sheet*", "Sheet[", "Sheet]", "'Sheet'",
  147. "My:Sheet"};
  148. for (String sheetName : invalidNames) {
  149. assertThrows(
  150. IllegalArgumentException.class,
  151. () -> wb.createSheet(sheetName),
  152. "should have thrown exception due to invalid sheet name: " + sheetName
  153. );
  154. }
  155. //still have 3 sheets
  156. assertEquals(3, wb.getNumberOfSheets());
  157. //change the name of the 3rd sheet
  158. wb.setSheetName(2, "I changed!");
  159. //try to assign an invalid name to the 2nd sheet
  160. assertThrows(IllegalArgumentException.class, () -> wb.setSheetName(1, "[I'm invalid]"));
  161. //try to assign an invalid name to the 2nd sheet
  162. assertThrows(IllegalArgumentException.class, () -> wb.createSheet(null));
  163. assertThrows(IllegalArgumentException.class, () -> wb.setSheetName(2, null));
  164. //check
  165. assertEquals(0, wb.getSheetIndex("sheet0"));
  166. assertEquals(1, wb.getSheetIndex("sheet1"));
  167. assertEquals(2, wb.getSheetIndex("I changed!"));
  168. assertSame(sheet0, wb.getSheet("sheet0"));
  169. assertSame(sheet1, wb.getSheet("sheet1"));
  170. assertSame(originalSheet, wb.getSheet("I changed!"));
  171. assertNull(wb.getSheet("unknown"));
  172. //serialize and read again
  173. try (Workbook wb2 = _testDataProvider.writeOutAndReadBack(wb)) {
  174. assertEquals(3, wb2.getNumberOfSheets());
  175. assertEquals(0, wb2.getSheetIndex("sheet0"));
  176. assertEquals(1, wb2.getSheetIndex("sheet1"));
  177. assertEquals(2, wb2.getSheetIndex("I changed!"));
  178. }
  179. }
  180. }
  181. /**
  182. * POI allows creating sheets with names longer than 31 characters.
  183. *
  184. * Excel opens files with long sheet names without error or warning.
  185. * However, long sheet names are silently truncated to 31 chars. In order to
  186. * avoid funny duplicate sheet name errors, POI enforces uniqueness on only the first 31 chars.
  187. * but for the purpose of uniqueness long sheet names are silently truncated to 31 chars.
  188. */
  189. @Test
  190. void createSheetWithLongNames() throws IOException {
  191. try (Workbook wb1 = _testDataProvider.createWorkbook()) {
  192. String sheetName1 = "My very long sheet name which is longer than 31 chars";
  193. String truncatedSheetName1 = sheetName1.substring(0, 31);
  194. Sheet sh1 = wb1.createSheet(sheetName1);
  195. assertEquals(truncatedSheetName1, sh1.getSheetName());
  196. assertSame(sh1, wb1.getSheet(truncatedSheetName1));
  197. // now via wb.setSheetName
  198. wb1.setSheetName(0, sheetName1);
  199. assertEquals(truncatedSheetName1, sh1.getSheetName());
  200. assertSame(sh1, wb1.getSheet(truncatedSheetName1));
  201. String sheetName2 = "My very long sheet name which is longer than 31 chars " +
  202. "and sheetName2.substring(0, 31) == sheetName1.substring(0, 31)";
  203. IllegalArgumentException e = assertThrows(IllegalArgumentException.class, () -> wb1.createSheet(sheetName2));
  204. assertEquals("The workbook already contains a sheet named 'My very long sheet name which is longer than 31 chars and sheetName2.substring(0, 31) == sheetName1.substring(0, 31)'", e.getMessage());
  205. String sheetName3 = "POI allows creating sheets with names longer than 31 characters";
  206. String truncatedSheetName3 = sheetName3.substring(0, 31);
  207. Sheet sh3 = wb1.createSheet(sheetName3);
  208. assertEquals(truncatedSheetName3, sh3.getSheetName());
  209. assertSame(sh3, wb1.getSheet(truncatedSheetName3));
  210. //serialize and read again
  211. try (Workbook wb2 = _testDataProvider.writeOutAndReadBack(wb1)) {
  212. assertEquals(2, wb2.getNumberOfSheets());
  213. assertEquals(0, wb2.getSheetIndex(truncatedSheetName1));
  214. assertEquals(1, wb2.getSheetIndex(truncatedSheetName3));
  215. }
  216. }
  217. }
  218. @Test
  219. void removeSheetAt() throws IOException {
  220. try (Workbook workbook = _testDataProvider.createWorkbook()) {
  221. workbook.createSheet("sheet1");
  222. workbook.createSheet("sheet2");
  223. workbook.createSheet("sheet3");
  224. assertEquals(3, workbook.getNumberOfSheets());
  225. assertEquals(0, workbook.getActiveSheetIndex());
  226. workbook.removeSheetAt(1);
  227. assertEquals(2, workbook.getNumberOfSheets());
  228. assertEquals("sheet3", workbook.getSheetName(1));
  229. assertEquals(0, workbook.getActiveSheetIndex());
  230. workbook.removeSheetAt(0);
  231. assertEquals(1, workbook.getNumberOfSheets());
  232. assertEquals("sheet3", workbook.getSheetName(0));
  233. assertEquals(0, workbook.getActiveSheetIndex());
  234. workbook.removeSheetAt(0);
  235. assertEquals(0, workbook.getNumberOfSheets());
  236. assertEquals(0, workbook.getActiveSheetIndex());
  237. //re-create the sheets
  238. workbook.createSheet("sheet1");
  239. workbook.createSheet("sheet2");
  240. workbook.createSheet("sheet3");
  241. workbook.createSheet("sheet4");
  242. assertEquals(4, workbook.getNumberOfSheets());
  243. assertEquals(0, workbook.getActiveSheetIndex());
  244. workbook.setActiveSheet(2);
  245. assertEquals(2, workbook.getActiveSheetIndex());
  246. workbook.removeSheetAt(2);
  247. assertEquals(2, workbook.getActiveSheetIndex());
  248. workbook.removeSheetAt(1);
  249. assertEquals(1, workbook.getActiveSheetIndex());
  250. workbook.removeSheetAt(0);
  251. assertEquals(0, workbook.getActiveSheetIndex());
  252. workbook.removeSheetAt(0);
  253. assertEquals(0, workbook.getActiveSheetIndex());
  254. }
  255. }
  256. @Test
  257. void testSetActiveCell() throws IOException {
  258. try (Workbook wb = _testDataProvider.createWorkbook()) {
  259. Sheet sheet = wb.createSheet("new sheet");
  260. final CellAddress initialActiveCell = sheet.getActiveCell();
  261. assertTrue(initialActiveCell == null || new CellAddress("A1").equals(initialActiveCell));
  262. sheet.setActiveCell(new CellAddress("E11"));
  263. assertEquals(new CellAddress("E11"), sheet.getActiveCell());
  264. Workbook wbr = _testDataProvider.writeOutAndReadBack(wb);
  265. sheet = wbr.getSheet("new sheet");
  266. assertEquals(new CellAddress("E11"), sheet.getActiveCell());
  267. }
  268. }
  269. @Test
  270. void defaultValues() throws IOException {
  271. try (Workbook b = _testDataProvider.createWorkbook()) {
  272. assertEquals(0, b.getActiveSheetIndex());
  273. assertEquals(0, b.getFirstVisibleTab());
  274. assertEquals(0, b.getNumberOfNames());
  275. assertEquals(0, b.getNumberOfSheets());
  276. }
  277. }
  278. @Test
  279. void sheetSelection() throws IOException {
  280. try (Workbook b = _testDataProvider.createWorkbook()) {
  281. b.createSheet("Sheet One");
  282. b.createSheet("Sheet Two");
  283. b.setActiveSheet(1);
  284. b.setSelectedTab(1);
  285. b.setFirstVisibleTab(1);
  286. assertEquals(1, b.getActiveSheetIndex());
  287. assertEquals(1, b.getFirstVisibleTab());
  288. }
  289. }
  290. @Test
  291. void printArea() throws IOException {
  292. try (Workbook workbook = _testDataProvider.createWorkbook()) {
  293. Sheet sheet1 = workbook.createSheet("Test Print Area");
  294. String sheetName1 = sheet1.getSheetName();
  295. // workbook.setPrintArea(0, reference);
  296. workbook.setPrintArea(0, 1, 5, 4, 9);
  297. String retrievedPrintArea = workbook.getPrintArea(0);
  298. assertEquals("'" + sheetName1 + "'!$B$5:$F$10", retrievedPrintArea);
  299. String reference = "$A$1:$B$1";
  300. workbook.setPrintArea(0, reference);
  301. retrievedPrintArea = workbook.getPrintArea(0);
  302. assertEquals("'" + sheetName1 + "'!" + reference, retrievedPrintArea);
  303. workbook.removePrintArea(0);
  304. assertNull(workbook.getPrintArea(0));
  305. }
  306. }
  307. @Test
  308. void getSetActiveSheet() throws IOException {
  309. try (Workbook workbook = _testDataProvider.createWorkbook()) {
  310. assertEquals(0, workbook.getActiveSheetIndex());
  311. workbook.createSheet("sheet1");
  312. workbook.createSheet("sheet2");
  313. workbook.createSheet("sheet3");
  314. // set second sheet
  315. workbook.setActiveSheet(1);
  316. // test if second sheet is set up
  317. assertEquals(1, workbook.getActiveSheetIndex());
  318. workbook.setActiveSheet(0);
  319. // test if second sheet is set up
  320. assertEquals(0, workbook.getActiveSheetIndex());
  321. }
  322. }
  323. @Test
  324. void setSheetOrder() throws IOException {
  325. try (Workbook wb = _testDataProvider.createWorkbook()) {
  326. for (int i = 0; i < 10; i++) {
  327. wb.createSheet("Sheet " + i);
  328. }
  329. // Check the initial order
  330. assertEquals(0, wb.getSheetIndex("Sheet 0"));
  331. assertEquals(1, wb.getSheetIndex("Sheet 1"));
  332. assertEquals(2, wb.getSheetIndex("Sheet 2"));
  333. assertEquals(3, wb.getSheetIndex("Sheet 3"));
  334. assertEquals(4, wb.getSheetIndex("Sheet 4"));
  335. assertEquals(5, wb.getSheetIndex("Sheet 5"));
  336. assertEquals(6, wb.getSheetIndex("Sheet 6"));
  337. assertEquals(7, wb.getSheetIndex("Sheet 7"));
  338. assertEquals(8, wb.getSheetIndex("Sheet 8"));
  339. assertEquals(9, wb.getSheetIndex("Sheet 9"));
  340. // check active sheet
  341. assertEquals(0, wb.getActiveSheetIndex());
  342. // Change
  343. wb.setSheetOrder("Sheet 6", 0);
  344. assertEquals(1, wb.getActiveSheetIndex());
  345. wb.setSheetOrder("Sheet 3", 7);
  346. wb.setSheetOrder("Sheet 1", 9);
  347. // now the first sheet is at index 1
  348. assertEquals(1, wb.getActiveSheetIndex());
  349. // Check they're currently right
  350. assertEquals(0, wb.getSheetIndex("Sheet 6"));
  351. assertEquals(1, wb.getSheetIndex("Sheet 0"));
  352. assertEquals(2, wb.getSheetIndex("Sheet 2"));
  353. assertEquals(3, wb.getSheetIndex("Sheet 4"));
  354. assertEquals(4, wb.getSheetIndex("Sheet 5"));
  355. assertEquals(5, wb.getSheetIndex("Sheet 7"));
  356. assertEquals(6, wb.getSheetIndex("Sheet 3"));
  357. assertEquals(7, wb.getSheetIndex("Sheet 8"));
  358. assertEquals(8, wb.getSheetIndex("Sheet 9"));
  359. assertEquals(9, wb.getSheetIndex("Sheet 1"));
  360. try (Workbook wbr = _testDataProvider.writeOutAndReadBack(wb)) {
  361. assertEquals(0, wbr.getSheetIndex("Sheet 6"));
  362. assertEquals(1, wbr.getSheetIndex("Sheet 0"));
  363. assertEquals(2, wbr.getSheetIndex("Sheet 2"));
  364. assertEquals(3, wbr.getSheetIndex("Sheet 4"));
  365. assertEquals(4, wbr.getSheetIndex("Sheet 5"));
  366. assertEquals(5, wbr.getSheetIndex("Sheet 7"));
  367. assertEquals(6, wbr.getSheetIndex("Sheet 3"));
  368. assertEquals(7, wbr.getSheetIndex("Sheet 8"));
  369. assertEquals(8, wbr.getSheetIndex("Sheet 9"));
  370. assertEquals(9, wbr.getSheetIndex("Sheet 1"));
  371. assertEquals(1, wb.getActiveSheetIndex());
  372. // Now get the index by the sheet, not the name
  373. for (int i = 0; i < 10; i++) {
  374. Sheet s = wbr.getSheetAt(i);
  375. assertEquals(i, wbr.getSheetIndex(s));
  376. }
  377. }
  378. }
  379. }
  380. @Test
  381. public void cloneSheet() throws IOException {
  382. try (Workbook book = _testDataProvider.createWorkbook()) {
  383. Sheet sheet = book.createSheet("TEST");
  384. sheet.createRow(0).createCell(0).setCellValue("Test");
  385. sheet.createRow(1).createCell(0).setCellValue(36.6);
  386. assertEquals(0, sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, 2)));
  387. assertEquals(1, sheet.addMergedRegion(new CellRangeAddress(2, 3, 0, 2)));
  388. assertTrue(sheet.isSelected());
  389. Sheet clonedSheet = book.cloneSheet(0);
  390. assertEquals("TEST (2)", clonedSheet.getSheetName());
  391. assertEquals(2, clonedSheet.getPhysicalNumberOfRows());
  392. assertEquals(2, clonedSheet.getNumMergedRegions());
  393. assertFalse(clonedSheet.isSelected());
  394. //cloned sheet is a deep copy, adding rows or merged regions in the original does not affect the clone
  395. sheet.createRow(2).createCell(0).setCellValue(1);
  396. assertEquals(2, sheet.addMergedRegion(new CellRangeAddress(4, 5, 0, 2)));
  397. assertEquals(2, clonedSheet.getPhysicalNumberOfRows());
  398. assertEquals(2, clonedSheet.getNumMergedRegions());
  399. clonedSheet.createRow(2).createCell(0).setCellValue(1);
  400. assertEquals(2, clonedSheet.addMergedRegion(new CellRangeAddress(6, 7, 0, 2)));
  401. assertEquals(3, clonedSheet.getPhysicalNumberOfRows());
  402. assertEquals(3, clonedSheet.getNumMergedRegions());
  403. }
  404. }
  405. @Test
  406. protected void parentReferences() throws IOException {
  407. try (Workbook wb1 = _testDataProvider.createWorkbook()) {
  408. Sheet sheet = wb1.createSheet();
  409. assertSame(wb1, sheet.getWorkbook());
  410. Row row = sheet.createRow(0);
  411. assertSame(sheet, row.getSheet());
  412. Cell cell = row.createCell(1);
  413. assertSame(sheet, cell.getSheet());
  414. assertSame(row, cell.getRow());
  415. try (Workbook wb2 = _testDataProvider.writeOutAndReadBack(wb1)) {
  416. wb1.close();
  417. sheet = wb2.getSheetAt(0);
  418. assertSame(wb2, sheet.getWorkbook());
  419. row = sheet.getRow(0);
  420. assertSame(sheet, row.getSheet());
  421. cell = row.getCell(1);
  422. assertSame(sheet, cell.getSheet());
  423. assertSame(row, cell.getRow());
  424. }
  425. }
  426. }
  427. /**
  428. * Test to validate that replacement for removed setRepeatingRowsAndColumns() methods
  429. * is still working correctly
  430. */
  431. @Test
  432. void setRepeatingRowsAndColumns() throws IOException {
  433. try (Workbook wb = _testDataProvider.createWorkbook()) {
  434. CellRangeAddress cra = new CellRangeAddress(0, 3, 0, 0);
  435. String expRows = "1:4", expCols = "A:A";
  436. Sheet sheet1 = wb.createSheet();
  437. sheet1.setRepeatingRows(cra);
  438. sheet1.setRepeatingColumns(cra);
  439. assertEquals(expRows, sheet1.getRepeatingRows().formatAsString());
  440. assertEquals(expCols, sheet1.getRepeatingColumns().formatAsString());
  441. //must handle sheets with quotas, see Bugzilla #47294
  442. Sheet sheet2 = wb.createSheet("My' Sheet");
  443. sheet2.setRepeatingRows(cra);
  444. sheet2.setRepeatingColumns(cra);
  445. assertEquals(expRows, sheet2.getRepeatingRows().formatAsString());
  446. assertEquals(expCols, sheet2.getRepeatingColumns().formatAsString());
  447. }
  448. }
  449. /**
  450. * Tests that all the unicode capable string fields can be set, written and then read back
  451. */
  452. @Test
  453. protected void unicodeInAll() throws IOException {
  454. try (Workbook wb1 = _testDataProvider.createWorkbook()) {
  455. CreationHelper factory = wb1.getCreationHelper();
  456. //Create a unicode dataformat (contains euro symbol)
  457. DataFormat df = wb1.createDataFormat();
  458. final String formatStr = "_([$\u20ac-2]\\\\\\ * #,##0.00_);_([$\u20ac-2]\\\\\\ * \\\\\\(#,##0.00\\\\\\);_([$\u20ac-2]\\\\\\ *\\\"\\-\\\\\"??_);_(@_)";
  459. short fmt = df.getFormat(formatStr);
  460. //Create a unicode sheet name (euro symbol)
  461. Sheet s = wb1.createSheet("\u20ac");
  462. //Set a unicode header (you guessed it the euro symbol)
  463. Header h = s.getHeader();
  464. h.setCenter("\u20ac");
  465. h.setLeft("\u20ac");
  466. h.setRight("\u20ac");
  467. //Set a unicode footer
  468. Footer f = s.getFooter();
  469. f.setCenter("\u20ac");
  470. f.setLeft("\u20ac");
  471. f.setRight("\u20ac");
  472. Row r = s.createRow(0);
  473. Cell c = r.createCell(1);
  474. c.setCellValue(12.34);
  475. c.getCellStyle().setDataFormat(fmt);
  476. /*Cell c2 =*/
  477. r.createCell(2); // TODO - c2 unused but changing next line ('c'->'c2') causes test to fail
  478. c.setCellValue(factory.createRichTextString("\u20ac"));
  479. Cell c3 = r.createCell(3);
  480. String formulaString = "TEXT(12.34,\"\u20ac###,##\")";
  481. c3.setCellFormula(formulaString);
  482. try (Workbook wb2 = _testDataProvider.writeOutAndReadBack(wb1)) {
  483. //Test the sheetname
  484. s = wb2.getSheet("\u20ac");
  485. assertNotNull(s);
  486. //Test the header
  487. h = s.getHeader();
  488. assertEquals("\u20ac", h.getCenter());
  489. assertEquals("\u20ac", h.getLeft());
  490. assertEquals("\u20ac", h.getRight());
  491. //Test the footer
  492. f = s.getFooter();
  493. assertEquals("\u20ac", f.getCenter());
  494. assertEquals("\u20ac", f.getLeft());
  495. assertEquals("\u20ac", f.getRight());
  496. //Test the dataformat
  497. r = s.getRow(0);
  498. c = r.getCell(1);
  499. df = wb2.createDataFormat();
  500. assertEquals(formatStr, df.getFormat(c.getCellStyle().getDataFormat()));
  501. //Test the cell string value
  502. /*c2 =*/
  503. r.getCell(2);
  504. assertEquals("\u20ac", c.getRichStringCellValue().getString());
  505. //Test the cell formula
  506. c3 = r.getCell(3);
  507. assertEquals(c3.getCellFormula(), formulaString);
  508. }
  509. }
  510. }
  511. private Workbook newSetSheetNameTestingWorkbook() {
  512. Workbook wb = _testDataProvider.createWorkbook();
  513. Sheet sh1 = wb.createSheet("Worksheet");
  514. Sheet sh2 = wb.createSheet("Testing 47100");
  515. Sheet sh3 = wb.createSheet("To be renamed");
  516. Name name1 = wb.createName();
  517. name1.setNameName("sale_1");
  518. name1.setRefersToFormula("Worksheet!$A$1");
  519. Name name2 = wb.createName();
  520. name2.setNameName("sale_2");
  521. name2.setRefersToFormula("'Testing 47100'!$A$1");
  522. Name name3 = wb.createName();
  523. name3.setNameName("sale_3");
  524. name3.setRefersToFormula("'Testing 47100'!$B$1");
  525. Name name4 = wb.createName();
  526. name4.setNameName("sale_4");
  527. name4.setRefersToFormula("'To be renamed'!$A$3");
  528. sh1.createRow(0).createCell(0).setCellFormula("SUM('Testing 47100'!A1:C1)");
  529. sh1.createRow(1).createCell(0).setCellFormula("SUM('Testing 47100'!A1:C1,'To be renamed'!A1:A5)");
  530. sh1.createRow(2).createCell(0).setCellFormula("sale_2+sale_3+'Testing 47100'!C1");
  531. sh2.createRow(0).createCell(0).setCellValue(1);
  532. sh2.getRow(0).createCell(1).setCellValue(2);
  533. sh2.getRow(0).createCell(2).setCellValue(3);
  534. sh3.createRow(0).createCell(0).setCellValue(1);
  535. sh3.createRow(1).createCell(0).setCellValue(2);
  536. sh3.createRow(2).createCell(0).setCellValue(3);
  537. sh3.createRow(3).createCell(0).setCellValue(4);
  538. sh3.createRow(4).createCell(0).setCellValue(5);
  539. sh3.createRow(5).createCell(0).setCellFormula("sale_3");
  540. sh3.createRow(6).createCell(0).setCellFormula("'Testing 47100'!C1");
  541. return wb;
  542. }
  543. /**
  544. * Ensure that Workbook#setSheetName updates all dependent formulas and named ranges
  545. *
  546. * @see <a href="https://issues.apache.org/bugzilla/show_bug.cgi?id=47100">Bugzilla 47100</a>
  547. */
  548. @Test
  549. protected void setSheetName() throws IOException {
  550. try (Workbook wb1 = newSetSheetNameTestingWorkbook()) {
  551. Sheet sh1 = wb1.getSheetAt(0);
  552. Name sale_2 = wb1.getName("sale_2");
  553. Name sale_3 = wb1.getName("sale_3");
  554. Name sale_4 = wb1.getName("sale_4");
  555. assertEquals("sale_2", sale_2.getNameName());
  556. assertEquals("'Testing 47100'!$A$1", sale_2.getRefersToFormula());
  557. assertEquals("sale_3", sale_3.getNameName());
  558. assertEquals("'Testing 47100'!$B$1", sale_3.getRefersToFormula());
  559. assertEquals("sale_4", sale_4.getNameName());
  560. assertEquals("'To be renamed'!$A$3", sale_4.getRefersToFormula());
  561. FormulaEvaluator evaluator = wb1.getCreationHelper().createFormulaEvaluator();
  562. Cell cell0 = sh1.getRow(0).getCell(0);
  563. Cell cell1 = sh1.getRow(1).getCell(0);
  564. Cell cell2 = sh1.getRow(2).getCell(0);
  565. assertEquals("SUM('Testing 47100'!A1:C1)", cell0.getCellFormula());
  566. assertEquals("SUM('Testing 47100'!A1:C1,'To be renamed'!A1:A5)", cell1.getCellFormula());
  567. assertEquals("sale_2+sale_3+'Testing 47100'!C1", cell2.getCellFormula());
  568. assertEquals(6.0, evaluator.evaluate(cell0).getNumberValue(), 0);
  569. assertEquals(21.0, evaluator.evaluate(cell1).getNumberValue(), 0);
  570. assertEquals(6.0, evaluator.evaluate(cell2).getNumberValue(), 0);
  571. wb1.setSheetName(1, "47100 - First");
  572. wb1.setSheetName(2, "47100 - Second");
  573. assertEquals("sale_2", sale_2.getNameName());
  574. assertEquals("'47100 - First'!$A$1", sale_2.getRefersToFormula());
  575. assertEquals("sale_3", sale_3.getNameName());
  576. assertEquals("'47100 - First'!$B$1", sale_3.getRefersToFormula());
  577. assertEquals("sale_4", sale_4.getNameName());
  578. assertEquals("'47100 - Second'!$A$3", sale_4.getRefersToFormula());
  579. assertEquals("SUM('47100 - First'!A1:C1)", cell0.getCellFormula());
  580. assertEquals("SUM('47100 - First'!A1:C1,'47100 - Second'!A1:A5)", cell1.getCellFormula());
  581. assertEquals("sale_2+sale_3+'47100 - First'!C1", cell2.getCellFormula());
  582. evaluator.clearAllCachedResultValues();
  583. assertEquals(6.0, evaluator.evaluate(cell0).getNumberValue(), 0);
  584. assertEquals(21.0, evaluator.evaluate(cell1).getNumberValue(), 0);
  585. assertEquals(6.0, evaluator.evaluate(cell2).getNumberValue(), 0);
  586. try (Workbook wb2 = _testDataProvider.writeOutAndReadBack(wb1)) {
  587. sh1 = wb2.getSheetAt(0);
  588. sale_2 = wb2.getName("sale_2");
  589. sale_3 = wb2.getName("sale_3");
  590. sale_4 = wb2.getName("sale_4");
  591. cell0 = sh1.getRow(0).getCell(0);
  592. cell1 = sh1.getRow(1).getCell(0);
  593. cell2 = sh1.getRow(2).getCell(0);
  594. assertEquals("sale_2", sale_2.getNameName());
  595. assertEquals("'47100 - First'!$A$1", sale_2.getRefersToFormula());
  596. assertEquals("sale_3", sale_3.getNameName());
  597. assertEquals("'47100 - First'!$B$1", sale_3.getRefersToFormula());
  598. assertEquals("sale_4", sale_4.getNameName());
  599. assertEquals("'47100 - Second'!$A$3", sale_4.getRefersToFormula());
  600. assertEquals("SUM('47100 - First'!A1:C1)", cell0.getCellFormula());
  601. assertEquals("SUM('47100 - First'!A1:C1,'47100 - Second'!A1:A5)", cell1.getCellFormula());
  602. assertEquals("sale_2+sale_3+'47100 - First'!C1", cell2.getCellFormula());
  603. evaluator = wb2.getCreationHelper().createFormulaEvaluator();
  604. assertEquals(6.0, evaluator.evaluate(cell0).getNumberValue(), 0);
  605. assertEquals(21.0, evaluator.evaluate(cell1).getNumberValue(), 0);
  606. assertEquals(6.0, evaluator.evaluate(cell2).getNumberValue(), 0);
  607. }
  608. }
  609. }
  610. @Test
  611. protected void changeSheetNameWithSharedFormulas() throws IOException {
  612. String sampleFile = "shared_formulas.xls" + (getClass().getName().contains("xssf") ? "x" : "");
  613. try (Workbook wb = _testDataProvider.openSampleWorkbook(sampleFile)) {
  614. FormulaEvaluator evaluator = wb.getCreationHelper().createFormulaEvaluator();
  615. Sheet sheet = wb.getSheetAt(0);
  616. for (int rownum = 1; rownum <= 40; rownum++) {
  617. Cell cellA = sheet.getRow(1).getCell(0);
  618. Cell cellB = sheet.getRow(1).getCell(1);
  619. assertEquals(cellB.getStringCellValue(), evaluator.evaluate(cellA).getStringValue());
  620. }
  621. wb.setSheetName(0, "Renamed by POI");
  622. evaluator.clearAllCachedResultValues();
  623. for (int rownum = 1; rownum <= 40; rownum++) {
  624. Cell cellA = sheet.getRow(1).getCell(0);
  625. Cell cellB = sheet.getRow(1).getCell(1);
  626. assertEquals(cellB.getStringCellValue(), evaluator.evaluate(cellA).getStringValue());
  627. }
  628. }
  629. }
  630. protected void assertSheetOrder(Workbook wb, String... sheets) {
  631. StringBuilder sheetNames = new StringBuilder();
  632. for(int i = 0;i < wb.getNumberOfSheets();i++) {
  633. sheetNames.append(wb.getSheetAt(i).getSheetName()).append(",");
  634. }
  635. assertEquals(sheets.length, wb.getNumberOfSheets(), "Had: " + sheetNames);
  636. for(int i = 0;i < wb.getNumberOfSheets();i++) {
  637. assertEquals(sheets[i], wb.getSheetAt(i).getSheetName(), "Had: " + sheetNames);
  638. }
  639. }
  640. @Test
  641. void test58499() throws IOException {
  642. try (Workbook workbook = _testDataProvider.createWorkbook()) {
  643. Sheet sheet = workbook.createSheet();
  644. for (int i = 0; i < 900; i++) {
  645. Row r = sheet.createRow(i);
  646. Cell c = r.createCell(0);
  647. CellStyle cs = workbook.createCellStyle();
  648. c.setCellStyle(cs);
  649. c.setCellValue("AAA");
  650. }
  651. assertDoesNotThrow(() -> workbook.write(NULL_OUTPUT_STREAM));
  652. }
  653. }
  654. @Test
  655. void windowOneDefaults() throws IOException {
  656. try (Workbook b = _testDataProvider.createWorkbook()) {
  657. assertEquals(0, b.getActiveSheetIndex());
  658. assertEquals(0, b.getFirstVisibleTab());
  659. // throws NullPointerException when WindowOneRecord in Workbook is not probably initialized
  660. }
  661. }
  662. @Test
  663. void getSpreadsheetVersion() throws IOException {
  664. try (Workbook wb = _testDataProvider.createWorkbook()) {
  665. assertEquals(_testDataProvider.getSpreadsheetVersion(), wb.getSpreadsheetVersion());
  666. }
  667. }
  668. /* FIXME copied from {@link org.apache.poi.ss.TestWorkbookFactory} */
  669. protected static void assertCloseDoesNotModifyFile(String filename, Workbook wb) throws IOException {
  670. final byte[] before = HSSFTestDataSamples.getTestDataFileContent(filename);
  671. wb.close();
  672. final byte[] after = HSSFTestDataSamples.getTestDataFileContent(filename);
  673. assertArrayEquals(before, after, filename + " sample file was modified as a result of closing the workbook");
  674. }
  675. @Test
  676. public void sheetClone() throws IOException {
  677. // First up, try a simple file
  678. try (Workbook b = _testDataProvider.createWorkbook();
  679. Workbook bBack = HSSFTestDataSamples.openSampleWorkbook("SheetWithDrawing.xls")) {
  680. assertEquals(0, b.getNumberOfSheets());
  681. b.createSheet("Sheet One");
  682. b.createSheet("Sheet Two");
  683. assertEquals(2, b.getNumberOfSheets());
  684. b.cloneSheet(0);
  685. assertEquals(3, b.getNumberOfSheets());
  686. // Now try a problem one with drawing records in it
  687. assertEquals(1, bBack.getNumberOfSheets());
  688. bBack.cloneSheet(0);
  689. assertEquals(2, bBack.getNumberOfSheets());
  690. }
  691. }
  692. @Test
  693. void getSheetIndex() throws IOException {
  694. try (Workbook wb = _testDataProvider.createWorkbook()) {
  695. Sheet sheet1 = wb.createSheet("Sheet1");
  696. Sheet sheet2 = wb.createSheet("Sheet2");
  697. Sheet sheet3 = wb.createSheet("Sheet3");
  698. Sheet sheet4 = wb.createSheet("Sheet4");
  699. assertEquals(0, wb.getSheetIndex(sheet1));
  700. assertEquals(1, wb.getSheetIndex(sheet2));
  701. assertEquals(2, wb.getSheetIndex(sheet3));
  702. assertEquals(3, wb.getSheetIndex(sheet4));
  703. // remove sheets
  704. wb.removeSheetAt(0);
  705. wb.removeSheetAt(2);
  706. // ensure that sheets are moved up and removed sheets are not found any more
  707. assertEquals(-1, wb.getSheetIndex(sheet1));
  708. assertEquals(0, wb.getSheetIndex(sheet2));
  709. assertEquals(1, wb.getSheetIndex(sheet3));
  710. assertEquals(-1, wb.getSheetIndex(sheet4));
  711. }
  712. }
  713. @Test
  714. void addSheetTwice() throws IOException {
  715. try (Workbook wb = _testDataProvider.createWorkbook()) {
  716. Sheet sheet1 = wb.createSheet("Sheet1");
  717. assertNotNull(sheet1);
  718. IllegalArgumentException ex = assertThrows(
  719. IllegalArgumentException.class,
  720. () -> wb.createSheet("Sheet1")
  721. );
  722. assertEquals("The workbook already contains a sheet named 'Sheet1'", ex.getMessage());
  723. }
  724. }
  725. // bug 51233 and 55075: correctly size image if added to a row with a custom height
  726. @Test
  727. void createDrawing() throws Exception {
  728. try (Workbook wb = _testDataProvider.createWorkbook()) {
  729. Sheet sheet = wb.createSheet("Main Sheet");
  730. Row row0 = sheet.createRow(0);
  731. Row row1 = sheet.createRow(1);
  732. row1.createCell(0);
  733. row0.createCell(1);
  734. row1.createCell(0);
  735. row1.createCell(1);
  736. byte[] pictureData = _testDataProvider.getTestDataFileContent("logoKarmokar4.png");
  737. int handle = wb.addPicture(pictureData, Workbook.PICTURE_TYPE_PNG);
  738. Drawing<?> drawing = sheet.createDrawingPatriarch();
  739. CreationHelper helper = wb.getCreationHelper();
  740. ClientAnchor anchor = helper.createClientAnchor();
  741. anchor.setAnchorType(AnchorType.DONT_MOVE_AND_RESIZE);
  742. anchor.setCol1(0);
  743. anchor.setRow1(0);
  744. Picture picture = drawing.createPicture(anchor, handle);
  745. row0.setHeightInPoints(144);
  746. // set a column width so that XSSF and SXSSF have the same width (default widths may be different otherwise)
  747. sheet.setColumnWidth(0, 100 * 256);
  748. picture.resize();
  749. // The actual dimensions don't matter as much as having XSSF and SXSSF produce the same size drawings
  750. // Check drawing height
  751. assertEquals(0, anchor.getRow1());
  752. assertEquals(0, anchor.getRow2());
  753. assertEquals(0, anchor.getDy1());
  754. assertEquals(1609725, anchor.getDy2()); //HSSF: 225
  755. // Check drawing width
  756. assertEquals(0, anchor.getCol1());
  757. assertEquals(0, anchor.getCol2());
  758. assertEquals(0, anchor.getDx1());
  759. assertEquals(1114425, anchor.getDx2()); //HSSF: 171
  760. }
  761. }
  762. @Test
  763. void testSheetNameTrimming() throws IOException {
  764. try (Workbook workbook = _testDataProvider.createWorkbook()) {
  765. Sheet sheet = workbook.createSheet("MyVeryLongSheetName_9999999999999999");
  766. assertNotNull(sheet);
  767. assertEquals("MyVeryLongSheetName_99999999999", workbook.getSheetName(0));
  768. assertThrows(IllegalArgumentException.class,
  769. () -> workbook.createSheet("MyVeryLongSheetName_9999999999999998")
  770. );
  771. }
  772. }
  773. }