def add_header_row(self, index=None):
if index is None:
index = 1
self.open_more_options_menu(index)
count = BasePage.count_elements(self, By.CSS_SELECTOR, self.row_number_prefix.format(index) +
self.header_name_field)
BasePage.click(self, By.CSS_SELECTOR, self.row_number_prefix.format(index) + self.add_header_button)
BasePage.wait_page(self)
assert BasePage.count_elements(self, By.CSS_SELECTOR, self.row_number_prefix.format(index) +
self.header_name_field) == count + 1, 'Failed to add new row'
def delete_header_row(self, index=None, header_index=None):
if index is None:
index = 1
if header_index is None:
header_index = 1
self.open_more_options_menu(index)
count = BasePage.count_elements(self, By.CSS_SELECTOR, self.row_number_prefix.format(index) +
self.header_name_field)
if count >= header_index and count is not 0:
BasePage.click(self, By.CSS_SELECTOR, self.row_header_prefix.format(index, header_index) +
self.header_delete_row)
BasePage.wait_page(self)
else:
raise NoSuchElementException('There is no such header')
assert BasePage.count_elements(self, By.CSS_SELECTOR, self.row_number_prefix.format(index) +
self.header_name_field) == count - 1, 'Failed to delete header row'
def add_url_field(self):
count = self.get_fields_count()
BasePage.click(self, By.CSS_SELECTOR, self.add_row_button)
BasePage.wait_page(self)
assert self.get_fields_count() == count + 1
def click_more_options(self, index=None):
if index is None:
index = 1
BasePage.click(self, By.CSS_SELECTOR, self.row_number_prefix.format(index) + self.more_options_button)
BasePage.wait_page(self)
