From fe09616703937b27c9774d844d95faaf90e7ba4d Mon Sep 17 00:00:00 2001
From: Trinh Anh Ngoc <atw1990@gmail.com>
Date: Thu, 13 Jun 2024 23:07:28 +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 d2081fe..6d99479 100644
--- a/odoo-stubs/tests/common.pyi
+++ b/odoo-stubs/tests/common.pyi
@@ -1,6 +1,7 @@
 import collections
 import logging
 import unittest
+from contextlib import contextmanager
 from itertools import count
 from re import Pattern
 from typing import Any, Callable, Generator, Generic, Iterator, Mapping, Match, TypeVar
@@ -74,12 +75,15 @@ class BaseCase(TreeCase):
     def ref(self, xid: str) -> int: ...
     def browse_ref(self, xid: str) -> BaseModel | None: ...
     def with_user(self, login: str) -> 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]: ...