def test_vimeo_links_get_resolved(self):
mock_response = MockResponse()
requests.head = lambda *args, **kwargs: mock_response
sl = ShortenedLink(shortened_url)
sl.do_resolution()
self.assertEquals('1234', sl.resolve())
def test_non_vimeo_links_get_marked_as_such(self):
mock_response = MockResponse()
mock_response.headers['location'] = 'http://google.com'
requests.head = lambda *args, **kwargs: mock_response
sl = ShortenedLink(shortened_url)
sl.do_resolution()
self.assertEquals(LINK_IS_NOT_VIMEO, sl.resolve())
