Working on test but am having issues with the try

etiennerichart · etiennerichart · commit 2edc977f46f7 · 2020-07-06T12:26:02.000-04:00
diff --git a/importtest.py b/importtest.py
@@ -1,7 +1,44 @@
 from bpython.importcompletion import find_modules
-import os
+import os, sys, tempfile
 
-foo = find_modules(os.path.abspath("./importtestfolder"))
+with tempfile.TemporaryDirectory() as import_test_folder:
+    try:
+        os.chdir(import_test_folder)
+    except:
+        print("Can't change the Current Working Directory")
+        sys.exit()
 
-for thing in foo:
-    print("YIELD:", thing)
+    os.mkdir(os.path.join(os.getcwd(), "Level0")
+    try:
+        os.chdir(Level0)
+        open("__init__.py", w)
+    except OSError:
+        print("Can't change the Current Working Directory")
+        sys.exit()
+
+    os.mkdir(os.path.join(os.getcwd(), "Level1")
+    try:
+        os.chdir(Level1)
+        open("__init__.py", w)
+    except OSError:
+        print("Can't change the Current Working Directory")
+        sys.exit()
+
+    os.mkdir(os.path.join(os.getcwd(), "Level2")
+    try:
+        os.chdir(Level2)
+        open("__init__.py", w)
+    except OSError:
+        print("Can't change the Current Working Directory")
+        sys.exit()
+
+    os.symlink(
+        "import_test_folder/Level0/Level1",
+        os.path.join(os.getcwd, "Level3"),
+        True,
+    )
+
+    foo = find_modules(os.path.abspath("import_test_folder"))
+
+    for thing in foo:
+        print("YIELD:", thing)
