From 6e15e75bbe9823d7dfdc9809529de470681b8594 Mon Sep 17 00:00:00 2001
From: Trinh Anh Ngoc <atw1990@gmail.com>
Date: Thu, 13 Jun 2024 23:02:24 +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 3ec08b6..9669af3 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
@@ -80,12 +81,15 @@ class BaseCase(unittest.TestCase, metaclass=MetaCase):
     def browse_ref(self, xid: str) -> BaseModel | None: ...
     def patch(self, obj, key, val) -> 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]: ...