Formatting.

airbytehq · Nov 20, 2024 · a1af53d · a1af53d
1 parent 18508a9
commit a1af53d
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 24 deletions.
diff --git a/...te-cdk/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/state/ReservationManagerTest.kt b/...te-cdk/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/state/ReservationManagerTest.kt
@@ -55,27 +55,27 @@ class ReservationManagerTest {
         Assertions.assertTrue(reserved.get())
     }
 
-//    @Test
-//    fun testReserveMultithreaded() = runTest {
-//        val manager = ReservationManager(1000)
-//        withContext(Dispatchers.IO) {
-//            manager.reserve(1000, this)
-//            Assertions.assertEquals(0, manager.remainingCapacityBytes)
-//            val nIterations = 100000
-//
-//            val jobs = (0 until nIterations).map { launch { manager.reserve(10, this) } }
-//
-//            repeat(nIterations) {
-//                manager.release(10)
-//                Assertions.assertTrue(
-//                    manager.remainingCapacityBytes >= 0,
-//                    "Remaining memory is negative: ${manager.remainingCapacityBytes}"
-//                )
-//            }
-//            jobs.forEach { it.join() }
-//            Assertions.assertEquals(0, manager.remainingCapacityBytes)
-//        }
-//    }
+    @Test
+    fun testReserveMultithreaded() = runTest {
+        val manager = ReservationManager(1000)
+        withContext(Dispatchers.IO) {
+            manager.reserve(1000, this)
+            Assertions.assertEquals(0, manager.remainingCapacityBytes)
+            val nIterations = 100000
+
+            val jobs = (0 until nIterations).map { launch { manager.reserve(10, this) } }
+
+            repeat(nIterations) {
+                manager.release(10)
+                Assertions.assertTrue(
+                    manager.remainingCapacityBytes >= 0,
+                    "Remaining memory is negative: ${manager.remainingCapacityBytes}"
+                )
+            }
+            jobs.forEach { it.join() }
+            Assertions.assertEquals(0, manager.remainingCapacityBytes)
+        }
+    }
 
     @Test
     fun testRequestingMoreThanAvailableThrows() = runTest {

diff --git a/...k/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/task/internal/SpillToDiskTaskTest.kt b/...k/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/task/internal/SpillToDiskTaskTest.kt
@@ -107,9 +107,8 @@ class SpillToDiskTaskTest {
         @Test
         fun `publishes 'spilled file' aggregates according to time window on stream flush event`() =
             runTest {
-
                 val timeWindow = TimeWindowTrigger(Clock.systemUTC(), 0)
-                task =
+                val task =
                     DefaultSpillToDiskTask(
                         spillFileProvider,
                         inputQueue,
@@ -141,7 +140,6 @@ class SpillToDiskTaskTest {
                 task.execute()
                 coVerify(exactly = 1) { taskLauncher.handleNewSpilledFile(any(), any()) }
             }
-
     }
 
     /**