Update matcher to allow ProfiledComponent as acceptable value

apollographql · Nov 28, 2023 · 5c76e4f · 5c76e4f
1 parent 01d6696
commit 5c76e4f
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/src/testing/matchers/ProfiledComponent.ts b/src/testing/matchers/ProfiledComponent.ts
@@ -3,12 +3,16 @@ import { WaitForRenderTimeoutError } from "../internal/index.js";
 import type {
   NextRenderOptions,
   Profiler,
+  ProfiledComponent,
   ProfiledHook,
 } from "../internal/index.js";
 
 export const toRerender: MatcherFunction<[options?: NextRenderOptions]> =
   async function (actual, options) {
-    const _profiler = actual as Profiler<any> | ProfiledHook<any, any>;
+    const _profiler = actual as
+      | Profiler<any>
+      | ProfiledComponent<any, any>
+      | ProfiledHook<any, any>;
     const profiler = "Profiler" in _profiler ? _profiler.Profiler : _profiler;
     const hint = this.utils.matcherHint("toRerender", "ProfiledComponent", "");
     let pass = true;
@@ -40,7 +44,10 @@ const failed = {};
 export const toRenderExactlyTimes: MatcherFunction<
   [times: number, options?: NextRenderOptions]
 > = async function (actual, times, optionsPerRender) {
-  const _profiler = actual as Profiler<any> | ProfiledHook<any, any>;
+  const _profiler = actual as
+    | Profiler<any>
+    | ProfiledComponent<any, any>
+    | ProfiledHook<any, any>;
   const profiler = "Profiler" in _profiler ? _profiler.Profiler : _profiler;
   const options = { timeout: 100, ...optionsPerRender };
   const hint = this.utils.matcherHint("toRenderExactlyTimes");

diff --git a/src/testing/matchers/index.d.ts b/src/testing/matchers/index.d.ts
@@ -6,6 +6,7 @@ import type {
 import {
   NextRenderOptions,
   Profiler,
+  ProfiledComponent,
   ProfiledHook,
 } from "../internal/index.js";
 
@@ -29,11 +30,17 @@ interface ApolloCustomMatchers<R = void, T = {}> {
       ) => R
     : { error: "matcher needs to be called on an ApolloClient instance" };
 
-  toRerender: T extends Profiler<any, any> | ProfiledHook<any, any>
+  toRerender: T extends
+    | Profiler<any, any>
+    | ProfiledComponent<any, any>
+    | ProfiledHook<any, any>
     ? (options?: NextRenderOptions) => Promise<R>
     : { error: "matcher needs to be called on a ProfiledComponent instance" };
 
-  toRenderExactlyTimes: T extends Profiler<any, any> | ProfiledHook<any, any>
+  toRenderExactlyTimes: T extends
+    | Profiler<any, any>
+    | ProfiledComponent<any, any>
+    | ProfiledHook<any, any>
     ? (count: number, options?: NextRenderOptions) => Promise<R>
     : { error: "matcher needs to be called on a ProfiledComponent instance" };
 }