django/tests/admin_scripts/test_django_admin_py.py

import os
import subprocess
import sys

import django
from django.test import SimpleTestCase


class DeprecationTests(SimpleTestCase):
    DEPRECATION_MESSAGE = (
        b'RemovedInDjango40Warning: django-admin.py is deprecated in favor of '
        b'django-admin.'
    )

    def setUp(self):
        script_dir = os.path.abspath(os.path.join(os.path.dirname(django.__file__), 'bin'))
        self.django_admin_py = os.path.join(script_dir, 'django-admin.py')

    def _run_test(self, args):
        p = subprocess.run(
            [sys.executable, *args],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            check=True,
        )
        return p.stdout, p.stderr

    def test_django_admin_py_deprecated(self):
        _, err = self._run_test(['-Wd', self.django_admin_py, '--version'])
        self.assertIn(self.DEPRECATION_MESSAGE, err)

    def test_main_not_deprecated(self):
        _, err = self._run_test(['-Wd', '-m', 'django', '--version'])
        self.assertNotIn(self.DEPRECATION_MESSAGE, err)

    def test_django_admin_py_equivalent_main(self):
        django_admin_py_out, _ = self._run_test([self.django_admin_py, '--version'])
        django_out, _ = self._run_test(['-m', 'django', '--version'])
        self.assertEqual(django_admin_py_out, django_out)