Skip to content

test(jetbrains): isolate workspace concurrency test #11629

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
markijbema merged 1 commit into from
Jun 24, 2026
Merged

test(jetbrains): isolate workspace concurrency test #11629

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 25 additions & 13 deletions ...tbrains/backend/src/test/kotlin/ai/kilocode/backend/workspace/KiloBackendWorkspaceTest.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,18 +2,23 @@ package ai.kilocode.backend.workspace

import ai.kilocode.backend.app.KiloAppState
import ai.kilocode.backend.app.KiloBackendAppService
import ai.kilocode.backend.app.KiloBackendSessionManager
import ai.kilocode.backend.app.SseEvent
import ai.kilocode.backend.cli.KiloBackendHttpClients
import ai.kilocode.backend.workspace.KiloBackendWorkspace
import ai.kilocode.backend.workspace.KiloWorkspaceState
import ai.kilocode.backend.testing.FakeCliServer
import ai.kilocode.backend.testing.MockCliServer
import ai.kilocode.backend.testing.TestLog
import ai.kilocode.jetbrains.api.client.DefaultApi
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.SupervisorJob
import kotlinx.coroutines.async
import kotlinx.coroutines.awaitAll
import kotlinx.coroutines.cancel
import kotlinx.coroutines.delay
import kotlinx.coroutines.flow.MutableSharedFlow
import kotlinx.coroutines.flow.first
import kotlinx.coroutines.launch
import kotlinx.coroutines.runBlocking
Expand Down Expand Up @@ -388,20 +393,27 @@ class KiloBackendWorkspaceTest {

@Test
fun `concurrent get for same directory returns same instance`() = runBlocking {
val app = setup()
app.connect()
withTimeout(10_000) { app.appState.first { it is KiloAppState.Ready } }

// Launch many concurrent get() calls for the same directory
val results = (1..10).map {
async(Dispatchers.Default) {
app.workspaces.get("/same/dir")
}
}.awaitAll()
val port = mock.start()
val http = KiloBackendHttpClients.api(mock.password)
val api = DefaultApi(basePath = "http://127.0.0.1:$port", client = http)
val events = MutableSharedFlow<SseEvent>()
val sessions = KiloBackendSessionManager(scope, log)
val manager = KiloBackendWorkspaceManager(scope, sessions, log)
manager.start(api, http, port, events)

try {
val results = (1..10).map {
async(Dispatchers.Default) {
manager.get("/same/dir")
}
}.awaitAll()

// All must return the exact same instance
val first = results[0]
results.forEach { assertTrue(it === first) }
val first = results[0]
results.forEach { assertTrue(it === first) }
} finally {
manager.stop()
KiloBackendHttpClients.shutdown(http)
}
}

@Test
Expand Down
Loading