def _mock_comment(self,
id=1,
author_id=1,
link_id=1,
sr_id=1,
can_comment=True,
can_view_promo=True,
is_moderator=False,
**kwargs):
kwargs['id'] = id
kwargs['author_id'] = author_id
kwargs['link_id'] = link_id
kwargs['sr_id'] = sr_id
comment = Comment(**kwargs)
self.autopatch(VByName, "run", return_value=comment)
link = Link(id=link_id, sr_id=sr_id)
self.autopatch(Link, "_byID", return_value=link)
sr = Subreddit(id=sr_id)
self.autopatch(Subreddit, "_byID", return_value=sr)
self.autopatch(Subreddit, "can_comment", return_value=can_comment)
self.autopatch(Link, "can_view_promo", return_value=can_view_promo)
self.autopatch(Subreddit, "is_moderator", return_value=is_moderator)
return comment
def _mock_comment(id=1,
author_id=1,
link_id=1,
sr_id=1,
can_comment=True,
can_view_promo=True,
is_moderator=False,
**kwargs):
kwargs['id'] = id
kwargs['author_id'] = author_id
kwargs['link_id'] = link_id
comment = Comment(**kwargs)
VByName.run = MagicMock(return_value=comment)
link = Link(id=link_id)
Link._byID = MagicMock(return_value=link)
sr = Subreddit(id=sr_id)
comment.subreddit_slow = sr
link.subreddit_slow = sr
Subreddit.can_comment = MagicMock(return_value=can_comment)
Link.can_view_promo = MagicMock(return_value=can_view_promo)
Subreddit.is_moderator = MagicMock(return_value=is_moderator)
return comment
def test_imgur_link(self):
post = Link(is_self=True, selftext='''
Some text here.
https://example.com
https://imgur.com''')
url = _get_scrape_url(post)
self.assertEqual(url, 'https://imgur.com')
def test_simple_self_post(self):
post = Link(is_self=True, selftext='''
Some text here.
https://example.com
https://reddit.com''')
url = _get_scrape_url(post)
self.assertEqual(url, 'https://example.com')
def test_image_link(self):
post = Link(is_self=True, selftext='''
Some text here.
https://example.com
https://reddit.com/a.jpg''')
url = _get_scrape_url(post)
self.assertEqual(url, 'https://reddit.com/a.jpg')
post = Link(is_self=True, selftext='''
Some text here.
https://example.com
https://reddit.com/a.PNG''')
url = _get_scrape_url(post)
self.assertEqual(url, 'https://reddit.com/a.PNG')
post = Link(is_self=True, selftext='''
Some text here.
https://example.com
https://reddit.com/a.jpg/b''')
url = _get_scrape_url(post)
self.assertEqual(url, 'https://example.com')
def _mock_link(self,
id=1,
author_id=1,
sr_id=1,
can_comment=True,
can_view_promo=True,
**kwargs):
kwargs['id'] = id
kwargs['author_id'] = author_id
kwargs['sr_id'] = sr_id
link = Link(**kwargs)
self.autopatch(VByName, "run", return_value=link)
sr = Subreddit(id=sr_id)
self.autopatch(Subreddit, "_byID", return_value=sr)
self.autopatch(Subreddit, "can_comment", return_value=can_comment)
self.autopatch(Link, "can_view_promo", return_value=can_view_promo)
return link
def _mock_link(id=1,
author_id=1,
sr_id=1,
can_comment=True,
can_view_promo=True,
**kwargs):
kwargs['id'] = id
kwargs['author_id'] = author_id
kwargs['sr_id'] = sr_id
link = Link(**kwargs)
VByName.run = MagicMock(return_value=link)
sr = Subreddit(id=sr_id)
link.subreddit_slow = sr
Subreddit.can_comment = MagicMock(return_value=can_comment)
Link.can_view_promo = MagicMock(return_value=can_view_promo)
return link
def test_link_post(self, Link):
post = Link()
post.url = 'https://example.com'
post.is_self = False
url = _get_scrape_url(post)
self.assertEqual(url, 'https://example.com')
def test_link_post(self):
post = Link(url='https://example.com')
url = _get_scrape_url(post)
self.assertEqual(url, 'https://example.com')
def test_more_marker(self):
test_cases = (('asdf<a id="more"></a>lkjh', 'asdf', 'lkjh'), )
for input_text, expected_summary, expected_more in test_cases:
link = Link(article=input_text)
self.assertEqual(link._summary(), expected_summary)
self.assertEqual(link._more(), expected_more)
def test_name(self):
link = Link(name='Link Name')
self.assertEqual(link.name, 'Link Name')
