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