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_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_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 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_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 _assert_models_present_under_names(self, group, group_col, name,
name_col, note, note_col,
system_user, unique_model,
permission, test_app_permission,
unique_test_app_model):
command = shell_plus.Command()
self.imported_objects = command.get_imported_objects({})
self.output = sys.stdout.getvalue()
self._assert_imported_under_names(Group, group)
self._assert_imported_under_names(Group_Col, group_col)
self._assert_imported_under_names(Name, name)
self._assert_imported_under_names(Name_Col, name_col)
self._assert_imported_under_names(Note, note)
self._assert_imported_under_names(Note_Col, note_col)
self._assert_imported_under_names(SystemUser, system_user)
self._assert_imported_under_names(UniqueModel, unique_model)
self._assert_imported_under_names(Permission, permission)
self._assert_imported_under_names(TAPermission, test_app_permission)
self._assert_imported_under_names(UniqueTestAppModel,
unique_test_app_model)
def run_shell_plus(self):
command = shell_plus.Command()
self.imported_objects = command.get_imported_objects({})
self.output = sys.stdout.getvalue()
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