def create_timestamp_pictures(event_id, timestamps):
event = Event.objects.get(id=event_id)
assert isinstance(timestamps, list)
pictures.create_timestamp_pictures(
event,
timestamps,
verbose=settings.DEBUG,
)
def test_create_timestamp_pictures(
self,
mock_popen,
rhead,
p_logging
):
def mocked_head(url, **options):
return Response(
'',
200
)
rhead.side_effect = mocked_head
ffmpeged_urls = []
sample_jpg = self.sample_jpg
def mocked_popen(command, **kwargs):
url = command[4]
ffmpeged_urls.append(url)
destination = command[-1]
assert os.path.isdir(os.path.dirname(destination))
class Inner:
def communicate(self):
out = err = ''
if 'xyz123' in url:
# 04.jpg because the chapter is at timestamp 4 seconds
if '03.jpg' in destination or '07.jpg' in destination:
shutil.copyfile(sample_jpg, destination)
else:
raise NotImplementedError(destination)
else:
raise NotImplementedError(url)
return out, err
return Inner()
mock_popen.side_effect = mocked_popen
event = Event.objects.get(title='Test event')
event.duration = 100
event.save()
template = Template.objects.create(
name='Vid.ly Something',
content="{{ tag }}"
)
event.template = template
event.template_environment = {'tag': 'xyz123'}
event.save()
# Nothing will happen because the event doesn't have a duration or
# Vid.ly template.
pictures.create_timestamp_pictures(event, [3, 7])
pic1, pic2 = Picture.objects.filter(event=event).order_by('timestamp')
eq_(pic1.timestamp, 3)
eq_(pic2.timestamp, 7)
assert ffmpeged_urls
assert len(ffmpeged_urls) == 2
# Run it again, it should not create more pictures
pictures.create_timestamp_pictures(event, [3, 7])
ok_(pic1 not in Picture.objects.all())
ok_(pic2 not in Picture.objects.all())
pic1, pic2 = Picture.objects.filter(event=event).order_by('timestamp')
eq_(pic1.timestamp, 3)
eq_(pic2.timestamp, 7)
def test_create_timestamp_pictures(self, mock_popen, rhead, p_logging):
def mocked_head(url, **options):
return Response('', 200)
rhead.side_effect = mocked_head
ffmpeged_urls = []
sample_jpg = self.sample_jpg
def mocked_popen(command, **kwargs):
url = command[4]
ffmpeged_urls.append(url)
destination = command[-1]
assert os.path.isdir(os.path.dirname(destination))
class Inner:
def communicate(self):
out = err = ''
if 'xyz123' in url:
# 04.jpg because the chapter is at timestamp 4 seconds
if '03.jpg' in destination or '07.jpg' in destination:
shutil.copyfile(sample_jpg, destination)
else:
raise NotImplementedError(destination)
else:
raise NotImplementedError(url)
return out, err
return Inner()
mock_popen.side_effect = mocked_popen
event = Event.objects.get(title='Test event')
event.duration = 100
event.save()
template = Template.objects.create(name='Vid.ly Something',
content="{{ tag }}")
event.template = template
event.template_environment = {'tag': 'xyz123'}
event.save()
# Nothing will happen because the event doesn't have a duration or
# Vid.ly template.
pictures.create_timestamp_pictures(event, [3, 7])
pic1, pic2 = Picture.objects.filter(event=event).order_by('timestamp')
eq_(pic1.timestamp, 3)
eq_(pic2.timestamp, 7)
assert ffmpeged_urls
assert len(ffmpeged_urls) == 2
# Run it again, it should not create more pictures
pictures.create_timestamp_pictures(event, [3, 7])
ok_(pic1 not in Picture.objects.all())
ok_(pic2 not in Picture.objects.all())
pic1, pic2 = Picture.objects.filter(event=event).order_by('timestamp')
eq_(pic1.timestamp, 3)
eq_(pic2.timestamp, 7)
