def test_twitter_link(self):
r1 = mock.Mock()
r1.text = '<div class="permalink-inner permalink-tweet-container"><div><p class="tweet-text">.@elonmusk Just turned on my ice machine. I wonder if I own a bucket..</p></div></div>'
r1.status_code = 200
with mock.patch("requests.get") as get:
get.side_effect = [r1]
result = receivers.twitter_link(None, content="https://twitter.com/notch/status/500690106991513603")
self.assertEqual(result['content'], '[ notch: .@elonmusk Just turned on my ice machine. I wonder if I own a bucket.. ]')
def test_404(self):
with mock.patch("requests.get") as get:
get.return_value.status_code = 404
r = receivers.twitter_link(None, content="https://twitter.com/notch/status/500690106991513603")
self.assertEqual(r, None)
