Topograhic place, parking, scheduled stop point and group of tariff zones are also versioned

Author: testower

src/main/java/org/entur/netex/index/api/NetexEntitiesIndex.java

@@ -252,19 +252,19 @@ public interface NetexEntitiesIndex {
      * Get an entity index of TopographicPlace
      * @return
      */
-    NetexEntityIndex<TopographicPlace> getTopographicPlaceIndex();
+    VersionedNetexEntityIndex<TopographicPlace> getTopographicPlaceIndex();
 
     /**
      * Get an entity index of Parking
      * @return
      */
-    NetexEntityIndex<Parking> getParkingIndex();
+    VersionedNetexEntityIndex<Parking> getParkingIndex();
 
     /**
      * Get an entity index of ScheduledStopPoint
      * @return
      */
-    NetexEntityIndex<ScheduledStopPoint> getScheduledStopPointIndex();
+    VersionedNetexEntityIndex<ScheduledStopPoint> getScheduledStopPointIndex();
 
     /**
      * Get an entity index of RoutePoint
@@ -284,7 +284,7 @@ public interface NetexEntitiesIndex {
      */
     VersionedNetexEntityIndex<FareZone> getFareZoneIndex();
 
-    NetexEntityIndex<GroupOfTariffZones> getGroupOfTariffZonesIndex();
+    VersionedNetexEntityIndex<GroupOfTariffZones> getGroupOfTariffZonesIndex();
 
     Collection<CompositeFrame> getCompositeFrames();
     Collection<ResourceFrame> getResourceFrames();

src/main/java/org/entur/netex/index/impl/NetexEntitiesIndexImpl.java

@@ -88,12 +88,12 @@ public class NetexEntitiesIndexImpl implements NetexEntitiesIndex {
     public final NetexEntityIndex<ServiceLink> serviceLinkById;
     public final VersionedNetexEntityIndex<StopPlace> stopPlaceById;
     public final VersionedNetexEntityIndex<TariffZone> tariffZonesById;
-    public final NetexEntityIndex<TopographicPlace> topographicPlaceById;
-    public final NetexEntityIndex<Parking> parkingById;
-    public final NetexEntityIndex<ScheduledStopPoint> scheduledStopPointById;
+    public final VersionedNetexEntityIndex<TopographicPlace> topographicPlaceById;
+    public final VersionedNetexEntityIndex<Parking> parkingById;
+    public final VersionedNetexEntityIndex<ScheduledStopPoint> scheduledStopPointById;
     public final NetexEntityIndex<RoutePoint> routePointById;
     public final VersionedNetexEntityIndex<FareZone> fareZoneById;
-    public final NetexEntityIndex<GroupOfTariffZones> groupOfTariffZonesById;
+    public final VersionedNetexEntityIndex<GroupOfTariffZones> groupOfTariffZonesById;
     public final Multimap<String, Parking> parkingsByParentSiteRef;
 
     // Relations between entities - The Netex XML sometimes rely on the the
@@ -147,12 +147,12 @@ public NetexEntitiesIndexImpl() {
         this.serviceLinkById = new NetexEntityMapByIdImpl<>();
         this.stopPlaceById = new VersionedNetexEntityIndexImpl<>();
         this.tariffZonesById = new VersionedNetexEntityIndexImpl<>();
-        this.topographicPlaceById = new NetexEntityMapByIdImpl<>();
-        this.parkingById = new NetexEntityMapByIdImpl<>();
-        this.scheduledStopPointById = new NetexEntityMapByIdImpl<>();
+        this.topographicPlaceById = new VersionedNetexEntityIndexImpl<>();
+        this.parkingById = new VersionedNetexEntityIndexImpl<>();
+        this.scheduledStopPointById = new VersionedNetexEntityIndexImpl<>();
         this.routePointById = new NetexEntityMapByIdImpl<>();
         this.fareZoneById = new VersionedNetexEntityIndexImpl<>();
-        this.groupOfTariffZonesById = new NetexEntityMapByIdImpl<>();
+        this.groupOfTariffZonesById = new VersionedNetexEntityIndexImpl<>();
         this.compositeFrames = new HashSet<>();
         this.siteFrames = new HashSet<>();
         this.resourceFrames = new HashSet<>();
@@ -328,17 +328,17 @@ public VersionedNetexEntityIndex<TariffZone> getTariffZoneIndex() {
     }
 
     @Override
-    public NetexEntityIndex<TopographicPlace> getTopographicPlaceIndex() {
+    public VersionedNetexEntityIndex<TopographicPlace> getTopographicPlaceIndex() {
         return topographicPlaceById;
     }
 
     @Override
-    public NetexEntityIndex<Parking> getParkingIndex() {
+    public VersionedNetexEntityIndex<Parking> getParkingIndex() {
         return parkingById;
     }
 
     @Override
-    public NetexEntityIndex<ScheduledStopPoint> getScheduledStopPointIndex() {
+    public VersionedNetexEntityIndex<ScheduledStopPoint> getScheduledStopPointIndex() {
         return scheduledStopPointById;
     }
 
@@ -353,7 +353,7 @@ public VersionedNetexEntityIndex<FareZone> getFareZoneIndex() {
     }
 
     @Override
-    public NetexEntityIndex<GroupOfTariffZones> getGroupOfTariffZonesIndex() {
+    public VersionedNetexEntityIndex<GroupOfTariffZones> getGroupOfTariffZonesIndex() {
         return groupOfTariffZonesById;
     }
 

src/test/java/org/entur/netex/TestGroupOfTariffZones.java

@@ -26,7 +26,7 @@ static void init() {
 
     @Test
     void testGetAllGroupsOfTariffZones() {
-        Collection<GroupOfTariffZones> groups = index.getGroupOfTariffZonesIndex().getAll();
+        Collection<GroupOfTariffZones> groups = index.getGroupOfTariffZonesIndex().getLatestVersions();
         Assertions.assertFalse(groups.isEmpty());
     }
 }

src/test/java/org/entur/netex/TestStopPlacesExport.java

@@ -62,26 +62,26 @@ void testGetTariffZone() {
 
     @Test
     void testGetTopographicPlace() {
-        TopographicPlace topographicPlace = index.getTopographicPlaceIndex().get("KVE:TopographicPlace:50");
+        TopographicPlace topographicPlace = index.getTopographicPlaceIndex().getLatestVersion("KVE:TopographicPlace:50");
         Assertions.assertEquals("Trøndelag", topographicPlace.getDescriptor().getName().getValue());
         Assertions.assertEquals("no", topographicPlace.getCountryRef().getRef().value());
     }
 
     @Test
     void testGetParking() {
-        Parking parking = index.getParkingIndex().get("NSR:Parking:1");
+        Parking parking = index.getParkingIndex().getLatestVersion("NSR:Parking:1");
         Assertions.assertEquals("Drammen", parking.getName().getValue());
     }
 
     @Test
     void testGetScheduledStopPoint() {
-        ScheduledStopPoint scheduledStopPoint = index.getScheduledStopPointIndex().get("NSR:ScheduledStopPoint:S5");
+        ScheduledStopPoint scheduledStopPoint = index.getScheduledStopPointIndex().getLatestVersion("NSR:ScheduledStopPoint:S5");
         Assertions.assertEquals("Gudå", scheduledStopPoint.getName().getValue());
     }
 
     @Test
     void testGetPassengerStopAssignment() {
-        ScheduledStopPoint scheduledStopPoint = index.getScheduledStopPointIndex().get("NSR:ScheduledStopPoint:S5");
+        ScheduledStopPoint scheduledStopPoint = index.getScheduledStopPointIndex().getLatestVersion("NSR:ScheduledStopPoint:S5");
         Collection<PassengerStopAssignment> passengerStopAssignments = index.getPassengerStopAssignmentsByStopPointRefIndex().get(scheduledStopPoint.getId());
 
         Assertions.assertEquals(1, passengerStopAssignments.size());