pass consistentReads through overload (#191)

* pass consistentReads through overload * fix code format
cashapp · Sep 3, 2024 · 9c4094e · 9c4094e
1 parent 64f40ef
commit 9c4094e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 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,13 @@ class LogicalDbBatchTest {
     musicTable.playlistInfo.save(playlistInfo)
 
     val loadedItems = musicDb.batchLoad(
-      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)
+      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)
+      ),
+      consistentReads = true
     )
     assertThat(loadedItems.getItems<AlbumTrack>()).containsAll(albumTracks)
     assertThat(loadedItems.getItems<PlaylistInfo>()).containsExactly(playlistInfo)