talk/test/e2e/utils/SortedWindowHandler.js

/**
 * SortedWindowHandler assists in making e2e tests more robust by returning
 * deterministic window handles. An instance must be created before new windows
 * are created and windowHandles must be called each time a window was created or
 * closed.
 */
class SortedWindowHandler {

  /**
   * Constructor, must be called before new windows were created.
   */
  constructor(client) {
    this.client = client;
    this.client.windowHandles((result) => {
      this.handles = result.value;
      if (this.handles.length > 2) {
        throw new Error('SortedWindowHandler must be created before new windows were created.');
      }
    });
  }

  /**
   * windowHandles will call given `callback` with an array of window handles.
   */
  windowHandles(callback) {
    this.client.windowHandles((result) => {

      if (result.value.message) {
        throw new Error(result.value.message);
      }

      this.handles = this.handles.filter((handle) => result.value.includes(handle));
      const remaining = result.value.filter((handle) => !this.handles.includes(handle));

      if (remaining.length === 1) {
        this.handles.push(remaining[0]);
      }
      if (remaining.length > 1) {
        throw new Error('Cannot detect new window handle, because more than one windows was created.');
      }
      callback(this.handles);
    });
  }

  /**
   *  Switch to main window handle and remove latest handle.
   */
  pop() {
    this.client.switchWindow(this.handles[0]);
    this.handles.splice(-1, 1);
  }
}

module.exports = SortedWindowHandler;