diff --git a/driver/lib/commands/execute.ts b/driver/lib/commands/execute.ts index d290163c..44acf8ae 100644 --- a/driver/lib/commands/execute.ts +++ b/driver/lib/commands/execute.ts @@ -162,6 +162,11 @@ const commandHandlers: CommandMap = { command: 'getTextWithCommandExtension', findBy: params.findBy, }), + customCommandExtension: async (driver, params) => + await driver.socket.executeSocketCommand({ + command: params.command, + ...params.args, + }), }; export const execute = async function ( this: FlutterDriver,