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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
# -*- coding: utf-8 -*-
import os
import inspect
from django.db.models import Model
from django.test import override_settings
from django_extensions.management.commands import shell_plus
from django_extensions.management.shells import SHELL_PLUS_DJANGO_IMPORTS
def test_shell_plus_plain_startup():
parser = shell_plus.Command().create_parser("test", "shell_plus")
args = ["--plain"]
options = parser.parse_args(args=args)
command = shell_plus.Command()
command.tests_mode = True
retcode = command.handle(**vars(options))
assert retcode == 130
def test_shell_plus_plain_startup_with_pythonrc(monkeypatch):
parser = shell_plus.Command().create_parser("test", "shell_plus")
args = ["--plain", "--use-pythonrc"]
options = parser.parse_args(args=args)
tests_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
pythonrc_file = os.path.join(tests_dir, 'pythonrc.py')
assert os.path.isfile(pythonrc_file)
monkeypatch.setenv('PYTHONSTARTUP', pythonrc_file)
command = shell_plus.Command()
command.tests_mode = True
retcode = command.handle(**vars(options))
assert retcode == 130
imported_objects = command.tests_imported_objects
assert 'pythonrc_test_func' in imported_objects
assert imported_objects['pythonrc_test_func']() == "pythonrc was loaded"
def test_shell_plus_plain_loading_standard_django_imports(monkeypatch):
parser = shell_plus.Command().create_parser("test", "shell_plus")
args = ["--plain"]
options = parser.parse_args(args=args)
command = shell_plus.Command()
command.tests_mode = True
retcode = command.handle(**vars(options))
assert retcode == 130
imported_objects = command.tests_imported_objects
assert 'get_user_model' in imported_objects
assert 'settings' in imported_objects
assert 'timezone' in imported_objects
def test_shell_plus_plain_loading_django_extensions_modules(monkeypatch):
parser = shell_plus.Command().create_parser("test", "shell_plus")
args = ["--plain"]
options = parser.parse_args(args=args)
command = shell_plus.Command()
command.tests_mode = True
retcode = command.handle(**vars(options))
assert retcode == 130
imported_objects = command.tests_imported_objects
assert 'Club' in imported_objects
assert 'UniqueTestAppModel' in imported_objects
assert 'RandomCharTestModel' in imported_objects
def test_shell_plus_get_imported_objects():
command = shell_plus.Command()
objs = command.get_imported_objects({})
for items in SHELL_PLUS_DJANGO_IMPORTS.values():
for item in items:
assert item in objs, "%s not loaded by get_imported_objects()" % item
def assert_should_models_be_imported(should_be, cli_arguments=None):
command = shell_plus.Command()
objs = command.get_imported_objects(cli_arguments or {})
imported_models = filter(lambda imported: inspect.isclass(imported) and issubclass(imported, Model), objs.values())
assert bool(list(imported_models)) == should_be
def test_shell_plus_loading_models():
assert_should_models_be_imported(True)
def test_shell_plus_skipping_models_import_cli():
assert_should_models_be_imported(False, cli_arguments={'dont_load': ['*']})
@override_settings(SHELL_PLUS_DONT_LOAD=['*'])
def test_shell_plus_skipping_models_import_settings():
assert_should_models_be_imported(False)
|
