@Test
public void readAirlineCompanyCSVFileWithOneRecordTest() throws IOException, ParseException {
Files.deleteIfExists(Paths.get(projectPath + CSVFileTest.AIRLINECOMPANYCSV.value()));
AirlineCompany airlineCompany = new AirlineCompany();
airlineCompany.setAirline("Cathay Pacific Airways");
ArrayList<AirlineCompany> airlineCompanies = new ArrayList<AirlineCompany>();
airlineCompanies.add(airlineCompany);
SourceWriter<List<AirlineCompany>> airlineCompanyCSVFileWriter =
new AirlineCompanyCSVFileWriter(projectPath + CSVFileTest.AIRLINECOMPANYCSV.value());
airlineCompanyCSVFileWriter.write(airlineCompanies);
airlineCompanyCSVFileWriter.close();
SourceReader<AirlineCompany> airlineCompanyCSVFileReader =
new AirlineCompanyCSVFileReader(projectPath + CSVFileTest.AIRLINECOMPANYCSV.value());
List<AirlineCompany> airlineCompanyResultList =
airlineCompanyCSVFileReader.read(new AirlineCompanyParser());
airlineCompanyCSVFileReader.close();
assertThat(airlineCompany.getAirline(), is(airlineCompanyResultList.get(0).getAirline()));
assertThat(
airlineCompany.getFlights().size(),
is(airlineCompanyResultList.get(0).getFlights().size()));
}
