selenium-webdriver 将函数数组作为参数传递给 executeScript

Question

import webdriver from 'selenium-webdriver';

const driver = new webdriver.Builder()
              .withCapabilities(webdriver.Capabilities.chrome())
              .build();

driver.get('https://www.google.com');
let foo = function(rules) {
  rules.forEach(rule => {
    rule();
  });
}
let bar = function() { return 'bar' };
let baz = function() { return 'baz' };
driver.executeScript(foo, [bar, baz]).then(function(result) {
  console.log(result);
});

driver.quit();

用

出错

WebDriverError: unknown error: rule is not a function


let foo = function(rules) {
  return rules;
  // rules.forEach(rule => {
  //   rule();
  // });
}
let bar = function() { return 'bar' };
let baz = function() { return 'baz' };
driver.executeScript(foo, [bar, baz]).then(function(result) {
  console.log(result); // refer the log pasted below
});

看起来函数正在被序列化为string

[ 'function bar() {\n  return \'bar\';\n}',
  'function baz() {\n  return \'baz\';\n}' ]

任何关于如何传递array of functions as arguments 的指针都会有所帮助。

Answer 1

我发现了一个不同的解决方法，不使用eval明确，但以类似的方式。

1. 要注入的函数必须是named function

2. 注入function.toString()作为<script>的内容

   function foo(rules) {
     var result = [];
     rules.forEach(rule => {
       result.push(rule());
     });
     return result;
   }
   function bar() { return 'bar' };
   function baz() { return 'baz' };
   
   function inject(content) {
     var script = document.createElement('script');
     script.innerHTML = content;
     document.head.appendChild(script);
   }
   let script = `${bar.toString()} ${baz.toString()} ${foo.toString()}`;
   
   driver.executeScript(inject, script);
   

3. 然后执行所需的函数为

   driver.executeScript('return foo([bar, baz])').then(function(result) {
     // use the result
   });
   

---

完整示例

// example.js
import webdriver from 'selenium-webdriver';

const driver = new webdriver.Builder()
  .withCapabilities(webdriver.Capabilities.chrome())
  .build();

driver.get('https://www.google.com');

function foo(rules) {
  var result = [];
  rules.forEach(rule => {
    result.push(rule());
  });
  return result;
}
function bar() { return 'bar' };
function baz() { return 'baz' };

function inject(content) {
  var script = document.createElement('script');
  script.innerHTML = content;
  document.head.appendChild(script);
}
let script = `${bar.toString()} ${baz.toString()} ${foo.toString()}`;

driver.executeScript(inject, script);

driver.executeScript('return foo([bar, baz])').then(function(result) {
  console.log(result);
});

driver.quit();

---

> babel-node example.js

[ 'bar', 'baz' ]

Answer 2

问题： 所有作为参数传递给函数的重要数据都将转换为字符串，因为这是将某些东西从 selenium 驱动程序注入浏览器的唯一方法。

参数必须是数字、布尔值、字符串、WebElement 或上述任意组合的列表。

解决方案 您可以将字符串转换为函数并使用eval 或函数构造函数new Function('...function body here...') 执行它。是的，这很糟糕，但实际上，没有其他方法可以将非平凡数据从驱动程序传递到浏览器。实际上，当你调用这个driver.executeScript(foo, [], ...) 时，foo 函数也会被转换为字符串并在浏览器中使用eval 执行。

如果我是你，我会尝试寻找另一种方法来实现结果而不将函数作为参数传递。

您可以阅读更多内容：https://selenium.googlecode.com/svn/trunk/docs/api/java/org/openqa/selenium/JavascriptExecutor.html

Answer 3

我认为你可以使用 eval。见以下代码：

driver.get('https://www.google.com');
let foo = function(rules) {
    var results = [];
    rules.forEach(rule => {
        results.push(eval(rule));
    });
    return results;
}

let bar = "(function() { return ' message returned from bar' })()";
let baz = "(function() { return 'message returned from baz' })()";

driver.executeScript(foo, [bar, baz]).then(function(result) {
    console.log(result);
});