def test_list_all_covered_modules_duplicate(self):
module_a_val = MagicMock(__name__='module_a', __file__='module_a.py')
module_b_val = MagicMock(__name__='module_b', __file__='module_b.py')
with patch.dict('sys.modules', module_a=module_a_val, module_b=module_b_val):
with patch(import_patch) as import_func:
returned_modules = _list_all_covered_modules(MagicMock(), ['module_a', 'module_b', 'module_a'], [])
self.assertEquals([module_a_val, module_b_val], returned_modules)
import_func.assert_not_called()
def test_list_all_covered_modules_load(self):
module_a_val = MagicMock(__name__='module_a', __file__='module_a.py')
module_b_val = MagicMock(__name__='module_b', __file__='module_b.py')
with patch.dict('sys.modules', module_a=module_a_val):
with patch(import_patch) as import_func:
import_func.return_value = module_b_val
returned_modules = _list_all_covered_modules(MagicMock(), ['module_a', 'module_b'], [])
self.assertEquals([module_a_val, module_b_val], returned_modules)
import_func.assert_called_once_with('module_b')
def test_list_all_covered_modules_all_loaded(self):
module_a_val = MagicMock(__name__='module_a', __file__='module_a.py')
module_b_val = MagicMock(__name__='module_b', __file__='module_b.py')
with patch.dict('sys.modules', module_a=module_a_val, module_b=module_b_val):
with patch(import_patch) as import_func:
returned_modules, non_imported_modules = _list_all_covered_modules(MagicMock(),
['module_a', 'module_b'], [], True)
self.assertEquals([module_a_val, module_b_val], returned_modules)
import_func.assert_not_called()
def test_list_all_covered_modules_load(self):
module_a_val = MagicMock(__name__='module_a', __file__='module_a.py')
module_b_val = MagicMock(__name__='module_b', __file__='module_b.py')
with patch.dict('sys.modules', module_a=module_a_val):
with patch(import_patch) as import_func:
import_func.return_value = module_b_val
returned_modules = _list_all_covered_modules(
MagicMock(), ['module_a', 'module_b'], [])
self.assertEquals([module_a_val, module_b_val], returned_modules)
import_func.assert_called_once_with('module_b')
def test_list_all_covered_modules_exclusions(self):
module_a_val = MagicMock(__name__='module_a', __file__='module_a.py')
module_b_val = MagicMock(__name__='module_b', __file__='module_b.py')
logger = MagicMock()
with patch.dict('sys.modules', module_a=module_a_val, module_b=module_b_val):
with patch(import_patch) as import_func:
returned_modules = _list_all_covered_modules(logger, ['module_a', 'module_b', 'module_c'],
['module_c'])
self.assertEquals([module_a_val, module_b_val], returned_modules)
import_func.assert_not_called()
logger.debug.assert_called_with("Module '%s' was excluded", 'module_c')
def test_list_all_covered_modules_exclusions(self):
module_a_val = MagicMock(__name__='module_a', __file__='module_a.py')
module_b_val = MagicMock(__name__='module_b', __file__='module_b.py')
logger = MagicMock()
with patch.dict('sys.modules', module_a=module_a_val, module_b=module_b_val):
with patch(import_patch) as import_func:
returned_modules, non_imported_modules = _list_all_covered_modules(logger,
['module_a', 'module_b', 'module_c'],
['module_c'], True)
self.assertEquals([module_a_val, module_b_val], returned_modules)
import_func.assert_not_called()
logger.debug.assert_called_with("Module '%s' was excluded", 'module_c')
