diff options
Diffstat (limited to 'tests/regressiontests/admin_scripts/tests.py')
| -rw-r--r-- | tests/regressiontests/admin_scripts/tests.py | 181 |
1 files changed, 104 insertions, 77 deletions
diff --git a/tests/regressiontests/admin_scripts/tests.py b/tests/regressiontests/admin_scripts/tests.py index 442f357782..f1b167eec7 100644 --- a/tests/regressiontests/admin_scripts/tests.py +++ b/tests/regressiontests/admin_scripts/tests.py @@ -1,18 +1,20 @@ """ -A series of tests to establish that the command-line managment tools work as +A series of tests to establish that the command-line managment tools work as advertised - especially with regards to the handling of the DJANGO_SETTINGS_MODULE and default settings.py files. """ import os import unittest import shutil +import sys +import re from django import conf, bin, get_version from django.conf import settings class AdminScriptTestCase(unittest.TestCase): def write_settings(self, filename, apps=None): - test_dir = os.path.dirname(os.path.dirname(__file__)) + test_dir = os.path.dirname(os.path.dirname(__file__)) settings_file = open(os.path.join(test_dir,filename), 'w') settings_file.write('# Settings file automatically generated by regressiontests.admin_scripts test case\n') exports = [ @@ -27,72 +29,93 @@ class AdminScriptTestCase(unittest.TestCase): for s in exports: if hasattr(settings,s): settings_file.write("%s = '%s'\n" % (s, str(getattr(settings,s)))) - + if apps is None: apps = ['django.contrib.auth', 'django.contrib.contenttypes', 'admin_scripts'] if apps: settings_file.write("INSTALLED_APPS = %s\n" % apps) - + settings_file.close() - + 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 + test_dir = os.path.dirname(os.path.dirname(__file__)) + full_name = os.path.join(test_dir, filename) + os.remove(full_name) + + # 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')) + if sys.platform.startswith('java'): + # Jython produces module$py.class files + os.remove(re.sub(r'\.py$', '$py.class', full_name)) + else: + # CPython produces module.pyc files + os.remove(full_name + 'c') except OSError: pass - + + def _sys_executable(self): + """ + Returns the command line needed to run a python interpreter, including + the options for setting sys.path on Jython, which doesn't recognize + PYTHONPATH. + """ + if sys.platform.startswith('java'): + return "%s -J-Dpython.path=%s" % \ + (sys.executable, os.environ['PYTHONPATH']) + else: + return 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 = 'python "%s"' % 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 if settings_file: os.environ['DJANGO_SETTINGS_MODULE'] = settings_file elif 'DJANGO_SETTINGS_MODULE' in os.environ: del os.environ['DJANGO_SETTINGS_MODULE'] - - os.environ['PYTHONPATH'] = os.pathsep.join([test_dir,base_dir]) + + 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(), script) + cmd += ''.join([' %s' % arg for arg in args]) # Move to the test directory and run os.chdir(test_dir) stdin, stdout, stderr = os.popen3(cmd) out, err = stdout.read(), stderr.read() - + # 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) - + return out, err - + def run_django_admin(self, args, settings_file=None): bin_dir = os.path.dirname(bin.__file__) return self.run_test(os.path.join(bin_dir,'django-admin.py'), args, settings_file) - + def run_manage(self, args, settings_file=None): conf_dir = os.path.dirname(conf.__file__) template_manage_py = os.path.join(conf_dir, 'project_template', 'manage.py') - test_dir = os.path.dirname(os.path.dirname(__file__)) + test_dir = os.path.dirname(os.path.dirname(__file__)) test_manage_py = os.path.join(test_dir, 'manage.py') shutil.copyfile(template_manage_py, test_manage_py) @@ -100,7 +123,7 @@ class AdminScriptTestCase(unittest.TestCase): # Cleanup - remove the generated manage.py script os.remove(test_manage_py) - + return stdout, stderr def assertNoOutput(self, stream): @@ -119,14 +142,14 @@ class AdminScriptTestCase(unittest.TestCase): class DjangoAdminNoSettings(AdminScriptTestCase): "A series of tests for django-admin.py when there is no settings.py file." - + def test_builtin_command(self): "no settings: django-admin builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] out, err = self.run_django_admin(args) self.assertNoOutput(out) self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined') - + def test_builtin_with_bad_settings(self): "no settings: django-admin builtin commands fail if settings file (from argument) doesn't exist" args = ['sqlall','--settings=bad_settings', 'admin_scripts'] @@ -148,17 +171,17 @@ class DjangoAdminDefaultSettings(AdminScriptTestCase): """ def setUp(self): self.write_settings('settings.py') - + def tearDown(self): self.remove_settings('settings.py') - + def test_builtin_command(self): "default: django-admin builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] out, err = self.run_django_admin(args) self.assertNoOutput(out) self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined') - + def test_builtin_with_settings(self): "default: django-admin builtin commands succeed if settings are provided as argument" args = ['sqlall','--settings=settings', 'admin_scripts'] @@ -193,14 +216,14 @@ class DjangoAdminDefaultSettings(AdminScriptTestCase): out, err = self.run_django_admin(args) self.assertNoOutput(out) self.assertOutput(err, "Unknown command: 'noargs_command'") - + def test_custom_command_with_settings(self): "default: django-admin can't execute user commands, even if settings are provided as argument" args = ['noargs_command', '--settings=settings'] out, err = self.run_django_admin(args) self.assertNoOutput(out) self.assertOutput(err, "Unknown command: 'noargs_command'") - + def test_custom_command_with_environment(self): "default: django-admin can't execute user commands, even if settings are provided in environment" args = ['noargs_command'] @@ -214,17 +237,17 @@ class DjangoAdminMinimalSettings(AdminScriptTestCase): """ def setUp(self): self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes']) - + def tearDown(self): self.remove_settings('settings.py') - + def test_builtin_command(self): "minimal: django-admin builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] out, err = self.run_django_admin(args) self.assertNoOutput(out) self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined') - + def test_builtin_with_settings(self): "minimal: django-admin builtin commands fail if settings are provided as argument" args = ['sqlall','--settings=settings', 'admin_scripts'] @@ -252,21 +275,21 @@ class DjangoAdminMinimalSettings(AdminScriptTestCase): out, err = self.run_django_admin(args,'bad_settings') self.assertNoOutput(out) self.assertOutput(err, "Could not import settings 'bad_settings'") - + def test_custom_command(self): "minimal: django-admin can't execute user commands" args = ['noargs_command'] out, err = self.run_django_admin(args) self.assertNoOutput(out) self.assertOutput(err, "Unknown command: 'noargs_command'") - + def test_custom_command_with_settings(self): "minimal: django-admin can't execute user commands, even if settings are provided as argument" args = ['noargs_command', '--settings=settings'] out, err = self.run_django_admin(args) self.assertNoOutput(out) self.assertOutput(err, "Unknown command: 'noargs_command'") - + def test_custom_command_with_environment(self): "minimal: django-admin can't execute user commands, even if settings are provided in environment" args = ['noargs_command'] @@ -280,10 +303,10 @@ class DjangoAdminAlternateSettings(AdminScriptTestCase): """ def setUp(self): self.write_settings('alternate_settings.py') - + def tearDown(self): self.remove_settings('alternate_settings.py') - + def test_builtin_command(self): "alternate: django-admin builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] @@ -344,17 +367,17 @@ class DjangoAdminAlternateSettings(AdminScriptTestCase): class DjangoAdminMultipleSettings(AdminScriptTestCase): """A series of tests for django-admin.py when multiple settings files (including the default 'settings.py') are available. The default settings - file is insufficient for performing the operations described, so the + file is insufficient for performing the operations described, so the alternate settings must be used by the running script. """ def setUp(self): self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes']) self.write_settings('alternate_settings.py') - + def tearDown(self): self.remove_settings('settings.py') self.remove_settings('alternate_settings.py') - + def test_builtin_command(self): "alternate: django-admin builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] @@ -409,7 +432,7 @@ class DjangoAdminMultipleSettings(AdminScriptTestCase): out, err = self.run_django_admin(args,'alternate_settings') self.assertNoOutput(out) self.assertOutput(err, "Unknown command: 'noargs_command'") - + ########################################################################## # MANAGE.PY TESTS # This next series of test classes checks the environment processing @@ -418,14 +441,14 @@ class DjangoAdminMultipleSettings(AdminScriptTestCase): class ManageNoSettings(AdminScriptTestCase): "A series of tests for manage.py when there is no settings.py file." - + def test_builtin_command(self): "no settings: manage.py builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] out, err = self.run_manage(args) self.assertNoOutput(out) self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'") - + def test_builtin_with_bad_settings(self): "no settings: manage.py builtin commands fail if settings file (from argument) doesn't exist" args = ['sqlall','--settings=bad_settings', 'admin_scripts'] @@ -447,17 +470,17 @@ class ManageDefaultSettings(AdminScriptTestCase): """ def setUp(self): self.write_settings('settings.py') - + def tearDown(self): self.remove_settings('settings.py') - + def test_builtin_command(self): "default: manage.py builtin commands succeed when default settings are appropriate" args = ['sqlall','admin_scripts'] out, err = self.run_manage(args) self.assertNoOutput(err) self.assertOutput(out, 'CREATE TABLE') - + def test_builtin_with_settings(self): "default: manage.py builtin commands succeed if settings are provided as argument" args = ['sqlall','--settings=settings', 'admin_scripts'] @@ -492,14 +515,14 @@ class ManageDefaultSettings(AdminScriptTestCase): out, err = self.run_manage(args) self.assertNoOutput(err) self.assertOutput(out, "EXECUTE:NoArgsCommand") - + def test_custom_command_with_settings(self): "default: manage.py can execute user commands when settings are provided as argument" args = ['noargs_command', '--settings=settings'] out, err = self.run_manage(args) self.assertNoOutput(err) self.assertOutput(out, "EXECUTE:NoArgsCommand") - + def test_custom_command_with_environment(self): "default: manage.py can execute user commands when settings are provided in environment" args = ['noargs_command'] @@ -513,17 +536,17 @@ class ManageMinimalSettings(AdminScriptTestCase): """ def setUp(self): self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes']) - + def tearDown(self): self.remove_settings('settings.py') - + def test_builtin_command(self): "minimal: manage.py builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] out, err = self.run_manage(args) self.assertNoOutput(out) self.assertOutput(err, 'App with label admin_scripts could not be found') - + def test_builtin_with_settings(self): "minimal: manage.py builtin commands fail if settings are provided as argument" args = ['sqlall','--settings=settings', 'admin_scripts'] @@ -551,21 +574,21 @@ class ManageMinimalSettings(AdminScriptTestCase): out, err = self.run_manage(args,'bad_settings') self.assertNoOutput(out) self.assertOutput(err, 'App with label admin_scripts could not be found') - + def test_custom_command(self): "minimal: manage.py can't execute user commands without appropriate settings" args = ['noargs_command'] out, err = self.run_manage(args) self.assertNoOutput(out) self.assertOutput(err, "Unknown command: 'noargs_command'") - + def test_custom_command_with_settings(self): "minimal: manage.py can't execute user commands, even if settings are provided as argument" args = ['noargs_command', '--settings=settings'] out, err = self.run_manage(args) self.assertNoOutput(out) self.assertOutput(err, "Unknown command: 'noargs_command'") - + def test_custom_command_with_environment(self): "minimal: manage.py can't execute user commands, even if settings are provided in environment" args = ['noargs_command'] @@ -579,10 +602,10 @@ class ManageAlternateSettings(AdminScriptTestCase): """ def setUp(self): self.write_settings('alternate_settings.py') - + def tearDown(self): self.remove_settings('alternate_settings.py') - + def test_builtin_command(self): "alternate: manage.py builtin commands fail with an import error when no default settings provided" args = ['sqlall','admin_scripts'] @@ -643,17 +666,17 @@ class ManageAlternateSettings(AdminScriptTestCase): class ManageMultipleSettings(AdminScriptTestCase): """A series of tests for manage.py when multiple settings files (including the default 'settings.py') are available. The default settings - file is insufficient for performing the operations described, so the + file is insufficient for performing the operations described, so the alternate settings must be used by the running script. """ def setUp(self): self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes']) self.write_settings('alternate_settings.py') - + def tearDown(self): self.remove_settings('settings.py') self.remove_settings('alternate_settings.py') - + def test_builtin_command(self): "multiple: manage.py builtin commands fail with an import error when no settings provided" args = ['sqlall','admin_scripts'] @@ -721,10 +744,10 @@ class CommandTypes(AdminScriptTestCase): "Tests for the various types of base command types that can be defined." def setUp(self): self.write_settings('settings.py') - + def tearDown(self): self.remove_settings('settings.py') - + def test_version(self): "--version is handled as a special case" args = ['--version'] @@ -737,7 +760,10 @@ class CommandTypes(AdminScriptTestCase): "--help is handled as a special case" args = ['--help'] out, err = self.run_manage(args) - self.assertOutput(out, "Usage: manage.py [options]") + if sys.version_info < (2, 5): + self.assertOutput(out, "usage: manage.py [options]") + else: + self.assertOutput(out, "Usage: manage.py [options]") self.assertOutput(err, "Type 'manage.py help <subcommand>' for help on a specific subcommand.") def test_specific_help(self): @@ -746,14 +772,14 @@ class CommandTypes(AdminScriptTestCase): out, err = self.run_manage(args) self.assertNoOutput(err) self.assertOutput(out, "Prints the CREATE TABLE, custom SQL and CREATE INDEX SQL statements for the given model module name(s).") - + def test_base_command(self): "User BaseCommands can execute when a label is provided" args = ['base_command','testlabel'] out, err = self.run_manage(args) self.assertNoOutput(err) self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', '1'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', None), ('traceback', None)]") - + def test_base_command_no_label(self): "User BaseCommands can execute when no labels are provided" args = ['base_command'] @@ -781,7 +807,7 @@ class CommandTypes(AdminScriptTestCase): out, err = self.run_manage(args) self.assertNoOutput(err) self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', 'y'), ('option_c', '3'), ('pythonpath', None), ('settings', None), ('traceback', None)]") - + def test_noargs(self): "NoArg Commands can be executed" args = ['noargs_command'] @@ -794,16 +820,16 @@ class CommandTypes(AdminScriptTestCase): args = ['noargs_command','argument'] out, err = self.run_manage(args) self.assertOutput(err, "Error: Command doesn't accept any arguments") - + def test_app_command(self): "User AppCommands can execute when a single app name is provided" args = ['app_command', 'auth'] 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.py'])) + self.assertOutput(out, os.sep.join(['django','contrib','auth','models.py'])) self.assertOutput(out, "'>, options=[('pythonpath', None), ('settings', None), ('traceback', None)]") - + def test_app_command_no_apps(self): "User AppCommands raise an error when no app name is provided" args = ['app_command'] @@ -816,7 +842,8 @@ class CommandTypes(AdminScriptTestCase): 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, "'>, options=[('pythonpath', None), ('settings', None), ('traceback', None)]") 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)]") @@ -826,7 +853,7 @@ class CommandTypes(AdminScriptTestCase): args = ['app_command', 'NOT_AN_APP'] out, err = self.run_manage(args) self.assertOutput(err, "App with label NOT_AN_APP could not be found") - + def test_app_command_some_invalid_appnames(self): "User AppCommands can execute when some of the provided app names are invalid" args = ['app_command', 'auth', 'NOT_AN_APP'] @@ -839,7 +866,7 @@ class CommandTypes(AdminScriptTestCase): out, err = self.run_manage(args) self.assertNoOutput(err) self.assertOutput(out, "EXECUTE:LabelCommand label=testlabel, options=[('pythonpath', None), ('settings', None), ('traceback', None)]") - + def test_label_command_no_label(self): "User LabelCommands raise an error if no label is provided" args = ['label_command'] @@ -859,14 +886,14 @@ class ArgumentOrder(AdminScriptTestCase): django-admin command arguments are parsed in 2 parts; the core arguments (--settings, --traceback and --pythonpath) are parsed using a Lax parser. - This Lax parser ignores any unknown options. Then the full settings are + This Lax parser ignores any unknown options. Then the full settings are passed to the command parser, which extracts commands of interest to the - individual command. + individual command. """ def setUp(self): self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes']) self.write_settings('alternate_settings.py') - + def tearDown(self): self.remove_settings('settings.py') self.remove_settings('alternate_settings.py') |