testsheethiding.java

来自「EXCEL read and write」· Java 代码 · 共 113 行

/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements.  See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License.  You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package org.apache.poi.hssf.usermodel;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;

import junit.framework.TestCase;

import org.apache.poi.hssf.HSSFTestDataSamples;

/**
 * Tests for how HSSFWorkbook behaves with XLS files
 *  with a WORKBOOK directory entry (instead of the more
 *  usual, Workbook)
 */
public final class TestSheetHiding extends TestCase {
	private HSSFWorkbook wbH;
	private HSSFWorkbook wbU;

	protected void setUp() {
		wbH = HSSFTestDataSamples.openSampleWorkbook("TwoSheetsOneHidden.xls");
		wbU = HSSFTestDataSamples.openSampleWorkbook("TwoSheetsNoneHidden.xls");
	}

	/**
	 * Test that we get the right number of sheets,
	 *  with the right text on them, no matter what
	 *  the hidden flags are
	 */
	public void testTextSheets() throws Exception {
		// Both should have two sheets
		assertEquals(2, wbH.getNumberOfSheets());
		assertEquals(2, wbU.getNumberOfSheets());

		// All sheets should have one row
		assertEquals(0, wbH.getSheetAt(0).getLastRowNum());
		assertEquals(0, wbH.getSheetAt(1).getLastRowNum());
		assertEquals(0, wbU.getSheetAt(0).getLastRowNum());
		assertEquals(0, wbU.getSheetAt(1).getLastRowNum());

		// All rows should have one column
		assertEquals(1, wbH.getSheetAt(0).getRow(0).getLastCellNum());
		assertEquals(1, wbH.getSheetAt(1).getRow(0).getLastCellNum());
		assertEquals(1, wbU.getSheetAt(0).getRow(0).getLastCellNum());
		assertEquals(1, wbU.getSheetAt(1).getRow(0).getLastCellNum());

		// Text should be sheet based
		assertEquals("Sheet1A1", wbH.getSheetAt(0).getRow(0).getCell(0).getRichStringCellValue().getString());
		assertEquals("Sheet2A1", wbH.getSheetAt(1).getRow(0).getCell(0).getRichStringCellValue().getString());
		assertEquals("Sheet1A1", wbU.getSheetAt(0).getRow(0).getCell(0).getRichStringCellValue().getString());
		assertEquals("Sheet2A1", wbU.getSheetAt(1).getRow(0).getCell(0).getRichStringCellValue().getString());
	}

	/**
	 * Check that we can get and set the hidden flags
	 *  as expected
	 */
	public void testHideUnHideFlags() throws Exception {
		assertTrue(wbH.isSheetHidden(0));
		assertFalse(wbH.isSheetHidden(1));
		assertFalse(wbU.isSheetHidden(0));
		assertFalse(wbU.isSheetHidden(1));
	}

	/**
	 * Turn the sheet with none hidden into the one with
	 *  one hidden
	 */
	public void testHide() throws Exception {
		wbU.setSheetHidden(0, true);
		assertTrue(wbU.isSheetHidden(0));
		assertFalse(wbU.isSheetHidden(1));
		ByteArrayOutputStream out = new ByteArrayOutputStream();
		wbU.write(out);
		out.close();
		HSSFWorkbook wb2 = new HSSFWorkbook(new ByteArrayInputStream(out.toByteArray()));
		assertTrue(wb2.isSheetHidden(0));
		assertFalse(wb2.isSheetHidden(1));
	}

	/**
	 * Turn the sheet with one hidden into the one with
	 *  none hidden
	 */
	public void testUnHide() throws Exception {
		wbH.setSheetHidden(0, false);
		assertFalse(wbH.isSheetHidden(0));
		assertFalse(wbH.isSheetHidden(1));
		ByteArrayOutputStream out = new ByteArrayOutputStream();
		wbH.write(out);
		out.close();
		HSSFWorkbook wb2 = new HSSFWorkbook(new ByteArrayInputStream(out.toByteArray()));
		assertFalse(wb2.isSheetHidden(0));
		assertFalse(wb2.isSheetHidden(1));
	}
}