From 2d02bd54e8d445aa2fcaa86189f548b484987285 Mon Sep 17 00:00:00 2001
From: Trinh Anh Ngoc <atw1990@gmail.com>
Date: Wed, 17 May 2023 14:32:45 +0700
Subject: [PATCH] Update stubs

---
 odoo-stubs/tests/loader.pyi | 6 ++++--
 odoo-stubs/tools/misc.pyi   | 3 ++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/odoo-stubs/tests/loader.pyi b/odoo-stubs/tests/loader.pyi
index c0f7e29..9103702 100644
--- a/odoo-stubs/tests/loader.pyi
+++ b/odoo-stubs/tests/loader.pyi
@@ -1,11 +1,13 @@
 from importlib.machinery import ModuleSpec
+from types import ModuleType
 from typing import Iterator
 
-from .runner import OdooTestResult
-from ..tests import OdooSuite
+from .result import OdooTestResult
+from .suite import OdooSuite
 
 def get_test_modules(module: str) -> list: ...
 def _get_tests_modules(mod: ModuleSpec) -> list: ...
+def _get_upgrade_test_modules(module) -> Iterator[ModuleType]: ...
 def make_suite(module_names: list[str], position: str = ...) -> OdooSuite: ...
 def run_suite(suite: OdooSuite, module_name: str | None = ...) -> OdooTestResult: ...
 def unwrap_suite(test) -> Iterator: ...
diff --git a/odoo-stubs/tools/misc.pyi b/odoo-stubs/tools/misc.pyi
index fe01946..c23e85c 100644
--- a/odoo-stubs/tools/misc.pyi
+++ b/odoo-stubs/tools/misc.pyi
@@ -257,4 +257,5 @@ ADDRESS_REGEX: Pattern
 
 def street_split(street: str) -> dict[str, str]: ...
 def is_list_of(values: list | tuple, type_: type | tuple[type, ...]) -> bool: ...
-def has_list_types(values: list | tuple, types: Collection[type | tuple[type, ...]]) -> bool: ...
\ No newline at end of file
+def has_list_types(values: list | tuple, types: Collection[type | tuple[type, ...]]) -> bool: ...
+def get_flag(country_code: str) -> str: ...