180 lines
6.7 KiB
Java
180 lines
6.7 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.hssf.eventmodel;
|
|
|
|
import java.io.ByteArrayInputStream;
|
|
|
|
import junit.framework.TestCase;
|
|
|
|
import org.apache.poi.hssf.record.BOFRecord;
|
|
import org.apache.poi.hssf.record.ContinueRecord;
|
|
import org.apache.poi.hssf.record.EOFRecord;
|
|
import org.apache.poi.hssf.record.Record;
|
|
import org.apache.poi.hssf.record.RecordFactory;
|
|
import org.apache.poi.hssf.record.TestcaseRecordInputStream;
|
|
import org.apache.poi.hssf.record.UnknownRecord;
|
|
|
|
/**
|
|
* enclosing_type describe the purpose here
|
|
*
|
|
* @author Andrew C. Oliver acoliver@apache.org
|
|
* @author Csaba Nagy (ncsaba at yahoo dot com)
|
|
*/
|
|
public final class TestEventRecordFactory extends TestCase {
|
|
|
|
/**
|
|
* tests that the records can be processed and properly return
|
|
* values.
|
|
*/
|
|
public void testProcessRecords()
|
|
{
|
|
final boolean[] wascalled = { false, }; // hack to pass boolean by ref into inner class
|
|
|
|
ERFListener listener = new ERFListener() {
|
|
public boolean processRecord(Record rec) {
|
|
wascalled[0] = true;
|
|
assertTrue("must be BOFRecord got SID="+rec.getSid(),
|
|
(rec.getSid() == BOFRecord.sid));
|
|
return true;
|
|
}
|
|
};
|
|
EventRecordFactory factory = new EventRecordFactory(listener, new short[] {BOFRecord.sid});
|
|
|
|
BOFRecord bof = new BOFRecord();
|
|
bof.setBuild((short)0);
|
|
bof.setBuildYear((short)1999);
|
|
bof.setRequiredVersion(123);
|
|
bof.setType(BOFRecord.TYPE_WORKBOOK);
|
|
bof.setVersion((short)0x06);
|
|
bof.setHistoryBitMask(BOFRecord.HISTORY_MASK);
|
|
|
|
EOFRecord eof = EOFRecord.instance;
|
|
byte[] bytes = new byte[bof.getRecordSize() + eof.getRecordSize()];
|
|
int offset = 0;
|
|
offset = bof.serialize(offset,bytes);
|
|
offset = eof.serialize(offset,bytes);
|
|
|
|
factory.processRecords(new ByteArrayInputStream(bytes));
|
|
assertTrue("The record listener must be called", wascalled[0]);
|
|
}
|
|
|
|
/**
|
|
* tests that the create record function returns a properly
|
|
* constructed record in the simple case.
|
|
*/
|
|
public void testCreateRecord() {
|
|
BOFRecord bof = new BOFRecord();
|
|
bof.setBuild((short)0);
|
|
bof.setBuildYear((short)1999);
|
|
bof.setRequiredVersion(123);
|
|
bof.setType(BOFRecord.TYPE_WORKBOOK);
|
|
bof.setVersion((short)0x06);
|
|
bof.setHistoryBitMask(BOFRecord.HISTORY_MASK);
|
|
|
|
byte[] bytes = bof.serialize();
|
|
|
|
Record[] records = RecordFactory.createRecord(TestcaseRecordInputStream.create(bytes));
|
|
|
|
assertTrue("record.length must be 1, was ="+records.length,records.length == 1);
|
|
assertTrue("record is the same", compareRec(bof,records[0]));
|
|
|
|
}
|
|
|
|
/**
|
|
* Compare the serialized bytes of two records are equal
|
|
* @param first the first record to compare
|
|
* @param second the second record to compare
|
|
* @return boolean whether or not the record where equal
|
|
*/
|
|
private static boolean compareRec(Record first, Record second) {
|
|
byte[] rec1 = first.serialize();
|
|
byte[] rec2 = second.serialize();
|
|
|
|
if (rec1.length != rec2.length) {
|
|
return false;
|
|
}
|
|
for (int k=0; k<rec1.length; k++) {
|
|
if (rec1[k] != rec2[k]) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
/**
|
|
* tests that the create record function returns a properly
|
|
* constructed record in the case of a continued record.
|
|
* TODO - need a real world example to put in a unit test
|
|
*/
|
|
public void testCreateContinuedRecord()
|
|
{
|
|
// fail("not implemented");
|
|
}
|
|
|
|
|
|
/**
|
|
* TEST NAME: Test Creating ContinueRecords After Unknown Records From An InputStream <P>
|
|
* OBJECTIVE: Test that the RecordFactory given an InputStream
|
|
* constructs the expected records.<P>
|
|
* SUCCESS: Record factory creates the expected records.<P>
|
|
* FAILURE: The wrong records are created or contain the wrong values <P>
|
|
*
|
|
*/
|
|
public void testContinuedUnknownRecord() {
|
|
final byte[] data = {
|
|
0, -1, 0, 0, // an unknown record with 0 length
|
|
0x3C , 0, 3, 0, 1, 2, 3, // a continuation record with 3 bytes of data
|
|
0x3C , 0, 1, 0, 4 // one more continuation record with 1 byte of data
|
|
};
|
|
|
|
final int[] recCnt = { 0 };
|
|
final int[] offset = { 0 };
|
|
ERFListener listener = new ERFListener() {
|
|
private String[] expectedRecordTypes = {
|
|
UnknownRecord.class.getName(),
|
|
ContinueRecord.class.getName(),
|
|
ContinueRecord.class.getName()
|
|
};
|
|
public boolean processRecord(Record rec)
|
|
{
|
|
// System.out.println(rec.toString());
|
|
assertEquals(
|
|
"Record type",
|
|
expectedRecordTypes[recCnt[0]],
|
|
rec.getClass().getName()
|
|
);
|
|
compareData(rec, "Record " + recCnt[0] + ": ");
|
|
recCnt[0]++;
|
|
return true;
|
|
}
|
|
private void compareData(Record record, String message) {
|
|
byte[] recData = record.serialize();
|
|
for (int i = 0; i < recData.length; i++) {
|
|
assertEquals(message + " data byte " + i, data[offset[0]++], recData[i]);
|
|
}
|
|
}
|
|
};
|
|
EventRecordFactory factory = new EventRecordFactory(listener, new short[] {-256, 0x3C});
|
|
|
|
factory.processRecords(new ByteArrayInputStream(data));
|
|
assertEquals("nr. of processed records", 3, recCnt[0]);
|
|
assertEquals("nr. of processed bytes", data.length, offset[0]);
|
|
}
|
|
}
|