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 37KB

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