Browse Source

Modify the data a bit to have a better test

feature-nr-renew
LO Kam Tao Leo 3 years ago
parent
commit
6fed025104
  1. 6
      src/main/java/org/leolo/nrdatad/model/TrainSchedule.java
  2. 29
      src/test/java/org/leolo/nrdatad/model/TrainScheduleTest.java
  3. 4
      src/test/resources/org/leolo/nrdatad/test/V56331_N.json

6
src/main/java/org/leolo/nrdatad/model/TrainSchedule.java

@ -2,6 +2,7 @@ package org.leolo.nrdatad.model;
import org.json.JSONObject;
import org.leolo.nrdatad.util.DateUtil;
import org.leolo.nrdatad.util.TUIDDateFormat;
import java.util.Date;
@ -23,6 +24,11 @@ public class TrainSchedule {
private Date endDate;
private TrainScheduleSector sector;
public String getSUID(){
TUIDDateFormat tuiddf = new TUIDDateFormat();
return trainUid+tuiddf.format(startDate)+tuiddf.format(endDate)+shortTermPlanningIndicator.getCode();
}
public String getRunsOnBankHoliday() {
return runsOnBankHoliday;
}

29
src/test/java/org/leolo/nrdatad/model/TrainScheduleTest.java

@ -6,14 +6,41 @@ import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import static org.junit.Assert.*;
public class TrainScheduleTest {
@Test public void testSchedule1() throws IOException{
@Test public void testSchedule1() throws IOException, ParseException {
String json = TestUtil.openResourceFileAsString("org/leolo/nrdatad/test/V56331_N.json");
TrainSchedule ts = TrainSchedule.parseJSON(json);
SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
//Schedule Info
assertEquals("V56331", ts.getTrainUid());
assertEquals("V56331M3M4N", ts.getSUID());
assertEquals("AW", ts.getAtocCode());
assertEquals(ShortTermPlanningIndicator.NEW, ts.getShortTermPlanningIndicator());
assertEquals(sdf.parse("20230104"), ts.getStartDate());
assertEquals(sdf.parse("20230105"), ts.getEndDate());
assertEquals("0011000", ts.getRunsOn());
assertEquals("",ts.getRunsOnBankHoliday());
//Sector Info
TrainScheduleSector tss = ts.getSector();
assertEquals("", tss.getServiceBranding());
assertEquals("OO", tss.getTrainCategory());
assertEquals("??", tss.getBusinessSector());
assertEquals(75, tss.getSpeed());
assertEquals("", tss.getReservation());
assertEquals("", tss.getCatering());
assertEquals("DMU", tss.getPowerType());
assertEquals("S", tss.getTimingLoad());
assertEquals("2F60", tss.getSignalId());
assertEquals("", tss.getHeadcode());
assertEquals("", tss.getOperatingCharacteristic());
assertEquals("", tss.getSleeper());
assertEquals("25441000", tss.getTrainServiceCode());
assertEquals("S", tss.getTrainClass());
//TODO: finish the function
}

4
src/test/resources/org/leolo/nrdatad/test/V56331_N.json

@ -2,7 +2,7 @@
"CIF_bank_holiday_running": null,
"train_status": "1",
"CIF_train_uid": "V56331",
"schedule_days_runs": "0010000",
"schedule_days_runs": "0011000",
"CIF_stp_indicator": "N",
"applicable_timetable": "Y",
"atoc_code": "AW",
@ -12,7 +12,7 @@
"traction_class": ""
},
"transaction_type": "Create",
"schedule_end_date": "2023-01-04",
"schedule_end_date": "2023-01-05",
"schedule_segment": {
"CIF_service_branding": "",
"CIF_train_category": "OO",

Loading…
Cancel
Save