Skip to content

Commit 3622a8f

Browse files
committed
Refactor StdioClientTransport tests
1 parent b486d61 commit 3622a8f

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed
Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@ package io.modelcontextprotocol.kotlin.sdk.client.stdio
33
import io.kotest.matchers.booleans.shouldBeFalse
44
import io.kotest.matchers.shouldBe
55
import io.modelcontextprotocol.kotlin.sdk.client.StdioClientTransport
6-
import kotlinx.atomicfu.atomic
7-
import kotlinx.atomicfu.update
86
import kotlinx.coroutines.delay
97
import kotlinx.coroutines.test.runTest
108
import kotlinx.io.Buffer
119
import kotlinx.io.writeString
10+
import kotlin.concurrent.atomics.AtomicBoolean
11+
import kotlin.concurrent.atomics.ExperimentalAtomicApi
1212
import kotlin.test.Test
1313
import kotlin.time.Duration.Companion.milliseconds
1414

@@ -19,6 +19,7 @@ class StdioClientTransportErrorHandlingTest {
1919

2020
private lateinit var transport: StdioClientTransport
2121

22+
@OptIn(ExperimentalAtomicApi::class)
2223
@Test
2324
fun `should continue on stderr EOF`() = runTest {
2425
val stderrBuffer = Buffer()
@@ -34,17 +35,17 @@ class StdioClientTransportErrorHandlingTest {
3435
error = stderrBuffer,
3536
)
3637

37-
val closeCalled = atomic(false)
38-
transport.onClose { closeCalled.update { true } }
38+
val closeCalled = AtomicBoolean(false)
39+
transport.onClose { closeCalled.store(true) }
3940

4041
transport.start()
4142
delay(200.milliseconds)
4243

4344
// Stderr EOF should not close transport
44-
closeCalled.value shouldBe false
45+
closeCalled.load() shouldBe false
4546

4647
transport.close()
47-
closeCalled.value shouldBe true
48+
closeCalled.load() shouldBe true
4849
}
4950

5051
@Test

0 commit comments

Comments
 (0)