Example #1
0
  /**
   * Tests CSVWriter, also tests a little CSVReader, as it reads back and validates the data, it has
   * written
   *
   * @throws ParseException when creating fake athlete something goes wrong with date, then throws
   *     it.
   */
  @Test
  public void writeDataToCSVTest() throws ParseException {
    ArrayList<Athlete> fakeAthletes = createFakeAthleteList();

    DataToCSVWriter csvWriter = new DataToCSVWriter();

    String testFile = "csvTest.csv";

    csvWriter.writeDataToCSV(fakeAthletes, testFile);

    assertTrue(new File(testFile).exists());

    CSVReader csvReader = new CSVReader();
    ArrayList<String> data = csvReader.readInData(testFile);
    String[] line1 = null;
    String[] line2 = null;

    for (String s : data) {
      if (line1 == null) line1 = s.split(",");
      else line2 = s.split(",");
    }

    assert line1 != null && line2 != null;
    assertTrue(line1[2].contains("Taavi"));
    assertTrue(line2[2].contains("James"));
  }