def get_favicon_url(self):
href = self.get_favicon_info()
if href is None:
return utils.get_root_url(self.current_url()) + "favicon.ico"
url_obj = urlparse(href)
if href.startwith('//'):
url_obj = urlparse(self.current_url())
return url_obj.scheme + href
elif url_obj.scheme == '':
return utils.get_root_url(self.current_url()) + href
def test_get_root_url_local_url_with_port(self):
self.assertEqual(get_root_url('http://localhost:9998/'),
'http://localhost:9998/')
def test_get_root_url_remote_url_with_query_and_port(self):
self.assertEqual(
get_root_url('http://www.hoverstate.com:8080/?favicon.ico'),
'http://www.hoverstate.com:8080/')
def test_get_root_url_local_url(self):
self.assertEqual(get_root_url('http://localhost/'),
'http://localhost/')
def test_get_root_url_remote_url_with_fragment(self):
self.assertEqual(get_root_url('http://www.hoverstate.com/#/form'),
'http://www.hoverstate.com/')
def test_get_root_url_remote_url_with_path(self):
self.assertEqual(get_root_url('http://www.hoverstate.com/favicon.ico'),
'http://www.hoverstate.com/')
def test_get_root_url_remote_url_with_port(self):
self.assertEqual(get_root_url('http://www.hoverstate.com:8080/'),
'http://www.hoverstate.com:8080/')
def test_get_root_url_remote_url(self):
self.assertEqual(get_root_url('http://www.hoverstate.com/'),
'http://www.hoverstate.com/')
def test_get_root_url_local_url_with_query_and_port(self):
self.assertEqual(get_root_url('http://localhost:9998/?favicon.ico'),
'http://localhost:9998/')
def test_get_root_url_local_url_with_query(self):
self.assertEqual(get_root_url('http://localhost/?favicon.ico'),
'http://localhost/')
def test_get_root_url_local_url_with_fragment(self):
self.assertEqual(get_root_url('http://localhost/#/form'),
'http://localhost/')