def testBadUrl(self, mock_logger):
image = ImageFactory()
initial_src = image.src
self.config.image_urls = '{src} not-a-url'.format(src=image.src)
self.config.save()
call_command('update_images')
mock_logger.warning.assert_called_with(
'Invalid image url: "not-a-url"')
image.refresh_from_db()
self.assertEqual(image.src, initial_src)
def setUp(self):
super(CorporateEndorsementTests, self).setUp()
self.corporation_name = 'test org'
self.individual_endorsements = CorporateEndorsement.objects.create(
corporation_name=self.corporation_name,
statement='test statement',
image=ImageFactory())
def test_delete_orphans(self):
""" Verify the delete_orphans method deletes orphaned instances. """
instances = (ImageFactory(), VideoFactory(),)
AbstractDataLoader.delete_orphans()
for instance in instances:
self.assertFalse(instance.__class__.objects.filter(pk=instance.pk).exists()) # pylint: disable=no-member
def testNormalRun(self):
image0 = ImageFactory()
image1 = ImageFactory()
self.config.image_urls = '''{src0} https://www.edx.org/bananas-in-pajamas.jpg
{src1} http://www.edx.org/are_coming_down_the_stairs.jpg'''.format(
src0=image0.src, src1=image1.src)
self.config.save()
call_command('update_images')
image0.refresh_from_db()
image1.refresh_from_db()
assert image0.src == 'https://www.edx.org/bananas-in-pajamas.jpg'
assert image1.src == 'http://www.edx.org/are_coming_down_the_stairs.jpg'
def test_data(self):
image = ImageFactory()
serializer = ImageSerializer(image)
expected = {
'src': image.src,
'description': image.description,
'height': image.height,
'width': image.width
}
self.assertDictEqual(serializer.data, expected)
