/*
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.odftoolkit.simple;
import java.awt.Rectangle;
import java.io.File;
import java.util.logging.Level;
import java.util.logging.Logger;
import junit.framework.Assert;
import org.junit.Test;
import org.odftoolkit.simple.Document.OdfMediaType;
import org.odftoolkit.simple.chart.Chart;
import org.odftoolkit.simple.chart.ChartType;
import org.odftoolkit.simple.chart.DataSet;
import org.odftoolkit.simple.table.Column;
import org.odftoolkit.simple.table.Table;
import org.odftoolkit.simple.utils.ResourceUtilities;
public class SpreadsheetDocumentTest {
private static final Logger LOG = Logger.getLogger(SpreadsheetDocumentTest.class.getName());
private static final String TEST_FILE = "spreadsheetTestTemplate.ots";
@Test
public void testGetMediaTypeString() throws Exception{
try {
String spreadDocPath = ResourceUtilities.getAbsolutePath(TEST_FILE);
SpreadsheetDocument spreadDoc = SpreadsheetDocument.loadDocument(spreadDocPath);
Assert.assertNotNull(spreadDoc);
OdfMediaType odfMediaType = spreadDoc.getOdfMediaType();
Assert.assertEquals(Document.OdfMediaType.SPREADSHEET_TEMPLATE, odfMediaType);
} catch (Exception e) {
LOG.log(Level.SEVERE, e.getMessage(), e);
Assert.fail(e.getMessage());
}
}
@Test
public void testGetSuffix() throws Exception{
try {
String spreadDocPath = ResourceUtilities.getAbsolutePath(TEST_FILE);
SpreadsheetDocument spreadDoc = SpreadsheetDocument.loadDocument(spreadDocPath);
Assert.assertNotNull(spreadDoc);
OdfMediaType odfMediaType = spreadDoc.getOdfMediaType();
String suffix = odfMediaType.getSuffix();
Assert.assertEquals("ots", suffix);
} catch (Exception e) {
LOG.log(Level.SEVERE, e.getMessage(), e);
Assert.fail(e.getMessage());
}
}
@Test
public void testGetOdfMediaType() throws Exception{
try {
String spreadDocPath = ResourceUtilities.getAbsolutePath(TEST_FILE);
SpreadsheetDocument spreadDoc = SpreadsheetDocument.loadDocument(spreadDocPath);
Assert.assertNotNull(spreadDoc);
Document.OdfMediaType odfMediaType = SpreadsheetDocument.OdfMediaType.getOdfMediaType(spreadDoc.getMediaTypeString());
Assert.assertEquals("ots", odfMediaType.getSuffix());
} catch (Exception e) {
LOG.log(Level.SEVERE, e.getMessage(), e);
Assert.fail(e.getMessage());
}
}
@Test
public void testLoadDocument() throws Exception{
try {
String filePath = ResourceUtilities.getAbsolutePath(TEST_FILE);
SpreadsheetDocument spreadDoc = SpreadsheetDocument.loadDocument(filePath);
Assert.assertNotNull(spreadDoc);
Assert.assertEquals(Document.OdfMediaType.SPREADSHEET_TEMPLATE, spreadDoc.getOdfMediaType());
} catch (Exception e) {
LOG.log(Level.SEVERE, e.getMessage(), e);
Assert.fail(e.getMessage());
}
}
@Test
public void testLoadDocumentFile() throws Exception{
try {
String filePath = ResourceUtilities.getAbsolutePath(TEST_FILE);
File fileDoc = new File(filePath);
SpreadsheetDocument spreadDoc = SpreadsheetDocument.loadDocument(fileDoc);;
Assert.assertNotNull(spreadDoc);
Assert.assertEquals(Document.OdfMediaType.SPREADSHEET_TEMPLATE, spreadDoc.getOdfMediaType());
} catch (Exception e) {
LOG.log(Level.SEVERE, e.getMessage(), e);
Assert.fail(e.getMessage());
}
}
@Test
public void testCreateChart() throws Exception{
try {
SpreadsheetDocument spDocument = SpreadsheetDocument.loadDocument(ResourceUtilities.getAbsolutePath(TEST_FILE));
String title = "XXXTitle";
String[] lables = new String[]{"spring","summer","autumn","autumn"};
String[] legends = new String[]{"hello1","hello2","hello3"};
double[][] data = new double[][]{{1.2,2.22,3},{2,3,4},{3,4,5},{4,5,6}};
DataSet dataset = new DataSet(lables, legends, data);
Rectangle rect = new Rectangle();
rect.x = 367;
rect.y = 389;
rect.width = 379;
rect.height = 424;
Chart spChart = spDocument.createChart(title, dataset, rect);
Assert.assertNotNull(spChart);
spChart.setChartType(ChartType.AREA);
//save
spDocument.save(ResourceUtilities.getTestOutput("Chart_"+TEST_FILE));
Assert.assertEquals(dataset, spChart.getChartData());
Assert.assertEquals("XXXTitle", spChart.getChartTitle());
Assert.assertEquals(ChartType.AREA, spChart.getChartType());
LOG.log(Level.INFO,"spChart--> " + spChart);
} catch (Exception e) {
LOG.log(Level.SEVERE, e.getMessage(), e);
Assert.fail(e.getMessage());
}
}
@Test
public void testGetSheetByIndex() throws Exception{
File file = new File(ResourceUtilities.getAbsolutePath("TestSpreadsheetTable.ods"));
SpreadsheetDocument spDocument = SpreadsheetDocument.loadDocument(file);
//index < 0 , Not expected, table ==null
Table tablenull = spDocument.getSheetByIndex(-1);
Assert.assertTrue((tablenull == null));
//index > 0
//index = 0
Table tableSheet0 = spDocument.getSheetByIndex(0);
Assert.assertTrue((tableSheet0 != null));
Assert.assertEquals("Sheet1", tableSheet0.getTableName());
Assert.assertEquals(29, tableSheet0.getColumnCount());
//index = 1
Table tableSheet1 = spDocument.getSheetByIndex(1);
Assert.assertTrue((tableSheet1 != null));
Assert.assertEquals("Sheet2", tableSheet1.getTableName());
Assert.assertEquals(1, tableSheet1.getColumnCount());
}
@Test
public void testInsertSheet() throws Exception{
File file = new File(ResourceUtilities.getAbsolutePath(TEST_FILE));
SpreadsheetDocument spDocument = SpreadsheetDocument.loadDocument(file);
//index <0 , Not expected
Table table = spDocument.insertSheet(-1);
Assert.assertNull(table);
//index >= sheet count
Table tableb = spDocument.insertSheet(11);
Assert.assertNull(tableb);
//index is within the law
Table tab = spDocument.getSheetByName("tabellDemo2");
if(tab != null){
for(int i=0;i<spDocument.getSheetCount();i++){
if(tab.equals(spDocument.getSheetByIndex(i)))
spDocument.removeSheet(i);
}
}
Table tablea = spDocument.insertSheet(0);
Column col = tablea.appendColumn();
col.setWidth(12.99);
Column col2 = tablea.appendColumn();
col.setWidth(12.);
tablea.setTableName("tabellDemo2");
Assert.assertEquals("tabellDemo2", tablea.getTableName());
spDocument.save(ResourceUtilities.getAbsolutePath(TEST_FILE));
}
@Test
public void testRemoveSheet() throws Exception{
File file = new File(ResourceUtilities.getAbsolutePath(TEST_FILE));
SpreadsheetDocument spDocument = SpreadsheetDocument.loadDocument(file);
//index <0 , Not expected
Table table = spDocument.insertSheet(-1);
Assert.assertNull(table);
//index >= sheet count
Table tableb = spDocument.insertSheet(11);
Assert.assertNull(tableb);
//index is within the law
Table tab = spDocument.getSheetByName("tabellDemo2");
if(tab != null){
for(int i=0;i<spDocument.getSheetCount();i++){
if(tab.equals(spDocument.getSheetByIndex(i)))
spDocument.removeSheet(i);
}
}
Table tablea = spDocument.insertSheet(0);
Column col = tablea.appendColumn();
col.setWidth(12.99);
Column col2 = tablea.appendColumn();
col.setWidth(12.);
tablea.setTableName("tabellDemo2");
Assert.assertEquals("tabellDemo2", tablea.getTableName());
spDocument.removeSheet(0);
Table tablem = spDocument.getSheetByIndex(0);
Assert.assertNotSame(tablea, tablem);
spDocument.save(ResourceUtilities.getAbsolutePath(TEST_FILE));
}
@Test
public void testGetSheetByIndex2() throws Exception{
File file = new File(ResourceUtilities.getAbsolutePath(TEST_FILE));
SpreadsheetDocument spDocument = SpreadsheetDocument.loadDocument(file);
//index <0 , Not expected
Table table = spDocument.insertSheet(-1);
Assert.assertNull(table);
//index >= sheet count
Table tableb = spDocument.insertSheet(11);
Assert.assertNull(tableb);
//index is within the law
Table tab = spDocument.getSheetByName("tabellDemo1");
if(tab != null)
tab.remove();
Table taba = spDocument.getSheetByName("Tabelle1");
Table tablea = spDocument.insertSheet(taba,0);
tablea.setTableName("tabellDemo1");
Assert.assertEquals("tabellDemo1", tablea.getTableName());
Table tablem = spDocument.getSheetByIndex(0);
Assert.assertEquals(tablea, tablem);
spDocument.save(ResourceUtilities.getAbsolutePath(TEST_FILE));
}
}