def test_get_stories_by_track_without_matches(self):
"""
L{TimePeriod.get_stories_by_track} returns an empty C{list} if no
L{Story}s match the specified track.
"""
start_date = datetime.utcnow()
end_date = start_date + timedelta(days=30)
time_period = TimePeriod("name", start_date, end_date)
time_period.add(Story("name", "description", "track1", QUEUED))
self.assertEqual([], time_period.get_stories_by_track("track2"))
def test_get_stories_by_track(self):
"""
L{TimePeriod.get_stories_by_track} returns all L{Story}s in the time
period that are in the specified track. The stories are returned in
the order they were added to the time period.
"""
start_date = datetime.utcnow()
end_date = start_date + timedelta(days=30)
time_period = TimePeriod("name", start_date, end_date)
time_period.add(Story("name", "description", "track2", QUEUED))
story1 = Story("name1", "description", "track1", QUEUED)
story2 = Story("name2", "description", "track1", QUEUED)
time_period.add(story1)
time_period.add(story2)
self.assertEqual([story1, story2],
time_period.get_stories_by_track("track1"))
