def test_custom_enhancement(self):
""" https://github.com/alisaifee/holmium.core/issues/14
"""
class CustomSelect(ElementEnhancer):
__TAG__ = "select"
def get_text_upper(self):
return self.element.text.upper()
register_enhancer(CustomSelect)
class SimplePage(Page):
id_el = Element(Locators.ID, "simple_id")
driver = mock.Mock()
driver.find_element.return_value.tag_name = "select"
driver.find_element.return_value.text = "fOo"
self.assertTrue(issubclass(SimplePage(driver).id_el.__class__, CustomSelect))
self.assertEquals(SimplePage(driver).id_el.get_text_upper(), "FOO")
def test_custom_enhancer(self):
class DivEnhancer(ElementEnhancer):
__TAG__ = "div"
def get_class(self):
return self.element.get_attribute("class")
register_enhancer(DivEnhancer)
class SimplePage(Page):
id_el = Element(Locators.ID, "simple_id")
self.driver.find_element.return_value.tag_name = "div"
self.driver.find_element.return_value.text = "foo"
self.driver.find_element.return_value.get_attribute.return_value = "div-class"
p = SimplePage(self.driver)
self.assertEquals(p.id_el.get_class(), "div-class")
def test_custom_enhancer(self):
class DivEnhancer(ElementEnhancer):
__TAG__ = "div"
def get_class(self):
return self.element.get_attribute("class")
register_enhancer(DivEnhancer)
class SimplePage(Page):
id_el = Element(Locators.ID, "simple_id")
self.driver.find_element.return_value.tag_name = "div"
self.driver.find_element.return_value.text = "foo"
self.driver.find_element.return_value.get_attribute.return_value = "div-class"
p = SimplePage(self.driver)
self.assertEquals(p.id_el.get_class(), "div-class")
def test_custom_enhancement(self):
""" https://github.com/alisaifee/holmium.core/issues/14
"""
class CustomSelect(ElementEnhancer):
__TAG__ = "select"
def get_text_upper(self):
return self.element.text.upper()
register_enhancer(CustomSelect)
class SimplePage(Page):
id_el = Element(Locators.ID, "simple_id")
driver = mock.Mock()
driver.find_element.return_value.tag_name = "select"
driver.find_element.return_value.text = "fOo"
self.assertTrue(
issubclass(SimplePage(driver).id_el.__class__, CustomSelect))
self.assertEquals(SimplePage(driver).id_el.get_text_upper(), "FOO")
