ow/source/lib/predicates/weak-map.ts

import {Predicate, Context} from './predicate';

export class WeakMapPredicate extends Predicate<WeakMap<any, any>> {
	constructor(context?: Context) {
		super('weakMap', context);
	}

	/**
	 * Test a WeakMap to include all the provided keys. The keys are tested by identity, not structure.
	 *
	 * @param keys The keys that should be a key in the WeakMap.
	 */
	hasKeys(...keys: any[]) {
		const missingKeys: any[] = [];

		return this.addValidator({
			message: () => `Expected WeakMap to have keys \`${JSON.stringify(missingKeys)}\``,
			validator: map => {
				for (const key of keys) {
					if (map.has(key)) {
						continue;
					}

					missingKeys.push(key);

					if (missingKeys.length === 5) {
						return false;
					}
				}

				return missingKeys.length === 0;
			}
		});
	}

	/**
	 * Test a WeakMap to include any of the provided keys. The keys are tested by identity, not structure.
	 *
	 * @param keys The keys that could be a key in the WeakMap.
	 */
	hasAnyKeys(...keys: any[]) {
		return this.addValidator({
			message: () => `Expected WeakMap to have any key of \`${JSON.stringify(keys)}\``,
			validator: map => keys.some(key => map.has(key))
		});
	}
}