def should_get_two_threads_different_length(self): comment4 = Comment('Comment 4') comment3 = Comment('Comment 3', replies=[comment4]) comment2 = Comment('Comment 2') comment1 = Comment('Comment 1', replies=[comment2, comment3]) assert [[comment1, comment2], [comment1, comment3, comment4]] == list(get_threads(comment1))
def should_get_multiple_children(self): comment3 = Comment('Comment 3') comment2 = Comment('Comment 2') comment1 = Comment('Comment 1', replies=[comment2, comment3]) assert [[comment1, comment2], [comment1, comment3]] == list(get_threads(comment1))
def should_get_three_thread(self): comment3 = Comment('Comment 3') comment2 = Comment('Comment 2', replies=[comment3]) comment1 = Comment('Comment 1', replies=[comment2]) assert [[comment1, comment2, comment3]] == list(get_threads(comment1))
def should_get_simple_thread(self): comment1 = Comment('Comment 1') assert [[comment1]] == list(get_threads(comment1))