@@ -54,14 +54,12 @@ class TestCodeGenTask(val request: TestCodeGenRequest, displayMessage: String) :
54
54
indicator.fraction = 0.1
55
55
indicator.text = AutoDevBundle .message(" intentions.chat.code.test.step.prepare-context" )
56
56
57
- // Check for cancellation early
58
57
indicator.checkCanceled()
59
58
60
59
AutoDevStatusService .notifyApplication(AutoDevStatus .InProgress )
61
60
val testContext = autoTestService.findOrCreateTestFile(request.file, request.project, request.element)
62
61
DumbService .getInstance(request.project).waitForSmartMode()
63
62
64
- // Check for cancellation after waiting for smart mode
65
63
indicator.checkCanceled()
66
64
67
65
if (testContext == null ) {
@@ -73,7 +71,6 @@ class TestCodeGenTask(val request: TestCodeGenRequest, displayMessage: String) :
73
71
indicator.text = AutoDevBundle .message(" intentions.chat.code.test.step.collect-context" )
74
72
indicator.fraction = 0.3
75
73
76
- // Check for cancellation before collecting context
77
74
indicator.checkCanceled()
78
75
79
76
val testPromptContext = TestCodeGenContext ()
@@ -90,8 +87,6 @@ class TestCodeGenTask(val request: TestCodeGenRequest, displayMessage: String) :
90
87
}
91
88
92
89
testPromptContext.frameworkContext = contextItems.joinToString(" \n " , transform = ChatContextItem ::text)
93
-
94
- // Check for cancellation before read actions
95
90
indicator.checkCanceled()
96
91
97
92
ReadAction .compute<Unit , Throwable > {
0 commit comments