def testSetProgramming(self):
collection = Collection(name='test')
collection.put()
media1 = fake_video(collection, 100)
media2 = fake_video(collection, 360)
channel = Channel.add('test')
ChannelCollection.add(channel=channel, collection=collection)
duration = media1[0].duration + 5
programs = Programming.set_programming(channel.id, duration=duration)
self.assertEquals(len(programs), 1)
self.assertEquals(programs[0].media.last_programmed, programs[0].time)
self.assertEquals(programs[0].media.id, media2[0].id)
# Check memcache
programming = memcache.get('programming') or {}
next_programs = Programming.next_programs(programming.get(channel.id), duration=1200, prelude=120)
self.assertEquals(len(next_programs), 1)
# Schedule second program
memcache.set('programming', []) # Sidestep next_program check
programs = Programming.set_programming(channel.id, duration=1200)
self.assertEquals(len(programs), 1)
self.assertEquals(programs[0].media.id, media1[0].id)
# Clear memcache
memcache.set('programming', [])
print programs[0].media.last_programmed
programs = Programming.set_programming(channel.id)
print programs[0].time
def testNextPrograms(self):
programming = memcache.get('programming') or {}
channel = Channel(name='test')
channel.put()
duration = 1200
self.assertEquals(len(Programming.next_programs(programming.get(channel.id), duration, prelude=120)), 0)
