Create JUnit test cases

class Flight {

private String flightNo;

private Calendar departureTime;

private Calendar arrivalTime;

public Flight(String flightNo, Calendar departureTime,

Calendar arrivalTime) {

this.flightNo = flightNo;

this.departureTime = departureTime;

this.arrivalTime = arrivalTime;

}

public int getFlightTime() {

return (int)(arrivalTime.getTimeInMillis() - departureTime.getTimeInMillis())

/ (1000 * 60);

}

public Calendar getDepartureTime() {

return departureTime;

}

public Calendar getArrivalTime() {

return arrivalTime;

}

public void setArrivalTime(Calendar arrivalTime) {

this.arrivalTime = arrivalTime;

}

public void setDepartureTime(Calendar departureTime) {

this.departureTime = departureTime;

}

}

class Itinerary {

private List flights;

public Itinerary(List flights) {

this.flights = flights;

}

public int getTotalTravelTime() {

int totalTime = getTotalFlightTime();

for (int i = 0; i < flights.size() - 1; i++) {

long time = flights.get(i + 1).getDepartureTime().getTimeInMillis() -

flights.get(i).getArrivalTime().getTimeInMillis();

totalTime += (int)time / (1000 * 60);

}

return totalTime;

}

public int getTotalFlightTime() {

int flightTime = 0;

for (Flight flight: flights)

flightTime += flight.getFlightTime();

return flightTime;

}

}