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_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")