def test_ensure_timeline_scheduled_script(self): client = StrictRedis(db=9) timeline = 'timeline' timestamp = 100.0 waiting_set_size = functools.partial(client.zcard, 'waiting') ready_set_size = functools.partial(client.zcard, 'ready') timeline_score_in_waiting_set = functools.partial(client.zscore, 'waiting', timeline) timeline_score_in_ready_set = functools.partial(client.zscore, 'ready', timeline) keys = ('waiting', 'ready', 'last-processed') # The first addition should cause the timeline to be added to the ready set. with self.assertChanges(ready_set_size, before=0, after=1), \ self.assertChanges(timeline_score_in_ready_set, before=None, after=timestamp): assert ensure_timeline_scheduled(keys, (timeline, timestamp, 1, 10), client) == 1 # Adding it again with a timestamp in the future should not change the schedule time. with self.assertDoesNotChange(waiting_set_size), \ self.assertDoesNotChange(ready_set_size), \ self.assertDoesNotChange(timeline_score_in_ready_set): assert ensure_timeline_scheduled(keys, (timeline, timestamp + 50, 1, 10), client) is None # Move the timeline from the ready set to the waiting set. client.zrem('ready', timeline) client.zadd('waiting', timestamp, timeline) client.set('last-processed', timestamp) increment = 1 with self.assertDoesNotChange(waiting_set_size), \ self.assertChanges(timeline_score_in_waiting_set, before=timestamp, after=timestamp + increment): assert ensure_timeline_scheduled(keys, (timeline, timestamp, increment, 10), client) is None # Make sure the schedule respects the maximum value. with self.assertDoesNotChange(waiting_set_size), \ self.assertChanges(timeline_score_in_waiting_set, before=timestamp + 1, after=timestamp): assert ensure_timeline_scheduled(keys, (timeline, timestamp, increment, 0), client) is None # Test to ensure a missing last processed timestamp can be handled # correctly (chooses minimum of schedule value and record timestamp.) client.zadd('waiting', timestamp, timeline) client.delete('last-processed') with self.assertDoesNotChange(waiting_set_size), \ self.assertDoesNotChange(timeline_score_in_waiting_set): assert ensure_timeline_scheduled(keys, (timeline, timestamp + 100, increment, 10), client) is None with self.assertDoesNotChange(waiting_set_size), \ self.assertChanges(timeline_score_in_waiting_set, before=timestamp, after=timestamp - 100): assert ensure_timeline_scheduled(keys, (timeline, timestamp - 100, increment, 10), client) is None
def test_ensure_timeline_scheduled_script(self): client = StrictRedis(db=9) timeline = 'timeline' timestamp = 100.0 waiting_set_size = functools.partial(client.zcard, 'waiting') ready_set_size = functools.partial(client.zcard, 'ready') timeline_score_in_waiting_set = functools.partial(client.zscore, 'waiting', timeline) timeline_score_in_ready_set = functools.partial(client.zscore, 'ready', timeline) # The first addition should cause the timeline to be added to the waiting set. with self.assertChanges(waiting_set_size, before=0, after=1), \ self.assertChanges(timeline_score_in_waiting_set, before=None, after=timestamp): ensure_timeline_scheduled(('waiting', 'ready'), (timeline, timestamp), client) # Adding it again with a timestamp in the future should not change the schedule time. with self.assertDoesNotChange(waiting_set_size), \ self.assertDoesNotChange(timeline_score_in_waiting_set): ensure_timeline_scheduled(('waiting', 'ready'), (timeline, timestamp + 50), client) # If we see a record with a timestamp earlier than the schedule time, # we should change the schedule. with self.assertDoesNotChange(waiting_set_size), \ self.assertChanges(timeline_score_in_waiting_set, before=timestamp, after=timestamp - 50): ensure_timeline_scheduled(('waiting', 'ready'), (timeline, timestamp - 50), client) # Move the timeline from the waiting set to the ready set. client.zrem('waiting', timeline) client.zadd('ready', timestamp, timeline) # Nothing should change. with self.assertDoesNotChange(waiting_set_size), \ self.assertDoesNotChange(ready_set_size), \ self.assertDoesNotChange(timeline_score_in_ready_set): ensure_timeline_scheduled(('waiting', 'ready'), (timeline, timestamp - 50), client)