Ticket #8047: 8047_admin_scripts_tests_on_jython.diff

tests/regressiontests/admin_scripts/tests.py

@@ -7 +7 @@
 import unittest
 import shutil
 import sys
+import re
 
 from django import conf, bin, get_version
 from django.conf import settings
@@ -40 +41 @@
     def remove_settings(self, filename):
         test_dir = os.path.dirname(os.path.dirname(__file__))
         os.remove(os.path.join(test_dir, filename))
-        # Also try to remove the pyc file; if it exists, it could
+        # Also try to remove the compiled file; if it exists, it could
         # mess up later tests that depend upon the .py file not existing
-        try:
-            os.remove(os.path.join(test_dir, filename + 'c'))
-        except OSError:
-            pass
+        def try_remove(name):
+            try:
+                os.remove(os.path.join(test_dir, name))
+            except OSError:
+                pass
+        # Remove *.pyc
+        try_remove(filename + 'c')
+        # Remove *$py.class (for Jython)
+        try_remove(re.sub(r'\.py$', '$py.class', filename))
+
+    def _sys_executable(self, python_path):
+        """
+        Returns the command line needed to run a python interpreter, with the specified PYTHONPATH.
+        """
+        if sys.platform.startswith('java'):
+            return "%s -J-Dpython.path=%s" % \
+                   (sys.executable, python_path)
+        else:
+            return "PYTHONPATH=%s %s" % (python_path, sys.executable)
 
     def run_test(self, script, args, settings_file=None, apps=None):
         test_dir = os.path.dirname(os.path.dirname(__file__))
         project_dir = os.path.dirname(test_dir)
         base_dir = os.path.dirname(project_dir)
 
-        # Build the command line
-        cmd = '%s "%s"' % (sys.executable, script)
-        cmd += ''.join([' %s' % arg for arg in args])
-
         # Remember the old environment
         old_django_settings_module = os.environ.get('DJANGO_SETTINGS_MODULE', None)
-        old_python_path = os.environ.get('PYTHONPATH', None)
         old_cwd = os.getcwd()
 
         # Set the test environment
@@ -66 +77 @@
             os.environ['DJANGO_SETTINGS_MODULE'] = settings_file
         elif 'DJANGO_SETTINGS_MODULE' in os.environ:
             del os.environ['DJANGO_SETTINGS_MODULE']
+        python_path = os.pathsep.join([test_dir, base_dir] +  sys.path)
 
-        if old_python_path:
-            os.environ['PYTHONPATH'] = os.pathsep.join([test_dir, base_dir, old_python_path])
-        else:
-            os.environ['PYTHONPATH'] = os.pathsep.join([test_dir, base_dir])
+        # Build the command line
+        cmd = '%s "%s"' % (self._sys_executable(python_path), script)
+        cmd += ''.join([' %s' % arg for arg in args])
 
         # Move to the test directory and run
         os.chdir(test_dir)
@@ -80 +91 @@
         # Restore the old environment
         if old_django_settings_module:
             os.environ['DJANGO_SETTINGS_MODULE'] = old_django_settings_module
-        if old_python_path:
-            os.environ['PYTHONPATH'] = old_python_path
 
         # Move back to the old working directory
         os.chdir(old_cwd)
@@ -823 +832 @@
         out, err = self.run_manage(args)
         self.assertNoOutput(err)
         self.assertOutput(out, "EXECUTE:AppCommand app=<module 'django.contrib.auth.models'")
-        self.assertOutput(out, os.sep.join(['django','contrib','auth','models.pyc']) + "'>, options=[('pythonpath', None), ('settings', None), ('traceback', None)]")
+        self.assertOutput(out, os.sep.join(['django','contrib','auth','models.py']))
         self.assertOutput(out, "EXECUTE:AppCommand app=<module 'django.contrib.contenttypes.models'")
         self.assertOutput(out, os.sep.join(['django','contrib','contenttypes','models.py']))
         self.assertOutput(out, "'>, options=[('pythonpath', None), ('settings', None), ('traceback', None)]")