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.

TestHSSFWorkbook.java 38KB

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