test: Add unit test for trip shape loading

Author: EmmaSimon

shared/src/commonMain/kotlin/com/mbta/tid/mbta_app/repositories/TripRepository.kt

@@ -100,6 +100,7 @@ constructor(
     internal var tripShape: MapFriendlyRouteResponse = MapFriendlyRouteResponse(emptyList()),
     internal val onGetTrip: (String) -> Unit = {},
     internal val onGetTripSchedules: (String) -> Unit = {},
+    internal val onGetTripShape: (String) -> Unit = {},
 ) : ITripRepository {
     override suspend fun getTripSchedules(tripId: String): ApiResult<TripSchedulesResponse> {
         onGetTripSchedules(tripId)
@@ -112,6 +113,7 @@ constructor(
     }
 
     override suspend fun getTripShape(tripId: String): ApiResult<MapFriendlyRouteResponse> {
+        onGetTripShape(tripId)
         return ApiResult.Ok(tripShape)
     }
 }

shared/src/commonTest/kotlin/com/mbta/tid/mbta_app/viewModel/MapViewModelTests.kt

@@ -16,6 +16,7 @@ import com.mbta.tid.mbta_app.model.response.AlertsStreamDataResponse
 import com.mbta.tid.mbta_app.model.response.MapFriendlyRouteResponse
 import com.mbta.tid.mbta_app.model.routeDetailsPage.RouteDetailsContext
 import com.mbta.tid.mbta_app.repositories.ISentryRepository
+import com.mbta.tid.mbta_app.repositories.MockTripRepository
 import com.mbta.tid.mbta_app.routes.SheetRoutes
 import com.mbta.tid.mbta_app.utils.EasternTimeInstant
 import com.mbta.tid.mbta_app.utils.IMapLayerManager
@@ -35,6 +36,7 @@ import kotlin.test.AfterTest
 import kotlin.test.Ignore
 import kotlin.test.Test
 import kotlin.test.assertEquals
+import kotlin.test.assertNull
 import kotlin.time.Clock
 import kotlin.time.Duration.Companion.hours
 import kotlin.time.Duration.Companion.seconds
@@ -298,6 +300,41 @@ internal class MapViewModelTests : KoinTest {
         }
     }
 
+    @Test
+    fun followingVehicleLoadsShape() = runTest {
+        val dispatcher = StandardTestDispatcher(testScheduler)
+        var loadedTripShapeId: String? = null
+        val tripRepository = MockTripRepository(onGetTripShape = { loadedTripShapeId = it })
+        setUpKoin(dispatcher) { trip = tripRepository }
+        val viewportProvider = MockViewportManager()
+        val viewModel: MapViewModel = get()
+        val vehicle = TestData.vehicle { currentStatus = Vehicle.CurrentStatus.StoppedAt }
+        val tripDetailsFilter = TripDetailsFilter("trip", vehicle.id, null, false)
+        testViewModelFlow(viewModel).test {
+            assertNull(loadedTripShapeId)
+            viewModel.setViewportManager(viewportProvider)
+            viewModel.densityChanged(1f)
+            assertEquals(MapViewModel.State(MapViewModel.LayerState.Overview, false), awaitItem())
+            val stop = TestData.stops["70113"]!!
+            viewModel.selectedTrip(null, stop, tripDetailsFilter, vehicle, true)
+            assertEquals(
+                MapViewModel.State(
+                    MapViewModel.LayerState.TripSelected(
+                        stop,
+                        null,
+                        tripDetailsFilter,
+                        vehicle,
+                        true,
+                    ),
+                    false,
+                ),
+                awaitItem(),
+            )
+            advanceUntilIdle()
+            assertEquals(tripDetailsFilter.tripId, loadedTripShapeId)
+        }
+    }
+
     @Test
     @Ignore // TODO: Address flakiness
     fun whenInStopDetailsNotResetToAllRailOnAlertChange() = runTest {