| /* ==================================================================== |
| 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.model; |
| |
| import junit.framework.TestCase; |
| import org.apache.poi.ddf.DefaultEscherRecordFactory; |
| import org.apache.poi.ddf.EscherContainerRecord; |
| import org.apache.poi.ddf.EscherTextboxRecord; |
| import org.apache.poi.hssf.HSSFTestDataSamples; |
| import org.apache.poi.hssf.record.EscherAggregate; |
| import org.apache.poi.hssf.record.Record; |
| import org.apache.poi.hssf.record.RecordBase; |
| import org.apache.poi.hssf.usermodel.HSSFSheet; |
| import org.apache.poi.hssf.usermodel.HSSFTestHelper; |
| import org.apache.poi.hssf.usermodel.HSSFWorkbook; |
| |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.util.Arrays; |
| import java.util.List; |
| import java.util.Random; |
| |
| /** |
| * @author Evgeniy Berlog |
| * @date 18.06.12 |
| */ |
| public class TestEscherRecordFactory extends TestCase{ |
| |
| private static byte[] toByteArray(List<RecordBase> records) { |
| ByteArrayOutputStream out = new ByteArrayOutputStream(); |
| for (RecordBase rb : records) { |
| Record r = (Record) rb; |
| try { |
| out.write(r.serialize()); |
| } catch (IOException e) { |
| throw new RuntimeException(e); |
| } |
| } |
| return out.toByteArray(); |
| } |
| |
| public void testDetectContainer() { |
| Random rnd = new Random(); |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0x0, EscherContainerRecord.DG_CONTAINER)); |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0x0, EscherContainerRecord.SOLVER_CONTAINER)); |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0x0, EscherContainerRecord.SP_CONTAINER)); |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0x0, EscherContainerRecord.DGG_CONTAINER)); |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0x0, EscherContainerRecord.BSTORE_CONTAINER)); |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0x0, EscherContainerRecord.SPGR_CONTAINER)); |
| |
| for (Short i=EscherContainerRecord.DGG_CONTAINER; i<= EscherContainerRecord.SOLVER_CONTAINER; i++){ |
| assertEquals(true, DefaultEscherRecordFactory.isContainer(Integer.valueOf(rnd.nextInt(Short.MAX_VALUE)).shortValue(), i)); |
| } |
| |
| assertEquals(false, DefaultEscherRecordFactory.isContainer((short) 0x0, Integer.valueOf(EscherContainerRecord.DGG_CONTAINER-1).shortValue())); |
| assertEquals(false, DefaultEscherRecordFactory.isContainer((short) 0x0, Integer.valueOf(EscherContainerRecord.SOLVER_CONTAINER+1).shortValue())); |
| |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0x000F, Integer.valueOf(EscherContainerRecord.DGG_CONTAINER-1).shortValue())); |
| assertEquals(true, DefaultEscherRecordFactory.isContainer((short) 0xFFFF, Integer.valueOf(EscherContainerRecord.DGG_CONTAINER-1).shortValue())); |
| assertEquals(false, DefaultEscherRecordFactory.isContainer((short) 0x000C, Integer.valueOf(EscherContainerRecord.DGG_CONTAINER-1).shortValue())); |
| assertEquals(false, DefaultEscherRecordFactory.isContainer((short) 0xCCCC, Integer.valueOf(EscherContainerRecord.DGG_CONTAINER-1).shortValue())); |
| assertEquals(false, DefaultEscherRecordFactory.isContainer((short) 0x000F, EscherTextboxRecord.RECORD_ID)); |
| assertEquals(false, DefaultEscherRecordFactory.isContainer((short) 0xCCCC, EscherTextboxRecord.RECORD_ID)); |
| } |
| |
| public void testDgContainerMustBeRootOfHSSFSheetEscherRecords() throws IOException { |
| HSSFWorkbook wb = HSSFTestDataSamples.openSampleWorkbook("47251.xls"); |
| HSSFSheet sh = wb.getSheetAt(0); |
| InternalSheet ish = HSSFTestHelper.getSheetForTest(sh); |
| List<RecordBase> records = ish.getRecords(); |
| // records to be aggregated |
| List<RecordBase> dgRecords = records.subList(19, 23); |
| byte[] dgBytes = toByteArray(dgRecords); |
| sh.getDrawingPatriarch(); |
| EscherAggregate agg = (EscherAggregate) ish.findFirstRecordBySid(EscherAggregate.sid); |
| assertEquals(true, agg.getEscherRecords().get(0) instanceof EscherContainerRecord); |
| assertEquals(EscherContainerRecord.DG_CONTAINER, agg.getEscherRecords().get(0).getRecordId()); |
| assertEquals((short) 0x0, agg.getEscherRecords().get(0).getOptions()); |
| agg = (EscherAggregate) ish.findFirstRecordBySid(EscherAggregate.sid); |
| byte[] dgBytesAfterSave = agg.serialize(); |
| assertEquals("different size of drawing data before and after save", dgBytes.length, dgBytesAfterSave.length); |
| assertTrue("drawing data before and after save is different", Arrays.equals(dgBytes, dgBytesAfterSave)); |
| } |
| } |