From 3dbaf0836da92e8596a3ed4202cafaadfc79589e Mon Sep 17 00:00:00 2001
From: AriPerkkio <ari.perkkio@gmail.com>
Date: Thu, 4 May 2023 08:58:40 +0300
Subject: [PATCH] fix: support new vitest typings

---
 src/vitest/register.ts | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/vitest/register.ts b/src/vitest/register.ts
index c74b2b9..5f3c779 100644
--- a/src/vitest/register.ts
+++ b/src/vitest/register.ts
@@ -3,22 +3,27 @@ import { afterEach, beforeEach, expect } from 'vitest';
 
 import { register as __register, toBeAnnounced } from '../to-be-announced';
 
-interface ToBeAnnouncedMatchers {
+interface ToBeAnnouncedMatchers<T = any> {
     /**
      * Assert whether given message was announced by ARIA live region.
      *
      * @param politenessSetting `POLITENESS_SETTING` of the announcement
      */
-    toBeAnnounced(politenessSetting?: Exclude<PolitenessSetting, 'off'>): void;
+    toBeAnnounced(politenessSetting?: Exclude<PolitenessSetting, 'off'>): T;
 }
 
+declare module '@vitest/expect' {
+    interface Assertion<T> extends ToBeAnnouncedMatchers<T> {}
+}
+
+// For Vitest below 0.31.x
 declare global {
     namespace Vi {
-        interface Assertion extends ToBeAnnouncedMatchers {}
+        interface Assertion<T> extends ToBeAnnouncedMatchers<T> {}
     }
 }
 
-expect.extend({ toBeAnnounced });
+(expect as any).extend({ toBeAnnounced });
 
 /**
  * Register `extend-to-be-expected` to track DOM nodes