updates
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
from typing import Union
|
||||
|
||||
_mock_module = None
|
||||
|
||||
|
||||
def get_mock_module(config):
|
||||
"""
|
||||
Import and return the actual "mock" module. By default this is
|
||||
"unittest.mock", but the user can force to always use "mock" using
|
||||
the mock_use_standalone_module ini option.
|
||||
"""
|
||||
global _mock_module
|
||||
if _mock_module is None:
|
||||
use_standalone_module = parse_ini_boolean(
|
||||
config.getini("mock_use_standalone_module")
|
||||
)
|
||||
if use_standalone_module:
|
||||
import mock
|
||||
|
||||
_mock_module = mock
|
||||
else:
|
||||
import unittest.mock
|
||||
|
||||
_mock_module = unittest.mock
|
||||
|
||||
return _mock_module
|
||||
|
||||
|
||||
def parse_ini_boolean(value: Union[bool, str]) -> bool:
|
||||
if isinstance(value, bool):
|
||||
return value
|
||||
if value.lower() == "true":
|
||||
return True
|
||||
if value.lower() == "false":
|
||||
return False
|
||||
raise ValueError("unknown string for bool: %r" % value)
|
||||
Reference in New Issue
Block a user