| 123456789101112131415161718192021 |
- from io import StringIO
- from django.core.management import call_command
- from django.test import TestCase
- from ..management.commands import loadavatargallery
- from ..models import AvatarGallery
- class LoadAvatarGalleryTests(TestCase):
- def test_regen_blank_avatar(self):
- """command regens blank avatar"""
- command = loadavatargallery.Command()
- out = StringIO()
- call_command(command, stdout=out)
- command_output = out.getvalue().splitlines()[0].strip()
- self.assertEqual(command_output, "New galleries have been loaded.")
- self.assertTrue(AvatarGallery.objects.exists())
|
