Test for updateTripsCache

core/test/build.gradle.kts

@@ -43,6 +43,7 @@ kotlin {
         commonTest {
             dependencies {
                 implementation(projects.core.analytics)
+                implementation(projects.core.dateTime)
                 implementation(projects.core.log)
                 implementation(projects.sandook)
                 implementation(projects.feature.tripPlanner.ui)
@@ -53,6 +54,7 @@ kotlin {
                 implementation(libs.test.kotlinxCoroutineTest)
                 implementation(libs.test.turbine)
                 implementation(libs.kotlinx.collections.immutable)
+                implementation(libs.kotlinx.datetime)
             }
         }
 

core/test/src/commonTest/kotlin/xyz/ksharma/core/test/fakes/FakeTripPlanningService.kt

@@ -1,12 +1,8 @@
 package xyz.ksharma.core.test.fakes
 
-import xyz.ksharma.core.test.fakes.FakeTripResponseBuilder.buildDestinationStopSequence
-import xyz.ksharma.core.test.fakes.FakeTripResponseBuilder.buildOriginStopSequence
-import xyz.ksharma.core.test.fakes.FakeTripResponseBuilder.buildTransportation
 import xyz.ksharma.krail.trip.planner.network.api.model.StopFinderResponse
 import xyz.ksharma.krail.trip.planner.network.api.model.StopType
 import xyz.ksharma.krail.trip.planner.network.api.model.TripResponse
-import xyz.ksharma.krail.trip.planner.network.api.model.TripResponse.StopSequence
 import xyz.ksharma.krail.trip.planner.network.api.service.DepArr
 import xyz.ksharma.krail.trip.planner.network.api.service.TripPlanningService
 
@@ -21,26 +17,7 @@ class FakeTripPlanningService : TripPlanningService {
         date: String?,
         time: String?,
     ): TripResponse {
-        return if (isSuccess)
-
-        // Return a fake TripResponse
-            TripResponse(
-                journeys = listOf(
-                    TripResponse.Journey(
-                        legs = listOf(
-                            TripResponse.Leg(
-                                origin = buildOriginStopSequence(),
-                                destination = buildDestinationStopSequence(),
-                                stopSequence = listOf(
-                                    buildOriginStopSequence(),
-                                ),
-                                transportation = buildTransportation(),
-                                duration = 100,
-                            ),
-                        ),
-                    ),
-                )
-            )
+        return if (isSuccess) FakeTripResponseBuilder.buildTripResponse()
         else throw IllegalStateException("Failed to fetch trip")
     }
 

core/test/src/commonTest/kotlin/xyz/ksharma/core/test/fakes/FakeTripResponseBuilder.kt

@@ -1,34 +1,62 @@
 package xyz.ksharma.core.test.fakes
 
+import kotlinx.datetime.Clock
 import xyz.ksharma.krail.trip.planner.network.api.model.StopType
 import xyz.ksharma.krail.trip.planner.network.api.model.TripResponse
 import xyz.ksharma.krail.trip.planner.network.api.model.TripResponse.StopSequence
+import kotlin.time.Duration.Companion.minutes
 
 object FakeTripResponseBuilder {
+    private var originStopSequence: StopSequence = buildStopSequence()
+    private var destinationStopSequence: StopSequence = buildStopSequence()
+    private var stopSequence: List<StopSequence> = listOf(buildStopSequence())
+    private var transportation: TripResponse.Transportation =
+        buildTransportation()
+    private var duration: Long = 200
 
-    fun buildOriginStopSequence() = StopSequence(
-        arrivalTimePlanned = "2024-09-24T19:00:00Z",
-        arrivalTimeEstimated = "2024-09-24T19:10:00Z",
-        departureTimePlanned = "2024-09-24T19:10:00Z",
-        departureTimeEstimated = "2024-09-24T19:10:00Z",
-        name = "Origin Stop",
-        disassembledName = "Origin Name",
-        id = "Origin_stop_id",
-        type = StopType.STOP.type,
-    )
+    fun setOriginStopSequence(originStopSequence: StopSequence) =
+        apply { this.originStopSequence = originStopSequence }
+
+    fun setDestinationStopSequence(destinationStopSequence: StopSequence) =
+        apply { this.destinationStopSequence = destinationStopSequence }
+
+    fun setStopSequence(stopSequence: List<StopSequence>) =
+        apply { this.stopSequence = stopSequence }
 
-    fun buildDestinationStopSequence() = StopSequence(
-        arrivalTimePlanned = "2024-09-24T20:00:00Z",
-        arrivalTimeEstimated = "2024-09-24T20:10:00Z",
-        departureTimePlanned = "2024-09-24T20:10:00Z",
-        departureTimeEstimated = "2024-09-24T20:10:00Z",
-        name = "Destination Stop",
-        disassembledName = "Destination Name",
-        id = "Destination_stop_id",
+    fun setTransportation(transportation: TripResponse.Transportation) =
+        apply { this.transportation = transportation }
+
+    fun setDuration(duration: Long) = apply { this.duration = duration }
+
+    fun build(): TripResponse.Leg {
+        return TripResponse.Leg(
+            origin = originStopSequence,
+            destination = destinationStopSequence,
+            stopSequence = stopSequence,
+            transportation = transportation,
+            duration = duration
+        )
+    }
+
+    private fun buildStopSequence(
+        arrivalTimePlanned: String = "2024-09-24T19:00:00Z",
+        arrivalTimeEstimated: String = "2024-09-24T19:00:00Z",
+        departureTimePlanned: String = "2024-09-24T19:10:00Z",
+        departureTimeEstimated: String = "2024-09-24T19:10:00Z",
+        name: String = "Stop",
+        id: String = "stop_id",
+    ) = StopSequence(
+        arrivalTimePlanned = arrivalTimePlanned,
+        arrivalTimeEstimated = arrivalTimeEstimated,
+        departureTimePlanned = departureTimePlanned,
+        departureTimeEstimated = departureTimeEstimated,
+        name = name,
+        disassembledName = name,
+        id = id,
         type = StopType.STOP.type,
     )
 
-    fun buildTransportation() = TripResponse.Transportation(
+    private fun buildTransportation(transportationId: String = "Transportation Id") = TripResponse.Transportation(
         disassembledName = "Transportation Name",
         product = TripResponse.Product(
             productClass = 1,
@@ -39,7 +67,77 @@ object FakeTripResponseBuilder {
             id = "Destination Operator Id",
         ),
         name = "Transportation Name",
-        id = "Transportation Id",
+        id = transportationId,
         description = "Transportation Description",
     )
+
+    private fun buildJourneyLeg(legIndex: Int, stops: Int, journeyIndex: Int = 0) =
+        TripResponse.Leg(
+        origin = buildStopSequence(
+            name = "Origin Stop $legIndex",
+            arrivalTimeEstimated = Clock.System.now().plus(5.minutes * journeyIndex).toString(),
+            arrivalTimePlanned = Clock.System.now().plus(5.minutes * journeyIndex).toString(),
+            departureTimeEstimated = Clock.System.now().plus(5.minutes * journeyIndex).toString(),
+            departureTimePlanned = Clock.System.now().plus(5.minutes * journeyIndex).toString(),
+        ),
+        destination = buildStopSequence(
+            name = "Destination Stop $legIndex",
+            arrivalTimeEstimated = Clock.System.now().plus(10.minutes * journeyIndex).toString(),
+            arrivalTimePlanned = Clock.System.now().plus(10.minutes * journeyIndex).toString(),
+            departureTimeEstimated = Clock.System.now().plus(10.minutes * journeyIndex).toString(),
+            departureTimePlanned = Clock.System.now().plus(10.minutes * journeyIndex).toString(),
+        ),
+        stopSequence = List(stops) { index ->
+            buildStopSequence(
+                id = "stop_id_${index + 1}",
+                name = "Stop ${index + 1}",
+                arrivalTimeEstimated = Clock.System.now().plus(5.minutes + journeyIndex.minutes)
+                    .toString(),
+                arrivalTimePlanned = Clock.System.now().plus(5.minutes + journeyIndex.minutes)
+                    .toString(),
+                departureTimeEstimated = Clock.System.now().plus(5.minutes + journeyIndex.minutes)
+                    .toString(),
+                departureTimePlanned = Clock.System.now().plus(5.minutes + journeyIndex.minutes)
+                    .toString(),
+            )
+        },
+        transportation = buildTransportation(
+            transportationId = "Transportation Id $journeyIndex",
+        ),
+        duration = 120,
+    )
+
+    private fun buildJourneyList(
+        numberOfJourney: Int = 1,
+        numberOfLegs: Int = 1,
+        reverseTimeOrder: Boolean = false,
+    ): List<TripResponse.Journey> {
+        val journeyList = List(numberOfJourney) { buildJourney(numberOfLegs = numberOfLegs, journeyIndex = it) }
+        return if (reverseTimeOrder) journeyList.reversed() else journeyList
+    }
+
+    private fun buildJourney(
+        numberOfLegs: Int = 1,
+        stops: Int = 1,
+        journeyIndex: Int = 0,
+    ): TripResponse.Journey {
+        return TripResponse.Journey(
+            legs = List(numberOfLegs) { index ->
+                buildJourneyLeg(legIndex = index, stops = stops, journeyIndex = journeyIndex)
+            },
+        )
+    }
+
+    fun buildTripResponse(
+        numberOfJourney: Int = 1, numberOfLegs: Int = 1,
+        reverseTimeOrder: Boolean = false,
+    ): TripResponse {
+        return TripResponse(
+            journeys = buildJourneyList(
+                numberOfJourney = numberOfJourney,
+                numberOfLegs = numberOfLegs,
+                reverseTimeOrder = reverseTimeOrder,
+            ),
+        )
+    }
 }