@applitools/driver 1.4.2 → 1.4.6

package/CHANGELOG.md

@@ -3,6 +3,24 @@
 ## Unreleased
 
 
+## 1.4.6 - 2021/12/22
+
+- extract device orientation in `Driver`'s `init` and provide readonly access through the `orientation` getter of the `Driver`
+- updated to @applitools/snippets@2.1.11 (from 2.1.10)
+- updated to @applitools/types@1.0.23 (from 1.0.22)
+- updated to @applitools/utils@1.2.5 (from 1.2.4)
+
+## 1.4.5 - 2021/12/20
+
+- improve native android app scrolling performance
+- fix bug with `Buffer.from` and base64
+- updated to @applitools/snippets@2.1.10 (from 2.1.9)
+
+## 1.4.3 - 2021/12/17
+
+- convert base64 result of `spec.takeScreenshot` to `Buffer` before return it from `Driver`'s `takeScreenshot` method
+- updated to @applitools/snippets@2.1.9 (from 2.1.8)
+
 ## 1.4.2 - 2021/12/16
 
 - fix exports subpathes

package/dist/debug/check-spec-driver.js

@@ -1,10 +1,7 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.checkSpecDriver = void 0;
-const strict_1 = __importDefault(require("assert/strict"));
+const assert_1 = require("assert");
 const snippets = require('@applitools/snippets');
 async function checkSpecDriver(options) {
     const { spec, driver } = options;
@@ -13,7 +10,7 @@ async function checkSpecDriver(options) {
         'execute script with echo snippet': async () => {
             const arg = [1, '23', false, { a: 4, b: [5] }, null];
             const result = await spec.executeScript(context, 'return arguments[0]', arg);
-            strict_1.default.deepEqual(result, arg, 'script returns an array of json data');
+            assert_1.strict.deepEqual(result, arg, 'script returns an array of json data');
         },
         'execute script with functional script': async () => {
             const arg = { a: 2, b: 1, c: 7 };
@@ -21,23 +18,23 @@ async function checkSpecDriver(options) {
                 return arg;
             }
             const result = await spec.executeScript(context, script, arg);
-            strict_1.default.deepEqual(result, arg, 'script returns an array of json data from functional script');
+            assert_1.strict.deepEqual(result, arg, 'script returns an array of json data from functional script');
         },
         'execute script with return value of dom element': async () => {
             const element = await spec.executeScript(context, 'return document.documentElement');
             const isHtmlElement = await spec.executeScript(context, 'return arguments[0] === document.documentElement', element);
-            strict_1.default.ok(isHtmlElement, 'script returns an element and could be executed with an element');
+            assert_1.strict.ok(isHtmlElement, 'script returns an element and could be executed with an element');
         },
         'execute script with nested element references': async () => {
             const elements = await spec.executeScript(context, 'return [{html: document.documentElement, body: document.body}]');
             const isElements = await spec.executeScript(context, 'return arguments[0][0].html === document.documentElement && arguments[0][0].html === document.documentElement', elements);
-            strict_1.default.ok(isElements, 'script returns elements inside nested structure and could be executed with a nested structure of elements');
+            assert_1.strict.ok(isElements, 'script returns elements inside nested structure and could be executed with a nested structure of elements');
         },
         'find element with string selector': async () => {
             const selector = transformSelector('html>body>h1');
             const element = await spec.findElement(context, selector);
             const isWantedElement = await spec.executeScript(context, `return arguments[0] === document.querySelector("${selector}")`, element);
-            strict_1.default.ok(isWantedElement, `returns element by string selector - "${selector}"`);
+            assert_1.strict.ok(isWantedElement, `returns element by string selector - "${selector}"`);
         },
         'find element with spec selector': async () => {
             const cssSelector = transformSelector({ type: 'css', selector: 'html>body>h1' });
@@ -45,21 +42,21 @@ async function checkSpecDriver(options) {
             const verificationScript = `return arguments[0] === document.querySelector('html>body>h1')`;
             const cssElement = await spec.findElement(context, cssSelector);
             const isCssElement = await spec.executeScript(context, verificationScript, cssElement);
-            strict_1.default.ok(isCssElement, `returns element by spec selector - ${JSON.stringify(cssSelector)}`);
+            assert_1.strict.ok(isCssElement, `returns element by spec selector - ${JSON.stringify(cssSelector)}`);
             const xpathElement = await spec.findElement(context, xpathSelector);
             const isXpathElement = await spec.executeScript(context, verificationScript, xpathElement);
-            strict_1.default.ok(isXpathElement, `returns element by spec selector - ${JSON.stringify(xpathSelector)}`);
+            assert_1.strict.ok(isXpathElement, `returns element by spec selector - ${JSON.stringify(xpathSelector)}`);
         },
         'find element with unresolvable selector': async () => {
             const selector = transformSelector('unresolvable_selector');
             const element = await spec.findElement(context, selector);
-            strict_1.default.equal(element, null, `returns null by unresolvable selector - "${selector}"`);
+            assert_1.strict.equal(element, null, `returns null by unresolvable selector - "${selector}"`);
         },
         'find elements with string selector': async () => {
             const selector = transformSelector('html p');
             const elements = await spec.findElements(context, selector);
             const isExpectedElements = await spec.executeScript(context, `var expected = arguments[0]; return Array.prototype.every.call(document.querySelectorAll("${selector}"), function(element, index) { return element === expected[index] })`, elements);
-            strict_1.default.ok(isExpectedElements, `returns elements by string selector - "${selector}"`);
+            assert_1.strict.ok(isExpectedElements, `returns elements by string selector - "${selector}"`);
         },
         'find elements with spec selector': async () => {
             const cssSelector = transformSelector({ type: 'css', selector: 'html p' });
@@ -67,22 +64,22 @@ async function checkSpecDriver(options) {
             const verificationScript = `var expected = arguments[0]; return Array.prototype.every.call(document.querySelectorAll('html p'), function(element, index) { return element === expected[index] })`;
             const cssElements = await spec.findElements(context, cssSelector);
             const isCssElements = await spec.executeScript(context, verificationScript, cssElements);
-            strict_1.default.ok(isCssElements, `returns elements by spec selector - ${JSON.stringify(cssSelector)}`);
+            assert_1.strict.ok(isCssElements, `returns elements by spec selector - ${JSON.stringify(cssSelector)}`);
             const xpathElements = await spec.findElements(context, xpathSelector);
             const isXpathElements = await spec.executeScript(context, verificationScript, xpathElements);
-            strict_1.default.ok(isXpathElements, `returns element by spec selector - ${JSON.stringify(xpathSelector)}`);
+            assert_1.strict.ok(isXpathElements, `returns element by spec selector - ${JSON.stringify(xpathSelector)}`);
         },
         'find elements with unresolvable selector': async () => {
             const selector = transformSelector('unresolvable_selector');
             const element = await spec.findElements(context, selector);
-            strict_1.default.deepEqual(element, [], `returns empty array by unresolvable selector - "${selector}"`);
+            assert_1.strict.deepEqual(element, [], `returns empty array by unresolvable selector - "${selector}"`);
         },
         'child context': async () => {
             const element = await spec.findElement(context, transformSelector('[name="frame1"]'));
             const childContext = await spec.childContext(context, element);
             const inFrame = await spec.executeScript(childContext, 'return window.frameElement.name === "frame1"');
-            strict_1.default.ok(inFrame, 'returns or switches to a child context');
-            strict_1.default.ok(typeof spec.mainContext === 'function', 'spec.mainContext also needs to be implemented in order to test spec.childContext');
+            assert_1.strict.ok(inFrame, 'returns or switches to a child context');
+            assert_1.strict.ok(typeof spec.mainContext === 'function', 'spec.mainContext also needs to be implemented in order to test spec.childContext');
             await spec.mainContext(context);
         },
         'is equal elements': async () => {
@@ -90,53 +87,53 @@ async function checkSpecDriver(options) {
                 return { skipped: true };
             const htmlEl = await spec.findElement(context, transformSelector('html'));
             const htmlEl2 = await spec.executeScript(context, 'return document.documentElement');
-            strict_1.default.ok(await spec.isEqualElements(context, htmlEl, htmlEl2), 'elements should be equal');
+            assert_1.strict.ok(await spec.isEqualElements(context, htmlEl, htmlEl2), 'elements should be equal');
             const bodyEl = await spec.executeScript(context, 'return document.body');
-            strict_1.default.ok(!(await spec.isEqualElements(context, htmlEl, bodyEl)), 'elements should not be equal');
-            strict_1.default.ok(!(await spec.isEqualElements(context, htmlEl, undefined)), 'isEqualElements should return false if one of the arguments is falsy');
-            strict_1.default.ok(!(await spec.isEqualElements(context, undefined, htmlEl)), 'isEqualElements should return false if one of the arguments is falsy');
+            assert_1.strict.ok(!(await spec.isEqualElements(context, htmlEl, bodyEl)), 'elements should not be equal');
+            assert_1.strict.ok(!(await spec.isEqualElements(context, htmlEl, undefined)), 'isEqualElements should return false if one of the arguments is falsy');
+            assert_1.strict.ok(!(await spec.isEqualElements(context, undefined, htmlEl)), 'isEqualElements should return false if one of the arguments is falsy');
         },
         'main context': async () => {
             const mainDocument1 = await spec.findElement(context, transformSelector('html'));
             const childContext1 = await spec.childContext(context, await spec.findElement(context, transformSelector('[name="frame1"]')));
             const childContext2 = await spec.childContext(childContext1, await spec.findElement(childContext1, transformSelector('[name="frame1-1"]')));
             const frameDocument = await spec.findElement(childContext2, transformSelector('html'));
-            strict_1.default.ok(!(await isEqualElements(childContext2, mainDocument1, frameDocument)));
+            assert_1.strict.ok(!(await isEqualElements(childContext2, mainDocument1, frameDocument)));
             const mainContext = await spec.mainContext(childContext2);
             const mainDocument2 = await spec.findElement(mainContext, transformSelector('html'));
-            strict_1.default.ok(await isEqualElements(mainContext, mainDocument2, mainDocument1));
+            assert_1.strict.ok(await isEqualElements(mainContext, mainDocument2, mainDocument1));
         },
         'parent context': async () => {
             const parentContext1 = await spec.childContext(context, await spec.findElement(context, transformSelector('[name="frame1"]')));
             const parentDocument1 = await spec.findElement(parentContext1, transformSelector('html'));
             const frameContext = await spec.childContext(parentContext1, await spec.findElement(parentContext1, transformSelector('[name="frame1-1"]')));
             const frameDocument = await spec.findElement(frameContext, transformSelector('html'));
-            strict_1.default.ok(!(await isEqualElements(frameContext, parentDocument1, frameDocument)));
+            assert_1.strict.ok(!(await isEqualElements(frameContext, parentDocument1, frameDocument)));
             const parentContext2 = await spec.parentContext(frameContext);
             const parentDocument2 = await spec.findElement(parentContext2, transformSelector('html'));
-            strict_1.default.ok(await isEqualElements(parentContext2, parentDocument2, parentDocument1));
+            assert_1.strict.ok(await isEqualElements(parentContext2, parentDocument2, parentDocument1));
             await spec.mainContext(context);
         },
         'get title': async () => {
             const title = await spec.getTitle(driver);
-            strict_1.default.equal(title, 'Cross SDK test', 'returns title of the current page');
+            assert_1.strict.equal(title, 'Cross SDK test', 'returns title of the current page');
         },
         'get url': async () => {
             const url = await spec.getUrl(driver);
-            strict_1.default.equal(url, 'https://applitools.github.io/demo/TestPages/FramesTestPage/', 'returns url of the current page');
+            assert_1.strict.equal(url, 'https://applitools.github.io/demo/TestPages/FramesTestPage/', 'returns url of the current page');
         },
         'is driver': async () => {
-            strict_1.default.ok(await spec.isDriver(driver), 'driver should be considered a driver :)');
-            strict_1.default.ok(!(await spec.isDriver(undefined)), 'undefined should not be considered a driver');
-            strict_1.default.ok(!(await spec.isDriver(3)), 'number should not be considered a driver');
-            strict_1.default.ok(!(await spec.isDriver('str')), 'string should not be considered a driver');
+            assert_1.strict.ok(await spec.isDriver(driver), 'driver should be considered a driver :)');
+            assert_1.strict.ok(!(await spec.isDriver(undefined)), 'undefined should not be considered a driver');
+            assert_1.strict.ok(!(await spec.isDriver(3)), 'number should not be considered a driver');
+            assert_1.strict.ok(!(await spec.isDriver('str')), 'string should not be considered a driver');
         },
         'is element': async () => {
             const el = await spec.findElement(context, transformSelector('html'));
-            strict_1.default.ok(await spec.isElement(el), 'element should be considered an element :)');
-            strict_1.default.ok(!(await spec.isElement(undefined)), 'undefined should not be considered an element');
-            strict_1.default.ok(!(await spec.isElement(3)), 'number should not be considered an element');
-            strict_1.default.ok(!(await spec.isElement('str')), 'str should not be considered an element');
+            assert_1.strict.ok(await spec.isElement(el), 'element should be considered an element :)');
+            assert_1.strict.ok(!(await spec.isElement(undefined)), 'undefined should not be considered an element');
+            assert_1.strict.ok(!(await spec.isElement(3)), 'number should not be considered an element');
+            assert_1.strict.ok(!(await spec.isElement('str')), 'str should not be considered an element');
         },
         // 'is selector': async () => {}, // hard to test this
         // 'set window size': async () => {}, // hard to test this

package/dist/driver.js

@@ -92,6 +92,9 @@ class Driver {
         var _a;
         return (_a = this._driverInfo) === null || _a === void 0 ? void 0 : _a.userAgent;
     }
+    get orientation() {
+        return this._driverInfo.orientation;
+    }
     get pixelRatio() {
         var _a;
         return (_a = this._driverInfo.pixelRatio) !== null && _a !== void 0 ? _a : 1;
@@ -131,31 +134,31 @@ class Driver {
         this._currentContext = context;
     }
     async init() {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
-        var _s, _t, _u, _v, _w, _x;
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s;
+        var _t, _u, _v, _w, _x, _y, _z;
         const capabilities = await ((_b = (_a = this._spec).getCapabilities) === null || _b === void 0 ? void 0 : _b.call(_a, this.target));
         this._logger.log('Driver capabilities', capabilities);
         const capabilitiesInfo = capabilities ? capabilities_1.parseCapabilities(capabilities) : undefined;
         const driverInfo = await ((_d = (_c = this._spec).getDriverInfo) === null || _d === void 0 ? void 0 : _d.call(_c, this.target));
         this._driverInfo = Object.assign(Object.assign({}, capabilitiesInfo), driverInfo);
         if (this.isWeb) {
-            (_e = (_s = this._driverInfo).pixelRatio) !== null && _e !== void 0 ? _e : (_s.pixelRatio = await this.execute(snippets.getPixelRatio));
-            (_f = (_t = this._driverInfo).userAgent) !== null && _f !== void 0 ? _f : (_t.userAgent = await this.execute(snippets.getUserAgent));
+            (_e = (_t = this._driverInfo).pixelRatio) !== null && _e !== void 0 ? _e : (_t.pixelRatio = await this.execute(snippets.getPixelRatio));
+            (_f = (_u = this._driverInfo).userAgent) !== null && _f !== void 0 ? _f : (_u.userAgent = await this.execute(snippets.getUserAgent));
             if (this._driverInfo.userAgent) {
                 const userAgentInfo = user_agent_1.parseUserAgent(this._driverInfo.userAgent);
                 this._driverInfo.browserName = (_g = userAgentInfo.browserName) !== null && _g !== void 0 ? _g : this._driverInfo.browserName;
                 this._driverInfo.browserVersion = (_h = userAgentInfo.browserVersion) !== null && _h !== void 0 ? _h : this._driverInfo.browserVersion;
                 if (this._driverInfo.isMobile) {
-                    (_j = (_u = this._driverInfo).platformName) !== null && _j !== void 0 ? _j : (_u.platformName = userAgentInfo.platformName);
-                    (_k = (_v = this._driverInfo).platformVersion) !== null && _k !== void 0 ? _k : (_v.platformVersion = userAgentInfo.platformVersion);
+                    (_j = (_v = this._driverInfo).platformName) !== null && _j !== void 0 ? _j : (_v.platformName = userAgentInfo.platformName);
+                    (_k = (_w = this._driverInfo).platformVersion) !== null && _k !== void 0 ? _k : (_w.platformVersion = userAgentInfo.platformVersion);
                 }
                 else {
                     this._driverInfo.platformName = (_l = userAgentInfo.platformName) !== null && _l !== void 0 ? _l : this._driverInfo.platformName;
                     this._driverInfo.platformVersion = (_m = userAgentInfo.platformVersion) !== null && _m !== void 0 ? _m : this._driverInfo.platformVersion;
                 }
             }
-            (_o = (_w = this._driverInfo).features) !== null && _o !== void 0 ? _o : (_w.features = {});
-            (_p = (_x = this._driverInfo.features).allCookies) !== null && _p !== void 0 ? _p : (_x.allCookies = /chrome/i.test(this._driverInfo.browserName) && !this._driverInfo.isMobile);
+            (_o = (_x = this._driverInfo).features) !== null && _o !== void 0 ? _o : (_x.features = {});
+            (_p = (_y = this._driverInfo.features).allCookies) !== null && _p !== void 0 ? _p : (_y.allCookies = /chrome/i.test(this._driverInfo.browserName) && !this._driverInfo.isMobile);
         }
         else {
             const barsHeight = await ((_r = (_q = this._spec).getBarsHeight) === null || _r === void 0 ? void 0 : _r.call(_q, this.target).catch(() => undefined));
@@ -201,6 +204,9 @@ class Driver {
                 ? await helper_ios_1.HelperIOS.make({ spec: this._spec, driver: this, logger: this._logger })
                 : await helper_android_1.HelperAndroid.make({ spec: this._spec, driver: this, logger: this._logger });
         }
+        if (this.isMobile) {
+            (_s = (_z = this._driverInfo).orientation) !== null && _s !== void 0 ? _s : (_z.orientation = await this.getOrientation().catch(() => undefined));
+        }
         this._logger.log('Combined driver info', this._driverInfo);
         return this;
     }
@@ -390,8 +396,11 @@ class Driver {
         return this.currentContext.execute(script, arg);
     }
     async takeScreenshot() {
-        const data = await this._spec.takeScreenshot(this.target);
-        return utils.types.isString(data) ? data.replace(/[\r\n]+/g, '') : data;
+        const image = await this._spec.takeScreenshot(this.target);
+        if (utils.types.isString(image)) {
+            return Buffer.from(image.replace(/[\r\n]+/g, ''), 'base64');
+        }
+        return image;
     }
     async getViewportSize() {
         var _a;
@@ -459,13 +468,13 @@ class Driver {
         throw new Error('Failed to set viewport size!');
     }
     async getDisplaySize() {
-        if (this.isWeb)
+        if (this.isWeb && !this.isMobile)
             return;
         const size = await this._spec.getWindowSize(this.target);
         return this.isAndroid ? utils.geometry.scale(size, 1 / this.pixelRatio) : size;
     }
     async getOrientation() {
-        if (this.isWeb)
+        if (this.isWeb && !this.isMobile)
             return;
         const orientation = this._spec.getOrientation(this.target);
         this._logger.log('Extracted device orientation:', orientation);

package/dist/element.js

@@ -289,6 +289,7 @@ class Element {
                 if (this.driver.isAndroid) {
                     remainingOffset = utils.geometry.scale(remainingOffset, this.driver.pixelRatio);
                 }
+                const actions = [];
                 const touchPadding = await this.getTouchPadding();
                 const xPadding = Math.max(Math.floor(effectiveRegion.width * 0.1), touchPadding);
                 const yTrack = Math.floor(effectiveRegion.y + effectiveRegion.height / 2); // center
@@ -299,7 +300,7 @@ class Element {
                 while (xRemaining > 0) {
                     const xRight = effectiveRegion.x + Math.min(xRemaining + xPadding, effectiveRegion.width - xPadding);
                     const [xStart, xEnd] = xDirection === 'right' ? [xRight, xLeft] : [xLeft, xRight];
-                    await this._spec.performAction(this.driver.target, [
+                    actions.push([
                         { action: 'press', y: yTrack, x: xStart },
                         { action: 'wait', ms: 100 },
                         { action: 'moveTo', y: yTrack, x: xStart + xGap },
@@ -320,7 +321,7 @@ class Element {
                 while (yRemaining > 0) {
                     const yTop = Math.max(yBottom - yRemaining, effectiveRegion.y + yPadding);
                     const [yStart, yEnd] = yDirection === 'down' ? [yBottom, yTop] : [yTop, yBottom];
-                    await this._spec.performAction(this.driver.target, [
+                    actions.push([
                         { action: 'press', x: xTrack, y: yStart },
                         { action: 'wait', ms: 100 },
                         { action: 'moveTo', x: xTrack, y: yStart + yGap },
@@ -332,6 +333,15 @@ class Element {
                     ]);
                     yRemaining -= yBottom - yTop;
                 }
+                // ios actions should be executed one-by-one sequentially, otherwise the result isn't stable
+                if (this.driver.isIOS) {
+                    for (const action of actions) {
+                        await this._spec.performAction(this.driver.target, action);
+                    }
+                }
+                else {
+                    await this._spec.performAction(this.driver.target, [].concat(...actions));
+                }
                 const actualScrollableRegion = await this.getClientRegion();
                 this._state.scrollOffset = utils.geometry.offsetNegative(requiredOffset, {
                     x: scrollableRegion.x - actualScrollableRegion.x,

package/dist/fake/spec-driver.js

@@ -19,7 +19,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.visit = exports.getTitle = exports.getUrl = exports.setWindowSize = exports.getWindowSize = exports.getDriverInfo = exports.takeScreenshot = exports.childContext = exports.parentContext = exports.mainContext = exports.findElements = exports.findElement = exports.executeScript = exports.isEqualElements = exports.isStaleElementError = exports.transformSelector = exports.isSelector = exports.isElement = exports.isDriver = void 0;
+exports.visit = exports.getTitle = exports.getUrl = exports.getOrientation = exports.setWindowSize = exports.getWindowSize = exports.getDriverInfo = exports.takeScreenshot = exports.childContext = exports.parentContext = exports.mainContext = exports.findElements = exports.findElement = exports.executeScript = exports.isEqualElements = exports.isStaleElementError = exports.transformSelector = exports.isSelector = exports.isElement = exports.isDriver = void 0;
 const utils = __importStar(require("@applitools/utils"));
 function isDriver(driver) {
     return driver && driver.constructor.name === 'MockDriver';
@@ -86,6 +86,10 @@ async function setWindowSize(driver, size) {
     await driver.setWindowRect(size);
 }
 exports.setWindowSize = setWindowSize;
+async function getOrientation(_driver) {
+    return 'portrait';
+}
+exports.getOrientation = getOrientation;
 async function getUrl(driver) {
     if (this._isNative)
         return null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@applitools/driver",
-  "version": "1.4.2",
+  "version": "1.4.6",
   "description": "Applitools universal framework wrapper",
   "keywords": [
     "applitools",
@@ -73,9 +73,9 @@
     }
   },
   "dependencies": {
-    "@applitools/snippets": "2.1.8",
-    "@applitools/types": "1.0.22",
-    "@applitools/utils": "1.2.4"
+    "@applitools/snippets": "2.1.11",
+    "@applitools/types": "1.0.23",
+    "@applitools/utils": "1.2.5"
   },
   "devDependencies": {
     "@applitools/sdk-release-kit": "0.13.4",

package/types/driver.d.ts

@@ -32,6 +32,7 @@ export declare class Driver<TDriver, TContext, TElement, TSelector> {
     get browserName(): string;
     get browserVersion(): string | number;
     get userAgent(): string;
+    get orientation(): 'portrait' | 'landscape';
     get pixelRatio(): number;
     get statusBarHeight(): number;
     get navigationBarHeight(): number;
@@ -54,7 +55,7 @@ export declare class Driver<TDriver, TContext, TElement, TSelector> {
     element(selector: types.Selector<TSelector>): Promise<Element<TDriver, TContext, TElement, TSelector>>;
     elements(selector: types.Selector<TSelector>): Promise<Element<TDriver, TContext, TElement, TSelector>[]>;
     execute(script: ((arg: any) => any) | string, arg?: any): Promise<any>;
-    takeScreenshot(): Promise<Buffer | string>;
+    takeScreenshot(): Promise<Buffer>;
     getViewportSize(): Promise<types.Size>;
     setViewportSize(size: types.Size): Promise<void>;
     getDisplaySize(): Promise<types.Size>;

package/types/fake/spec-driver.d.ts

@@ -24,6 +24,7 @@ export declare function takeScreenshot(driver: Driver): Promise<Buffer | string>
 export declare function getDriverInfo(driver: Driver): Promise<DriverInfo>;
 export declare function getWindowSize(driver: Driver): Promise<Size>;
 export declare function setWindowSize(driver: Driver, size: Size): Promise<void>;
+export declare function getOrientation(_driver: Driver): Promise<'portrait' | 'landscape'>;
 export declare function getUrl(driver: Driver): Promise<string>;
 export declare function getTitle(driver: Driver): Promise<string>;
 export declare function visit(driver: Driver, url: string): Promise<void>;