def __init__(self, driver):
super(AdminDashboard, self).__init__(driver)
self.tab_people = widget_bar.Tab(self._driver,
locator.WidgetBar.ADMIN_PEOPLE)
self.tab_roles = widget_bar.Tab(self._driver,
locator.WidgetBar.ADMIN_ROLES)
self.tab_events = widget_bar.Tab(self._driver,
locator.WidgetBar.ADMIN_EVENTS)
self.tab_custom_attributes = widget_bar.Tab(
self._driver, locator.WidgetBar.ADMIN_CUSTOM_ATTRIBUTE)
def _get_widget(self, widget_name):
"""Add attribute, click on relevant tab and return relevant widget."""
attr_name = "tab_" + widget_name
setattr(
self, attr_name,
widget_bar.Tab(self._driver, factory.get_locator_widget(widget_name)))
getattr(self, attr_name).click()
return factory.get_cls_widget(widget_name)(self._driver, widget_name)
