1# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
2# For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
3# Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
4
5"""Astroid hooks for unittest module."""
6from astroid.brain.helpers import register_module_extender
7from astroid.builder import parse
8from astroid.manager import AstroidManager
9
10
11def IsolatedAsyncioTestCaseImport():
12 """
13 In the unittest package, the IsolatedAsyncioTestCase class is imported lazily.
14
15 I.E. only when the ``__getattr__`` method of the unittest module is called with
16 'IsolatedAsyncioTestCase' as argument. Thus the IsolatedAsyncioTestCase
17 is not imported statically (during import time).
18 This function mocks a classical static import of the IsolatedAsyncioTestCase.
19
20 (see https://github.com/pylint-dev/pylint/issues/4060)
21 """
22 return parse(
23 """
24 from .async_case import IsolatedAsyncioTestCase
25 """
26 )
27
28
29def register(manager: AstroidManager) -> None:
30 register_module_extender(manager, "unittest", IsolatedAsyncioTestCaseImport)