expect-webdriverio/lib/matchers/elements/toBeElementsArrayOfSize.js

WebdriverIO Assertion Library

import { waitUntil, enhanceError, compareNumbers, numberError } from '../../utils.js';
import { refetchElements } from '../../util/refetchElements.js';
import { DEFAULT_OPTIONS } from '../../constants.js';
export async function toBeElementsArrayOfSize(received, expectedValue, options = DEFAULT_OPTIONS) {
    const isNot = this.isNot;
    const { expectation = 'elements array of size', verb = 'be' } = this;
    await options.beforeAssertion?.({
        matcherName: 'toBeElementsArrayOfSize',
        expectedValue,
        options,
    });
    let numberOptions;
    if (typeof expectedValue === 'number') {
        numberOptions = { eq: expectedValue };
    }
    else if (!expectedValue || (typeof expectedValue.eq !== 'number' && typeof expectedValue.gte !== 'number' && typeof expectedValue.lte !== 'number')) {
        throw new Error('Invalid params passed to toBeElementsArrayOfSize.');
    }
    else {
        numberOptions = expectedValue;
    }
    let elements = await received;
    const originalLength = elements.length;
    const pass = await waitUntil(async () => {
        const isPassing = compareNumbers(elements.length, numberOptions);
        if (isPassing) {
            return isPassing;
        }
        elements = await refetchElements(elements, numberOptions.wait, true);
        return false;
    }, isNot, { ...numberOptions, ...options });
    if (Array.isArray(received) && pass) {
        for (let index = originalLength; index < elements.length; index++) {
            received.push(elements[index]);
        }
    }
    const error = numberError(numberOptions);
    const message = enhanceError(elements, error, originalLength, this, verb, expectation, '', numberOptions);
    const result = {
        pass,
        message: () => message
    };
    await options.afterAssertion?.({
        matcherName: 'toBeElementsArrayOfSize',
        expectedValue,
        options,
        result
    });
    return result;
}