Used call_command return value in staticfiles tests

Refs #26190.
This commit is contained in:
Claude Paroz 2016-02-13 18:20:29 +01:00
parent b46c0ea6c8
commit 269b5f262c
3 changed files with 11 additions and 20 deletions

View File

@ -194,7 +194,7 @@ class Command(BaseCommand):
', %s post-processed' ', %s post-processed'
% post_processed_count or ''), % post_processed_count or ''),
} }
self.stdout.write(summary) return summary
def log(self, msg, level=2): def log(self, msg, level=2):
""" """

View File

@ -22,7 +22,7 @@ class Command(LabelCommand):
result = finders.find(path, all=options['all']) result = finders.find(path, all=options['all'])
path = force_text(path) path = force_text(path)
if verbosity >= 2: if verbosity >= 2:
searched_locations = ("Looking in the following locations:\n %s" % searched_locations = ("\nLooking in the following locations:\n %s" %
"\n ".join(force_text(location) "\n ".join(force_text(location)
for location in finders.searched_locations)) for location in finders.searched_locations))
else: else:
@ -33,7 +33,7 @@ class Command(LabelCommand):
result = (force_text(os.path.realpath(path)) for path in result) result = (force_text(os.path.realpath(path)) for path in result)
if verbosity >= 1: if verbosity >= 1:
file_list = '\n '.join(result) file_list = '\n '.join(result)
return ("Found '%s' here:\n %s\n%s" % return ("Found '%s' here:\n %s%s" %
(path, file_list, searched_locations)) (path, file_list, searched_locations))
else: else:
return '\n'.join(result) return '\n'.join(result)

View File

@ -39,21 +39,16 @@ class TestFindStatic(CollectionTestCase, TestDefaults):
Test ``findstatic`` management command. Test ``findstatic`` management command.
""" """
def _get_file(self, filepath): def _get_file(self, filepath):
out = six.StringIO() path = call_command('findstatic', filepath, all=False, verbosity=0, stdout=six.StringIO())
call_command('findstatic', filepath, all=False, verbosity=0, stdout=out) with codecs.open(force_text(path), "r", "utf-8") as f:
out.seek(0)
lines = [l.strip() for l in out.readlines()]
with codecs.open(force_text(lines[0].strip()), "r", "utf-8") as f:
return f.read() return f.read()
def test_all_files(self): def test_all_files(self):
""" """
Test that findstatic returns all candidate files if run without --first and -v1. Test that findstatic returns all candidate files if run without --first and -v1.
""" """
out = six.StringIO() result = call_command('findstatic', 'test/file.txt', verbosity=1, stdout=six.StringIO())
call_command('findstatic', 'test/file.txt', verbosity=1, stdout=out) lines = [l.strip() for l in result.split('\n')]
out.seek(0)
lines = [l.strip() for l in out.readlines()]
self.assertEqual(len(lines), 3) # three because there is also the "Found <file> here" line self.assertEqual(len(lines), 3) # three because there is also the "Found <file> here" line
self.assertIn('project', force_text(lines[1])) self.assertIn('project', force_text(lines[1]))
self.assertIn('apps', force_text(lines[2])) self.assertIn('apps', force_text(lines[2]))
@ -62,10 +57,8 @@ class TestFindStatic(CollectionTestCase, TestDefaults):
""" """
Test that findstatic returns all candidate files if run without --first and -v0. Test that findstatic returns all candidate files if run without --first and -v0.
""" """
out = six.StringIO() result = call_command('findstatic', 'test/file.txt', verbosity=0, stdout=six.StringIO())
call_command('findstatic', 'test/file.txt', verbosity=0, stdout=out) lines = [l.strip() for l in result.split('\n')]
out.seek(0)
lines = [l.strip() for l in out.readlines()]
self.assertEqual(len(lines), 2) self.assertEqual(len(lines), 2)
self.assertIn('project', force_text(lines[0])) self.assertIn('project', force_text(lines[0]))
self.assertIn('apps', force_text(lines[1])) self.assertIn('apps', force_text(lines[1]))
@ -75,10 +68,8 @@ class TestFindStatic(CollectionTestCase, TestDefaults):
Test that findstatic returns all candidate files if run without --first and -v2. Test that findstatic returns all candidate files if run without --first and -v2.
Also, test that findstatic returns the searched locations with -v2. Also, test that findstatic returns the searched locations with -v2.
""" """
out = six.StringIO() result = call_command('findstatic', 'test/file.txt', verbosity=2, stdout=six.StringIO())
call_command('findstatic', 'test/file.txt', verbosity=2, stdout=out) lines = [l.strip() for l in result.split('\n')]
out.seek(0)
lines = [l.strip() for l in out.readlines()]
self.assertIn('project', force_text(lines[1])) self.assertIn('project', force_text(lines[1]))
self.assertIn('apps', force_text(lines[2])) self.assertIn('apps', force_text(lines[2]))
self.assertIn("Looking in the following locations:", force_text(lines[3])) self.assertIn("Looking in the following locations:", force_text(lines[3]))