added some assertions

sourcegraph · Jun 26, 2024 · c56438e · c56438e
1 parent ae54763
commit c56438e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/integrationTest/kotlin/com/sourcegraph/cody/util/CodyIntegrationTestFixture.kt b/src/integrationTest/kotlin/com/sourcegraph/cody/util/CodyIntegrationTestFixture.kt
@@ -218,6 +218,9 @@ abstract class CodyIntegrationTestFixture : BasePlatformTestCase() {
   protected fun assertInlayIsShown() {
     runInEdtAndWait {
       PlatformTestUtil.dispatchAllEventsInIdeEventQueue()
+      assertNotNull(myFixture)
+      assertNotNull(myFixture.editor)
+      assertNotNull(myFixture.editor.inlayModel)
       assertTrue(
           "Lens group inlay should be displayed", myFixture.editor.inlayModel.hasBlockElements())
     }

diff --git a/src/integrationTest/kotlin/com/sourcegraph/cody/util/DocumentSynchronizationTestFixture.kt b/src/integrationTest/kotlin/com/sourcegraph/cody/util/DocumentSynchronizationTestFixture.kt
@@ -9,8 +9,8 @@ import com.sourcegraph.cody.agent.protocol.ProtocolTextDocument
 import java.util.concurrent.CompletableFuture
 
 /**
- * Lets you specify before/after tests that modify the document and check the Agent's copy.
- * You can specify the starting caret with "^" and optionally the selection with "@".
+ * Lets you specify before/after tests that modify the document and check the Agent's copy. You can
+ * specify the starting caret with "^" and optionally the selection with "@".
  */
 abstract class DocumentSynchronizationTestFixture : CodyIntegrationTestFixture() {