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. 必须命名要注入的函数功能

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 驱动程序注入浏览器的唯一方法。

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

溶液您可以将字符串转换为函数并使用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);
});