feat: add auto-mock support for async generators

Author: Sam Hewitt

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ### Features
 
+- `[jest-mock]` Add support for auto-mocking async generator functions ([#11080](https://github.com/facebook/jest/pull/11080))
 - `[jest-circus]` [**BREAKING**] Fail tests when multiple `done()` calls are made ([#10624](https://github.com/facebook/jest/pull/10624))
 - `[jest-circus, jest-jasmine2]` [**BREAKING**] Fail the test instead of just warning when describe returns a value ([#10947](https://github.com/facebook/jest/pull/10947))
 - `[jest-config]` [**BREAKING**] Default to Node testing environment instead of browser (JSDOM) ([#9874](https://github.com/facebook/jest/pull/9874))

e2e/generator-mock/__tests__/generatorMock.test.js

@@ -12,3 +12,7 @@ const methods = require('../index');
 test('mock works with generator', () => {
   expect(methods.generatorMethod).toBeDefined();
 });
+
+test('mock works with asyncGenerator', () => {
+  expect(methods.asyncGeneratorMethod).toBeDefined();
+});

e2e/generator-mock/index.js

@@ -9,4 +9,9 @@ function* generatorMethod() {
   yield 42;
 }
 
+async function* asyncGeneratorMethod() {
+  yield 42;
+}
+
 module.exports.generatorMethod = generatorMethod;
+module.exports.asyncGeneratorMethod = asyncGeneratorMethod;

packages/jest-mock/src/index.ts

@@ -304,7 +304,8 @@ function getType(ref?: unknown): MockFunctionMetadataType | null {
   if (
     typeName === 'Function' ||
     typeName === 'AsyncFunction' ||
-    typeName === 'GeneratorFunction'
+    typeName === 'GeneratorFunction' ||
+    typeName === 'AsyncGeneratorFunction'
   ) {
     return 'function';
   } else if (Array.isArray(ref)) {
@@ -347,7 +348,8 @@ function isReadonlyProp(object: any, prop: string): boolean {
     return (
       typeName === 'Function' ||
       typeName === 'AsyncFunction' ||
-      typeName === 'GeneratorFunction'
+      typeName === 'GeneratorFunction' ||
+      typeName === 'AsyncGeneratorFunction'
     );
   }