def test_load_all_modules_flat(self): '''helpers.load_all_modules('foo', TEMPDIR)—flat directory''' self.prepare_temporary_directory() self.populate_temporary_directory_files( { '/foo': [ 'a.py', 'b.py', 'c.py', ], } ) self.mock_importlib_import_module() helpers.load_all_modules('foo', self.temporary_directory_path) self.assertEqual(3, self.recursive_file_count('/')) _ = [ mock.call('foo.c'), mock.call('foo.b'), mock.call('foo.a'), ] self.mocked_importlib_import_module.assert_has_calls(_)
def test_load_all_modules_empty(self): '''helpers.load_all_modules('foo', TEMPDIR)—empty directory''' self.prepare_temporary_directory() self.mock_importlib_import_module() helpers.load_all_modules('foo', self.temporary_directory_path) self.assertEqual(0, self.recursive_file_count('/')) self.assertFalse(self.mocked_importlib_import_module.called)
# Copyright (C) 2014 by Alex Brandt <*****@*****.**> # # upkern is freely distributable under the terms of an MIT-style license. # See COPYING or http://www.opensource.org/licenses/mit-license.php. import logging import os from upkern import helpers logger = logging.getLogger(__name__) PREPARERS = {} helpers.load_all_modules(__name__, os.path.dirname(__file__)) class InitialRAMFileSystem(object): def __init__(self, preparer, *args, **kwargs): self.preparer = PREPARERS[preparer](*args, **kwargs) def __getattr__(self, name): return getattr(self.preparer, name)