40d49a54d2
git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1496675 13f79535-47bb-0310-9956-ffa450edef68
173 lines
5.8 KiB
Java
173 lines
5.8 KiB
Java
/* ====================================================================
|
|
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;
|
|
|
|
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
|
|
import junit.framework.TestCase;
|
|
|
|
import org.apache.poi.hssf.HSSFTestDataSamples;
|
|
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
|
|
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
|
|
|
|
/**
|
|
* Tests that POIDocument correctly loads and saves the common
|
|
* (hspf) Document Properties.
|
|
*
|
|
* This is part 1 of 2 of the tests - it only does the POIDocuments
|
|
* which are part of the Main (not scratchpad)
|
|
*
|
|
* @author Nick Burch (nick at torchbox dot com)
|
|
*/
|
|
public final class TestPOIDocumentMain extends TestCase {
|
|
// The POI Documents to work on
|
|
private POIDocument doc;
|
|
private POIDocument doc2;
|
|
|
|
/**
|
|
* Set things up, two spreadsheets for our testing
|
|
*/
|
|
public void setUp() {
|
|
doc = HSSFTestDataSamples.openSampleWorkbook("DateFormats.xls");
|
|
doc2 = HSSFTestDataSamples.openSampleWorkbook("StringFormulas.xls");
|
|
}
|
|
|
|
public void testReadProperties() {
|
|
// We should have both sets
|
|
assertNotNull(doc.getDocumentSummaryInformation());
|
|
assertNotNull(doc.getSummaryInformation());
|
|
|
|
// Check they are as expected for the test doc
|
|
assertEquals("Administrator", doc.getSummaryInformation().getAuthor());
|
|
assertEquals(0, doc.getDocumentSummaryInformation().getByteCount());
|
|
}
|
|
|
|
public void testReadProperties2() {
|
|
// Check again on the word one
|
|
assertNotNull(doc2.getDocumentSummaryInformation());
|
|
assertNotNull(doc2.getSummaryInformation());
|
|
|
|
assertEquals("Avik Sengupta", doc2.getSummaryInformation().getAuthor());
|
|
assertEquals(null, doc2.getSummaryInformation().getKeywords());
|
|
assertEquals(0, doc2.getDocumentSummaryInformation().getByteCount());
|
|
}
|
|
|
|
public void testWriteProperties() throws Exception {
|
|
// Just check we can write them back out into a filesystem
|
|
POIFSFileSystem outFS = new POIFSFileSystem();
|
|
doc.readProperties();
|
|
doc.writeProperties(outFS);
|
|
|
|
// Should now hold them
|
|
assertNotNull(
|
|
outFS.createDocumentInputStream("\005SummaryInformation")
|
|
);
|
|
assertNotNull(
|
|
outFS.createDocumentInputStream("\005DocumentSummaryInformation")
|
|
);
|
|
}
|
|
|
|
public void testWriteReadProperties() throws Exception {
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
|
|
// Write them out
|
|
POIFSFileSystem outFS = new POIFSFileSystem();
|
|
doc.readProperties();
|
|
doc.writeProperties(outFS);
|
|
outFS.writeFilesystem(baos);
|
|
|
|
// Create a new version
|
|
ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
|
|
POIFSFileSystem inFS = new POIFSFileSystem(bais);
|
|
|
|
// Check they're still there
|
|
doc.directory = inFS.getRoot();
|
|
doc.readProperties();
|
|
|
|
// Delegate test
|
|
testReadProperties();
|
|
}
|
|
|
|
public void testCreateNewProperties() throws IOException {
|
|
POIDocument doc = new HSSFWorkbook();
|
|
|
|
// New document won't have them
|
|
assertNull(doc.getSummaryInformation());
|
|
assertNull(doc.getDocumentSummaryInformation());
|
|
|
|
// Add them in
|
|
doc.createInformationProperties();
|
|
assertNotNull(doc.getSummaryInformation());
|
|
assertNotNull(doc.getDocumentSummaryInformation());
|
|
|
|
// Write out and back in again, no change
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
doc.write(baos);
|
|
ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
|
|
|
|
doc = new HSSFWorkbook(bais);
|
|
|
|
assertNotNull(doc.getSummaryInformation());
|
|
assertNotNull(doc.getDocumentSummaryInformation());
|
|
}
|
|
|
|
public void testCreateNewPropertiesOnExistingFile() throws IOException {
|
|
POIDocument doc = new HSSFWorkbook();
|
|
|
|
// New document won't have them
|
|
assertNull(doc.getSummaryInformation());
|
|
assertNull(doc.getDocumentSummaryInformation());
|
|
|
|
// Write out and back in again, no change
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
doc.write(baos);
|
|
ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
|
|
doc = new HSSFWorkbook(bais);
|
|
|
|
assertNull(doc.getSummaryInformation());
|
|
assertNull(doc.getDocumentSummaryInformation());
|
|
|
|
// Create, and change
|
|
doc.createInformationProperties();
|
|
doc.getSummaryInformation().setAuthor("POI Testing");
|
|
doc.getDocumentSummaryInformation().setCompany("ASF");
|
|
|
|
// Save and re-load
|
|
baos = new ByteArrayOutputStream();
|
|
doc.write(baos);
|
|
bais = new ByteArrayInputStream(baos.toByteArray());
|
|
doc = new HSSFWorkbook(bais);
|
|
|
|
// Check
|
|
assertNotNull(doc.getSummaryInformation());
|
|
assertNotNull(doc.getDocumentSummaryInformation());
|
|
assertEquals("POI Testing", doc.getSummaryInformation().getAuthor());
|
|
assertEquals("ASF", doc.getDocumentSummaryInformation().getCompany());
|
|
|
|
// Asking to re-create will make no difference now
|
|
doc.createInformationProperties();
|
|
assertNotNull(doc.getSummaryInformation());
|
|
assertNotNull(doc.getDocumentSummaryInformation());
|
|
assertEquals("POI Testing", doc.getSummaryInformation().getAuthor());
|
|
assertEquals("ASF", doc.getDocumentSummaryInformation().getCompany());
|
|
}
|
|
}
|