fix: map current_range_meters as provided

Fixing upstream feed issues (missing current_range_meters) would need to check the vehicle_type's propulsion_type, which is not available here
entur · Jul 24, 2024 · e6b2c71 · e6b2c71
1 parent 115e467
commit e6b2c71
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/src/main/java/org/entur/lamassu/mapper/feedmapper/v2/FreeBikeStatusFeedMapper.java b/src/main/java/org/entur/lamassu/mapper/feedmapper/v2/FreeBikeStatusFeedMapper.java
@@ -80,9 +80,7 @@ protected GBFSBike mapBike(GBFSBike bike, FeedProvider feedProvider) {
     mapped.setRentalUris(bike.getRentalUris());
     mapped.setVehicleTypeId(mapVehicleTypeId(bike.getVehicleTypeId(), feedProvider));
     mapped.setLastReported(bike.getLastReported());
-    mapped.setCurrentRangeMeters(
-      bike.getCurrentRangeMeters() != null ? bike.getCurrentRangeMeters() : 0
-    );
+    mapped.setCurrentRangeMeters(bike.getCurrentRangeMeters());
     mapped.setCurrentFuelPercent(bike.getCurrentFuelPercent());
     mapped.setStationId(mapStationId(bike.getStationId(), feedProvider));
     mapped.setHomeStationId(mapStationId(bike.getHomeStationId(), feedProvider));

diff --git a/src/main/java/org/entur/lamassu/mapper/feedmapper/v3/VehicleStatusFeedMapper.java b/src/main/java/org/entur/lamassu/mapper/feedmapper/v3/VehicleStatusFeedMapper.java
@@ -78,9 +78,7 @@ protected GBFSVehicle mapVehicle(GBFSVehicle vehicle, FeedProvider feedProvider)
     mapped.setRentalUris(vehicle.getRentalUris());
     mapped.setVehicleTypeId(mapVehicleTypeId(vehicle.getVehicleTypeId(), feedProvider));
     mapped.setLastReported(vehicle.getLastReported());
-    mapped.setCurrentRangeMeters(
-      vehicle.getCurrentRangeMeters() != null ? vehicle.getCurrentRangeMeters() : 0
-    );
+    mapped.setCurrentRangeMeters(vehicle.getCurrentRangeMeters());
     mapped.setCurrentFuelPercent(vehicle.getCurrentFuelPercent());
     mapped.setStationId(mapStationId(vehicle.getStationId(), feedProvider));
     mapped.setHomeStationId(mapStationId(vehicle.getHomeStationId(), feedProvider));

diff --git a/src/test/java/org/entur/lamassu/mapper/feedmapper/v2/FreeBikeStatusFeedMapperTest.java b/src/test/java/org/entur/lamassu/mapper/feedmapper/v2/FreeBikeStatusFeedMapperTest.java
@@ -41,7 +41,8 @@ void prepare() {
   void testMissingCurrentRangeMeters() {
     var feedProvider = getTestProvider();
     var mapped = mapper.mapBike(new GBFSBike(), feedProvider);
-    Assertions.assertNotNull(mapped.getCurrentRangeMeters());
+    // if no current_range_meters is provided, we explicitly don't want Lamassu to fill it in
+    Assertions.assertNull(mapped.getCurrentRangeMeters());
   }
 
   @Test