Fixed Django tests discovery

pythonFiles/testing_tools/unittest_discovery.py

@@ -4,11 +4,27 @@
 import traceback
 import unittest
 
+import os.path
+
+sys.path.insert(
+    1,
+    os.path.dirname(  # pythonFiles
+        os.path.dirname(  # pythonFiles/testing_tools
+            os.path.abspath(__file__)  # this file
+        )
+    ),
+)
+
+from unittestadapter.utils import setup_django_test_env
+
 start_dir = sys.argv[1]
 pattern = sys.argv[2]
 top_level_dir = sys.argv[3] if len(sys.argv) >= 4 else None
 sys.path.insert(0, os.getcwd())
 
+# Setup django env to prevent missing django tests
+setup_django_test_env(start_dir)
+
 
 def get_sourceline(obj):
     try:

pythonFiles/unittestadapter/discovery.py

@@ -23,7 +23,12 @@
 from testing_tools import socket_manager
 
 # If I use from utils then there will be an import error in test_discovery.py.
-from unittestadapter.utils import TestNode, build_test_tree, parse_unittest_args
+from unittestadapter.utils import (
+    TestNode,
+    build_test_tree,
+    parse_unittest_args,
+    setup_django_test_env,
+)
 
 # Add the lib path to sys.path to find the typing_extensions module.
 sys.path.insert(0, os.path.join(PYTHON_FILES, "lib", "python"))
@@ -121,6 +126,9 @@ def discover_tests(
 
     start_dir, pattern, top_level_dir = parse_unittest_args(argv[index + 1 :])
 
+    # Setup django env to prevent missing django tests
+    setup_django_test_env(start_dir)
+
     # Perform test discovery.
     port, uuid = parse_discovery_cli_args(argv[:index])
     payload = discover_tests(start_dir, pattern, top_level_dir, uuid)

pythonFiles/unittestadapter/execution.py

@@ -25,7 +25,7 @@
 sys.path.insert(0, os.path.join(PYTHON_FILES, "lib", "python"))
 from testing_tools import socket_manager
 from typing_extensions import NotRequired, TypeAlias, TypedDict
-from unittestadapter.utils import parse_unittest_args
+from unittestadapter.utils import parse_unittest_args, setup_django_test_env
 
 DEFAULT_PORT = "45454"
 
@@ -232,6 +232,9 @@ def run_tests(
 
     start_dir, pattern, top_level_dir = parse_unittest_args(argv[index + 1 :])
 
+    # Setup django env to prevent missing django tests
+    setup_django_test_env(start_dir)
+
     run_test_ids_port = os.environ.get("RUN_TEST_IDS_PORT")
     run_test_ids_port_int = (
         int(run_test_ids_port) if run_test_ids_port is not None else 0

pythonFiles/unittestadapter/utils.py

@@ -233,3 +233,42 @@ def parse_unittest_args(args: List[str]) -> Tuple[str, str, Union[str, None]]:
         parsed_args.pattern,
         parsed_args.top_level_directory,
     )
+
+
+def setup_django_test_env(root):
+    """Configure Django environment for running Django tests.
+
+    It checks if Django is installed by attempting to import the `django` module.
+    Looks for `manage.py` file to extract the value of `DJANGO_SETTINGS_MODULE`.
+    Sets `DJANGO_SETTINGS_MODULE` environment variable and initializes Django setup.
+    If couldn't find the file or import django during this process, the function fails silently.
+
+    Args:
+        root (str): The root directory of the Django project.
+
+    Returns:
+        None
+    """
+    import os, re
+
+    try:
+        # Check if Django is installed
+        import django
+
+        # Check if manage.py exists
+        with open(os.path.join(root, "manage.py"), "r") as manage_py:
+            # Look for a line that sets the DJANGO_SETTINGS_MODULE environment variable
+            pattern = r"^os\.environ\.setdefault\((\'|\")DJANGO_SETTINGS_MODULE(\'|\"), (\'|\")(?P<settings_path>[\w.]+)(\'|\")\)$"
+            for line in manage_py.readlines():
+                pattern_matched = re.match(pattern, line.strip())
+                if pattern_matched is not None:
+                    # Extract value for DJANGO_SETTINGS_MODULE
+                    settings_path = str(
+                        pattern_matched.groupdict().get("settings_path", "")
+                    )
+                    # Set the DJANGO_SETTINGS_MODULE environment variable and initialize Django's settings
+                    os.environ.setdefault("DJANGO_SETTINGS_MODULE", settings_path)
+                    django.setup()
+                    return
+    except (ModuleNotFoundError, FileNotFoundError):
+        return

pythonFiles/visualstudio_py_testlauncher.py

@@ -25,6 +25,10 @@
 import traceback
 import unittest
 
+sys.path.insert(1, os.path.abspath(__file__))  # this file
+
+from unittestadapter.utils import setup_django_test_env
+
 try:
     import thread
 except:
@@ -327,11 +331,15 @@ def main():
                 opts.us = "."
             if opts.up is None:
                 opts.up = "test*.py"
+            # Setup django env to prevent missing django tests
+            setup_django_test_env(opts.us)
             tests = unittest.defaultTestLoader.discover(opts.us, opts.up)
         else:
             # loadTestsFromNames doesn't work well (with duplicate file names or class names)
             # Easier approach is find the test suite and use that for running
             loader = unittest.TestLoader()
+            # Setup django env to prevent missing django tests
+            setup_django_test_env(opts.us)
             # opts.us will be passed in
             suites = loader.discover(
                 opts.us, pattern=os.path.basename(opts.testFile), top_level_dir=opts.ut