Пример #1
0
  @Test
  public void testZipFileWrite() throws Exception {

    String outputFileName = null;

    try {
      // Get somewhere temporary to write out to
      outputFileName = File.createTempFile("ItemWriterTest-", ".csv.zip").getAbsolutePath();

      // Configure the ItemWriter
      C24ItemWriter itemWriter = new C24ItemWriter();
      itemWriter.setSink(new TextualSink());
      itemWriter.setWriterSource(new ZipFileWriterSource());
      itemWriter.setup(getStepExecution(outputFileName));
      // Write the employees out
      itemWriter.write(employees);
      // Close the file
      itemWriter.cleanup();

      // Check that we wrote out what was expected
      ZipFile zipFile = new ZipFile(outputFileName);
      Enumeration<? extends ZipEntry> entries = zipFile.entries();
      assertNotNull(entries);
      // Make sure there's at least one entry
      assertTrue(entries.hasMoreElements());
      ZipEntry entry = entries.nextElement();
      // Make sure that the trailing .zip has been removed and the leading path has been removed
      assertFalse(entry.getName().contains(System.getProperty("file.separator")));
      assertFalse(entry.getName().endsWith(".zip"));
      // Make sure that there aren't any other entries
      assertFalse(entries.hasMoreElements());

      try {
        compareCsv(zipFile.getInputStream(entry), employees);
      } finally {
        if (zipFile != null) {
          zipFile.close();
        }
      }

    } finally {
      if (outputFileName != null) {
        // Clear up our temporary file
        File file = new File(outputFileName);
        file.delete();
      }
    }
  }
Пример #2
0
  @Test
  public void testFileWrite() throws Exception {

    String outputFileName = null;

    try {
      // Get somewhere temporary to write out to
      outputFileName = File.createTempFile("ItemWriterTest-", ".csv").getAbsolutePath();

      // Configure the ItemWriter
      C24ItemWriter itemWriter = new C24ItemWriter();
      itemWriter.setSink(new TextualSink());
      itemWriter.setWriterSource(new FileWriterSource());
      itemWriter.setup(getStepExecution(outputFileName));
      // Write the employees out
      itemWriter.write(employees);
      // Close the file
      itemWriter.cleanup();

      // Check that we wrote out what was expected
      FileInputStream inputStream = new FileInputStream(outputFileName);
      try {
        compareCsv(inputStream, employees);
      } finally {
        if (inputStream != null) {
          inputStream.close();
        }
      }

    } finally {
      if (outputFileName != null) {
        // Clear up our temporary file
        File file = new File(outputFileName);
        file.delete();
      }
    }
  }