Fixed bug where submission indicies were duplicated when paging.

This commit is contained in:
Michael Lazar
2016-04-15 17:47:45 -07:00
parent e69c59baa8
commit 6c130001b1
2 changed files with 7 additions and 3 deletions

View File

@@ -439,9 +439,10 @@ class SubredditContent(Content):
raise IndexError raise IndexError
else: else:
data = self.strip_praw_submission(submission) data = self.strip_praw_submission(submission)
data['index'] = index #data['index'] = len(self._submission_data)
data['index'] = len(self._submission_data) + 1
# Add the post number to the beginning of the title # Add the post number to the beginning of the title
data['title'] = '{0}. {1}'.format(index+1, data['title']) data['title'] = '{0}. {1}'.format(data['index'], data['title'])
self._submission_data.append(data) self._submission_data.append(data)
# Modifies the original dict, faster than copying # Modifies the original dict, faster than copying

View File

@@ -216,13 +216,16 @@ def test_content_subreddit_load_more(reddit, terminal):
assert content.get(50)['type'] == 'Submission' assert content.get(50)['type'] == 'Submission'
assert len(content._submission_data) == 51 assert len(content._submission_data) == 51
for data in islice(content.iterate(0, 1), 0, 50): for i, data in enumerate(islice(content.iterate(0, 1), 0, 50)):
assert all(k in data for k in ('object', 'n_rows', 'offset', 'type', assert all(k in data for k in ('object', 'n_rows', 'offset', 'type',
'index', 'title', 'split_title')) 'index', 'title', 'split_title'))
# All text should be converted to unicode by this point # All text should be converted to unicode by this point
for val in data.values(): for val in data.values():
assert not isinstance(val, six.binary_type) assert not isinstance(val, six.binary_type)
# Index be appended to each title, starting at "1." and incrementing
assert data['index'] == i + 1
assert data['title'].startswith(six.text_type(i + 1))
def test_content_subreddit_from_name(reddit, terminal): def test_content_subreddit_from_name(reddit, terminal):