Fix CodyActionPromoter

sourcegraph · Jul 31, 2024 · a402e6d · a402e6d
1 parent bb4d043
commit a402e6d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/integrationTest/kotlin/com/sourcegraph/cody/util/CodyIntegrationTextFixture.kt b/src/integrationTest/kotlin/com/sourcegraph/cody/util/CodyIntegrationTextFixture.kt
@@ -39,9 +39,9 @@ open class CodyIntegrationTextFixture : BasePlatformTestCase(), LensListener {
 
   override fun setUp() {
     super.setUp()
+    myProject = project
     configureFixture()
     checkInitialConditions()
-    myProject = project
     LensesService.getInstance(project).addListener(this)
   }
 

diff --git a/src/main/kotlin/com/sourcegraph/cody/edit/EditCommandPrompt.kt b/src/main/kotlin/com/sourcegraph/cody/edit/EditCommandPrompt.kt
@@ -279,6 +279,8 @@ class EditCommandPrompt(
     shape = makeCornerShape(width, height)
     updateDialogPosition()
     isVisible = true
+
+    project.putUserData(EDIT_COMMAND_PROMPT_KEY, this)
   }
 
   fun isOkActionEnabled() = okButtonGroup.isEnabled && model != null