Add test for #847

sebastinas · sebastinas · commit 6438d2786892 · 2021-01-10T00:12:36.000+01:00
diff --git a/bpython/test/test_importcompletion.py b/bpython/test/test_importcompletion.py
@@ -199,5 +199,32 @@ def test_simple_symbolic_link_loop(self):
         self.assertFalse(filepaths)
 
 
+class TestBugReports(unittest.TestCase):
+    def test_issue_847(self):
+        with tempfile.TemporaryDirectory() as import_test_folder:
+            #   ./xyzzy
+            #   ./xyzzy/__init__.py
+            #   ./xyzzy/plugh
+            #   ./xyzzy/plugh/__init__.py
+            #   ./xyzzy/plugh/bar.py
+            #   ./xyzzy/plugh/foo.py
+
+            base_path = Path(import_test_folder)
+            (base_path / "xyzzy" / "plugh").mkdir(parents=True)
+            (base_path / "xyzzy" / "__init__.py").touch()
+            (base_path / "xyzzy" / "plugh" / "__init__.py").touch()
+            (base_path / "xyzzy" / "plugh" / "bar.py").touch()
+            (base_path / "xyzzy" / "plugh" / "foo.py").touch()
+
+            module_gatherer = ModuleGatherer([base_path.absolute()])
+            while module_gatherer.find_coroutine():
+                pass
+
+            self.assertSetEqual(
+                module_gatherer.complete(17, "from xyzzy.plugh."),
+                {"xyzzy.plugh.bar", "xyzzy.plugh.foo"},
+            )
+
+
 if __name__ == "__main__":
     unittest.main()
