import missing functions from vitest

Author: timdeschryver

projects/testing-library/src/tests/bindings.spec.ts

@@ -1,5 +1,5 @@
 import { Component, input, output, inputBinding, outputBinding, twoWayBinding, signal, model } from '@angular/core';
-import { vi, describe, test } from 'vitest';
+import { vi, describe, test, expect } from 'vitest';
 import { render, screen, aliasedInput } from '../public_api';
 
 describe('Bindings API Support', () => {

projects/testing-library/src/tests/config.spec.ts

@@ -1,6 +1,6 @@
 import { Component, InjectionToken, NgModule } from '@angular/core';
 import { TestBed } from '@angular/core/testing';
-import { test } from 'vitest';
+import { test, afterEach, beforeEach } from 'vitest';
 import { render, configure, Config } from '../public_api';
 
 const TEST_TOKEN = new InjectionToken<string>('TEST_TOKEN');
@@ -23,21 +23,16 @@ beforeEach(() => {
   // it at the end of the test
   configure((existingConfig) => {
     originalConfig = existingConfig as Config;
-    // Don't change the existing config
-    return {};
+    return {
+      defaultImports: [TestModule],
+    };
   });
 });
 
 afterEach(() => {
   configure(originalConfig);
 });
 
-beforeEach(() => {
-  configure({
-    defaultImports: [TestModule],
-  });
-});
-
 test('adds default imports to the testbed', async () => {
   await render(TestComponent);
 

projects/testing-library/src/tests/integration.spec.ts

@@ -1,6 +1,6 @@
 import { Component, EventEmitter, inject, Injectable, Input, Output } from '@angular/core';
 import { TestBed } from '@angular/core/testing';
-import { vi, test, expect } from 'vitest';
+import { vi, test, expect, afterEach } from 'vitest';
 import { of, BehaviorSubject } from 'rxjs';
 import { debounceTime, switchMap, map, startWith } from 'rxjs/operators';
 import { render, screen, waitFor, waitForElementToBeRemoved, within } from '../lib/testing-library';
@@ -92,6 +92,10 @@ const entities = [
   },
 ];
 
+afterEach(() => {
+  vi.useRealTimers();
+});
+
 async function setup() {
   vi.useFakeTimers();
   const user = userEvent.setup();

projects/testing-library/src/tests/issues/issue-318.spec.ts

@@ -1,7 +1,7 @@
 import { Component, inject, OnDestroy, OnInit } from '@angular/core';
 import { Router } from '@angular/router';
 import { RouterTestingModule } from '@angular/router/testing';
-import { vi, test } from 'vitest';
+import { vi, test, expect } from 'vitest';
 import { Subject, takeUntil } from 'rxjs';
 import { render } from '../../public_api';
 

projects/testing-library/src/tests/issues/issue-386.spec.ts

@@ -1,5 +1,5 @@
 import { Component } from '@angular/core';
-import { vi, describe, test } from 'vitest';
+import { vi, describe, test, afterEach, beforeEach } from 'vitest';
 import { throwError } from 'rxjs';
 import { render, screen, fireEvent } from '../../public_api';
 

projects/testing-library/src/tests/issues/issue-435.spec.ts

@@ -2,6 +2,7 @@ import { CommonModule } from '@angular/common';
 import { BehaviorSubject } from 'rxjs';
 import { Component, inject, Injectable } from '@angular/core';
 import { screen, render } from '../../public_api';
+import { expect, test } from 'vitest';
 
 // Service
 @Injectable()

projects/testing-library/src/tests/issues/issue-437.spec.ts

@@ -1,6 +1,6 @@
 import userEvent from '@testing-library/user-event';
 import { MatSidenavModule } from '@angular/material/sidenav';
-import { vi, test } from 'vitest';
+import { vi, test, afterEach } from 'vitest';
 import { screen, render } from '../../public_api';
 
 afterEach(() => {
@@ -54,4 +54,5 @@ test('issue #437 with fakeTimers', async () => {
   await screen.findByTestId('test-button');
 
   await user.click(screen.getByTestId('test-button'));
+  vi.useRealTimers();
 });

projects/testing-library/src/tests/rerender.spec.ts

@@ -1,8 +1,8 @@
 import { Component, Input, OnChanges, SimpleChanges } from '@angular/core';
-import { vi, test, type Mock } from 'vitest';
+import { vi, test, afterEach } from 'vitest';
 import { render, screen } from '../public_api';
 
-let ngOnChangesSpy: Mock;
+const ngOnChangesSpy = vi.fn();
 @Component({
   selector: 'atl-fixture',
   template: ` {{ firstName }} {{ lastName }} `,
@@ -15,8 +15,8 @@ class FixtureComponent implements OnChanges {
   }
 }
 
-beforeEach(() => {
-  ngOnChangesSpy = vi.fn();
+afterEach(() => {
+  ngOnChangesSpy.mockReset();
 });
 
 test('rerenders the component with updated props', async () => {