// (C) Copyright 2015 Moodle Pty Ltd. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. import { Component, CUSTOM_ELEMENTS_SCHEMA, Type, ViewChild } from '@angular/core'; import { ComponentFixture, TestBed } from '@angular/core/testing'; import { CoreSingletonProxy } from '@singletons'; import { CoreTextUtilsProvider } from '@services/utils/text'; abstract class WrapperComponent { child!: U; }; export interface RenderConfig { declarations: unknown[]; providers: unknown[]; } export type WrapperComponentFixture = ComponentFixture>; const textUtils = new CoreTextUtilsProvider(); export function mock(instance?: Record): T; export function mock(methods: string[], instance?: Record): T; export function mock( methodsOrInstance: string[] | Record = [], instance: Record = {}, ): T { instance = Array.isArray(methodsOrInstance) ? instance : methodsOrInstance; const methods = Array.isArray(methodsOrInstance) ? methodsOrInstance : []; for (const property of Object.getOwnPropertyNames(instance)) { const value = instance[property]; if (typeof value !== 'function') { continue; } instance[property] = jest.fn((...args) => value.call(instance, ...args)); } for (const method of methods) { instance[method] = jest.fn(); } return instance as T; } export function mockSingleton(singletonClass: CoreSingletonProxy, instance: T): T; export function mockSingleton(singletonClass: CoreSingletonProxy, instance?: Record): T; export function mockSingleton( singletonClass: CoreSingletonProxy, methods: string[], instance?: Record, ): T; export function mockSingleton( singleton: CoreSingletonProxy, methodsOrInstance: string[] | Record = [], instance: Record = {}, ): T { instance = Array.isArray(methodsOrInstance) ? instance : methodsOrInstance; const methods = Array.isArray(methodsOrInstance) ? methodsOrInstance : []; const mockInstance = mock(methods, instance); singleton.setInstance(mockInstance); return mockInstance; } export async function renderComponent(component: Type, config: Partial = {}): Promise> { return renderAngularComponent(component, { declarations: [], providers: [], ...config, }); } export async function renderTemplate( component: Type, template: string, config: Partial = {}, ): Promise> { config.declarations = config.declarations ?? []; config.declarations.push(component); return renderAngularComponent( createWrapperComponent(template, component), { declarations: [], providers: [], ...config, }, ); } export async function renderWrapperComponent( component: Type, tag: string, inputs: Record = {}, config: Partial = {}, ): Promise> { const inputAttributes = Object .entries(inputs) .map(([name, value]) => `[${name}]="${textUtils.escapeHTML(JSON.stringify(value)).replace(/\//g, '\\/')}"`) .join(' '); return renderTemplate(component, `<${tag} ${inputAttributes}>`, config); } async function renderAngularComponent(component: Type, config: RenderConfig): Promise> { config.declarations.push(component); TestBed.configureTestingModule({ declarations: config.declarations, schemas: [CUSTOM_ELEMENTS_SCHEMA], providers: config.providers, }); await TestBed.compileComponents(); const fixture = TestBed.createComponent(component); fixture.autoDetectChanges(true); await fixture.whenRenderingDone(); await fixture.whenStable(); return fixture; } function createWrapperComponent(template: string, componentClass: Type): Type> { @Component({ template }) class HostComponent extends WrapperComponent { @ViewChild(componentClass) child!: U; } return HostComponent; }