Dont use extension.

airbytehq · Nov 20, 2024 · dd5c2ef · dd5c2ef
1 parent 4ed50bd
commit dd5c2ef
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/...yte-cdk/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/state/TimeWindowTriggerTest.kt b/...yte-cdk/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/state/TimeWindowTriggerTest.kt
@@ -5,6 +5,7 @@
 package io.airbyte.cdk.load.state
 
 import io.airbyte.cdk.load.state.TimeWindowTriggerTest.Fixtures.TIME_WINDOW_WIDTH_MS
+import io.mockk.MockKAnnotations
 import io.mockk.every
 import io.mockk.impl.annotations.MockK
 import io.mockk.junit5.MockKExtension
@@ -18,14 +19,16 @@ import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.Arguments
 import org.junit.jupiter.params.provider.MethodSource
 
-@ExtendWith(MockKExtension::class)
+@MockKExtension.RequireParallelTesting
 class TimeWindowTriggerTest {
     @MockK lateinit var clock: Clock
 
     private lateinit var timeWindow: TimeWindowTrigger
 
     @BeforeEach
     fun setup() {
+        MockKAnnotations.init(this)
+
         timeWindow = TimeWindowTrigger(clock, TIME_WINDOW_WIDTH_MS)
     }
 

diff --git a/...cdk/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/task/internal/FlushTickTaskTest.kt b/...cdk/bulk/core/load/src/test/kotlin/io/airbyte/cdk/load/task/internal/FlushTickTaskTest.kt
@@ -17,6 +17,7 @@ import io.airbyte.cdk.load.message.MessageQueue
 import io.airbyte.cdk.load.message.MessageQueueSupplier
 import io.airbyte.cdk.load.message.StreamFlushEvent
 import io.airbyte.cdk.load.state.Reserved
+import io.mockk.MockKAnnotations
 import io.mockk.coVerify
 import io.mockk.every
 import io.mockk.impl.annotations.MockK
@@ -33,7 +34,6 @@ import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.Arguments
 import org.junit.jupiter.params.provider.MethodSource
 
-@ExtendWith(MockKExtension::class)
 class FlushTickTaskTest {
     @MockK(relaxed = true) lateinit var clock: Clock
     @MockK(relaxed = true) lateinit var coroutineTimeUtils: TimeProvider
@@ -48,6 +48,8 @@ class FlushTickTaskTest {
 
     @BeforeEach
     fun setup() {
+        MockKAnnotations.init(this)
+
         task =
             FlushTickTask(
                 tickIntervalMs,

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
@@ -27,6 +27,7 @@ import io.airbyte.cdk.load.task.DestinationTaskLauncher
 import io.airbyte.cdk.load.task.MockTaskLauncher
 import io.airbyte.cdk.load.test.util.StubDestinationMessageFactory
 import io.airbyte.cdk.load.util.lineSequence
+import io.mockk.MockKAnnotations
 import io.mockk.coEvery
 import io.mockk.coVerify
 import io.mockk.every
@@ -45,7 +46,7 @@ import org.junit.jupiter.api.extension.ExtendWith
 class SpillToDiskTaskTest {
     /** Validates task delegates to dependencies as expected. Does not test dependency behavior. */
     @Nested
-    @ExtendWith(MockKExtension::class)
+//    @MockKExtension.RequireParallelTesting
     inner class UnitTests {
         @MockK(relaxed = true) lateinit var spillFileProvider: SpillFileProvider
 
@@ -63,7 +64,7 @@ class SpillToDiskTaskTest {
 
         @BeforeEach
         fun setup() {
-            every { timeWindow.isComplete() } returns true
+            MockKAnnotations.init(this)
 
             inputQueue = DestinationStreamEventQueue()
             task =