コード例 #1
0
  @Test
  public void test() throws ParseException {
    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");

    String routeInfo = "1";
    String date = "2015-12-01";
    String time = "00:00";
    Route r = new Route();
    r.setRouteId(1);
    r.setRouteName("route");
    Direction d = new Direction();
    d.setDirectionId(2);
    Shedule s = new Shedule();
    s.setDirection(d);
    s.setStep(0);
    List<Shedule> steps = new ArrayList<Shedule>();
    steps.add(s);
    Train train = new Train();
    train.setTrainId(3);
    List<Train> trains = new ArrayList<Train>();
    trains.add(train);
    Journey j = new Journey(4, r, train, sdf.parse(date));

    Mockito.when(dao.getRoute(Integer.parseInt(routeInfo))).thenReturn(r);
    Mockito.when(dao.getShedulesOfRoute(r.getRouteId())).thenReturn(steps);
    Mockito.when(dao.getAllTrains()).thenReturn(trains);
    Mockito.when(dao.getAllJourneysOfTrain(train.getTrainId()))
        .thenReturn(new ArrayList<Journey>());
    Mockito.when(dao.createJourney(train, r, sdf.parse(date))).thenReturn(j);

    NewJourneyInfo info1 = new NewJourneyInfo(routeInfo, date, time, null, null, null, false);
    NewJourneyInfo info2 =
        new NewJourneyInfo(
            routeInfo,
            date,
            time,
            String.valueOf(j.getJourneyId()),
            r.getRouteName(),
            String.valueOf(train.getTrainId()),
            false);
    NewJourneyInfo info3 = planner.plan(info1);

    Assert.assertTrue(info3.getJourneyId().equals(info2.getJourneyId()));
    Assert.assertTrue(info3.getRouteInfo().equals(info2.getRouteInfo()));
    Assert.assertTrue(info3.getRouteName().equals(info2.getRouteName()));
    Assert.assertTrue(info3.getTime().equals(info2.getTime()));
    Assert.assertTrue(info3.getTrain().equals(info2.getTrain()));
  }
コード例 #2
0
  @Test
  public void test8() throws ParseException {

    String routeInfo = "1";
    String date = "2015-12-01";
    String time = "00:00";
    Route r = new Route();
    r.setRouteId(1);
    r.setRouteName("route");
    Direction d = new Direction();
    d.setDirectionId(2);
    Shedule s = new Shedule();
    s.setDirection(d);
    s.setStep(0);
    List<Shedule> steps = new ArrayList<Shedule>();
    steps.add(s);

    List<Train> trains = new ArrayList<Train>();

    Mockito.when(dao.getRoute(Integer.parseInt(routeInfo))).thenReturn(r);
    Mockito.when(dao.getShedulesOfRoute(r.getRouteId())).thenReturn(steps);
    Mockito.when(dao.getAllTrains()).thenReturn(trains);

    NewJourneyInfo info1 = new NewJourneyInfo(routeInfo, date, time, null, null, null, false);
    NewJourneyInfo info3 = planner.plan(info1);

    Assert.assertTrue(info3.getJourneyId() == null);
    Assert.assertTrue(info3.getRouteName() == null);
    Assert.assertTrue(info3.getTrain() == null);
    Assert.assertTrue(info3.isTrainsLack());
  }