Added tests

tests/integration/models/reddit/test_subreddit.py

@@ -954,6 +954,16 @@ def test_comments__with_skip_existing(self, reddit):
         # that there are at least 400 comments in the stream.
         assert count < 400
 
+    def test_comments__with_start_after(self, reddit):
+        subreddit = reddit.subreddit("kakapo")
+        initial_stream = subreddit.stream.comments()
+        first_ten = [next(initial_stream) for _ in range(10)]
+        generator = subreddit.stream.comments(start_after=first_ten[4].fullname)
+        for i in range(5):
+            comment = next(generator)
+            assert isinstance(comment, Comment)
+            assert comment.fullname == first_ten[i + 5].fullname
+
     def test_submissions(self, reddit):
         subreddit = reddit.subreddit("all")
         generator = subreddit.stream.submissions()

tests/unit/models/test_util.py

@@ -124,3 +124,33 @@ def generate(limit, **kwargs):
             thing = next(stream)
             assert thing not in seen
             seen.add(thing)
+
+    def test_stream_start_after(
+        self,
+    ):
+        Thing = namedtuple("Thing", ["fullname"])
+        initial_things = [Thing(n) for n in reversed(range(100))]
+        counter = 99
+
+        def generate(limit, params=None, **kwargs):
+            nonlocal counter
+            counter += 1
+            sliced_things = initial_things
+            if params:
+                sliced_things = initial_things[
+                    : next(
+                        i
+                        for i, thing in enumerate(initial_things)
+                        if thing.fullname == params["before"]
+                    )
+                ]
+            if counter % 2 == 0:
+                return sliced_things
+            return [Thing(counter)] + sliced_things[:-1]
+
+        stream = stream_generator(generate, start_after=49)
+        expected_fullname = 50
+        for _ in range(50):
+            thing = next(stream)
+            assert thing.fullname == expected_fullname, thing
+            expected_fullname += 1