PageRenderTime 54ms CodeModel.GetById 21ms RepoModel.GetById 1ms app.codeStats 0ms

/tests/regressiontests/admin_scripts/tests.py

https://code.google.com/p/mango-py/
Python | 1293 lines | 1192 code | 67 blank | 34 comment | 75 complexity | b59da4d4d2c3806b16567cc87888cd86 MD5 | raw file
Possible License(s): BSD-3-Clause
  1. """
  2. A series of tests to establish that the command-line managment tools work as
  3. advertised - especially with regards to the handling of the DJANGO_SETTINGS_MODULE
  4. and default settings.py files.
  5. """
  6. import os
  7. import shutil
  8. import sys
  9. import re
  10. from django import conf, bin, get_version
  11. from django.conf import settings
  12. from django.utils import unittest
  13. class AdminScriptTestCase(unittest.TestCase):
  14. def write_settings(self, filename, apps=None, is_dir=False, sdict=None):
  15. test_dir = os.path.dirname(os.path.dirname(__file__))
  16. if is_dir:
  17. settings_dir = os.path.join(test_dir,filename)
  18. os.mkdir(settings_dir)
  19. settings_file = open(os.path.join(settings_dir,'__init__.py'), 'w')
  20. else:
  21. settings_file = open(os.path.join(test_dir, filename), 'w')
  22. settings_file.write('# Settings file automatically generated by regressiontests.admin_scripts test case\n')
  23. exports = [
  24. 'DATABASES',
  25. 'ROOT_URLCONF'
  26. ]
  27. for s in exports:
  28. if hasattr(settings, s):
  29. o = getattr(settings, s)
  30. if not isinstance(o, dict):
  31. o = "'%s'" % o
  32. settings_file.write("%s = %s\n" % (s, o))
  33. if apps is None:
  34. apps = ['django.contrib.auth', 'django.contrib.contenttypes', 'admin_scripts']
  35. if apps:
  36. settings_file.write("INSTALLED_APPS = %s\n" % apps)
  37. if sdict:
  38. for k, v in sdict.items():
  39. settings_file.write("%s = %s\n" % (k, v))
  40. settings_file.close()
  41. def remove_settings(self, filename, is_dir=False):
  42. test_dir = os.path.dirname(os.path.dirname(__file__))
  43. full_name = os.path.join(test_dir, filename)
  44. if is_dir:
  45. shutil.rmtree(full_name)
  46. else:
  47. os.remove(full_name)
  48. # Also try to remove the compiled file; if it exists, it could
  49. # mess up later tests that depend upon the .py file not existing
  50. try:
  51. if sys.platform.startswith('java'):
  52. # Jython produces module$py.class files
  53. os.remove(re.sub(r'\.py$', '$py.class', full_name))
  54. else:
  55. # CPython produces module.pyc files
  56. os.remove(full_name + 'c')
  57. except OSError:
  58. pass
  59. def _ext_backend_paths(self):
  60. """
  61. Returns the paths for any external backend packages.
  62. """
  63. paths = []
  64. first_package_re = re.compile(r'(^[^\.]+)\.')
  65. for backend in settings.DATABASES.values():
  66. result = first_package_re.findall(backend['ENGINE'])
  67. if result and result != 'django':
  68. backend_pkg = __import__(result[0])
  69. backend_dir = os.path.dirname(backend_pkg.__file__)
  70. paths.append(os.path.dirname(backend_dir))
  71. return paths
  72. def run_test(self, script, args, settings_file=None, apps=None):
  73. test_dir = os.path.dirname(os.path.dirname(__file__))
  74. project_dir = os.path.dirname(test_dir)
  75. base_dir = os.path.dirname(project_dir)
  76. ext_backend_base_dirs = self._ext_backend_paths()
  77. # Remember the old environment
  78. old_django_settings_module = os.environ.get('DJANGO_SETTINGS_MODULE', None)
  79. if sys.platform.startswith('java'):
  80. python_path_var_name = 'JYTHONPATH'
  81. else:
  82. python_path_var_name = 'PYTHONPATH'
  83. old_python_path = os.environ.get(python_path_var_name, None)
  84. old_cwd = os.getcwd()
  85. # Set the test environment
  86. if settings_file:
  87. os.environ['DJANGO_SETTINGS_MODULE'] = settings_file
  88. elif 'DJANGO_SETTINGS_MODULE' in os.environ:
  89. del os.environ['DJANGO_SETTINGS_MODULE']
  90. python_path = [test_dir, base_dir]
  91. python_path.extend(ext_backend_base_dirs)
  92. os.environ[python_path_var_name] = os.pathsep.join(python_path)
  93. # Build the command line
  94. executable = sys.executable
  95. arg_string = ' '.join(['%s' % arg for arg in args])
  96. if ' ' in executable:
  97. cmd = '""%s" "%s" %s"' % (executable, script, arg_string)
  98. else:
  99. cmd = '%s "%s" %s' % (executable, script, arg_string)
  100. # Move to the test directory and run
  101. os.chdir(test_dir)
  102. try:
  103. from subprocess import Popen, PIPE
  104. p = Popen(cmd, shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE)
  105. stdin, stdout, stderr = (p.stdin, p.stdout, p.stderr)
  106. p.wait()
  107. except ImportError:
  108. stdin, stdout, stderr = os.popen3(cmd)
  109. out, err = stdout.read(), stderr.read()
  110. # Restore the old environment
  111. if old_django_settings_module:
  112. os.environ['DJANGO_SETTINGS_MODULE'] = old_django_settings_module
  113. if old_python_path:
  114. os.environ[python_path_var_name] = old_python_path
  115. # Move back to the old working directory
  116. os.chdir(old_cwd)
  117. return out, err
  118. def run_django_admin(self, args, settings_file=None):
  119. bin_dir = os.path.abspath(os.path.dirname(bin.__file__))
  120. return self.run_test(os.path.join(bin_dir,'django-admin.py'), args, settings_file)
  121. def run_manage(self, args, settings_file=None):
  122. conf_dir = os.path.dirname(conf.__file__)
  123. template_manage_py = os.path.join(conf_dir, 'project_template', 'manage.py')
  124. test_dir = os.path.dirname(os.path.dirname(__file__))
  125. test_manage_py = os.path.join(test_dir, 'manage.py')
  126. shutil.copyfile(template_manage_py, test_manage_py)
  127. stdout, stderr = self.run_test('./manage.py', args, settings_file)
  128. # Cleanup - remove the generated manage.py script
  129. os.remove(test_manage_py)
  130. return stdout, stderr
  131. def assertNoOutput(self, stream):
  132. "Utility assertion: assert that the given stream is empty"
  133. self.assertEqual(len(stream), 0, "Stream should be empty: actually contains '%s'" % stream)
  134. def assertOutput(self, stream, msg):
  135. "Utility assertion: assert that the given message exists in the output"
  136. self.assertTrue(msg in stream, "'%s' does not match actual output text '%s'" % (msg, stream))
  137. ##########################################################################
  138. # DJANGO ADMIN TESTS
  139. # This first series of test classes checks the environment processing
  140. # of the django-admin.py script
  141. ##########################################################################
  142. class DjangoAdminNoSettings(AdminScriptTestCase):
  143. "A series of tests for django-admin.py when there is no settings.py file."
  144. def test_builtin_command(self):
  145. "no settings: django-admin builtin commands fail with an import error when no settings provided"
  146. args = ['sqlall','admin_scripts']
  147. out, err = self.run_django_admin(args)
  148. self.assertNoOutput(out)
  149. self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined')
  150. def test_builtin_with_bad_settings(self):
  151. "no settings: django-admin builtin commands fail if settings file (from argument) doesn't exist"
  152. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  153. out, err = self.run_django_admin(args)
  154. self.assertNoOutput(out)
  155. self.assertOutput(err, "Could not import settings 'bad_settings'")
  156. def test_builtin_with_bad_environment(self):
  157. "no settings: django-admin builtin commands fail if settings file (from environment) doesn't exist"
  158. args = ['sqlall','admin_scripts']
  159. out, err = self.run_django_admin(args,'bad_settings')
  160. self.assertNoOutput(out)
  161. self.assertOutput(err, "Could not import settings 'bad_settings'")
  162. class DjangoAdminDefaultSettings(AdminScriptTestCase):
  163. """A series of tests for django-admin.py when using a settings.py file that
  164. contains the test application.
  165. """
  166. def setUp(self):
  167. self.write_settings('settings.py')
  168. def tearDown(self):
  169. self.remove_settings('settings.py')
  170. def test_builtin_command(self):
  171. "default: django-admin builtin commands fail with an import error when no settings provided"
  172. args = ['sqlall','admin_scripts']
  173. out, err = self.run_django_admin(args)
  174. self.assertNoOutput(out)
  175. self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined')
  176. def test_builtin_with_settings(self):
  177. "default: django-admin builtin commands succeed if settings are provided as argument"
  178. args = ['sqlall','--settings=settings', 'admin_scripts']
  179. out, err = self.run_django_admin(args)
  180. self.assertNoOutput(err)
  181. self.assertOutput(out, 'CREATE TABLE')
  182. def test_builtin_with_environment(self):
  183. "default: django-admin builtin commands succeed if settings are provided in the environment"
  184. args = ['sqlall','admin_scripts']
  185. out, err = self.run_django_admin(args,'settings')
  186. self.assertNoOutput(err)
  187. self.assertOutput(out, 'CREATE TABLE')
  188. def test_builtin_with_bad_settings(self):
  189. "default: django-admin builtin commands fail if settings file (from argument) doesn't exist"
  190. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  191. out, err = self.run_django_admin(args)
  192. self.assertNoOutput(out)
  193. self.assertOutput(err, "Could not import settings 'bad_settings'")
  194. def test_builtin_with_bad_environment(self):
  195. "default: django-admin builtin commands fail if settings file (from environment) doesn't exist"
  196. args = ['sqlall','admin_scripts']
  197. out, err = self.run_django_admin(args,'bad_settings')
  198. self.assertNoOutput(out)
  199. self.assertOutput(err, "Could not import settings 'bad_settings'")
  200. def test_custom_command(self):
  201. "default: django-admin can't execute user commands if it isn't provided settings"
  202. args = ['noargs_command']
  203. out, err = self.run_django_admin(args)
  204. self.assertNoOutput(out)
  205. self.assertOutput(err, "Unknown command: 'noargs_command'")
  206. def test_custom_command_with_settings(self):
  207. "default: django-admin can execute user commands if settings are provided as argument"
  208. args = ['noargs_command', '--settings=settings']
  209. out, err = self.run_django_admin(args)
  210. self.assertNoOutput(err)
  211. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  212. def test_custom_command_with_environment(self):
  213. "default: django-admin can execute user commands if settings are provided in environment"
  214. args = ['noargs_command']
  215. out, err = self.run_django_admin(args,'settings')
  216. self.assertNoOutput(err)
  217. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  218. class DjangoAdminFullPathDefaultSettings(AdminScriptTestCase):
  219. """A series of tests for django-admin.py when using a settings.py file that
  220. contains the test application specified using a full path.
  221. """
  222. def setUp(self):
  223. self.write_settings('settings.py', ['django.contrib.auth', 'django.contrib.contenttypes', 'regressiontests.admin_scripts'])
  224. def tearDown(self):
  225. self.remove_settings('settings.py')
  226. def test_builtin_command(self):
  227. "fulldefault: django-admin builtin commands fail with an import error when no settings provided"
  228. args = ['sqlall','admin_scripts']
  229. out, err = self.run_django_admin(args)
  230. self.assertNoOutput(out)
  231. self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined')
  232. def test_builtin_with_settings(self):
  233. "fulldefault: django-admin builtin commands succeed if a settings file is provided"
  234. args = ['sqlall','--settings=settings', 'admin_scripts']
  235. out, err = self.run_django_admin(args)
  236. self.assertNoOutput(err)
  237. self.assertOutput(out, 'CREATE TABLE')
  238. def test_builtin_with_environment(self):
  239. "fulldefault: django-admin builtin commands succeed if the environment contains settings"
  240. args = ['sqlall','admin_scripts']
  241. out, err = self.run_django_admin(args,'settings')
  242. self.assertNoOutput(err)
  243. self.assertOutput(out, 'CREATE TABLE')
  244. def test_builtin_with_bad_settings(self):
  245. "fulldefault: django-admin builtin commands fail if settings file (from argument) doesn't exist"
  246. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  247. out, err = self.run_django_admin(args)
  248. self.assertNoOutput(out)
  249. self.assertOutput(err, "Could not import settings 'bad_settings'")
  250. def test_builtin_with_bad_environment(self):
  251. "fulldefault: django-admin builtin commands fail if settings file (from environment) doesn't exist"
  252. args = ['sqlall','admin_scripts']
  253. out, err = self.run_django_admin(args,'bad_settings')
  254. self.assertNoOutput(out)
  255. self.assertOutput(err, "Could not import settings 'bad_settings'")
  256. def test_custom_command(self):
  257. "fulldefault: django-admin can't execute user commands unless settings are provided"
  258. args = ['noargs_command']
  259. out, err = self.run_django_admin(args)
  260. self.assertNoOutput(out)
  261. self.assertOutput(err, "Unknown command: 'noargs_command'")
  262. def test_custom_command_with_settings(self):
  263. "fulldefault: django-admin can execute user commands if settings are provided as argument"
  264. args = ['noargs_command', '--settings=settings']
  265. out, err = self.run_django_admin(args)
  266. self.assertNoOutput(err)
  267. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  268. def test_custom_command_with_environment(self):
  269. "fulldefault: django-admin can execute user commands if settings are provided in environment"
  270. args = ['noargs_command']
  271. out, err = self.run_django_admin(args,'settings')
  272. self.assertNoOutput(err)
  273. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  274. class DjangoAdminMinimalSettings(AdminScriptTestCase):
  275. """A series of tests for django-admin.py when using a settings.py file that
  276. doesn't contain the test application.
  277. """
  278. def setUp(self):
  279. self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes'])
  280. def tearDown(self):
  281. self.remove_settings('settings.py')
  282. def test_builtin_command(self):
  283. "minimal: django-admin builtin commands fail with an import error when no settings provided"
  284. args = ['sqlall','admin_scripts']
  285. out, err = self.run_django_admin(args)
  286. self.assertNoOutput(out)
  287. self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined')
  288. def test_builtin_with_settings(self):
  289. "minimal: django-admin builtin commands fail if settings are provided as argument"
  290. args = ['sqlall','--settings=settings', 'admin_scripts']
  291. out, err = self.run_django_admin(args)
  292. self.assertNoOutput(out)
  293. self.assertOutput(err, 'App with label admin_scripts could not be found')
  294. def test_builtin_with_environment(self):
  295. "minimal: django-admin builtin commands fail if settings are provided in the environment"
  296. args = ['sqlall','admin_scripts']
  297. out, err = self.run_django_admin(args,'settings')
  298. self.assertNoOutput(out)
  299. self.assertOutput(err, 'App with label admin_scripts could not be found')
  300. def test_builtin_with_bad_settings(self):
  301. "minimal: django-admin builtin commands fail if settings file (from argument) doesn't exist"
  302. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  303. out, err = self.run_django_admin(args)
  304. self.assertNoOutput(out)
  305. self.assertOutput(err, "Could not import settings 'bad_settings'")
  306. def test_builtin_with_bad_environment(self):
  307. "minimal: django-admin builtin commands fail if settings file (from environment) doesn't exist"
  308. args = ['sqlall','admin_scripts']
  309. out, err = self.run_django_admin(args,'bad_settings')
  310. self.assertNoOutput(out)
  311. self.assertOutput(err, "Could not import settings 'bad_settings'")
  312. def test_custom_command(self):
  313. "minimal: django-admin can't execute user commands unless settings are provided"
  314. args = ['noargs_command']
  315. out, err = self.run_django_admin(args)
  316. self.assertNoOutput(out)
  317. self.assertOutput(err, "Unknown command: 'noargs_command'")
  318. def test_custom_command_with_settings(self):
  319. "minimal: django-admin can't execute user commands, even if settings are provided as argument"
  320. args = ['noargs_command', '--settings=settings']
  321. out, err = self.run_django_admin(args)
  322. self.assertNoOutput(out)
  323. self.assertOutput(err, "Unknown command: 'noargs_command'")
  324. def test_custom_command_with_environment(self):
  325. "minimal: django-admin can't execute user commands, even if settings are provided in environment"
  326. args = ['noargs_command']
  327. out, err = self.run_django_admin(args,'settings')
  328. self.assertNoOutput(out)
  329. self.assertOutput(err, "Unknown command: 'noargs_command'")
  330. class DjangoAdminAlternateSettings(AdminScriptTestCase):
  331. """A series of tests for django-admin.py when using a settings file
  332. with a name other than 'settings.py'.
  333. """
  334. def setUp(self):
  335. self.write_settings('alternate_settings.py')
  336. def tearDown(self):
  337. self.remove_settings('alternate_settings.py')
  338. def test_builtin_command(self):
  339. "alternate: django-admin builtin commands fail with an import error when no settings provided"
  340. args = ['sqlall','admin_scripts']
  341. out, err = self.run_django_admin(args)
  342. self.assertNoOutput(out)
  343. self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined')
  344. def test_builtin_with_settings(self):
  345. "alternate: django-admin builtin commands succeed if settings are provided as argument"
  346. args = ['sqlall','--settings=alternate_settings', 'admin_scripts']
  347. out, err = self.run_django_admin(args)
  348. self.assertNoOutput(err)
  349. self.assertOutput(out, 'CREATE TABLE')
  350. def test_builtin_with_environment(self):
  351. "alternate: django-admin builtin commands succeed if settings are provided in the environment"
  352. args = ['sqlall','admin_scripts']
  353. out, err = self.run_django_admin(args,'alternate_settings')
  354. self.assertNoOutput(err)
  355. self.assertOutput(out, 'CREATE TABLE')
  356. def test_builtin_with_bad_settings(self):
  357. "alternate: django-admin builtin commands fail if settings file (from argument) doesn't exist"
  358. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  359. out, err = self.run_django_admin(args)
  360. self.assertNoOutput(out)
  361. self.assertOutput(err, "Could not import settings 'bad_settings'")
  362. def test_builtin_with_bad_environment(self):
  363. "alternate: django-admin builtin commands fail if settings file (from environment) doesn't exist"
  364. args = ['sqlall','admin_scripts']
  365. out, err = self.run_django_admin(args,'bad_settings')
  366. self.assertNoOutput(out)
  367. self.assertOutput(err, "Could not import settings 'bad_settings'")
  368. def test_custom_command(self):
  369. "alternate: django-admin can't execute user commands unless settings are provided"
  370. args = ['noargs_command']
  371. out, err = self.run_django_admin(args)
  372. self.assertNoOutput(out)
  373. self.assertOutput(err, "Unknown command: 'noargs_command'")
  374. def test_custom_command_with_settings(self):
  375. "alternate: django-admin can execute user commands if settings are provided as argument"
  376. args = ['noargs_command', '--settings=alternate_settings']
  377. out, err = self.run_django_admin(args)
  378. self.assertNoOutput(err)
  379. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  380. def test_custom_command_with_environment(self):
  381. "alternate: django-admin can execute user commands if settings are provided in environment"
  382. args = ['noargs_command']
  383. out, err = self.run_django_admin(args,'alternate_settings')
  384. self.assertNoOutput(err)
  385. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  386. class DjangoAdminMultipleSettings(AdminScriptTestCase):
  387. """A series of tests for django-admin.py when multiple settings files
  388. (including the default 'settings.py') are available. The default settings
  389. file is insufficient for performing the operations described, so the
  390. alternate settings must be used by the running script.
  391. """
  392. def setUp(self):
  393. self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes'])
  394. self.write_settings('alternate_settings.py')
  395. def tearDown(self):
  396. self.remove_settings('settings.py')
  397. self.remove_settings('alternate_settings.py')
  398. def test_builtin_command(self):
  399. "alternate: django-admin builtin commands fail with an import error when no settings provided"
  400. args = ['sqlall','admin_scripts']
  401. out, err = self.run_django_admin(args)
  402. self.assertNoOutput(out)
  403. self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined')
  404. def test_builtin_with_settings(self):
  405. "alternate: django-admin builtin commands succeed if settings are provided as argument"
  406. args = ['sqlall','--settings=alternate_settings', 'admin_scripts']
  407. out, err = self.run_django_admin(args)
  408. self.assertNoOutput(err)
  409. self.assertOutput(out, 'CREATE TABLE')
  410. def test_builtin_with_environment(self):
  411. "alternate: django-admin builtin commands succeed if settings are provided in the environment"
  412. args = ['sqlall','admin_scripts']
  413. out, err = self.run_django_admin(args,'alternate_settings')
  414. self.assertNoOutput(err)
  415. self.assertOutput(out, 'CREATE TABLE')
  416. def test_builtin_with_bad_settings(self):
  417. "alternate: django-admin builtin commands fail if settings file (from argument) doesn't exist"
  418. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  419. out, err = self.run_django_admin(args)
  420. self.assertOutput(err, "Could not import settings 'bad_settings'")
  421. def test_builtin_with_bad_environment(self):
  422. "alternate: django-admin builtin commands fail if settings file (from environment) doesn't exist"
  423. args = ['sqlall','admin_scripts']
  424. out, err = self.run_django_admin(args,'bad_settings')
  425. self.assertNoOutput(out)
  426. self.assertOutput(err, "Could not import settings 'bad_settings'")
  427. def test_custom_command(self):
  428. "alternate: django-admin can't execute user commands unless settings are provided"
  429. args = ['noargs_command']
  430. out, err = self.run_django_admin(args)
  431. self.assertNoOutput(out)
  432. self.assertOutput(err, "Unknown command: 'noargs_command'")
  433. def test_custom_command_with_settings(self):
  434. "alternate: django-admin can't execute user commands, even if settings are provided as argument"
  435. args = ['noargs_command', '--settings=alternate_settings']
  436. out, err = self.run_django_admin(args)
  437. self.assertNoOutput(err)
  438. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  439. def test_custom_command_with_environment(self):
  440. "alternate: django-admin can't execute user commands, even if settings are provided in environment"
  441. args = ['noargs_command']
  442. out, err = self.run_django_admin(args,'alternate_settings')
  443. self.assertNoOutput(err)
  444. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  445. class DjangoAdminSettingsDirectory(AdminScriptTestCase):
  446. """
  447. A series of tests for django-admin.py when the settings file is in a
  448. directory. (see #9751).
  449. """
  450. def setUp(self):
  451. self.write_settings('settings', is_dir=True)
  452. def tearDown(self):
  453. self.remove_settings('settings', is_dir=True)
  454. def test_setup_environ(self):
  455. "directory: startapp creates the correct directory"
  456. test_dir = os.path.dirname(os.path.dirname(__file__))
  457. args = ['startapp','settings_test']
  458. out, err = self.run_django_admin(args,'settings')
  459. self.assertNoOutput(err)
  460. self.assertTrue(os.path.exists(os.path.join(test_dir, 'settings_test')))
  461. shutil.rmtree(os.path.join(test_dir, 'settings_test'))
  462. def test_builtin_command(self):
  463. "directory: django-admin builtin commands fail with an import error when no settings provided"
  464. args = ['sqlall','admin_scripts']
  465. out, err = self.run_django_admin(args)
  466. self.assertNoOutput(out)
  467. self.assertOutput(err, 'environment variable DJANGO_SETTINGS_MODULE is undefined')
  468. def test_builtin_with_bad_settings(self):
  469. "directory: django-admin builtin commands fail if settings file (from argument) doesn't exist"
  470. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  471. out, err = self.run_django_admin(args)
  472. self.assertOutput(err, "Could not import settings 'bad_settings'")
  473. def test_builtin_with_bad_environment(self):
  474. "directory: django-admin builtin commands fail if settings file (from environment) doesn't exist"
  475. args = ['sqlall','admin_scripts']
  476. out, err = self.run_django_admin(args,'bad_settings')
  477. self.assertNoOutput(out)
  478. self.assertOutput(err, "Could not import settings 'bad_settings'")
  479. def test_custom_command(self):
  480. "directory: django-admin can't execute user commands unless settings are provided"
  481. args = ['noargs_command']
  482. out, err = self.run_django_admin(args)
  483. self.assertNoOutput(out)
  484. self.assertOutput(err, "Unknown command: 'noargs_command'")
  485. def test_builtin_with_settings(self):
  486. "directory: django-admin builtin commands succeed if settings are provided as argument"
  487. args = ['sqlall','--settings=settings', 'admin_scripts']
  488. out, err = self.run_django_admin(args)
  489. self.assertNoOutput(err)
  490. self.assertOutput(out, 'CREATE TABLE')
  491. def test_builtin_with_environment(self):
  492. "directory: django-admin builtin commands succeed if settings are provided in the environment"
  493. args = ['sqlall','admin_scripts']
  494. out, err = self.run_django_admin(args,'settings')
  495. self.assertNoOutput(err)
  496. self.assertOutput(out, 'CREATE TABLE')
  497. ##########################################################################
  498. # MANAGE.PY TESTS
  499. # This next series of test classes checks the environment processing
  500. # of the generated manage.py script
  501. ##########################################################################
  502. class ManageNoSettings(AdminScriptTestCase):
  503. "A series of tests for manage.py when there is no settings.py file."
  504. def test_builtin_command(self):
  505. "no settings: manage.py builtin commands fail with an import error when no settings provided"
  506. args = ['sqlall','admin_scripts']
  507. out, err = self.run_manage(args)
  508. self.assertNoOutput(out)
  509. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  510. def test_builtin_with_bad_settings(self):
  511. "no settings: manage.py builtin commands fail if settings file (from argument) doesn't exist"
  512. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  513. out, err = self.run_manage(args)
  514. self.assertNoOutput(out)
  515. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  516. def test_builtin_with_bad_environment(self):
  517. "no settings: manage.py builtin commands fail if settings file (from environment) doesn't exist"
  518. args = ['sqlall','admin_scripts']
  519. out, err = self.run_manage(args,'bad_settings')
  520. self.assertNoOutput(out)
  521. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  522. class ManageDefaultSettings(AdminScriptTestCase):
  523. """A series of tests for manage.py when using a settings.py file that
  524. contains the test application.
  525. """
  526. def setUp(self):
  527. self.write_settings('settings.py')
  528. def tearDown(self):
  529. self.remove_settings('settings.py')
  530. def test_builtin_command(self):
  531. "default: manage.py builtin commands succeed when default settings are appropriate"
  532. args = ['sqlall','admin_scripts']
  533. out, err = self.run_manage(args)
  534. self.assertNoOutput(err)
  535. self.assertOutput(out, 'CREATE TABLE')
  536. def test_builtin_with_settings(self):
  537. "default: manage.py builtin commands succeed if settings are provided as argument"
  538. args = ['sqlall','--settings=settings', 'admin_scripts']
  539. out, err = self.run_manage(args)
  540. self.assertNoOutput(err)
  541. self.assertOutput(out, 'CREATE TABLE')
  542. def test_builtin_with_environment(self):
  543. "default: manage.py builtin commands succeed if settings are provided in the environment"
  544. args = ['sqlall','admin_scripts']
  545. out, err = self.run_manage(args,'settings')
  546. self.assertNoOutput(err)
  547. self.assertOutput(out, 'CREATE TABLE')
  548. def test_builtin_with_bad_settings(self):
  549. "default: manage.py builtin commands succeed if settings file (from argument) doesn't exist"
  550. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  551. out, err = self.run_manage(args)
  552. self.assertNoOutput(out)
  553. self.assertOutput(err, "Could not import settings 'bad_settings'")
  554. def test_builtin_with_bad_environment(self):
  555. "default: manage.py builtin commands fail if settings file (from environment) doesn't exist"
  556. args = ['sqlall','admin_scripts']
  557. out, err = self.run_manage(args,'bad_settings')
  558. self.assertNoOutput(err)
  559. self.assertOutput(out, 'CREATE TABLE')
  560. def test_custom_command(self):
  561. "default: manage.py can execute user commands when default settings are appropriate"
  562. args = ['noargs_command']
  563. out, err = self.run_manage(args)
  564. self.assertNoOutput(err)
  565. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  566. def test_custom_command_with_settings(self):
  567. "default: manage.py can execute user commands when settings are provided as argument"
  568. args = ['noargs_command', '--settings=settings']
  569. out, err = self.run_manage(args)
  570. self.assertNoOutput(err)
  571. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  572. def test_custom_command_with_environment(self):
  573. "default: manage.py can execute user commands when settings are provided in environment"
  574. args = ['noargs_command']
  575. out, err = self.run_manage(args,'settings')
  576. self.assertNoOutput(err)
  577. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  578. class ManageFullPathDefaultSettings(AdminScriptTestCase):
  579. """A series of tests for manage.py when using a settings.py file that
  580. contains the test application specified using a full path.
  581. """
  582. def setUp(self):
  583. self.write_settings('settings.py', ['django.contrib.auth', 'django.contrib.contenttypes', 'regressiontests.admin_scripts'])
  584. def tearDown(self):
  585. self.remove_settings('settings.py')
  586. def test_builtin_command(self):
  587. "fulldefault: manage.py builtin commands succeed when default settings are appropriate"
  588. args = ['sqlall','admin_scripts']
  589. out, err = self.run_manage(args)
  590. self.assertNoOutput(err)
  591. self.assertOutput(out, 'CREATE TABLE')
  592. def test_builtin_with_settings(self):
  593. "fulldefault: manage.py builtin commands succeed if settings are provided as argument"
  594. args = ['sqlall','--settings=settings', 'admin_scripts']
  595. out, err = self.run_manage(args)
  596. self.assertNoOutput(err)
  597. self.assertOutput(out, 'CREATE TABLE')
  598. def test_builtin_with_environment(self):
  599. "fulldefault: manage.py builtin commands succeed if settings are provided in the environment"
  600. args = ['sqlall','admin_scripts']
  601. out, err = self.run_manage(args,'settings')
  602. self.assertNoOutput(err)
  603. self.assertOutput(out, 'CREATE TABLE')
  604. def test_builtin_with_bad_settings(self):
  605. "fulldefault: manage.py builtin commands succeed if settings file (from argument) doesn't exist"
  606. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  607. out, err = self.run_manage(args)
  608. self.assertNoOutput(out)
  609. self.assertOutput(err, "Could not import settings 'bad_settings'")
  610. def test_builtin_with_bad_environment(self):
  611. "fulldefault: manage.py builtin commands fail if settings file (from environment) doesn't exist"
  612. args = ['sqlall','admin_scripts']
  613. out, err = self.run_manage(args,'bad_settings')
  614. self.assertNoOutput(err)
  615. self.assertOutput(out, 'CREATE TABLE')
  616. def test_custom_command(self):
  617. "fulldefault: manage.py can execute user commands when default settings are appropriate"
  618. args = ['noargs_command']
  619. out, err = self.run_manage(args)
  620. self.assertNoOutput(err)
  621. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  622. def test_custom_command_with_settings(self):
  623. "fulldefault: manage.py can execute user commands when settings are provided as argument"
  624. args = ['noargs_command', '--settings=settings']
  625. out, err = self.run_manage(args)
  626. self.assertNoOutput(err)
  627. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  628. def test_custom_command_with_environment(self):
  629. "fulldefault: manage.py can execute user commands when settings are provided in environment"
  630. args = ['noargs_command']
  631. out, err = self.run_manage(args,'settings')
  632. self.assertNoOutput(err)
  633. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  634. class ManageMinimalSettings(AdminScriptTestCase):
  635. """A series of tests for manage.py when using a settings.py file that
  636. doesn't contain the test application.
  637. """
  638. def setUp(self):
  639. self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes'])
  640. def tearDown(self):
  641. self.remove_settings('settings.py')
  642. def test_builtin_command(self):
  643. "minimal: manage.py builtin commands fail with an import error when no settings provided"
  644. args = ['sqlall','admin_scripts']
  645. out, err = self.run_manage(args)
  646. self.assertNoOutput(out)
  647. self.assertOutput(err, 'App with label admin_scripts could not be found')
  648. def test_builtin_with_settings(self):
  649. "minimal: manage.py builtin commands fail if settings are provided as argument"
  650. args = ['sqlall','--settings=settings', 'admin_scripts']
  651. out, err = self.run_manage(args)
  652. self.assertNoOutput(out)
  653. self.assertOutput(err, 'App with label admin_scripts could not be found')
  654. def test_builtin_with_environment(self):
  655. "minimal: manage.py builtin commands fail if settings are provided in the environment"
  656. args = ['sqlall','admin_scripts']
  657. out, err = self.run_manage(args,'settings')
  658. self.assertNoOutput(out)
  659. self.assertOutput(err, 'App with label admin_scripts could not be found')
  660. def test_builtin_with_bad_settings(self):
  661. "minimal: manage.py builtin commands fail if settings file (from argument) doesn't exist"
  662. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  663. out, err = self.run_manage(args)
  664. self.assertNoOutput(out)
  665. self.assertOutput(err, "Could not import settings 'bad_settings'")
  666. def test_builtin_with_bad_environment(self):
  667. "minimal: manage.py builtin commands fail if settings file (from environment) doesn't exist"
  668. args = ['sqlall','admin_scripts']
  669. out, err = self.run_manage(args,'bad_settings')
  670. self.assertNoOutput(out)
  671. self.assertOutput(err, 'App with label admin_scripts could not be found')
  672. def test_custom_command(self):
  673. "minimal: manage.py can't execute user commands without appropriate settings"
  674. args = ['noargs_command']
  675. out, err = self.run_manage(args)
  676. self.assertNoOutput(out)
  677. self.assertOutput(err, "Unknown command: 'noargs_command'")
  678. def test_custom_command_with_settings(self):
  679. "minimal: manage.py can't execute user commands, even if settings are provided as argument"
  680. args = ['noargs_command', '--settings=settings']
  681. out, err = self.run_manage(args)
  682. self.assertNoOutput(out)
  683. self.assertOutput(err, "Unknown command: 'noargs_command'")
  684. def test_custom_command_with_environment(self):
  685. "minimal: manage.py can't execute user commands, even if settings are provided in environment"
  686. args = ['noargs_command']
  687. out, err = self.run_manage(args,'settings')
  688. self.assertNoOutput(out)
  689. self.assertOutput(err, "Unknown command: 'noargs_command'")
  690. class ManageAlternateSettings(AdminScriptTestCase):
  691. """A series of tests for manage.py when using a settings file
  692. with a name other than 'settings.py'.
  693. """
  694. def setUp(self):
  695. self.write_settings('alternate_settings.py')
  696. def tearDown(self):
  697. self.remove_settings('alternate_settings.py')
  698. def test_builtin_command(self):
  699. "alternate: manage.py builtin commands fail with an import error when no default settings provided"
  700. args = ['sqlall','admin_scripts']
  701. out, err = self.run_manage(args)
  702. self.assertNoOutput(out)
  703. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  704. def test_builtin_with_settings(self):
  705. "alternate: manage.py builtin commands fail if settings are provided as argument but no defaults"
  706. args = ['sqlall','--settings=alternate_settings', 'admin_scripts']
  707. out, err = self.run_manage(args)
  708. self.assertNoOutput(out)
  709. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  710. def test_builtin_with_environment(self):
  711. "alternate: manage.py builtin commands fail if settings are provided in the environment but no defaults"
  712. args = ['sqlall','admin_scripts']
  713. out, err = self.run_manage(args,'alternate_settings')
  714. self.assertNoOutput(out)
  715. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  716. def test_builtin_with_bad_settings(self):
  717. "alternate: manage.py builtin commands fail if settings file (from argument) doesn't exist"
  718. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  719. out, err = self.run_manage(args)
  720. self.assertNoOutput(out)
  721. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  722. def test_builtin_with_bad_environment(self):
  723. "alternate: manage.py builtin commands fail if settings file (from environment) doesn't exist"
  724. args = ['sqlall','admin_scripts']
  725. out, err = self.run_manage(args,'bad_settings')
  726. self.assertNoOutput(out)
  727. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  728. def test_custom_command(self):
  729. "alternate: manage.py can't execute user commands"
  730. args = ['noargs_command']
  731. out, err = self.run_manage(args)
  732. self.assertNoOutput(out)
  733. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  734. def test_custom_command_with_settings(self):
  735. "alternate: manage.py can't execute user commands, even if settings are provided as argument"
  736. args = ['noargs_command', '--settings=alternate_settings']
  737. out, err = self.run_manage(args)
  738. self.assertNoOutput(out)
  739. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  740. def test_custom_command_with_environment(self):
  741. "alternate: manage.py can't execute user commands, even if settings are provided in environment"
  742. args = ['noargs_command']
  743. out, err = self.run_manage(args,'alternate_settings')
  744. self.assertNoOutput(out)
  745. self.assertOutput(err, "Can't find the file 'settings.py' in the directory containing './manage.py'")
  746. class ManageMultipleSettings(AdminScriptTestCase):
  747. """A series of tests for manage.py when multiple settings files
  748. (including the default 'settings.py') are available. The default settings
  749. file is insufficient for performing the operations described, so the
  750. alternate settings must be used by the running script.
  751. """
  752. def setUp(self):
  753. self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes'])
  754. self.write_settings('alternate_settings.py')
  755. def tearDown(self):
  756. self.remove_settings('settings.py')
  757. self.remove_settings('alternate_settings.py')
  758. def test_builtin_command(self):
  759. "multiple: manage.py builtin commands fail with an import error when no settings provided"
  760. args = ['sqlall','admin_scripts']
  761. out, err = self.run_manage(args)
  762. self.assertNoOutput(out)
  763. self.assertOutput(err, 'App with label admin_scripts could not be found.')
  764. def test_builtin_with_settings(self):
  765. "multiple: manage.py builtin commands succeed if settings are provided as argument"
  766. args = ['sqlall','--settings=alternate_settings', 'admin_scripts']
  767. out, err = self.run_manage(args)
  768. self.assertNoOutput(err)
  769. self.assertOutput(out, 'CREATE TABLE')
  770. def test_builtin_with_environment(self):
  771. "multiple: manage.py builtin commands fail if settings are provided in the environment"
  772. # FIXME: This doesn't seem to be the correct output.
  773. args = ['sqlall','admin_scripts']
  774. out, err = self.run_manage(args,'alternate_settings')
  775. self.assertNoOutput(out)
  776. self.assertOutput(err, 'App with label admin_scripts could not be found.')
  777. def test_builtin_with_bad_settings(self):
  778. "multiple: manage.py builtin commands fail if settings file (from argument) doesn't exist"
  779. args = ['sqlall','--settings=bad_settings', 'admin_scripts']
  780. out, err = self.run_manage(args)
  781. self.assertNoOutput(out)
  782. self.assertOutput(err, "Could not import settings 'bad_settings'")
  783. def test_builtin_with_bad_environment(self):
  784. "multiple: manage.py builtin commands fail if settings file (from environment) doesn't exist"
  785. args = ['sqlall','admin_scripts']
  786. out, err = self.run_manage(args,'bad_settings')
  787. self.assertNoOutput(out)
  788. self.assertOutput(err, "App with label admin_scripts could not be found")
  789. def test_custom_command(self):
  790. "multiple: manage.py can't execute user commands using default settings"
  791. args = ['noargs_command']
  792. out, err = self.run_manage(args)
  793. self.assertNoOutput(out)
  794. self.assertOutput(err, "Unknown command: 'noargs_command'")
  795. def test_custom_command_with_settings(self):
  796. "multiple: manage.py can execute user commands if settings are provided as argument"
  797. args = ['noargs_command', '--settings=alternate_settings']
  798. out, err = self.run_manage(args)
  799. self.assertNoOutput(err)
  800. self.assertOutput(out, "EXECUTE:NoArgsCommand")
  801. def test_custom_command_with_environment(self):
  802. "multiple: manage.py can execute user commands if settings are provided in environment"
  803. args = ['noargs_command']
  804. out, err = self.run_manage(args,'alternate_settings')
  805. self.assertNoOutput(out)
  806. self.assertOutput(err, "Unknown command: 'noargs_command'")
  807. class ManageSettingsWithImportError(AdminScriptTestCase):
  808. """Tests for manage.py when using the default settings.py file
  809. with an import error. Ticket #14130.
  810. """
  811. def setUp(self):
  812. self.write_settings_with_import_error('settings.py')
  813. def tearDown(self):
  814. self.remove_settings('settings.py')
  815. def write_settings_with_import_error(self, filename, apps=None, is_dir=False, sdict=None):
  816. test_dir = os.path.dirname(os.path.dirname(__file__))
  817. if is_dir:
  818. settings_dir = os.path.join(test_dir,filename)
  819. os.mkdir(settings_dir)
  820. settings_file = open(os.path.join(settings_dir,'__init__.py'), 'w')
  821. else:
  822. settings_file = open(os.path.join(test_dir, filename), 'w')
  823. settings_file.write('# Settings file automatically generated by regressiontests.admin_scripts test case\n')
  824. settings_file.write('# The next line will cause an import error:\nimport foo42bar\n')
  825. settings_file.close()
  826. def test_builtin_command(self):
  827. "import error: manage.py builtin commands shows useful diagnostic info when settings with import errors is provided"
  828. args = ['sqlall','admin_scripts']
  829. out, err = self.run_manage(args)
  830. self.assertNoOutput(out)
  831. self.assertOutput(err, "ImportError: No module named foo42bar")
  832. class ManageValidate(AdminScriptTestCase):
  833. def tearDown(self):
  834. self.remove_settings('settings.py')
  835. def test_nonexistent_app(self):
  836. "manage.py validate reports an error on a non-existent app in INSTALLED_APPS"
  837. self.write_settings('settings.py', apps=['admin_scriptz.broken_app'], sdict={'USE_I18N': False})
  838. args = ['validate']
  839. out, err = self.run_manage(args)
  840. self.assertNoOutput(out)
  841. self.assertOutput(err, 'No module named admin_scriptz')
  842. def test_broken_app(self):
  843. "manage.py validate reports an ImportError if an app's models.py raises one on import"
  844. self.write_settings('settings.py', apps=['admin_scripts.broken_app'])
  845. args = ['validate']
  846. out, err = self.run_manage(args)
  847. self.assertNoOutput(out)
  848. self.assertOutput(err, 'ImportError')
  849. def test_complex_app(self):
  850. "manage.py validate does not raise an ImportError validating a complex app with nested calls to load_app"
  851. self.write_settings('settings.py',
  852. apps=['admin_scripts.complex_app', 'admin_scripts.simple_app'],
  853. sdict={'DEBUG': True})
  854. args = ['validate']
  855. out, err = self.run_manage(args)
  856. self.assertNoOutput(err)
  857. self.assertOutput(out, '0 errors found')
  858. def test_app_with_import(self):
  859. "manage.py validate does not raise errors when an app imports a base class that itself has an abstract base"
  860. self.write_settings('settings.py',
  861. apps=['admin_scripts.app_with_import', 'django.contrib.comments'],
  862. sdict={'DEBUG': True})
  863. args = ['validate']
  864. out, err = self.run_manage(args)
  865. self.assertNoOutput(err)
  866. self.assertOutput(out, '0 errors found')
  867. class ManageRunserver(AdminScriptTestCase):
  868. def setUp(self):
  869. from django.core.management.commands.runserver import BaseRunserverCommand
  870. def monkey_run(*args, **options): return
  871. self.cmd = BaseRunserverCommand()
  872. self.cmd.run = monkey_run
  873. def assertServerSettings(self, addr, port, ipv6=None, raw_ipv6=False):
  874. self.assertEqual(self.cmd.addr, addr)
  875. self.assertEqual(self.cmd.port, port)
  876. self.assertEqual(self.cmd.use_ipv6, ipv6)
  877. self.assertEqual(self.cmd._raw_ipv6, raw_ipv6)
  878. def test_runserver_addrport(self):
  879. self.cmd.handle()
  880. self.assertServerSettings('127.0.0.1', '8000')
  881. self.cmd.handle(addrport="1.2.3.4:8000")
  882. self.assertServerSettings('1.2.3.4', '8000')
  883. self.cmd.handle(addrport="7000")
  884. self.assertServerSettings('127.0.0.1', '7000')
  885. # IPv6
  886. self.cmd.handle(addrport="", use_ipv6=True)
  887. self.assertServerSettings('::1', '8000', ipv6=True, raw_ipv6=True)
  888. self.cmd.handle(addrport="7000", use_ipv6=True)
  889. self.assertServerSettings('::1', '7000', ipv6=True, raw_ipv6=True)
  890. self.cmd.handle(addrport="[2001:0db8:1234:5678::9]:7000")
  891. self.assertServerSettings('2001:0db8:1234:5678::9', '7000', ipv6=True, raw_ipv6=True)
  892. # Hostname
  893. self.cmd.handle(addrport="localhost:8000")
  894. self.assertServerSettings('localhost', '8000')
  895. self.cmd.handle(addrport="test.domain.local:7000")
  896. self.assertServerSettings('test.domain.local', '7000')
  897. self.cmd.handle(addrport="test.domain.local:7000", use_ipv6=True)
  898. self.assertServerSettings('test.domain.local', '7000', ipv6=True)
  899. # Potentially ambiguous
  900. # Only 4 characters, all of which could be in an ipv6 address
  901. self.cmd.handle(addrport="beef:7654")
  902. self.assertServerSettings('beef', '7654')
  903. # Uses only characters that could be in an ipv6 address
  904. self.cmd.handle(addrport="deadbeef:7654")
  905. self.assertServerSettings('deadbeef', '7654')
  906. ##########################################################################
  907. # COMMAND PROCESSING TESTS
  908. # Check that user-space commands are correctly handled - in particular,
  909. # that arguments to the commands are correctly parsed and processed.
  910. ##########################################################################
  911. class CommandTypes(AdminScriptTestCase):
  912. "Tests for the various types of base command types that can be defined."
  913. def setUp(self):
  914. self.write_settings('settings.py')
  915. def tearDown(self):
  916. self.remove_settings('settings.py')
  917. def test_version(self):
  918. "--version is handled as a special case"
  919. args = ['--version']
  920. out, err = self.run_manage(args)
  921. self.assertNoOutput(err)
  922. # Only check the first part of the version number
  923. self.assertOutput(out, get_version().split('-')[0])
  924. def test_help(self):
  925. "--help is handled as a special case"
  926. args = ['--help']
  927. out, err = self.run_manage(args)
  928. if sys.version_info < (2, 5):
  929. self.assertOutput(out, "usage: manage.py subcommand [options] [args]")
  930. else:
  931. self.assertOutput(out, "Usage: manage.py subcommand [options] [args]")
  932. self.assertOutput(err, "Type 'manage.py help <subcommand>' for help on a specific subcommand.")
  933. def test_short_help(self):
  934. "-h is handled as a short form of --help"
  935. args = ['-h']
  936. out, err = self.run_manage(args)
  937. if sys.version_info < (2, 5):
  938. self.assertOutput(out, "usage: manage.py subcommand [options] [args]")
  939. else:
  940. self.assertOutput(out, "Usage: manage.py subcommand [options] [args]")
  941. self.assertOutput(err, "Type 'manage.py help <subcommand>' for help on a specific subcommand.")
  942. def test_specific_help(self):
  943. "--help can be used on a specific command"
  944. args = ['sqlall','--help']
  945. out, err = self.run_manage(args)
  946. self.assertNoOutput(err)
  947. self.assertOutput(out, "Prints the CREATE TABLE, custom SQL and CREATE INDEX SQL statements for the given model module name(s).")
  948. def test_base_command(self):
  949. "User BaseCommands can execute when a label is provided"
  950. args = ['base_command','testlabel']
  951. out, err = self.run_manage(args)
  952. self.assertNoOutput(err)
  953. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', '1'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  954. def test_base_command_no_label(self):
  955. "User BaseCommands can execute when no labels are provided"
  956. args = ['base_command']
  957. out, err = self.run_manage(args)
  958. self.assertNoOutput(err)
  959. self.assertOutput(out, "EXECUTE:BaseCommand labels=(), options=[('option_a', '1'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  960. def test_base_command_multiple_label(self):
  961. "User BaseCommands can execute when no labels are provided"
  962. args = ['base_command','testlabel','anotherlabel']
  963. out, err = self.run_manage(args)
  964. self.assertNoOutput(err)
  965. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel', 'anotherlabel'), options=[('option_a', '1'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  966. def test_base_command_with_option(self):
  967. "User BaseCommands can execute with options when a label is provided"
  968. args = ['base_command','testlabel','--option_a=x']
  969. out, err = self.run_manage(args)
  970. self.assertNoOutput(err)
  971. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  972. def test_base_command_with_options(self):
  973. "User BaseCommands can execute with multiple options when a label is provided"
  974. args = ['base_command','testlabel','-a','x','--option_b=y']
  975. out, err = self.run_manage(args)
  976. self.assertNoOutput(err)
  977. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', 'y'), ('option_c', '3'), ('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  978. def test_noargs(self):
  979. "NoArg Commands can be executed"
  980. args = ['noargs_command']
  981. out, err = self.run_manage(args)
  982. self.assertNoOutput(err)
  983. self.assertOutput(out, "EXECUTE:NoArgsCommand options=[('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  984. def test_noargs_with_args(self):
  985. "NoArg Commands raise an error if an argument is provided"
  986. args = ['noargs_command','argument']
  987. out, err = self.run_manage(args)
  988. self.assertOutput(err, "Error: Command doesn't accept any arguments")
  989. def test_app_command(self):
  990. "User AppCommands can execute when a single app name is provided"
  991. args = ['app_command', 'auth']
  992. out, err = self.run_manage(args)
  993. self.assertNoOutput(err)
  994. self.assertOutput(out, "EXECUTE:AppCommand app=<module 'django.contrib.auth.models'")
  995. self.assertOutput(out, os.sep.join(['django','contrib','auth','models.py']))
  996. self.assertOutput(out, "'>, options=[('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  997. def test_app_command_no_apps(self):
  998. "User AppCommands raise an error when no app name is provided"
  999. args = ['app_command']
  1000. out, err = self.run_manage(args)
  1001. self.assertOutput(err, 'Error: Enter at least one appname.')
  1002. def test_app_command_multiple_apps(self):
  1003. "User AppCommands raise an error when multiple app names are provided"
  1004. args = ['app_command','auth','contenttypes']
  1005. out, err = self.run_manage(args)
  1006. self.assertNoOutput(err)
  1007. self.assertOutput(out, "EXECUTE:AppCommand app=<module 'django.contrib.auth.models'")
  1008. self.assertOutput(out, os.sep.join(['django','contrib','auth','models.py']))
  1009. self.assertOutput(out, "'>, options=[('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  1010. self.assertOutput(out, "EXECUTE:AppCommand app=<module 'django.contrib.contenttypes.models'")
  1011. self.assertOutput(out, os.sep.join(['django','contrib','contenttypes','models.py']))
  1012. self.assertOutput(out, "'>, options=[('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  1013. def test_app_command_invalid_appname(self):
  1014. "User AppCommands can execute when a single app name is provided"
  1015. args = ['app_command', 'NOT_AN_APP']
  1016. out, err = self.run_manage(args)
  1017. self.assertOutput(err, "App with label NOT_AN_APP could not be found")
  1018. def test_app_command_some_invalid_appnames(self):
  1019. "User AppCommands can execute when some of the provided app names are invalid"
  1020. args = ['app_command', 'auth', 'NOT_AN_APP']
  1021. out, err = self.run_manage(args)
  1022. self.assertOutput(err, "App with label NOT_AN_APP could not be found")
  1023. def test_label_command(self):
  1024. "User LabelCommands can execute when a label is provided"
  1025. args = ['label_command','testlabel']
  1026. out, err = self.run_manage(args)
  1027. self.assertNoOutput(err)
  1028. self.assertOutput(out, "EXECUTE:LabelCommand label=testlabel, options=[('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  1029. def test_label_command_no_label(self):
  1030. "User LabelCommands raise an error if no label is provided"
  1031. args = ['label_command']
  1032. out, err = self.run_manage(args)
  1033. self.assertOutput(err, 'Enter at least one label')
  1034. def test_label_command_multiple_label(self):
  1035. "User LabelCommands are executed multiple times if multiple labels are provided"
  1036. args = ['label_command','testlabel','anotherlabel']
  1037. out, err = self.run_manage(args)
  1038. self.assertNoOutput(err)
  1039. self.assertOutput(out, "EXECUTE:LabelCommand label=testlabel, options=[('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  1040. self.assertOutput(out, "EXECUTE:LabelCommand label=anotherlabel, options=[('pythonpath', None), ('settings', None), ('traceback', None), ('verbosity', '1')]")
  1041. class ArgumentOrder(AdminScriptTestCase):
  1042. """Tests for 2-stage argument parsing scheme.
  1043. django-admin command arguments are parsed in 2 parts; the core arguments
  1044. (--settings, --traceback and --pythonpath) are parsed using a Lax parser.
  1045. This Lax parser ignores any unknown options. Then the full settings are
  1046. passed to the command parser, which extracts commands of interest to the
  1047. individual command.
  1048. """
  1049. def setUp(self):
  1050. self.write_settings('settings.py', apps=['django.contrib.auth','django.contrib.contenttypes'])
  1051. self.write_settings('alternate_settings.py')
  1052. def tearDown(self):
  1053. self.remove_settings('settings.py')
  1054. self.remove_settings('alternate_settings.py')
  1055. def test_setting_then_option(self):
  1056. "Options passed after settings are correctly handled"
  1057. args = ['base_command','testlabel','--settings=alternate_settings','--option_a=x']
  1058. out, err = self.run_manage(args)
  1059. self.assertNoOutput(err)
  1060. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', 'alternate_settings'), ('traceback', None), ('verbosity', '1')]")
  1061. def test_setting_then_short_option(self):
  1062. "Short options passed after settings are correctly handled"
  1063. args = ['base_command','testlabel','--settings=alternate_settings','--option_a=x']
  1064. out, err = self.run_manage(args)
  1065. self.assertNoOutput(err)
  1066. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', 'alternate_settings'), ('traceback', None), ('verbosity', '1')]")
  1067. def test_option_then_setting(self):
  1068. "Options passed before settings are correctly handled"
  1069. args = ['base_command','testlabel','--option_a=x','--settings=alternate_settings']
  1070. out, err = self.run_manage(args)
  1071. self.assertNoOutput(err)
  1072. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', 'alternate_settings'), ('traceback', None), ('verbosity', '1')]")
  1073. def test_short_option_then_setting(self):
  1074. "Short options passed before settings are correctly handled"
  1075. args = ['base_command','testlabel','-a','x','--settings=alternate_settings']
  1076. out, err = self.run_manage(args)
  1077. self.assertNoOutput(err)
  1078. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', '2'), ('option_c', '3'), ('pythonpath', None), ('settings', 'alternate_settings'), ('traceback', None), ('verbosity', '1')]")
  1079. def test_option_then_setting_then_option(self):
  1080. "Options are correctly handled when they are passed before and after a setting"
  1081. args = ['base_command','testlabel','--option_a=x','--settings=alternate_settings','--option_b=y']
  1082. out, err = self.run_manage(args)
  1083. self.assertNoOutput(err)
  1084. self.assertOutput(out, "EXECUTE:BaseCommand labels=('testlabel',), options=[('option_a', 'x'), ('option_b', 'y'), ('option_c', '3'), ('pythonpath', None), ('settings', 'alternate_settings'), ('traceback', None), ('verbosity', '1')]")