|
|
@@ -0,0 +1,218 @@ |
|
|
|
/* ==================================================================== |
|
|
|
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.xssf.usermodel; |
|
|
|
|
|
|
|
import org.apache.poi.ss.usermodel.CellType; |
|
|
|
import org.apache.poi.ss.util.CellAddress; |
|
|
|
import org.apache.poi.ss.util.CellRangeAddress; |
|
|
|
import org.junit.After; |
|
|
|
import org.junit.Before; |
|
|
|
import org.junit.Ignore; |
|
|
|
import org.junit.Test; |
|
|
|
import org.openxmlformats.schemas.spreadsheetml.x2006.main.CTCell; |
|
|
|
|
|
|
|
import java.io.File; |
|
|
|
import java.io.FileOutputStream; |
|
|
|
import java.io.IOException; |
|
|
|
import java.io.OutputStream; |
|
|
|
import java.util.Locale; |
|
|
|
|
|
|
|
import static org.junit.Assert.assertEquals; |
|
|
|
|
|
|
|
public class TestXSSFSheetShiftRowsAndColumns { |
|
|
|
final File resultDir=new File("build/custom-reports-test"); |
|
|
|
XSSFWorkbook workbook=null; |
|
|
|
XSSFSheet sheet=null; |
|
|
|
String fileName=null; |
|
|
|
final int numRows=4; |
|
|
|
final int numCols=4; |
|
|
|
final int INSERT_ROW=1; |
|
|
|
final int INSERT_COLUMN=1; |
|
|
|
final int FIRST_MERGE_ROW=INSERT_ROW+1; |
|
|
|
final int LAST_MERGE_ROW=numRows-1; |
|
|
|
final int FIRST_MERGE_COL=INSERT_COLUMN+1; |
|
|
|
final int LAST_MERGE_COL=numCols-1; |
|
|
|
|
|
|
|
public TestXSSFSheetShiftRowsAndColumns() { |
|
|
|
resultDir.mkdirs(); |
|
|
|
} |
|
|
|
|
|
|
|
/** |
|
|
|
* This creates a workbook with one worksheet. It then puts data in rows 0 to numRows-1 and colulmns |
|
|
|
* 0 to numCols-1. |
|
|
|
*/ |
|
|
|
@Before |
|
|
|
public void setup() { |
|
|
|
final String procName="TestXSSFSheetShiftRowsAndColumns.setup"; |
|
|
|
workbook = new XSSFWorkbook(); |
|
|
|
sheet = workbook.createSheet(); |
|
|
|
|
|
|
|
for (int nRow = 0; nRow < numRows; ++nRow) { |
|
|
|
final XSSFRow row = sheet.createRow(nRow); |
|
|
|
for (int nCol = 0; nCol < numCols; ++nCol) { |
|
|
|
final XSSFCell cell = row.createCell(nCol); |
|
|
|
cell.setCellType(CellType.STRING); |
|
|
|
cell.setCellValue(String.format("Row %d col %d", nRow, nCol)); |
|
|
|
} |
|
|
|
} |
|
|
|
/* |
|
|
|
* Add a merge area |
|
|
|
*/ |
|
|
|
final CellRangeAddress range=new CellRangeAddress(FIRST_MERGE_ROW,LAST_MERGE_ROW,FIRST_MERGE_COL,LAST_MERGE_COL); |
|
|
|
sheet.addMergedRegion(range); |
|
|
|
System.out.println(String.format("\n%s: mergeArea=%s", procName,range)); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
* This method writes the workbook to resultDir/fileName. |
|
|
|
*/ |
|
|
|
@After |
|
|
|
public void cleanup() { |
|
|
|
final String procName="TestXSSFSheetRemoveTable.cleanup"; |
|
|
|
if (workbook == null) { |
|
|
|
System.out.println(String.format(Locale.ROOT,"%s: workbook==null",procName)); |
|
|
|
return; |
|
|
|
} |
|
|
|
|
|
|
|
if(fileName==null) { |
|
|
|
System.out.println(String.format(Locale.ROOT, "%s: fileName==null",procName)); |
|
|
|
return; |
|
|
|
} |
|
|
|
|
|
|
|
final File file=new File(resultDir,fileName); |
|
|
|
try (OutputStream fileOut = new FileOutputStream(file)) { |
|
|
|
workbook.write(fileOut); |
|
|
|
System.out.println(String.format(Locale.ROOT, "%s: test file written to %s",procName,file.getAbsolutePath())); |
|
|
|
} catch (Exception e) { |
|
|
|
System.err.println(e.getMessage()); |
|
|
|
} finally { |
|
|
|
try { |
|
|
|
workbook.close(); |
|
|
|
} catch (IOException e) { |
|
|
|
// TODO Auto-generated catch block |
|
|
|
e.printStackTrace(); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
/** |
|
|
|
* Apply no shift. The purpose of this is to test {@code testCellAddresses} and {@code testMergeRegion}. |
|
|
|
*/ |
|
|
|
@Test |
|
|
|
public void testNoShift() { |
|
|
|
final String procName="testNoShift"; |
|
|
|
fileName=procName+".xlsx"; |
|
|
|
|
|
|
|
testCellAddresses(procName,0,0); |
|
|
|
testMergeRegion(procName,0,0); |
|
|
|
|
|
|
|
System.out.println(String.format("%s: finished without error", procName)); |
|
|
|
} |
|
|
|
|
|
|
|
@Test |
|
|
|
public void testShiftOneRowAndTestAddresses() { |
|
|
|
final String procName="testShiftOneRowAndTestAddresses"; |
|
|
|
fileName=procName+".xlsx"; |
|
|
|
final int nRowsToShift=1; |
|
|
|
|
|
|
|
sheet.shiftRows(INSERT_ROW, numRows-1, nRowsToShift); |
|
|
|
testCellAddresses(procName,nRowsToShift,0); |
|
|
|
System.out.println(String.format("%s: finished without error", procName)); |
|
|
|
} |
|
|
|
|
|
|
|
@Ignore("currently fails") |
|
|
|
@Test |
|
|
|
public void testShiftOneRowAndTestMergeRegion() { |
|
|
|
final String procName="testShiftOneRowAndTestMergeRegion"; |
|
|
|
fileName=procName+".xlsx"; |
|
|
|
final int nRowsToShift=1; |
|
|
|
|
|
|
|
sheet.shiftRows(INSERT_ROW, numRows-1, nRowsToShift); |
|
|
|
testMergeRegion(procName,nRowsToShift,0); |
|
|
|
System.out.println(String.format("%s: finished without error", procName)); |
|
|
|
} |
|
|
|
|
|
|
|
@Test |
|
|
|
public void testShiftOneColumnAndTestAddresses() { |
|
|
|
final String procName="testShiftOneColumnAndTestAddresses"; |
|
|
|
fileName=procName+".xlsx"; |
|
|
|
final int nShift=1; |
|
|
|
|
|
|
|
sheet.shiftColumns(INSERT_COLUMN, numCols-1, nShift); |
|
|
|
testCellAddresses(procName,0,nShift); |
|
|
|
System.out.println(String.format("%s: finished without error", procName)); |
|
|
|
} |
|
|
|
|
|
|
|
@Ignore("currently fails") |
|
|
|
@Test |
|
|
|
public void testShiftOneColumnAndTestMergeRegion() { |
|
|
|
final String procName="testShiftOneColumnAndTestMergeRegion"; |
|
|
|
fileName=procName+".xlsx"; |
|
|
|
final int nShift=1; |
|
|
|
|
|
|
|
sheet.shiftColumns(INSERT_COLUMN, numCols-1, nShift); |
|
|
|
testMergeRegion(procName,0,nShift); |
|
|
|
System.out.println(String.format("%s: finished without error", procName)); |
|
|
|
} |
|
|
|
|
|
|
|
/** |
|
|
|
* Verify that the cell addresses are consistent |
|
|
|
*/ |
|
|
|
private void testCellAddresses(String procName,int nRowsToShift,int nColsToShift) { |
|
|
|
final int nNumRows=nRowsToShift+this.numCols; |
|
|
|
final int nNumCols=nColsToShift+this.numCols; |
|
|
|
for(int nRow=0;nRow<nNumRows;++nRow) { |
|
|
|
final XSSFRow row=sheet.getRow(nRow); |
|
|
|
if(row==null) { |
|
|
|
System.out.println(String.format("%s: Row %d is empty", procName,nRow)); |
|
|
|
continue; |
|
|
|
} |
|
|
|
for(int nCol=0;nCol<nNumCols;++nCol) { |
|
|
|
final String address=new CellAddress(nRow,nCol).formatAsString(); |
|
|
|
final XSSFCell cell=row.getCell(nCol); |
|
|
|
if(cell==null) { |
|
|
|
System.out.println(String.format("%s: Cell %s is empty", procName,address)); |
|
|
|
continue; |
|
|
|
} |
|
|
|
final CTCell ctCell=cell.getCTCell(); |
|
|
|
final Object cellAddress=cell.getAddress().formatAsString(); |
|
|
|
final Object r=ctCell.getR(); |
|
|
|
|
|
|
|
if(nCol==0 || nRow==0) { |
|
|
|
System.out.println(String.format("%s: Row %d col %d address=%s cell.address=%s cell.getR=%s", procName, nRow, |
|
|
|
nCol, address, cellAddress, ctCell.getR())); |
|
|
|
} |
|
|
|
assertEquals(String.format("%s: Testing cell.getAddress",procName),address,cellAddress); |
|
|
|
assertEquals(String.format("%s: Testing ctCell.getR",procName),address,r); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
/** |
|
|
|
* Verify that the merge area is consistent |
|
|
|
*/ |
|
|
|
private void testMergeRegion(String procName,int nRowsToShift,int nColsToShift) { |
|
|
|
final CellRangeAddress range=sheet.getMergedRegion(0); |
|
|
|
assertEquals(String.format("%s: Testing merge area %s",procName,range),range, |
|
|
|
new CellRangeAddress(FIRST_MERGE_ROW,LAST_MERGE_ROW+nRowsToShift,FIRST_MERGE_COL,LAST_MERGE_COL+nColsToShift)); |
|
|
|
} |
|
|
|
|
|
|
|
} |