def test_shortenurl_failed(self): """ APIの実行が失敗したとき、元のURLを返す """ def dummy_urlopen(request): raise Exception("Something went wrong") shortenurl.urlopen = dummy_urlopen url = shortenurl.shorten(URL) self.assertEqual(url, URL)
def test_shortenurl(self): """ GoogleのURL短縮APIを使ってURLが短縮できる """ def dummy_urlopen(request): mock = MagicMock() json_string = json.dumps({'id': 'http://goo.gl/hogehoge'}) mock.read.return_value = json_string.encode('utf-8') return mock shortenurl.urlopen = dummy_urlopen url = shortenurl.shorten(URL) self.assertRegex(url, r'^http:\/\/goo.\gl\/.+$')
def repl(m): return shorten(m.group())