mediawiki-extensions-VisualEditor/build/screenshots.js

const accessKey = process.env.SAUCE_ONDEMAND_ACCESS_KEY,
	Builder = require( 'selenium-webdriver' ).Builder,
	fs = require( 'fs' ),
	Jimp = require( 'jimp' ),
	username = process.env.SAUCE_ONDEMAND_USERNAME,
	webdriver = require( 'selenium-webdriver' ),
	TIMEOUT = 40 * 1000;

function createScreenshotEnvironment( test, beforeEach ) {
	let clientSize, driver;

	test.beforeEach( function () {
		const lang = this.currentTest.parent.lang || 'en';

		// Use Sauce Labs when running on Jenins
		if ( process.env.JENKINS_URL ) {
			driver = new webdriver.Builder().withCapabilities( {
				browserName: process.env.BROWSER,
				platform: process.env.PLATFORM,
				username: username,
				accessKey: accessKey
			} ).usingServer( 'http://' + username + ':' + accessKey +
				'@ondemand.saucelabs.com:80/wd/hub' ).build();
		} else {
			// If not running on Jenkins, use local browser
			driver = new Builder().forBrowser( 'chrome' ).build();
		}

		driver.manage().timeouts().setScriptTimeout( TIMEOUT );
		driver.manage().window().setSize( 1200, 1000 );

		driver.get( 'https://en.wikipedia.org/wiki/Help:Sample_page?veaction=edit&uselang=' + lang );
		driver.wait(
			driver.executeAsyncScript(
				require( './screenshots-client/utils.js' )
			).then( function ( cs ) {
				clientSize = cs;
			}, function ( e ) {
				// Log error (timeout)
				console.error( e.message );
			} )
		);
		if ( beforeEach ) {
			driver.manage().timeouts().setScriptTimeout( TIMEOUT );
			driver.wait(
				driver.executeAsyncScript( beforeEach ).then(
					function () {},
					function ( e ) {
						// Log error (timeout)
						console.error( e.message );
					}
				)
			);
		}
	} );

	test.afterEach( function () {
		driver.quit();
	} );

	function cropScreenshot( filename, imageBuffer, rect, padding ) {
		if ( padding === undefined ) {
			padding = 5;
		}

		const left = Math.max( 0, rect.left - padding );
		const top = Math.max( 0, rect.top - padding );
		const right = Math.min( clientSize.width, rect.left + rect.width + padding );
		const bottom = Math.min( clientSize.height, rect.top + rect.height + padding );

		return Jimp.read( imageBuffer ).then( function ( jimpImage ) {
			jimpImage
				.crop( left, top, right - left, bottom - top )
				.write( filename );
		} );
	}

	function runScreenshotTest( name, lang, clientScript, padding ) {
		const filename = './screenshots/' + name + '-' + lang + '.png';

		driver.manage().timeouts().setScriptTimeout( TIMEOUT );
		driver.wait(
			driver.executeAsyncScript( clientScript ).then( function ( rect ) {
				return driver.takeScreenshot().then( function ( base64Image ) {
					if ( rect ) {
						const imageBuffer = Buffer.from( base64Image, 'base64' );
						return cropScreenshot( filename, imageBuffer, rect, padding );
					} else {
						fs.writeFile( filename, base64Image, 'base64' );
					}
				} );
			}, function ( e ) {
				// Log error (timeout)
				console.error( e );
			} )
		);
	}

	return runScreenshotTest;
}

module.exports.createScreenshotEnvironment = createScreenshotEnvironment;