feat: add configureTestBed callback method

Author: timdeschryver

projects/testing-library/src/lib/models.ts

@@ -1,5 +1,5 @@
 import { Type, DebugElement } from '@angular/core';
-import { ComponentFixture } from '@angular/core/testing';
+import { ComponentFixture, TestBed } from '@angular/core/testing';
 import { Routes } from '@angular/router';
 import { BoundFunction, Queries, queries, Config as dtlConfig, PrettyDOMOptions } from '@testing-library/dom';
 
@@ -349,6 +349,20 @@ export interface RenderComponentOptions<ComponentType, Q extends Queries = typeo
    * })
    */
   removeAngularAttributes?: boolean;
+
+  /**
+   * @description
+   * Callback to configure the testbed before the compilation.
+   *
+   * @default
+   * () => {}
+   *
+   * @example
+   * const component = await render(AppComponent, {
+   *  configureTestBed: (testBed) => { }
+   * })
+   */
+  configureTestBed?: (testbed: TestBed) => void;
 }
 
 export interface ComponentOverride<T> {

projects/testing-library/src/lib/testing-library.ts

@@ -65,6 +65,9 @@ export async function render<SutType, WrapperType = SutType>(
     removeAngularAttributes = false,
     defaultImports = [],
     initialRoute = '',
+    configureTestBed = () => {
+      /* noop*/
+    },
   } = { ...globalConfig, ...renderOptions };
 
   dtlConfigure({
@@ -94,6 +97,8 @@ export async function render<SutType, WrapperType = SutType>(
   overrideComponentImports(sut, componentImports);
   overrideChildComponentProviders(childComponentOverrides);
 
+  configureTestBed(TestBed);
+
   await TestBed.compileComponents();
 
   componentProviders

projects/testing-library/tests/issues/issue-396-standalone-stub-child.spec.ts

@@ -0,0 +1,54 @@
+import { Component } from '@angular/core';
+import { render, screen } from '../../src/public_api';
+
+test('child', async () => {
+  await render(FixtureComponent);
+  expect(screen.getByText('Hello from child')).toBeInTheDocument();
+});
+
+test('stub', async () => {
+  await render(FixtureComponent, {
+    componentImports: [StubComponent],
+  });
+
+  expect(screen.getByText('Hello from stub')).toBeInTheDocument();
+});
+
+test('configure', async () => {
+  await render(FixtureComponent, {
+    configureTestBed: (testBed) => {
+      testBed.overrideComponent(FixtureComponent, {
+        add: {
+          imports: [StubComponent],
+        },
+        remove: {
+          imports: [ChildComponent],
+        },
+      });
+    },
+  });
+
+  expect(screen.getByText('Hello from stub')).toBeInTheDocument();
+});
+
+@Component({
+  selector: 'atl-child',
+  template: `Hello from child`,
+  standalone: true,
+})
+class ChildComponent {}
+
+@Component({
+  selector: 'atl-child',
+  template: `Hello from stub`,
+  standalone: true,
+})
+class StubComponent {}
+
+@Component({
+  selector: 'atl-fixture',
+  template: `<atl-child />`,
+  standalone: true,
+  imports: [ChildComponent],
+})
+class FixtureComponent {}

projects/testing-library/tests/render.spec.ts

@@ -366,3 +366,14 @@ describe('initialRoute', () => {
     expect(screen.getByText('button')).toBeInTheDocument();
   });
 });
+
+describe('configureTestBed', () => {
+  it('invokes configureTestBed', async () => {
+    const configureTestBedFn = jest.fn();
+    await render(FixtureComponent, {
+      configureTestBed: configureTestBedFn,
+    });
+
+    expect(configureTestBedFn).toHaveBeenCalledTimes(1);
+  });
+});