sizeOfElementToBe.ts

import type { StringOrElement } from './../utils/element.types';
import { getElement } from './../utils';

/**
 * A condition for checking size of element with given selector
 *
 * @example
 * browser.waitUntil(sizeOfElementsToBe('button', { width: 200, height: 200 }));
 *
 * @param {!string | ChainablePromiseElement<Element<'async'>>} selectorOrElement The selector or element to check
 * @param {!{ width: number, height: number }} expectedSize The selector to check
 *
 * @returns {!function} An expected condition that returns a promise
 *     representing whether the element size.
 */

export function sizeOfElementsToBe(
  selectorOrElement: StringOrElement,
  expectedSize: { width: number; height: number },
): () => Promise<boolean> {
  return async function (): Promise<boolean> {
    const element = await getElement(selectorOrElement);
    const actualSize = await element.getSize();

    return actualSize.width === expectedSize.width && actualSize.height === expectedSize.height;
  };
}