def test_webscraper_set_current_path(self):
"""
Test Webscraper get object will return a parsed webpage
:return: None
"""
websc = Webscraper()
websc.read_site("https://www.google.com")
websc.read_site("https://www.google.com/doodles")
websc.set_current_path("")
web_obj = websc.get_web_object()
self.assertEqual(web_obj.title.string, "Google")
websc.set_current_path("/doodles")
web_obj = websc.get_web_object()
self.assertEqual(web_obj.title.string, "Google Doodles")
def test_webscraper_get_object(self):
"""
Test Webscraper get object will return a parsed webpage
:return: None
"""
websc = Webscraper()
websc.read_site("http://www.google.com")
web_obj = websc.get_web_object()
self.assertEqual(web_obj.title.string, "Google")
def test_webscraper_get_child_elements(self):
"""
Test Webscraper get object will return a parsed webpage
:return: None
"""
websc = Webscraper()
websc.read_site("http://www.google.com/doodles")
web_obj = websc.get_web_object()
web_child_obj = websc.get_child_elements(web_obj, "div", ids=["content"])[0]
self.assertEqual(web_child_obj.get('id'), "content")