mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2024-12-22 17:16:24 +00:00
Code Issues 25 Releases Wiki Activity

Fixed #35517, Refs #35515 -- Improved test coverage of shell command.

Browse Source
This commit is contained in:
Salvo Polizzi 2024-06-11 10:08:02 +02:00 committed by Sarah Boyce
parent f0d05a747f
commit dfac15d570
1 changed files with 29 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tests/shell/tests.py
View File

@ -4,6 +4,7 @@ from unittest import mock
from django import __version__ from django import __version__
from django.core.management import CommandError, call_command from django.core.management import CommandError, call_command
from django.core.management.commands import shell
from django.test import SimpleTestCase from django.test import SimpleTestCase
from django.test.utils import captured_stdin, captured_stdout from django.test.utils import captured_stdin, captured_stdout
@ -70,6 +71,15 @@ class ShellCommandTestCase(SimpleTestCase):
call_command("shell") call_command("shell")
self.assertEqual(stdout.getvalue().strip(), __version__) self.assertEqual(stdout.getvalue().strip(), __version__)
def test_ipython(self):
cmd = shell.Command()
mock_ipython = mock.Mock(start_ipython=mock.MagicMock())
with mock.patch.dict(sys.modules, {"IPython": mock_ipython}):
cmd.ipython({})
self.assertEqual(mock_ipython.start_ipython.mock_calls, [mock.call(argv=[])])
@mock.patch("django.core.management.commands.shell.select.select") # [1] @mock.patch("django.core.management.commands.shell.select.select") # [1]
@mock.patch.dict("sys.modules", {"IPython": None}) @mock.patch.dict("sys.modules", {"IPython": None})
def test_shell_with_ipython_not_installed(self, select): def test_shell_with_ipython_not_installed(self, select):
@ -79,6 +89,15 @@ class ShellCommandTestCase(SimpleTestCase):
): ):
call_command("shell", interface="ipython") call_command("shell", interface="ipython")
def test_bpython(self):
cmd = shell.Command()
mock_bpython = mock.Mock(embed=mock.MagicMock())
with mock.patch.dict(sys.modules, {"bpython": mock_bpython}):
cmd.bpython({})
self.assertEqual(mock_bpython.embed.mock_calls, [mock.call()])
@mock.patch("django.core.management.commands.shell.select.select") # [1] @mock.patch("django.core.management.commands.shell.select.select") # [1]
@mock.patch.dict("sys.modules", {"bpython": None}) @mock.patch.dict("sys.modules", {"bpython": None})
def test_shell_with_bpython_not_installed(self, select): def test_shell_with_bpython_not_installed(self, select):
@ -88,7 +107,16 @@ class ShellCommandTestCase(SimpleTestCase):
): ):
call_command("shell", interface="bpython") call_command("shell", interface="bpython")
# [1] Patch select to prevent tests failing when when the test suite is run def test_python(self):
cmd = shell.Command()
mock_code = mock.Mock(interact=mock.MagicMock())
with mock.patch.dict(sys.modules, {"code": mock_code}):
cmd.python({"no_startup": True})
self.assertEqual(mock_code.interact.mock_calls, [mock.call(local={})])
# [1] Patch select to prevent tests failing when the test suite is run
# in parallel mode. The tests are run in a subprocess and the subprocess's # in parallel mode. The tests are run in a subprocess and the subprocess's
# stdin is closed and replaced by /dev/null. Reading from /dev/null always # stdin is closed and replaced by /dev/null. Reading from /dev/null always
# returns EOF and so select always shows that sys.stdin is ready to read. # returns EOF and so select always shows that sys.stdin is ready to read.