n8n/packages/cli/test/integration/shared/license.ts

import type { BooleanLicenseFeature, NumericLicenseFeature } from '@/Interfaces';
import type { License } from '@/license';

export interface LicenseMockDefaults {
	features?: BooleanLicenseFeature[];
	quotas?: Partial<{ [K in NumericLicenseFeature]: number }>;
}

export class LicenseMocker {
	private _enabledFeatures: Set<BooleanLicenseFeature> = new Set();

	private _defaultFeatures: Set<BooleanLicenseFeature> = new Set();

	private _featureQuotas: Map<NumericLicenseFeature, number> = new Map();

	private _defaultQuotas: Map<NumericLicenseFeature, number> = new Map();

	mock(license: License) {
		license.isFeatureEnabled = this.isFeatureEnabled.bind(this);
		license.getFeatureValue = this.getFeatureValue.bind(this);
	}

	reset() {
		this._enabledFeatures = new Set(this._defaultFeatures);
		this._featureQuotas = new Map(this._defaultQuotas);
	}

	setDefaults(defaults: LicenseMockDefaults) {
		this._defaultFeatures = new Set(defaults.features ?? []);
		this._defaultQuotas = new Map(
			Object.entries(defaults.quotas ?? {}) as Array<[NumericLicenseFeature, number]>,
		);
	}

	isFeatureEnabled(feature: BooleanLicenseFeature): boolean {
		return this._enabledFeatures.has(feature);
	}

	getFeatureValue(feature: string): boolean | number | undefined {
		if (this._featureQuotas.has(feature as NumericLicenseFeature)) {
			return this._featureQuotas.get(feature as NumericLicenseFeature);
		} else if (this._enabledFeatures.has(feature as BooleanLicenseFeature)) {
			return true;
		}
		return undefined;
	}

	enable(feature: BooleanLicenseFeature) {
		this._enabledFeatures.add(feature);
	}

	disable(feature: BooleanLicenseFeature) {
		this._enabledFeatures.delete(feature);
	}

	setQuota(feature: NumericLicenseFeature, quota: number) {
		this._featureQuotas.set(feature, quota);
	}
}