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))