MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-03-22 15:19:59 +07:00
Code Issues Packages Projects Releases Wiki Activity

[ai-completion] LLM-11785: fix tests that relied on that after insert no session is possible

Browse Source 
GitOrigin-RevId: e77d1221daf5af09c3ac5154fd54089292f6179a
This commit is contained in:
Kirill.Karnaukhov
2024-10-08 21:13:39 +02:00
committed by intellij-monorepo-bot
parent 0fffcefa5b
commit 0d4b48560f
2 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
platform/platform-tests/testSrc/com/intellij/codeInsight/inline/completion/logs/InlineCompletionSessionLogsTest.kt
View File

@@ -71,6 +71,7 @@ internal class InlineCompletionSessionLogsTest : InlineCompletionTestCase() {
callInlineCompletion()
// now cancel it
typeChar('a')
// it doesn't create the second session log because it hangs infinitely
}
singleSessionLog.assertRequestIdPresent()
singleSessionLog.assertSomeContextLogsPresent()
@@ -94,6 +95,7 @@ internal class InlineCompletionSessionLogsTest : InlineCompletionTestCase() {
provider.computeNextElement()
delay()
typeChar('a')
// it doesn't create the second session log because it hangs infinitely
}
singleSessionLog.assertRequestIdPresent()
singleSessionLog.assertSomeContextLogsPresent()

19
platform/platform-tests/testSrc/com/intellij/codeInsight/inline/completion/logs/TestInlineCompletionLogs.kt
View File

@@ -10,6 +10,7 @@ import com.jetbrains.fus.reporting.model.lion3.LogEvent
import io.kotest.common.runBlocking
import org.jetbrains.annotations.ApiStatus
import kotlin.test.assertEquals
import kotlin.test.assertNotNull
import kotlin.test.assertTrue
/**
@@ -42,6 +43,20 @@ object TestInlineCompletionLogs {
* Also see [noSessionLogs].
*/
fun InlineCompletionLifecycleTestDSL.singleSessionLog(block: suspend InlineCompletionLifecycleTestDSL.() -> Unit): SingleSessionLog {
val sessionsLogs = sessionsLog(block)
assertEquals(1, sessionsLogs.size, "There should be exactly session log, but found: $sessionsLogs")
return sessionsLogs.single()
}
fun InlineCompletionLifecycleTestDSL.firstSessionLog(block: suspend InlineCompletionLifecycleTestDSL.() -> Unit): SingleSessionLog {
val sessionLog = sessionsLog(block).firstOrNull()
assertNotNull(sessionLog, "There should be at least one session log, but found nothing.")
return sessionLog
}
private fun InlineCompletionLifecycleTestDSL.sessionsLog(
block: suspend InlineCompletionLifecycleTestDSL.() -> Unit
): List<SingleSessionLog> {
val disposable = Disposer.newDisposable()
val events = try {
FUCollectorTestCase.collectLogEvents("ML", disposable) {
@@ -53,9 +68,7 @@ object TestInlineCompletionLogs {
finally {
Disposer.dispose(disposable)
}
val sessionLogs = getSessionLogs(events)
assertEquals(1, sessionLogs.size, "There should be exactly session log, but found: $sessionLogs")
return SingleSessionLog(sessionLogs.single())
return getSessionLogs(events).map { SingleSessionLog(it) }
}
class SingleSessionLog internal constructor(private val logEvent: LogEvent) {