def test_failed_request(self, mock_get): """Assert a failed request results in the original link.""" mock_get.side_effect = requests.exceptions.RequestException() self.assertEqual('original link', formatters.shorten('original link'))
def test_success(self, mock_get): mock_get.return_value.text = '\n http://so.short/abc \n' shortened_link = formatters.shorten('http://www.example.com/') self.assertEqual('http://so.short/abc', shortened_link)
def test_no_link(self): """Assert an empty string is returned when no link is provided.""" self.assertEqual('', formatters.shorten(None))