[remote-driver] fix JCefUI

GitOrigin-RevId: 78f642297735a0625bee4341698d0340524b6247

platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/JCefUI.kt

@@ -44,6 +44,7 @@ fun Finder.jcef(@Language("xpath") xpath: String? = null, action: JCefUI.() -> U
 class JCefUI(data: ComponentData) : UiComponent(data) {
   private val jcefWorker by lazy {
     driver.new(JcefComponentWrapper::class, component).apply {
+      waitFor("Document exists") { hasDocument() }
       runJs(initScript)
     }
   }
@@ -94,7 +95,6 @@ class JCefUI(data: ComponentData) : UiComponent(data) {
   fun getUrl(): String =  jcefWorker.getUrl()
 
   fun getHtml(): String {
-    waitFor("Document exists") { hasDocument() }
     return callJs("""document.documentElement.outerHTML""")
   }
 

plugins/performanceTesting/remote-driver/src/com/jetbrains/performancePlugin/remotedriver/jcef/JcefComponentWrapper.kt

@@ -49,10 +49,10 @@ internal class JcefComponentWrapper(private val component: Component) {
       jbCefBrowser = currentComponent.getClientProperty(JB_BROWSER_KEY)
       currentComponent = currentComponent.parent as JComponent
     }
-    require(jbCefBrowser != null) {
+    check(jbCefBrowser != null) {
       "Failed to retrieve jbCefBrowser from $component"
     }
-    require(jbCefBrowser is JBCefBrowserBase) {
+    check(jbCefBrowser is JBCefBrowserBase) {
       "$jbCefBrowser is not JBCefBrowser"
     }
     return jbCefBrowser