fixup: reloadUntilTestExtensionLoaded

Author: jsjoeio

test/e2e/models/CodeServer.ts

@@ -302,7 +302,7 @@ export class CodeServerPage {
         this.codeServer.logger.debug(`test extension loaded after ${reloadCount} reloads`)
         break
       }
-      await this.page.reload()
+      await this.reloadUntilEditorIsReady()
     }
 
     this.codeServer.logger.debug("Test extenion has loaded!")
@@ -328,10 +328,9 @@ export class CodeServerPage {
   async isTestExtensionLoaded() {
     const selector = "text=test extension loaded"
     this.codeServer.logger.debug("Waiting for test extension to load...")
-    await this.page.waitForSelector(selector, {
-      timeout: 3000,
-    })
+    // await this.page.waitForSelector(selector)
 
+    await this.page.waitForLoadState("load")
     const loaded = await this.page.isVisible(selector)
 
     this.codeServer.logger.debug(`Test extension has ${loaded ? "" : "not"} loaded`)