1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
from __future__ import annotations
import sys
from code import InteractiveConsole
from unittest import mock
import django
import pytest
from django.core.management import call_command
from django.test import SimpleTestCase
from django.test.utils import captured_stdin, captured_stdout
from django_rich.management.commands.shell import Command
class ShellCommandTestCase(SimpleTestCase):
@pytest.mark.skipif(
sys.platform == "win32",
reason="Windows select() doesn't support file descriptors.",
)
@mock.patch("django.core.management.commands.shell.select")
def test_pretty(self, select):
console = InteractiveConsole()
with captured_stdin() as _, captured_stdout() as stdout:
call_command("shell", "-i", "python")
console.push("from rich.panel import Panel\n")
console.push('Panel.fit("hi!")\n')
lines = stdout.getvalue().splitlines()
if django.VERSION >= (6, 0):
assert lines == [
"10 objects imported automatically (use -v 2 for details).",
"",
"╭─────╮",
"│ hi! │",
"╰─────╯",
]
elif django.VERSION >= (5, 2):
assert lines == [
"4 objects imported automatically (use -v 2 for details).",
"",
"╭─────╮",
"│ hi! │",
"╰─────╯",
]
else:
assert lines == [
"╭─────╮",
"│ hi! │",
"╰─────╯",
]
@pytest.mark.skipif(
django.VERSION < (5, 2),
reason="Django 5.2 added the get_auto_imports() method",
)
def test_get_auto_imports(self):
command = Command()
auto_imports = command.get_auto_imports()
assert auto_imports[-4:] == [
"rich.inspect",
"rich.print",
"rich.print_json",
"rich.pretty.pprint",
]
|
