| /* ==================================================================== |
| 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.poifs.property; |
| |
| import static org.junit.jupiter.api.Assertions.assertArrayEquals; |
| import static org.junit.jupiter.api.Assertions.assertEquals; |
| import static org.junit.jupiter.api.Assertions.assertFalse; |
| |
| import java.io.IOException; |
| import java.util.Arrays; |
| |
| import org.apache.commons.io.output.UnsynchronizedByteArrayOutputStream; |
| import org.apache.poi.hpsf.ClassIDPredefined; |
| import org.apache.poi.poifs.common.POIFSConstants; |
| import org.apache.poi.poifs.storage.RawDataUtil; |
| import org.apache.poi.util.LocaleUtil; |
| import org.junit.jupiter.api.Test; |
| |
| /** |
| * Class to test RootProperty functionality |
| */ |
| final class TestRootProperty { |
| private RootProperty _property; |
| private byte[] _testblock; |
| |
| @Test |
| void testConstructor() throws IOException { |
| createBasicRootProperty(); |
| |
| UnsynchronizedByteArrayOutputStream stream = new UnsynchronizedByteArrayOutputStream(512); |
| _property.writeData(stream); |
| assertArrayEquals(_testblock, stream.toByteArray()); |
| |
| } |
| |
| private void createBasicRootProperty() { |
| _property = new RootProperty(); |
| _testblock = new byte[128]; |
| int index = 0; |
| |
| for (; index < 0x40; index++) { |
| _testblock[index] = (byte) 0; |
| } |
| String name = "Root Entry"; |
| int limit = name.length(); |
| |
| _testblock[index++] = (byte) (2 * (limit + 1)); |
| _testblock[index++] = (byte) 0; |
| _testblock[index++] = (byte) 5; |
| _testblock[index++] = (byte) 1; |
| for (; index < 0x50; index++) { |
| _testblock[index] = (byte) 0xff; |
| } |
| for (; index < 0x74; index++) { |
| _testblock[index] = (byte) 0; |
| } |
| _testblock[index++] = (byte) POIFSConstants.END_OF_CHAIN; |
| for (; index < 0x78; index++) { |
| _testblock[index] = (byte) 0xff; |
| } |
| for (; index < 0x80; index++) { |
| _testblock[index] = (byte) 0; |
| } |
| byte[] name_bytes = name.getBytes(LocaleUtil.CHARSET_1252); |
| |
| for (index = 0; index < limit; index++) { |
| _testblock[index * 2] = name_bytes[index]; |
| } |
| } |
| |
| @Test |
| void testSetSize() { |
| for (int j = 0; j < 10; j++) { |
| createBasicRootProperty(); |
| _property.setSize(j); |
| assertEquals(j * 64, _property.getSize(), "trying block count of " + j); |
| } |
| } |
| |
| @Test |
| void testReadingConstructor() throws IOException { |
| String[] inputBytes = { |
| "52 00 6F 00 6F 00 74 00 20 00 45 00 6E 00 74 00 72 00 79 00 00 00 00 00 00 00 00 00 00 00 00 00", |
| "00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00", |
| "16 00 05 01 FF FF FF FF FF FF FF FF 02 00 00 00 20 08 02 00 00 00 00 00 C0 00 00 00 00 00 00 46", |
| "00 00 00 00 00 00 00 00 00 00 00 00 C0 5C E8 23 9E 6B C1 01 FE FF FF FF 00 00 00 00 00 00 00 00", |
| }; |
| int index = 0; |
| byte[] input = RawDataUtil.decode(inputBytes); |
| int offset = 0; |
| |
| RootProperty property = new RootProperty(index, input, offset); |
| UnsynchronizedByteArrayOutputStream stream = new UnsynchronizedByteArrayOutputStream(128); |
| byte[] expected = Arrays.copyOfRange(input, offset, offset+128); |
| property.writeData(stream); |
| byte[] output = stream.toByteArray(); |
| |
| assertEquals(128, output.length); |
| for (int j = 0; j < 128; j++) { |
| assertEquals(expected[j], output[j], "mismatch at offset " + j); |
| } |
| assertEquals(index, property.getIndex()); |
| assertEquals("Root Entry", property.getName()); |
| assertFalse(property.getChildren().hasNext()); |
| assertEquals(ClassIDPredefined.EXCEL_V8.getClassID(), property.getStorageClsid()); |
| } |
| } |