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)