1234567891011121314151617 |
- from django.core.management import call_command
- from django.test import TestCase
- from django.utils.six import StringIO
- from ..management.commands import updateblankavatar
- class UpdateBlankAvatarTests(TestCase):
- def test_regen_blank_avatar(self):
- """command regens blank avatar"""
- command = updateblankavatar.Command()
- out = StringIO()
- call_command(command, stdout=out)
- command_output = out.getvalue().splitlines()[0].strip()
- self.assertEqual(command_output, 'Blank avatar was updated')
|