pass consistentReads through overload

cashapp · Sep 3, 2024 · c1b4b82 · c1b4b82
1 parent 64f40ef
commit c1b4b82
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tempest2/src/main/kotlin/app/cash/tempest2/LogicalDb.kt b/tempest2/src/main/kotlin/app/cash/tempest2/LogicalDb.kt
@@ -111,7 +111,7 @@ interface LogicalDb : LogicalTable.Factory {
   fun batchLoad(
     keys: Iterable<Any>,
     consistentReads: Boolean
-  ) = batchLoad(keys, consistentReads = false, MAX_BATCH_READ)
+  ) = batchLoad(keys, consistentReads, MAX_BATCH_READ)
 
   fun batchLoad(
     vararg keys: Any,

diff --git a/tempest2/src/test/kotlin/app/cash/tempest2/LogicalDbBatchTest.kt b/tempest2/src/test/kotlin/app/cash/tempest2/LogicalDbBatchTest.kt
@@ -54,10 +54,11 @@ class LogicalDbBatchTest {
     musicTable.playlistInfo.save(playlistInfo)
 
     val loadedItems = musicDb.batchLoad(
-      PlaylistInfo.Key("PLAYLIST_1"),
+      keys = listOf( PlaylistInfo.Key("PLAYLIST_1"),
       AlbumTrack.Key("ALBUM_1", track_number = 1),
       AlbumTrack.Key("ALBUM_1", track_number = 2),
-      AlbumTrack.Key("ALBUM_1", track_number = 3)
+      AlbumTrack.Key("ALBUM_1", track_number = 3)),
+      consistentReads = true
     )
     assertThat(loadedItems.getItems<AlbumTrack>()).containsAll(albumTracks)
     assertThat(loadedItems.getItems<PlaylistInfo>()).containsExactly(playlistInfo)