Selenium异步脚本在自己的线程中阻塞其他脚本



我有这样的场景:

不同的脚本必须在浏览器中执行。其中之一是从一个浏览器向另一个浏览器发送消息(WebRTC)。我想测量每个操作的延迟,特别是发送消息。

为了实现这一点,我为每个selenium驱动程序创建了一个自己的线程,该线程正在执行脚本。此外,我还创建了另一个线程,该线程正在等待来自其他浏览器的消息到达观察到的浏览器。等待消息的线程执行如下异步脚本:

String message = (String) ((JavascriptExecutor) driver).executeAsyncScript(
 "window.receivedMessage = function(message_id){" +
                                  "arguments[arguments.length - 1](message_id);" +
                           "}");

如您所见,如果浏览器收到消息,我正在等待调用的回调。

不幸的是,在这个异步脚本在浏览器中注册后,这会阻止我想要执行的所有其他脚本(无论是.executeScript()还是.executeSyncScript())。

有人知道如何不断等待消息到达并使用网络驱动程序执行其他脚本吗?

编辑

顺便说一句:忙于等待不应该是解决办法。我更喜欢一种方法,以某种方式通知selenium操作已经完成,而不需要不断轮询驱动程序的结果。

一种方法是将消息存储在全局变量中,并在执行其他脚本后进行处理。这里有一个例子:

JavascriptExecutor js = (JavascriptExecutor)driver;
driver.manage().timeouts().setScriptTimeout(20, TimeUnit.SECONDS);
// set a listener for the messages
js.executeScript(
    "window._asyncResult = [];                        " +
    "window.receivedMessage = function(message_id) {  " +
    "  window._asyncResult.push(message_id);          " +
    "};                                               " );
// send a message asynchroniously every second
js.executeScript(
    "window._asyncTimer = setInterval(function(){     " +
    "  window.receivedMessage(Math.random());         " +
    "}, 1000);                                        " );
// wait for 5 messages and return the list
List messages = (ArrayList)js.executeAsyncScript(
    "var callback = arguments[arguments.length - 1];  " +
    "(function fn(){                                  " +
    "  if(window._asyncResult.length > 5) {           " +
    "    clearInterval(window._asyncTimer);           " +
    "    return callback(window._asyncResult);        " +
    "  }                                              " +
    "  setTimeout(fn, 30);                            " +
    "})();                                            " );

最新更新