nightwatch/lib/api/client-commands/setGeolocation.js

Easy to use Node.js based end-to-end testing solution for web applications using the W3C WebDriver API.

const ClientCommand = require('./_base-command.js');
const {Logger} = require('../../utils');

/**
 * Mock the geolocation of the browser. Call without any arguments to reset the geolocation back to original.
 *
 * @example
 *  describe('mock geolocation', function() {
 *    it('sets the geolocation to Tokyo, Japan and then resets it', () => {
 *      browser
 *        .setGeolocation({
 *          latitude: 35.689487,
 *          longitude: 139.691706,
 *          accuracy: 100
 *        })  // sets the geolocation to Tokyo, Japan
 *        .navigateTo('https://www.gps-coordinates.net/my-location')
 *        .pause(3000)
 *        .setGeolocation()  // resets the geolocation
 *        .navigateTo('https://www.gps-coordinates.net/my-location')
 *        .pause(3000);
 *    });
 *  });
 *
 * @method setGeolocation
 * @syntax .setGeolocation({latitude, longitude, accuracy}, [callback])
 * @param {object} [coordinates] Latitude, longitude, and accuracy.
 * @param {function} [callback] Callback function to be called when the command finishes.
 * @api protocol.cdp
 * @since 2.2.0
 * @moreinfo nightwatchjs.org/guide/network-requests/mock-geolocation.html
 */
class SetGeolocation extends ClientCommand {
  static get isTraceable() {
    return true;
  }

  performAction(callback) {
    if (!this.api.isChrome()  && !this.api.isEdge()) {
      const error =  new Error('The command .setGeolocation() is only supported in Chrome and Edge drivers');
      Logger.error(error);

      return callback(error);
    }

    const {latitude, longitude, accuracy} = this.coordinates;

    if (latitude !== undefined && longitude !== undefined) {
      // Both latitude and longitude are provided

      const coordinates = {latitude, longitude, accuracy};
      // Set accuracy as 100 if not provided.
      if (accuracy === undefined) {coordinates.accuracy = 100}

      this.transportActions.setGeolocation(coordinates, callback);

      return;
    }

    if (latitude === undefined && longitude === undefined) {
      // Clear geolocation override.
      this.transportActions.clearGeolocation(callback);

      return;
    }

    // Exactly one of them is undefined, throw error.
    const error =  new Error('Please provide both latitude and longitude while using setGeolocation.');
    Logger.error(error);

    return callback(error);
  }

  command(coordinates, callback) {
    this.coordinates = coordinates || {};

    return super.command(callback);
  }
}

module.exports = SetGeolocation;