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
|
# -*- coding: utf-8 -*-
import os
import shutil
from django.conf import settings
from django.core.management import call_command
from django.test import TestCase
from io import StringIO
from unittest.mock import patch
TEST_APP = "testapp_with_appconfig"
class CreateCommandTests(TestCase):
"""Tests for create_command command."""
def setUp(self): # noqa
self.management_command_path = os.path.join(
settings.BASE_DIR, "tests/{}/management".format(TEST_APP)
)
self.command_template_path = os.path.join(
settings.BASE_DIR, "django_extensions/conf/command_template"
)
self.files = [
"__init__.py",
"commands/__init__.py",
"commands/sample.py",
]
def tearDown(self): # noqa
shutil.rmtree(self.management_command_path, ignore_errors=True)
shutil.rmtree(
os.path.join(self.command_template_path, ".hidden"), ignore_errors=True
)
test_pyc_path = os.path.join(self.command_template_path, "test.pyc")
if os.path.isfile(test_pyc_path):
os.remove(test_pyc_path)
def _create_management_command_with_empty_files(self):
os.mkdir(self.management_command_path)
os.mkdir(os.path.join(self.management_command_path, "commands"))
for f in self.files:
with open(os.path.join(self.management_command_path, f), "a"):
pass
def _create__pycache__in_command_template_directory(self):
with open(os.path.join(self.command_template_path, "test.pyc"), "a"):
pass
def _create_hidden_directory_in_command_template_directory(self):
os.mkdir(os.path.join(self.command_template_path, ".hidden"))
@patch("sys.stdout", new_callable=StringIO)
def test_should_print_management_command_files_only_on_dry_run(self, m_stdout): # noqa
call_command("create_command", TEST_APP, "--dry-run", verbosity=2)
for f in self.files:
filepath = os.path.join(self.management_command_path, f)
self.assertIn(filepath, m_stdout.getvalue())
self.assertFalse(os.path.isfile(filepath))
@patch("sys.stdout", new_callable=StringIO)
def test_should_create_management_command_files_and_print_filepaths(self, m_stdout): # noqa
call_command("create_command", TEST_APP, verbosity=2)
for f in self.files:
filepath = os.path.join(self.management_command_path, f)
self.assertIn(filepath, m_stdout.getvalue())
self.assertTrue(os.path.isfile(filepath))
@patch("sys.stdout", new_callable=StringIO)
def test_should_print_that_filepaths_already_exists(self, m_stdout): # noqa
self._create_management_command_with_empty_files()
call_command("create_command", TEST_APP, verbosity=2)
for f in self.files:
filepath = os.path.join(self.management_command_path, f)
self.assertIn("{} already exists".format(filepath), m_stdout.getvalue())
self.assertTrue(os.path.isfile(filepath))
self.assertEqual(os.path.getsize(filepath), 0)
@patch("sys.stderr", new_callable=StringIO)
@patch("django_extensions.management.commands.create_command._make_writeable") # noqa
def test_should_print_error_on_OSError_exception(self, m__make_writeable, m_stderr): # noqa
m__make_writeable.side_effect = OSError
self._create__pycache__in_command_template_directory()
self._create_hidden_directory_in_command_template_directory()
call_command("create_command", TEST_APP)
for f in self.files:
filepath = os.path.join(self.management_command_path, f)
self.assertIn(
"Notice: Couldn't set permission bits on {}. You're probably using an uncommon filesystem setup. No problem.\n".format(
filepath
), # noqa
m_stderr.getvalue(),
)
|
