def test_get_instance_return_None(self): driver = WebDriverMock() element = Element(identifier="identifier") element.driver = driver assert element.get_instance() == None
def test_get_instance_return_instance_by_id(self): instance = WebElementMock() instance.expect("tag_name", "input") driver = WebDriverMock() driver.expect("find_element_by_id", instance) element = Element(identifier="identifier") element.driver = driver assert element.get_instance().tag_name == instance.tag_name