allow runtests.py to discover tests in django_mongodb/tests

Author: timgraham

tests/runtests.py

@@ -13,6 +13,8 @@
 import warnings
 from pathlib import Path
 
+import django_mongodb_backend
+
 try:
     import django
 except ImportError as e:
@@ -57,6 +59,9 @@
 
 RUNTESTS_DIR = os.path.abspath(os.path.dirname(__file__))
 
+MONGODB_TEST_DIR = Path(django_mongodb_backend.__file__).parent.parent / "tests"
+sys.path.append(str(MONGODB_TEST_DIR))
+
 TEMPLATE_DIR = os.path.join(RUNTESTS_DIR, "templates")
 
 # Create a specific subdirectory for the duration of the test suite.
@@ -139,6 +144,21 @@ def get_test_modules(gis_enabled):
                     test_module = dirname + "." + test_module
                 yield test_module
 
+    # Discover tests in django_mongodb_backend/tests.
+    dirpath = os.path.join(MONGODB_TEST_DIR, dirname)
+    with os.scandir(dirpath) as entries:
+        for f in entries:
+            if (
+                "." in f.name
+                or f.is_file()
+                or not os.path.exists(os.path.join(f.path, "__init__.py"))
+            ):
+                continue
+            test_module = f.name
+            if dirname:
+                test_module = dirname + "." + test_module
+            yield test_module
+
 
 def get_label_module(label):
     """Return the top-level module part for a test label."""