Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.

TestHSSFWorkbook.java 43KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189
  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.hssf.usermodel;
  16. import static org.apache.poi.POITestCase.assertContains;
  17. import static org.apache.poi.hssf.HSSFTestDataSamples.openSampleWorkbook;
  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.assertThrows;
  24. import static org.junit.jupiter.api.Assertions.assertTrue;
  25. import static org.junit.jupiter.api.Assertions.fail;
  26. import java.io.ByteArrayInputStream;
  27. import java.io.ByteArrayOutputStream;
  28. import java.io.File;
  29. import java.io.FileInputStream;
  30. import java.io.FileOutputStream;
  31. import java.io.IOException;
  32. import java.io.InputStream;
  33. import java.util.ArrayList;
  34. import java.util.Collection;
  35. import java.util.List;
  36. import java.util.Map;
  37. import java.util.function.Supplier;
  38. import org.apache.poi.POIDataSamples;
  39. import org.apache.poi.ddf.EscherBSERecord;
  40. import org.apache.poi.hpsf.ClassID;
  41. import org.apache.poi.hssf.HSSFITestDataProvider;
  42. import org.apache.poi.hssf.HSSFTestDataSamples;
  43. import org.apache.poi.hssf.OldExcelFormatException;
  44. import org.apache.poi.hssf.model.HSSFFormulaParser;
  45. import org.apache.poi.hssf.model.InternalSheet;
  46. import org.apache.poi.hssf.model.InternalWorkbook;
  47. import org.apache.poi.hssf.record.CFRuleRecord;
  48. import org.apache.poi.hssf.record.HSSFRecordTypes;
  49. import org.apache.poi.hssf.record.NameRecord;
  50. import org.apache.poi.hssf.record.Record;
  51. import org.apache.poi.hssf.record.RecordBase;
  52. import org.apache.poi.hssf.record.WindowOneRecord;
  53. import org.apache.poi.poifs.filesystem.DirectoryEntry;
  54. import org.apache.poi.poifs.filesystem.DirectoryNode;
  55. import org.apache.poi.poifs.filesystem.POIFSFileSystem;
  56. import org.apache.poi.ss.formula.ptg.Area3DPtg;
  57. import org.apache.poi.ss.usermodel.BaseTestWorkbook;
  58. import org.apache.poi.ss.usermodel.Cell;
  59. import org.apache.poi.ss.usermodel.ConditionalFormatting;
  60. import org.apache.poi.ss.usermodel.ConditionalFormattingRule;
  61. import org.apache.poi.ss.usermodel.Name;
  62. import org.apache.poi.ss.usermodel.Row;
  63. import org.apache.poi.ss.usermodel.Sheet;
  64. import org.apache.poi.ss.usermodel.SheetConditionalFormatting;
  65. import org.apache.poi.ss.usermodel.Workbook;
  66. import org.apache.poi.ss.util.CellRangeAddress;
  67. import org.apache.poi.util.IOUtils;
  68. import org.apache.poi.util.TempFile;
  69. import org.junit.jupiter.api.Test;
  70. import org.junit.jupiter.params.ParameterizedTest;
  71. import org.junit.jupiter.params.provider.CsvSource;
  72. /**
  73. * Tests for {@link HSSFWorkbook}
  74. */
  75. public final class TestHSSFWorkbook extends BaseTestWorkbook {
  76. private static final HSSFITestDataProvider _testDataProvider = HSSFITestDataProvider.instance;
  77. private static final POIDataSamples samples = POIDataSamples.getSpreadSheetInstance();
  78. public TestHSSFWorkbook() {
  79. super(_testDataProvider);
  80. }
  81. /**
  82. * gives test code access to the {@link InternalWorkbook} within {@link HSSFWorkbook}
  83. */
  84. public static InternalWorkbook getInternalWorkbook(HSSFWorkbook wb) {
  85. return wb.getWorkbook();
  86. }
  87. /**
  88. * Tests for {@link HSSFWorkbook#isHidden()} etc
  89. */
  90. @Test
  91. public void hidden() throws IOException {
  92. HSSFWorkbook wb = new HSSFWorkbook();
  93. WindowOneRecord w1 = wb.getWorkbook().getWindowOne();
  94. assertFalse(wb.isHidden());
  95. assertFalse(w1.getHidden());
  96. wb.setHidden(true);
  97. assertTrue(wb.isHidden());
  98. assertTrue(w1.getHidden());
  99. HSSFWorkbook wbBack = HSSFTestDataSamples.writeOutAndReadBack(wb);
  100. w1 = wbBack.getWorkbook().getWindowOne();
  101. wbBack.setHidden(true);
  102. assertTrue(wbBack.isHidden());
  103. assertTrue(w1.getHidden());
  104. wbBack.setHidden(false);
  105. assertFalse(wbBack.isHidden());
  106. assertFalse(w1.getHidden());
  107. wbBack.close();
  108. wb.close();
  109. }
  110. @Test
  111. public void readWriteWithCharts() throws IOException {
  112. Sheet s;
  113. // Single chart, two sheets
  114. HSSFWorkbook b1 = openSampleWorkbook("44010-SingleChart.xls");
  115. assertEquals(2, b1.getNumberOfSheets());
  116. assertEquals("Graph2", b1.getSheetName(1));
  117. s = b1.getSheetAt(1);
  118. assertEquals(0, s.getFirstRowNum());
  119. assertEquals(8, s.getLastRowNum());
  120. // Has chart on 1st sheet??
  121. // FIXME
  122. assertNotNull(b1.getSheetAt(0).getDrawingPatriarch());
  123. assertNull(b1.getSheetAt(1).getDrawingPatriarch());
  124. assertFalse(b1.getSheetAt(0).getDrawingPatriarch().containsChart());
  125. b1.close();
  126. // We've now called getDrawingPatriarch() so
  127. // everything will be all screwy
  128. // So, start again
  129. HSSFWorkbook b2 = openSampleWorkbook("44010-SingleChart.xls");
  130. HSSFWorkbook b3 = HSSFTestDataSamples.writeOutAndReadBack(b2);
  131. b2.close();
  132. assertEquals(2, b3.getNumberOfSheets());
  133. s = b3.getSheetAt(1);
  134. assertEquals(0, s.getFirstRowNum());
  135. assertEquals(8, s.getLastRowNum());
  136. b3.close();
  137. // Two charts, three sheets
  138. HSSFWorkbook b4 = openSampleWorkbook("44010-TwoCharts.xls");
  139. assertEquals(3, b4.getNumberOfSheets());
  140. s = b4.getSheetAt(1);
  141. assertEquals(0, s.getFirstRowNum());
  142. assertEquals(8, s.getLastRowNum());
  143. s = b4.getSheetAt(2);
  144. assertEquals(0, s.getFirstRowNum());
  145. assertEquals(8, s.getLastRowNum());
  146. // Has chart on 1st sheet??
  147. // FIXME
  148. assertNotNull(b4.getSheetAt(0).getDrawingPatriarch());
  149. assertNull(b4.getSheetAt(1).getDrawingPatriarch());
  150. assertNull(b4.getSheetAt(2).getDrawingPatriarch());
  151. assertFalse(b4.getSheetAt(0).getDrawingPatriarch().containsChart());
  152. b4.close();
  153. // We've now called getDrawingPatriarch() so
  154. // everything will be all screwy
  155. // So, start again
  156. HSSFWorkbook b5 = openSampleWorkbook("44010-TwoCharts.xls");
  157. Workbook b6 = HSSFTestDataSamples.writeOutAndReadBack(b5);
  158. b5.close();
  159. assertEquals(3, b6.getNumberOfSheets());
  160. s = b6.getSheetAt(1);
  161. assertEquals(0, s.getFirstRowNum());
  162. assertEquals(8, s.getLastRowNum());
  163. s = b6.getSheetAt(2);
  164. assertEquals(0, s.getFirstRowNum());
  165. assertEquals(8, s.getLastRowNum());
  166. b6.close();
  167. }
  168. @Test
  169. public void selectedSheet_bug44523() throws IOException {
  170. HSSFWorkbook wb=new HSSFWorkbook();
  171. HSSFSheet sheet1 = wb.createSheet("Sheet1");
  172. HSSFSheet sheet2 = wb.createSheet("Sheet2");
  173. HSSFSheet sheet3 = wb.createSheet("Sheet3");
  174. HSSFSheet sheet4 = wb.createSheet("Sheet4");
  175. confirmActiveSelected(sheet1, true);
  176. confirmActiveSelected(sheet2, false);
  177. confirmActiveSelected(sheet3, false);
  178. confirmActiveSelected(sheet4, false);
  179. wb.setSelectedTab(1);
  180. // Demonstrate bug 44525:
  181. // Well... not quite, since isActive + isSelected were also added in the same bug fix
  182. assertFalse(sheet1.isSelected(), "Identified bug 44523 a");
  183. wb.setActiveSheet(1);
  184. assertFalse(sheet1.isActive(), "Identified bug 44523 b");
  185. confirmActiveSelected(sheet1, false);
  186. confirmActiveSelected(sheet2, true);
  187. confirmActiveSelected(sheet3, false);
  188. confirmActiveSelected(sheet4, false);
  189. wb.close();
  190. }
  191. private static List<Integer> arrayToList(int[] array) {
  192. List<Integer> list = new ArrayList<>(array.length);
  193. for ( Integer element : array ) {
  194. list.add(element);
  195. }
  196. return list;
  197. }
  198. private static void assertCollectionsEquals(Collection<Integer> expected, Collection<Integer> actual) {
  199. assertEquals(expected.size(), actual.size());
  200. for (int e : expected) {
  201. assertTrue(actual.contains(e));
  202. }
  203. for (int a : actual) {
  204. assertTrue(expected.contains(a));
  205. }
  206. }
  207. @Test
  208. public void selectMultiple() throws IOException {
  209. HSSFWorkbook wb = new HSSFWorkbook();
  210. HSSFSheet sheet0 = wb.createSheet("Sheet0");
  211. HSSFSheet sheet1 = wb.createSheet("Sheet1");
  212. HSSFSheet sheet2 = wb.createSheet("Sheet2");
  213. HSSFSheet sheet3 = wb.createSheet("Sheet3");
  214. HSSFSheet sheet4 = wb.createSheet("Sheet4");
  215. HSSFSheet sheet5 = wb.createSheet("Sheet5");
  216. List<Integer> selected = arrayToList(new int[] { 0, 2, 3 });
  217. wb.setSelectedTabs(selected);
  218. assertCollectionsEquals(selected, wb.getSelectedTabs());
  219. assertTrue(sheet0.isSelected());
  220. assertFalse(sheet1.isSelected());
  221. assertTrue(sheet2.isSelected());
  222. assertTrue(sheet3.isSelected());
  223. assertFalse(sheet4.isSelected());
  224. assertFalse(sheet5.isSelected());
  225. selected = arrayToList(new int[] { 1, 3, 5 });
  226. wb.setSelectedTabs(selected);
  227. // previous selection should be cleared
  228. assertCollectionsEquals(selected, wb.getSelectedTabs());
  229. assertFalse(sheet0.isSelected());
  230. assertTrue(sheet1.isSelected());
  231. assertFalse(sheet2.isSelected());
  232. assertTrue(sheet3.isSelected());
  233. assertFalse(sheet4.isSelected());
  234. assertTrue(sheet5.isSelected());
  235. assertTrue(sheet0.isActive());
  236. assertFalse(sheet2.isActive());
  237. wb.setActiveSheet(2);
  238. assertFalse(sheet0.isActive());
  239. assertTrue(sheet2.isActive());
  240. wb.close();
  241. }
  242. @Test
  243. public void activeSheetAfterDelete_bug40414() throws IOException {
  244. HSSFWorkbook wb=new HSSFWorkbook();
  245. HSSFSheet sheet0 = wb.createSheet("Sheet0");
  246. HSSFSheet sheet1 = wb.createSheet("Sheet1");
  247. HSSFSheet sheet2 = wb.createSheet("Sheet2");
  248. HSSFSheet sheet3 = wb.createSheet("Sheet3");
  249. HSSFSheet sheet4 = wb.createSheet("Sheet4");
  250. // confirm default activation/selection
  251. confirmActiveSelected(sheet0, true);
  252. confirmActiveSelected(sheet1, false);
  253. confirmActiveSelected(sheet2, false);
  254. confirmActiveSelected(sheet3, false);
  255. confirmActiveSelected(sheet4, false);
  256. wb.setActiveSheet(3);
  257. wb.setSelectedTab(3);
  258. confirmActiveSelected(sheet0, false);
  259. confirmActiveSelected(sheet1, false);
  260. confirmActiveSelected(sheet2, false);
  261. confirmActiveSelected(sheet3, true);
  262. confirmActiveSelected(sheet4, false);
  263. wb.removeSheetAt(3);
  264. // after removing the only active/selected sheet, another should be active/selected in its place
  265. assertTrue(sheet4.isSelected(), "identified bug 40414 a");
  266. assertTrue(sheet4.isActive(), "identified bug 40414 b");
  267. confirmActiveSelected(sheet0, false);
  268. confirmActiveSelected(sheet1, false);
  269. confirmActiveSelected(sheet2, false);
  270. confirmActiveSelected(sheet4, true);
  271. sheet3 = sheet4; // re-align local vars in this test case
  272. // Some more cases of removing sheets
  273. // Starting with a multiple selection, and different active sheet
  274. wb.setSelectedTabs(new int[] { 1, 3, });
  275. wb.setActiveSheet(2);
  276. confirmActiveSelected(sheet0, false, false);
  277. confirmActiveSelected(sheet1, false, true);
  278. confirmActiveSelected(sheet2, true, false);
  279. confirmActiveSelected(sheet3, false, true);
  280. // removing a sheet that is not active, and not the only selected sheet
  281. wb.removeSheetAt(3);
  282. confirmActiveSelected(sheet0, false, false);
  283. confirmActiveSelected(sheet1, false, true);
  284. confirmActiveSelected(sheet2, true, false);
  285. // removing the only selected sheet
  286. wb.removeSheetAt(1);
  287. confirmActiveSelected(sheet0, false, false);
  288. confirmActiveSelected(sheet2, true, true);
  289. // The last remaining sheet should always be active+selected
  290. wb.removeSheetAt(1);
  291. confirmActiveSelected(sheet0, true, true);
  292. wb.close();
  293. }
  294. private static void confirmActiveSelected(HSSFSheet sheet, boolean expected) {
  295. confirmActiveSelected(sheet, expected, expected);
  296. }
  297. private static void confirmActiveSelected(HSSFSheet sheet,
  298. boolean expectedActive, boolean expectedSelected) {
  299. assertEquals(expectedActive, sheet.isActive(), "active");
  300. assertEquals(expectedSelected, sheet.isSelected(), "selected");
  301. }
  302. /**
  303. * If Sheet.getSize() returns a different result to Sheet.serialize(), this will cause the BOF
  304. * records to be written with invalid offset indexes. Excel does not like this, and such
  305. * errors are particularly hard to track down. This test ensures that HSSFWorkbook throws
  306. * a specific exception as soon as the situation is detected. See bugzilla 45066
  307. */
  308. @Test
  309. public void sheetSerializeSizeMismatch_bug45066() throws IOException {
  310. try (HSSFWorkbook wb = new HSSFWorkbook()) {
  311. InternalSheet sheet = wb.createSheet("Sheet1").getSheet();
  312. List<RecordBase> sheetRecords = sheet.getRecords();
  313. // one way (of many) to cause the discrepancy is with a badly behaved record:
  314. sheetRecords.add(new BadlyBehavedRecord());
  315. // There is also much logic inside Sheet that (if buggy) might also cause the discrepancy
  316. IllegalStateException e = assertThrows(IllegalStateException.class, wb::getBytes, "Identified bug 45066 a");
  317. assertTrue(e.getMessage().startsWith("Actual serialized sheet size"));
  318. }
  319. }
  320. /**
  321. * Checks that us and HSSFName play nicely with named ranges
  322. * that point to deleted sheets
  323. */
  324. @Test
  325. public void namesToDeleteSheets() throws IOException {
  326. HSSFWorkbook b = openSampleWorkbook("30978-deleted.xls");
  327. assertEquals(3, b.getNumberOfNames());
  328. // Sheet 2 is deleted
  329. assertEquals("Sheet1", b.getSheetName(0));
  330. assertEquals("Sheet3", b.getSheetName(1));
  331. Area3DPtg ptg;
  332. NameRecord nr;
  333. HSSFName n;
  334. /* ======= Name pointing to deleted sheet ====== */
  335. // First at low level
  336. nr = b.getWorkbook().getNameRecord(0);
  337. assertEquals("On2", nr.getNameText());
  338. assertEquals(0, nr.getSheetNumber());
  339. assertEquals(1, nr.getExternSheetNumber());
  340. assertEquals(1, nr.getNameDefinition().length);
  341. ptg = (Area3DPtg)nr.getNameDefinition()[0];
  342. assertEquals(1, ptg.getExternSheetIndex());
  343. assertEquals(0, ptg.getFirstColumn());
  344. assertEquals(0, ptg.getFirstRow());
  345. assertEquals(0, ptg.getLastColumn());
  346. assertEquals(2, ptg.getLastRow());
  347. // Now at high level
  348. n = b.getNameAt(0);
  349. assertEquals("On2", n.getNameName());
  350. assertEquals("", n.getSheetName());
  351. assertEquals("#REF!$A$1:$A$3", n.getRefersToFormula());
  352. /* ======= Name pointing to 1st sheet ====== */
  353. // First at low level
  354. nr = b.getWorkbook().getNameRecord(1);
  355. assertEquals("OnOne", nr.getNameText());
  356. assertEquals(0, nr.getSheetNumber());
  357. assertEquals(0, nr.getExternSheetNumber());
  358. assertEquals(1, nr.getNameDefinition().length);
  359. ptg = (Area3DPtg)nr.getNameDefinition()[0];
  360. assertEquals(0, ptg.getExternSheetIndex());
  361. assertEquals(0, ptg.getFirstColumn());
  362. assertEquals(2, ptg.getFirstRow());
  363. assertEquals(0, ptg.getLastColumn());
  364. assertEquals(3, ptg.getLastRow());
  365. // Now at high level
  366. n = b.getNameAt(1);
  367. assertEquals("OnOne", n.getNameName());
  368. assertEquals("Sheet1", n.getSheetName());
  369. assertEquals("Sheet1!$A$3:$A$4", n.getRefersToFormula());
  370. /* ======= Name pointing to 3rd sheet ====== */
  371. // First at low level
  372. nr = b.getWorkbook().getNameRecord(2);
  373. assertEquals("OnSheet3", nr.getNameText());
  374. assertEquals(0, nr.getSheetNumber());
  375. assertEquals(2, nr.getExternSheetNumber());
  376. assertEquals(1, nr.getNameDefinition().length);
  377. ptg = (Area3DPtg)nr.getNameDefinition()[0];
  378. assertEquals(2, ptg.getExternSheetIndex());
  379. assertEquals(0, ptg.getFirstColumn());
  380. assertEquals(0, ptg.getFirstRow());
  381. assertEquals(0, ptg.getLastColumn());
  382. assertEquals(1, ptg.getLastRow());
  383. // Now at high level
  384. n = b.getNameAt(2);
  385. assertEquals("OnSheet3", n.getNameName());
  386. assertEquals("Sheet3", n.getSheetName());
  387. assertEquals("Sheet3!$A$1:$A$2", n.getRefersToFormula());
  388. b.close();
  389. }
  390. /**
  391. * result returned by getRecordSize() differs from result returned by serialize()
  392. */
  393. private static final class BadlyBehavedRecord extends Record {
  394. BadlyBehavedRecord() {
  395. //
  396. }
  397. @Override
  398. public short getSid() {
  399. return 0x777;
  400. }
  401. @Override
  402. public int serialize(int offset, byte[] data) {
  403. return 4;
  404. }
  405. @Override
  406. public int getRecordSize() {
  407. return 8;
  408. }
  409. @Override
  410. public BadlyBehavedRecord copy() {
  411. return null;
  412. }
  413. @Override
  414. public HSSFRecordTypes getGenericRecordType() {
  415. return null;
  416. }
  417. @Override
  418. public Map<String, Supplier<?>> getGenericProperties() {
  419. return null;
  420. }
  421. }
  422. /**
  423. * The sample file provided with bug 45582 seems to have one extra byte after the EOFRecord
  424. */
  425. @Test
  426. public void extraDataAfterEOFRecord() throws IOException {
  427. // bug 45582 - RecordFormatException - getCause() instanceof LittleEndian.BufferUnderrunException
  428. try (HSSFWorkbook wb = openSampleWorkbook("ex45582-22397.xls")) {
  429. assertNotNull(wb);
  430. }
  431. }
  432. /**
  433. * Test to make sure that NameRecord.getSheetNumber() is interpreted as a
  434. * 1-based sheet tab index (not a 1-based extern sheet index)
  435. */
  436. @Test
  437. public void findBuiltInNameRecord() throws IOException {
  438. // testRRaC has multiple (3) built-in name records
  439. // The second print titles name record has getSheetNumber()==4
  440. HSSFWorkbook wb1 = openSampleWorkbook("testRRaC.xls");
  441. NameRecord nr;
  442. assertEquals(3, wb1.getWorkbook().getNumNames());
  443. nr = wb1.getWorkbook().getNameRecord(2);
  444. // TODO - render full row and full column refs properly
  445. assertEquals("Sheet2!$A$1:$IV$1", HSSFFormulaParser.toFormulaString(wb1, nr.getNameDefinition())); // 1:1
  446. try {
  447. wb1.getSheetAt(3).setRepeatingRows(CellRangeAddress.valueOf("9:12"));
  448. wb1.getSheetAt(3).setRepeatingColumns(CellRangeAddress.valueOf("E:F"));
  449. } catch (RuntimeException e) {
  450. if (e.getMessage().equals("Builtin (7) already exists for sheet (4)")) {
  451. // there was a problem in the code which locates the existing print titles name record
  452. fail("Identified bug 45720b");
  453. }
  454. wb1.close();
  455. throw e;
  456. }
  457. HSSFWorkbook wb2 = HSSFTestDataSamples.writeOutAndReadBack(wb1);
  458. wb1.close();
  459. assertEquals(3, wb2.getWorkbook().getNumNames());
  460. nr = wb2.getWorkbook().getNameRecord(2);
  461. assertEquals("Sheet2!E:F,Sheet2!$A$9:$IV$12", HSSFFormulaParser.toFormulaString(wb2, nr.getNameDefinition())); // E:F,9:12
  462. wb2.close();
  463. }
  464. /**
  465. * Test that the storage clsid property is preserved
  466. */
  467. @Test
  468. public void bug47920() throws IOException {
  469. POIFSFileSystem fs1 = new POIFSFileSystem(samples.openResourceAsStream("47920.xls"));
  470. HSSFWorkbook wb = new HSSFWorkbook(fs1);
  471. ClassID clsid1 = fs1.getRoot().getStorageClsid();
  472. ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
  473. wb.write(out);
  474. byte[] bytes = out.toByteArray();
  475. POIFSFileSystem fs2 = new POIFSFileSystem(new ByteArrayInputStream(bytes));
  476. ClassID clsid2 = fs2.getRoot().getStorageClsid();
  477. assertEquals(clsid1, clsid2);
  478. fs2.close();
  479. wb.close();
  480. fs1.close();
  481. }
  482. /**
  483. * If we try to open an old (pre-97) workbook, we get a helpful
  484. * Exception give to explain what we've done wrong
  485. */
  486. @ParameterizedTest
  487. @CsvSource({
  488. "testEXCEL_4.xls,BIFF4",
  489. "testEXCEL_5.xls,BIFF5",
  490. "testEXCEL_95.xls,BIFF5"
  491. })
  492. public void helpfulExceptionOnOldFiles(String file, String format) throws Exception {
  493. POIDataSamples xlsData = samples;
  494. try (InputStream is = xlsData.openResourceAsStream(file)) {
  495. OldExcelFormatException e = assertThrows(OldExcelFormatException.class, () -> new HSSFWorkbook(is),
  496. "Shouldn't be able to load an Excel " + format + " file");
  497. assertContains(e.getMessage(), format);
  498. }
  499. }
  500. /**
  501. * Tests that we can work with both {@link POIFSFileSystem}
  502. * and {@link POIFSFileSystem}
  503. */
  504. @Test
  505. public void differentPOIFS() throws Exception {
  506. // Open the two filesystems
  507. DirectoryNode[] files = new DirectoryNode[2];
  508. try (POIFSFileSystem poifsFileSystem = new POIFSFileSystem(HSSFTestDataSamples.openSampleFileStream("Simple.xls"))) {
  509. files[0] = poifsFileSystem.getRoot();
  510. try (POIFSFileSystem poifsFileSystem2 = new POIFSFileSystem(HSSFTestDataSamples.getSampleFile("Simple.xls"))) {
  511. files[1] = poifsFileSystem2.getRoot();
  512. // Open without preserving nodes
  513. for (DirectoryNode dir : files) {
  514. HSSFWorkbook workbook = new HSSFWorkbook(dir, false);
  515. HSSFSheet sheet = workbook.getSheetAt(0);
  516. HSSFCell cell = sheet.getRow(0).getCell(0);
  517. assertEquals("replaceMe", cell.getRichStringCellValue().getString());
  518. workbook.close();
  519. }
  520. // Now re-check with preserving
  521. for (DirectoryNode dir : files) {
  522. HSSFWorkbook workbook = new HSSFWorkbook(dir, true);
  523. HSSFSheet sheet = workbook.getSheetAt(0);
  524. HSSFCell cell = sheet.getRow(0).getCell(0);
  525. assertEquals("replaceMe", cell.getRichStringCellValue().getString());
  526. workbook.close();
  527. }
  528. }
  529. }
  530. }
  531. @Test
  532. public void wordDocEmbeddedInXls() throws IOException {
  533. // Open the two filesystems
  534. DirectoryNode[] files = new DirectoryNode[2];
  535. try (POIFSFileSystem poifsFileSystem = new POIFSFileSystem(HSSFTestDataSamples.openSampleFileStream("WithEmbeddedObjects.xls"))) {
  536. files[0] = poifsFileSystem.getRoot();
  537. try (POIFSFileSystem poifsFileSystem2 = new POIFSFileSystem(HSSFTestDataSamples.getSampleFile("WithEmbeddedObjects.xls"))) {
  538. files[1] = poifsFileSystem2.getRoot();
  539. // Check the embedded parts
  540. for (DirectoryNode root : files) {
  541. HSSFWorkbook hw = new HSSFWorkbook(root, true);
  542. List<HSSFObjectData> objects = hw.getAllEmbeddedObjects();
  543. boolean found = false;
  544. for (HSSFObjectData embeddedObject : objects) {
  545. if (embeddedObject.hasDirectoryEntry()) {
  546. DirectoryEntry dir = embeddedObject.getDirectory();
  547. if (dir instanceof DirectoryNode) {
  548. DirectoryNode dNode = (DirectoryNode) dir;
  549. if (dNode.hasEntry("WordDocument")) {
  550. found = true;
  551. break;
  552. }
  553. }
  554. }
  555. }
  556. assertTrue(found);
  557. hw.close();
  558. }
  559. }
  560. }
  561. }
  562. /**
  563. * Checks that we can open a workbook with POIFS, and write it out
  564. * again (via POIFS) and have it be valid
  565. */
  566. @Test
  567. public void writeWorkbookFromPOIFS() throws IOException {
  568. try (InputStream is = HSSFTestDataSamples.openSampleFileStream("WithEmbeddedObjects.xls");
  569. POIFSFileSystem fs = new POIFSFileSystem(is)) {
  570. HSSFWorkbook wb = new HSSFWorkbook(fs.getRoot(), true);
  571. assertEquals(3, wb.getNumberOfSheets());
  572. assertEquals("Root xls", wb.getSheetAt(0).getRow(0).getCell(0).getStringCellValue());
  573. HSSFWorkbook wbBack = HSSFTestDataSamples.writeOutAndReadBack(wb);
  574. assertEquals(3, wbBack.getNumberOfSheets());
  575. assertEquals("Root xls", wbBack.getSheetAt(0).getRow(0).getCell(0).getStringCellValue());
  576. wbBack.close();
  577. wb.close();
  578. }
  579. }
  580. @Test
  581. public void cellStylesLimit() throws IOException {
  582. try (Workbook wb = new HSSFWorkbook()) {
  583. int numBuiltInStyles = wb.getNumCellStyles();
  584. int MAX_STYLES = 4030;
  585. int limit = MAX_STYLES - numBuiltInStyles;
  586. for (int i = 0; i < limit; i++) {
  587. wb.createCellStyle();
  588. }
  589. assertEquals(MAX_STYLES, wb.getNumCellStyles());
  590. IllegalStateException e = assertThrows(IllegalStateException.class, wb::createCellStyle);
  591. assertEquals("The maximum number of cell styles was exceeded. " +
  592. "You can define up to 4000 styles in a .xls workbook", e.getMessage());
  593. assertEquals(MAX_STYLES, wb.getNumCellStyles());
  594. }
  595. }
  596. @Test
  597. public void setSheetOrderHSSF() throws IOException{
  598. Workbook wb = new HSSFWorkbook();
  599. Sheet s1 = wb.createSheet("first sheet");
  600. Sheet s2 = wb.createSheet("other sheet");
  601. Name name1 = wb.createName();
  602. name1.setNameName("name1");
  603. name1.setRefersToFormula("'first sheet'!D1");
  604. Name name2 = wb.createName();
  605. name2.setNameName("name2");
  606. name2.setRefersToFormula("'other sheet'!C1");
  607. Row s1r1 = s1.createRow(2);
  608. Cell c1 = s1r1.createCell(3);
  609. c1.setCellValue(30);
  610. Cell c2 = s1r1.createCell(2);
  611. c2.setCellFormula("SUM('other sheet'!C1,'first sheet'!C1)");
  612. Row s2r1 = s2.createRow(0);
  613. Cell c3 = s2r1.createCell(1);
  614. c3.setCellFormula("'first sheet'!D3");
  615. Cell c4 = s2r1.createCell(2);
  616. c4.setCellFormula("'other sheet'!D3");
  617. // conditional formatting
  618. SheetConditionalFormatting sheetCF = s1.getSheetConditionalFormatting();
  619. ConditionalFormattingRule rule1 = sheetCF.createConditionalFormattingRule(
  620. CFRuleRecord.ComparisonOperator.BETWEEN, "'first sheet'!D1", "'other sheet'!D1");
  621. ConditionalFormattingRule [] cfRules = { rule1 };
  622. CellRangeAddress[] regions = {
  623. new CellRangeAddress(2, 4, 0, 0), // A3:A5
  624. };
  625. sheetCF.addConditionalFormatting(regions, cfRules);
  626. wb.setSheetOrder("other sheet", 0);
  627. // names
  628. //noinspection ConstantConditions
  629. assertEquals("'first sheet'!D1", wb.getName("name1").getRefersToFormula());
  630. //noinspection ConstantConditions
  631. assertEquals("'other sheet'!C1", wb.getName("name2").getRefersToFormula());
  632. // cells
  633. assertEquals("SUM('other sheet'!C1,'first sheet'!C1)", c2.getCellFormula());
  634. assertEquals("'first sheet'!D3", c3.getCellFormula());
  635. assertEquals("'other sheet'!D3", c4.getCellFormula());
  636. // conditional formatting
  637. ConditionalFormatting cf = sheetCF.getConditionalFormattingAt(0);
  638. assertEquals("'first sheet'!D1", cf.getRule(0).getFormula1());
  639. assertEquals("'other sheet'!D1", cf.getRule(0).getFormula2());
  640. wb.close();
  641. }
  642. @Test
  643. public void clonePictures() throws IOException {
  644. HSSFWorkbook wb = openSampleWorkbook("SimpleWithImages.xls");
  645. InternalWorkbook iwb = wb.getWorkbook();
  646. iwb.findDrawingGroup();
  647. for(int pictureIndex=1; pictureIndex <= 4; pictureIndex++){
  648. EscherBSERecord bse = iwb.getBSERecord(pictureIndex);
  649. assertEquals(1, bse.getRef());
  650. }
  651. wb.cloneSheet(0);
  652. for(int pictureIndex=1; pictureIndex <= 4; pictureIndex++){
  653. EscherBSERecord bse = iwb.getBSERecord(pictureIndex);
  654. assertEquals(2, bse.getRef());
  655. }
  656. wb.cloneSheet(0);
  657. for(int pictureIndex=1; pictureIndex <= 4; pictureIndex++){
  658. EscherBSERecord bse = iwb.getBSERecord(pictureIndex);
  659. assertEquals(3, bse.getRef());
  660. }
  661. wb.close();
  662. }
  663. // Should throw exception about invalid POIFSFileSystem
  664. @Test
  665. public void emptyDirectoryNode() throws IOException {
  666. try (POIFSFileSystem fs = new POIFSFileSystem()) {
  667. assertThrows(IllegalArgumentException.class, () -> new HSSFWorkbook(fs).close());
  668. }
  669. }
  670. @Test
  671. public void selectedSheetShort() throws IOException {
  672. HSSFWorkbook wb=new HSSFWorkbook();
  673. HSSFSheet sheet1 = wb.createSheet("Sheet1");
  674. HSSFSheet sheet2 = wb.createSheet("Sheet2");
  675. HSSFSheet sheet3 = wb.createSheet("Sheet3");
  676. HSSFSheet sheet4 = wb.createSheet("Sheet4");
  677. confirmActiveSelected(sheet1, true);
  678. confirmActiveSelected(sheet2, false);
  679. confirmActiveSelected(sheet3, false);
  680. confirmActiveSelected(sheet4, false);
  681. wb.setSelectedTab((short)1);
  682. // Demonstrate bug 44525:
  683. // Well... not quite, since isActive + isSelected were also added in the same bug fix
  684. assertFalse(sheet1.isSelected(), "Identified bug 44523 a");
  685. wb.setActiveSheet(1);
  686. assertFalse(sheet1.isActive(), "Identified bug 44523 b");
  687. confirmActiveSelected(sheet1, false);
  688. confirmActiveSelected(sheet2, true);
  689. confirmActiveSelected(sheet3, false);
  690. confirmActiveSelected(sheet4, false);
  691. assertEquals(0, wb.getFirstVisibleTab());
  692. wb.setFirstVisibleTab((short)2);
  693. assertEquals(2, wb.getFirstVisibleTab());
  694. wb.close();
  695. }
  696. @Test
  697. public void names() throws IOException {
  698. HSSFWorkbook wb=new HSSFWorkbook();
  699. IllegalStateException ex1 = assertThrows(IllegalStateException.class, () -> wb.getNameAt(0));
  700. assertTrue(ex1.getMessage().contains("no defined names"));
  701. HSSFName name = wb.createName();
  702. assertNotNull(name);
  703. assertNull(wb.getName("somename"));
  704. name.setNameName("myname");
  705. assertNotNull(wb.getName("myname"));
  706. assertEquals(0, wb.getNameIndex(name));
  707. assertEquals(0, wb.getNameIndex("myname"));
  708. IllegalArgumentException ex2 = assertThrows(IllegalArgumentException.class, () -> wb.getNameAt(5));
  709. assertTrue(ex2.getMessage().contains("outside the allowable range"));
  710. IllegalArgumentException ex = assertThrows(IllegalArgumentException.class, () -> wb.getNameAt(-3));
  711. assertTrue(ex.getMessage().contains("outside the allowable range"));
  712. wb.close();
  713. }
  714. @Test
  715. public void testMethods() throws IOException {
  716. try (HSSFWorkbook wb=new HSSFWorkbook()) {
  717. assertDoesNotThrow(wb::insertChartRecord);
  718. //wb.dumpDrawingGroupRecords(true);
  719. //wb.dumpDrawingGroupRecords(false);
  720. }
  721. }
  722. @Test
  723. public void writeProtection() throws IOException {
  724. HSSFWorkbook wb=new HSSFWorkbook();
  725. assertFalse(wb.isWriteProtected());
  726. wb.writeProtectWorkbook("mypassword", "myuser");
  727. assertTrue(wb.isWriteProtected());
  728. wb.unwriteProtectWorkbook();
  729. assertFalse(wb.isWriteProtected());
  730. wb.close();
  731. }
  732. @Test
  733. public void bug50298() throws Exception {
  734. HSSFWorkbook wb = openSampleWorkbook("50298.xls");
  735. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received");
  736. HSSFSheet sheet = wb.cloneSheet(0);
  737. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received", "Invoice (2)");
  738. wb.setSheetName(wb.getSheetIndex(sheet), "copy");
  739. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received", "copy");
  740. wb.setSheetOrder("copy", 0);
  741. assertSheetOrder(wb, "copy", "Invoice", "Invoice1", "Digest", "Deferred", "Received");
  742. wb.removeSheetAt(0);
  743. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received");
  744. // check that the overall workbook serializes with its correct size
  745. int expected = wb.getWorkbook().getSize();
  746. int written = wb.getWorkbook().serialize(0, new byte[expected*2]);
  747. assertEquals(expected, written, "Did not have the expected size when writing the workbook");
  748. HSSFWorkbook read = HSSFTestDataSamples.writeOutAndReadBack(wb);
  749. assertSheetOrder(read, "Invoice", "Invoice1", "Digest", "Deferred", "Received");
  750. read.close();
  751. wb.close();
  752. }
  753. @Test
  754. public void bug50298a() throws Exception {
  755. HSSFWorkbook wb = openSampleWorkbook("50298.xls");
  756. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received");
  757. HSSFSheet sheet = wb.cloneSheet(0);
  758. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received", "Invoice (2)");
  759. wb.setSheetName(wb.getSheetIndex(sheet), "copy");
  760. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received", "copy");
  761. wb.setSheetOrder("copy", 0);
  762. assertSheetOrder(wb, "copy", "Invoice", "Invoice1", "Digest", "Deferred", "Received");
  763. wb.removeSheetAt(0);
  764. assertSheetOrder(wb, "Invoice", "Invoice1", "Digest", "Deferred", "Received");
  765. wb.removeSheetAt(1);
  766. assertSheetOrder(wb, "Invoice", "Digest", "Deferred", "Received");
  767. wb.setSheetOrder("Digest", 3);
  768. assertSheetOrder(wb, "Invoice", "Deferred", "Received", "Digest");
  769. // check that the overall workbook serializes with its correct size
  770. int expected = wb.getWorkbook().getSize();
  771. int written = wb.getWorkbook().serialize(0, new byte[expected*2]);
  772. assertEquals(expected, written, "Did not have the expected size when writing the workbook");
  773. HSSFWorkbook read = HSSFTestDataSamples.writeOutAndReadBack(wb);
  774. assertSheetOrder(read, "Invoice", "Deferred", "Received", "Digest");
  775. read.close();
  776. wb.close();
  777. }
  778. @Test
  779. public void bug54500() throws Exception {
  780. String nameName = "AName";
  781. String sheetName = "ASheet";
  782. HSSFWorkbook wb = openSampleWorkbook("54500.xls");
  783. assertSheetOrder(wb, "Sheet1", "Sheet2", "Sheet3");
  784. wb.createSheet(sheetName);
  785. assertSheetOrder(wb, "Sheet1", "Sheet2", "Sheet3", "ASheet");
  786. Name n = wb.createName();
  787. n.setNameName(nameName);
  788. n.setSheetIndex(3);
  789. n.setRefersToFormula(sheetName + "!A1");
  790. assertSheetOrder(wb, "Sheet1", "Sheet2", "Sheet3", "ASheet");
  791. final HSSFName name = wb.getName(nameName);
  792. assertNotNull(name);
  793. assertEquals("ASheet!A1", name.getRefersToFormula());
  794. ByteArrayOutputStream stream = new ByteArrayOutputStream();
  795. wb.write(stream);
  796. assertSheetOrder(wb, "Sheet1", "Sheet2", "Sheet3", "ASheet");
  797. assertEquals("ASheet!A1", name.getRefersToFormula());
  798. wb.removeSheetAt(1);
  799. assertSheetOrder(wb, "Sheet1", "Sheet3", "ASheet");
  800. assertEquals("ASheet!A1", name.getRefersToFormula());
  801. ByteArrayOutputStream stream2 = new ByteArrayOutputStream();
  802. wb.write(stream2);
  803. assertSheetOrder(wb, "Sheet1", "Sheet3", "ASheet");
  804. assertEquals("ASheet!A1", name.getRefersToFormula());
  805. HSSFWorkbook wb2 = new HSSFWorkbook(new ByteArrayInputStream(stream.toByteArray()));
  806. expectName(wb2, nameName, "ASheet!A1");
  807. HSSFWorkbook wb3 = new HSSFWorkbook(new ByteArrayInputStream(stream2.toByteArray()));
  808. expectName(wb3, nameName, "ASheet!A1");
  809. wb3.close();
  810. wb2.close();
  811. wb.close();
  812. }
  813. @SuppressWarnings("SameParameterValue")
  814. private void expectName(HSSFWorkbook wb, String name, String expect) {
  815. final HSSFName hssfName = wb.getName(name);
  816. assertNotNull(hssfName);
  817. assertEquals(expect, hssfName.getRefersToFormula());
  818. }
  819. @Test
  820. public void test49423() throws Exception
  821. {
  822. HSSFWorkbook workbook = openSampleWorkbook("49423.xls");
  823. boolean found = false;
  824. int numSheets = workbook.getNumberOfSheets();
  825. for (int i = 0; i < numSheets; i++) {
  826. HSSFSheet sheet = workbook.getSheetAt(i);
  827. List<HSSFShape> shapes = sheet.getDrawingPatriarch().getChildren();
  828. for(HSSFShape shape : shapes){
  829. HSSFAnchor anchor = shape.getAnchor();
  830. if(anchor instanceof HSSFClientAnchor){
  831. // absolute coordinates
  832. HSSFClientAnchor clientAnchor = (HSSFClientAnchor)anchor;
  833. assertNotNull(clientAnchor);
  834. //System.out.println(clientAnchor.getRow1() + "," + clientAnchor.getRow2());
  835. found = true;
  836. } else if (anchor instanceof HSSFChildAnchor){
  837. // shape is grouped and the anchor is expressed in the coordinate system of the group
  838. HSSFChildAnchor childAnchor = (HSSFChildAnchor)anchor;
  839. assertNotNull(childAnchor);
  840. //System.out.println(childAnchor.getDy1() + "," + childAnchor.getDy2());
  841. found = true;
  842. }
  843. }
  844. }
  845. assertTrue(found, "Should find some images via Client or Child anchors, but did not find any at all");
  846. workbook.close();
  847. }
  848. @Test
  849. public void testRewriteFileBug58480() throws IOException {
  850. final File file = TempFile.createTempFile("TestHSSFWorkbook", ".xls");
  851. try {
  852. // create new workbook
  853. {
  854. final Workbook workbook = new HSSFWorkbook();
  855. final Sheet sheet = workbook.createSheet("foo");
  856. final Row row = sheet.createRow(1);
  857. row.createCell(1).setCellValue("bar");
  858. writeAndCloseWorkbook(workbook, file);
  859. }
  860. // edit the workbook
  861. {
  862. try (POIFSFileSystem fs = new POIFSFileSystem(file, false)) {
  863. DirectoryNode root = fs.getRoot();
  864. final Workbook workbook = new HSSFWorkbook(root, true);
  865. final Sheet sheet = workbook.getSheet("foo");
  866. sheet.getRow(1).createCell(2).setCellValue("baz");
  867. writeAndCloseWorkbook(workbook, file);
  868. }
  869. }
  870. } finally {
  871. assertTrue(file.exists());
  872. assertTrue(file.delete());
  873. }
  874. }
  875. private void writeAndCloseWorkbook(Workbook workbook, File file)
  876. throws IOException {
  877. final ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
  878. workbook.write(bytesOut);
  879. workbook.close();
  880. final byte[] byteArray = bytesOut.toByteArray();
  881. bytesOut.close();
  882. final FileOutputStream fileOut = new FileOutputStream(file);
  883. fileOut.write(byteArray);
  884. fileOut.close();
  885. }
  886. @Test
  887. public void closeDoesNotModifyWorkbook() throws IOException {
  888. final String filename = "SampleSS.xls";
  889. final File file = samples.getFile(filename);
  890. Workbook wb;
  891. // File via POIFileStream (java.nio)
  892. wb = new HSSFWorkbook(new POIFSFileSystem(file));
  893. assertCloseDoesNotModifyFile(filename, wb);
  894. // InputStream
  895. wb = new HSSFWorkbook(new FileInputStream(file));
  896. assertCloseDoesNotModifyFile(filename, wb);
  897. }
  898. @Test
  899. public void setSheetOrderToEnd() throws Exception {
  900. try (HSSFWorkbook workbook = new HSSFWorkbook()) {
  901. workbook.createSheet("A");
  902. workbook.createSheet("B");
  903. assertEquals("A", workbook.getSheetName(0));
  904. workbook.setSheetOrder("A", 1);
  905. assertEquals("A", workbook.getSheetName(1));
  906. }
  907. }
  908. @Test
  909. public void invalidInPlaceWrite() throws Exception {
  910. // Can't work for new files
  911. try (HSSFWorkbook wb = new HSSFWorkbook()) {
  912. assertThrows(IllegalStateException.class, wb::write, "Shouldn't work for new files");
  913. }
  914. // Can't work for InputStream opened files
  915. try (InputStream is = samples.openResourceAsStream("SampleSS.xls");
  916. HSSFWorkbook wb = new HSSFWorkbook(is)) {
  917. assertThrows(IllegalStateException.class, wb::write, "Shouldn't work for InputStream");
  918. }
  919. // Can't work for Read-Only files
  920. try (POIFSFileSystem fs = new POIFSFileSystem(samples.getFile("SampleSS.xls"), true);
  921. HSSFWorkbook wb = new HSSFWorkbook(fs)) {
  922. assertThrows(IllegalStateException.class, wb::write, "Shouldn't work for Read Only");
  923. }
  924. }
  925. @Test
  926. public void inPlaceWrite() throws Exception {
  927. // Setup as a copy of a known-good file
  928. final File file = TempFile.createTempFile("TestHSSFWorkbook", ".xls");
  929. try (InputStream inputStream = samples.openResourceAsStream("SampleSS.xls");
  930. FileOutputStream outputStream = new FileOutputStream(file)) {
  931. IOUtils.copy(inputStream, outputStream);
  932. }
  933. // Open from the temp file in read-write mode
  934. HSSFWorkbook wb = new HSSFWorkbook(new POIFSFileSystem(file, false));
  935. assertEquals(3, wb.getNumberOfSheets());
  936. // Change
  937. wb.removeSheetAt(2);
  938. wb.removeSheetAt(1);
  939. wb.getSheetAt(0).getRow(0).getCell(0).setCellValue("Changed!");
  940. // Save in-place, close, re-open and check
  941. wb.write();
  942. wb.close();
  943. wb = new HSSFWorkbook(new POIFSFileSystem(file));
  944. assertEquals(1, wb.getNumberOfSheets());
  945. assertEquals("Changed!", wb.getSheetAt(0).getRow(0).getCell(0).toString());
  946. wb.close();
  947. }
  948. @Test
  949. public void testWriteToNewFile() throws Exception {
  950. // Save to a new temp file
  951. final File file = TempFile.createTempFile("TestHSSFWorkbook", ".xls");
  952. // Open from a Stream
  953. try (HSSFWorkbook wb = new HSSFWorkbook(
  954. samples.openResourceAsStream("SampleSS.xls"))) {
  955. wb.write(file);
  956. }
  957. // Read and check
  958. try (HSSFWorkbook wb = new HSSFWorkbook(new POIFSFileSystem(file))) {
  959. assertEquals(3, wb.getNumberOfSheets());
  960. }
  961. }
  962. public void createDrawing() throws Exception {
  963. // the dimensions for this image are different than for XSSF and SXSSF
  964. }
  965. }