Roll typ to the latest version

third_party/typ/typ/tests/main_test.py

 # Copyright 2014 Dirk Pranke. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #    http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
@@ skipping 336 matching lines @@
         self.check(['-l', '../foo/pass_test.py'], files=files, cwd='bar',
                    ret=0, err='',
                    out='foo.pass_test.PassingTest.test_pass\n')
         self.check(['-l', 'foo'], files=files, cwd='bar',
                    ret=0, err='',
                    out='foo.pass_test.PassingTest.test_pass\n')
         self.check(['-l', '--path', '../foo', 'pass_test'],
                    files=files, cwd='bar', ret=0, err='',
                    out='pass_test.PassingTest.test_pass\n')
 
+    def test_multiple_top_level_dirs(self):
+        files = {
+            'foo/bar/__init__.py': '',
+            'foo/bar/pass_test.py': PASS_TEST_PY,
+            'baz/quux/__init__.py': '',
+            'baz/quux/second_test.py': PASS_TEST_PY,
+        }
+        self.check(['-l', 'foo/bar', 'baz/quux'], files=files,
+                   ret=0, err='',
+                   out=(
+                       'bar.pass_test.PassingTest.test_pass\n'
+                       'quux.second_test.PassingTest.test_pass\n'
+                       ))
+        self.check(['-l', 'foo/bar/pass_test.py', 'baz/quux'], files=files,
+                   ret=0, err='',
+                   out=(
+                       'bar.pass_test.PassingTest.test_pass\n'
+                       'quux.second_test.PassingTest.test_pass\n'
+                       ))
+        self.check(['-l', '--top-level-dirs', 'foo', '--top-level-dirs', 'baz'],
+                   files=files,
+                   ret=0, err='',
+                   out=(
+                       'bar.pass_test.PassingTest.test_pass\n'
+                       'quux.second_test.PassingTest.test_pass\n'
+                       ))
+
+    def test_single_top_level_dir(self):
+        files = {
+            'foo/bar/__init__.py': '',
+            'foo/bar/pass_test.py': PASS_TEST_PY,
+            'baz/quux/__init__.py': '',
+            'baz/quux/second_test.py': PASS_TEST_PY,
+        }
+        self.check(['-l', '--top-level-dir', 'foo'],
+                   files=files,
+                   ret=0, err='',
+                   out=(
+                       'bar.pass_test.PassingTest.test_pass\n'
+                       ))
+
+    def test_can_not_have_both_top_level_flags(self):
+        files = {
+            'foo/bar/__init__.py': '',
+            'foo/bar/pass_test.py': PASS_TEST_PY,
+            'baz/quux/__init__.py': '',
+            'baz/quux/second_test.py': PASS_TEST_PY,
+        }
+        self.check(
+            ['-l', '--top-level-dir', 'foo', '--top-level-dirs', 'bar'],
+            files=files,
+            ret=1, out='',
+            err='Cannot specify both --top-level-dir and --top-level-dirs\n')
+
     def test_help(self):
         self.check(['--help'], ret=0, rout='.*', err='')
 
     def test_import_failure_missing_file(self):
         _, out, _, _ = self.check(['-l', 'foo'], ret=1, err='')
         self.assertIn('Failed to load "foo" in find_tests', out)
-        self.assertIn('No module named foo', out)
+        self.assertIn('No module named', out)
 
     def test_import_failure_missing_package(self):
         files = {'foo.py': d("""\
                              import unittest
                              import package_that_does_not_exist
 
                              class ImportFailureTest(unittest.TestCase):
                                 def test_case(self):
                                     pass
                              """)}
         _, out, _, _ = self.check(['-l', 'foo.py'], files=files, ret=1, err='')
         self.assertIn('Failed to load "foo.py" in find_tests', out)
-        self.assertIn('No module named package_that_does_not_exist', out)
+        self.assertIn('No module named', out)
 
     def test_import_failure_no_tests(self):
         files = {'foo.py': 'import unittest'}
         self.check(['-l', 'foo'], files=files, ret=0, err='',
                    out='\n')
 
     def test_import_failure_syntax_error(self):
         files = {'syn_test.py': d("""\
                              import unittest
 
@@ skipping 379 matching lines @@
                     modules_to_unload.append(k)
             for k in modules_to_unload:
                 del sys.modules[k]
             sys.path = orig_sys_path
 
         return ret, out, err
 
     def test_debugger(self):
         # TODO: this test seems to hang under coverage.
         pass