From da66ee0204192c217b0f8e75c78f59515c961de6 Mon Sep 17 00:00:00 2001
From: Trinh Anh Ngoc <atw1990@gmail.com>
Date: Thu, 13 Jun 2024 22:59:08 +0700
Subject: [PATCH] Update stubs

---
 odoo-stubs/tests/common.pyi | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/odoo-stubs/tests/common.pyi b/odoo-stubs/tests/common.pyi
index a6dbaa4..4ed9802 100644
--- a/odoo-stubs/tests/common.pyi
+++ b/odoo-stubs/tests/common.pyi
@@ -1,5 +1,6 @@
 import collections
 from concurrent.futures import Future
+from contextlib import contextmanager
 from itertools import count
 from typing import Any, Callable, Generator, Generic, Iterator, Mapping, Match, TypeVar
 from unittest.mock import Mock
@@ -83,12 +84,15 @@ class BaseCase(case.TestCase, metaclass=MetaCase):
     def startClassPatcher(cls, patcher): ...
     def with_user(self, login: str) -> None: ...
     def debug_mode(self) -> Generator[None, None, None]: ...
+    @contextmanager
     def assertRaises(
         self, exception, func: Any | None = ..., *args, **kwargs
     ) -> Generator[Any, None, None] | None: ...
+    @contextmanager
     def assertQueries(
         self, expected, flush: bool = ...
     ) -> Generator[list, None, None]: ...
+    @contextmanager
     def assertQueryCount(
         self, default: int = ..., flush: bool = ..., **counters
     ) -> Generator[None, None, None]: ...