Spring/spring-batch
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

IN PROGRESS - BATCH-826: Create callback for header and footer reading in xml and flat files

Browse Source 
changed header and footer lines to callbacks
This commit is contained in:
robokaso
2008-09-11 08:11:27 +00:00
parent e80fdae0aa
commit 4b5cbef820
3 changed files with 75 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/FlatFileItemWriterTests.java
View File

@@ -25,6 +25,7 @@ import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.Writer;
import java.nio.charset.UnsupportedCharsetException;
import java.util.Arrays;
import java.util.Collections;
@@ -203,7 +204,13 @@ public class FlatFileItemWriterTests {
@Test
public void testRestart() throws Exception {
writer.setFooterLines(new String[] { "footer" });
writer.setFooterCallback(new FileWriterCallback() {
public void write(Writer writer) throws IOException {
writer.write("footer");
}
});
writer.open(executionContext);
// write some lines
@@ -309,7 +316,13 @@ public class FlatFileItemWriterTests {
@Test
public void testWriteFooter() throws Exception {
writer.setFooterLines(new String[] { "a", "b" });
writer.setFooterCallback(new FileWriterCallback() {
public void write(Writer writer) throws IOException {
writer.write("a\nb");
}
});
writer.open(executionContext);
writer.write(Collections.singletonList(TEST_STRING));
writer.close(executionContext);
@@ -320,7 +333,13 @@ public class FlatFileItemWriterTests {
@Test
public void testWriteHeader() throws Exception {
writer.setHeaderLines(new String[] { "a", "b" });
writer.setHeaderCallback(new FileWriterCallback() {
public void write(Writer writer) throws IOException {
writer.write("a\nb");
}
});
writer.open(executionContext);
writer.write(Collections.singletonList(TEST_STRING));
writer.close(null);
@@ -334,7 +353,13 @@ public class FlatFileItemWriterTests {
@Test
public void testWriteHeaderAfterRestartOnFirstChunk() throws Exception {
writer.setHeaderLines(new String[] { "a", "b" });
writer.setHeaderCallback(new FileWriterCallback() {
public void write(Writer writer) throws IOException {
writer.write("a\nb");
}
});
writer.open(executionContext);
writer.write(Collections.singletonList(TEST_STRING));
writer.close(executionContext);
@@ -353,7 +378,13 @@ public class FlatFileItemWriterTests {
@Test
public void testWriteHeaderAfterRestartOnSecondChunk() throws Exception {
writer.setHeaderLines(new String[] { "a", "b" });
writer.setHeaderCallback(new FileWriterCallback() {
public void write(Writer writer) throws IOException {
writer.write("a\nb");
}
});
writer.open(executionContext);
writer.write(Collections.singletonList(TEST_STRING));
writer.update(executionContext);