def verify_command(self):
status = test_utils.get_watcher_dialogue_status(self)
self.page.retry_click(
(By.CSS_SELECTOR, NavMenuLocators.status_alertifier_more_btn_css),
(By.XPATH, NavMenuLocators.process_watcher_alertfier))
self.page.wait_for_element_to_disappear(
lambda driver: driver.find_element_by_css_selector(".loading-logs"
))
if status != "Successfully completed.":
self.assertEquals(status, "Successfully completed.")
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_canvas_css).text
if self.test_level == 'database':
self.assertEquals(
command, "VACUUM (VERBOSE)\nRunning Query:"
"\nVACUUM VERBOSE;")
elif self.is_xss_check and self.test_level == 'table':
# Check for XSS in the dialog
source_code = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_canvas_css
).get_attribute('innerHTML')
self.check_escaped_characters(source_code,
'<h1>test_me</h1>',
'Maintenance detailed window')
else:
self.assertEquals(
command, "VACUUM "
"(VERBOSE)\nRunning Query:"
"\nVACUUM VERBOSE"
" public." + self.table_name + ";")
test_gui_helper.close_process_watcher(self)
def verify_command(self):
test_gui_helper.open_process_details(self)
message = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_message_css).text
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_css).text
vacuum_details = \
"VACUUM (VERBOSE) on database '{0}' of server " \
"{1} ({2}:{3})".format(self.database_name, self.server['name'],
self.server['host'], self.server['port'])
if self.test_level == 'database':
self.assertEqual(message, vacuum_details)
self.assertEqual(command, "VACUUM VERBOSE;")
elif self.is_xss_check and self.test_level == 'table':
# Check for XSS in the dialog
source_code = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_css
).get_attribute('innerHTML')
self.check_escaped_characters(
source_code,
'<h1>test_me</h1>',
'Maintenance detailed window'
)
else:
self.assertEqual(message, vacuum_details)
self.assertEqual(command, "VACUUM VERBOSE"
" public." + self.table_name + ";")
test_gui_helper.close_process_watcher(self)
def after(self):
test_gui_helper.close_process_watcher(self)
test_gui_helper.close_bgprocess_popup(self)
self.page.remove_server(self.server)
connection = test_utils.get_db_connection(self.server['db'],
self.server['username'],
self.server['db_password'],
self.server['host'],
self.server['port'],
self.server['sslmode'])
test_utils.drop_database(connection, self.database_name)
def after(self):
try:
test_gui_helper.close_process_watcher(self)
test_gui_helper.close_bgprocess_popup(self)
self.page.remove_server(self.server)
except Exception:
print("PGUtilitiesBackupFeatureTest - "
"Exception occurred in after method")
finally:
connection = test_utils.get_db_connection(
self.server['db'], self.server['username'],
self.server['db_password'], self.server['host'],
self.server['port'], self.server['sslmode'])
test_utils.drop_database(connection, self.database_name)
def runTest(self):
self.page.expand_database_node(
self.server['name'],
self.server['db_password'], self.database_name)
# Backup
self.initiate_backup()
# Wait for the backup status alertfier
self.wait.until(EC.visibility_of_element_located(
(By.CSS_SELECTOR,
NavMenuLocators.bcg_process_status_alertifier_css)))
status = test_utils.get_watcher_dialogue_status(self)
self.page.retry_click(
(By.CSS_SELECTOR,
NavMenuLocators.status_alertifier_more_btn_css),
(By.XPATH,
NavMenuLocators.process_watcher_alertfier))
self.page.wait_for_element_to_disappear(
lambda driver: driver.find_element_by_css_selector(
".loading-logs"), 10)
expected_backup_success_msg = "Successfully completed."
self.assertEqual(status, expected_backup_success_msg)
backup_file = None
# Check for XSS in Backup details
if self.is_xss_check:
self._check_detailed_window_for_xss('Backup')
else:
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_canvas_css). \
text
self.assertIn(self.server['name'], str(command))
self.assertIn("from database 'pg_utility_test_db'", str(command))
# On windows a modified path may be shown so skip this test
if os.name != 'nt':
self.assertIn("test_backup", str(command))
self.assertIn("pg_dump", str(command))
if command:
backup_file = command[int(command.find('--file')) +
8:int(command.find('--host')) - 2]
test_gui_helper.close_process_watcher(self)
# Restore
self.initiate_restore()
# Wait for the backup status alertfier
self.wait.until(EC.visibility_of_element_located(
(By.CSS_SELECTOR,
NavMenuLocators.bcg_process_status_alertifier_css)))
status = test_utils.get_watcher_dialogue_status(self)
self.page.retry_click(
(By.CSS_SELECTOR,
NavMenuLocators.status_alertifier_more_btn_css),
(By.XPATH,
NavMenuLocators.process_watcher_alertfier))
self.page.wait_for_element_to_disappear(
lambda driver: driver.find_element_by_css_selector(
".loading-logs"), 10)
self.assertEqual(status, expected_backup_success_msg)
# Check for XSS in Restore details
if self.is_xss_check:
self._check_detailed_window_for_xss('Restore')
else:
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_canvas_css). \
text
self.assertIn(self.server['name'], str(command))
if os.name != 'nt':
self.assertIn("test_backup", str(command))
self.assertIn("pg_restore", str(command))
test_gui_helper.close_process_watcher(self)
if backup_file is not None and os.path.isfile(backup_file):
os.remove(backup_file)
def runTest(self):
self.page.expand_database_node(self.server['name'],
self.server['db_password'],
self.database_name)
# Backup
self.page.retry_click(
(By.LINK_TEXT, NavMenuLocators.tools_menu_link_text),
(By.CSS_SELECTOR, NavMenuLocators.backup_obj_css))
backup_object = self.wait.until(
EC.visibility_of_element_located(
(By.CSS_SELECTOR, NavMenuLocators.backup_obj_css)))
backup_object.click()
# Enter the file name of the backup to be taken
self.wait.until(
EC.visibility_of_element_located(
(By.NAME, NavMenuLocators.backup_filename_txt_box_name)))
element = self.wait.until(
EC.element_to_be_clickable(
(By.NAME, NavMenuLocators.backup_filename_txt_box_name)))
element.click()
self.page.fill_input_by_field_name(
NavMenuLocators.backup_filename_txt_box_name,
"test_backup",
loose_focus=True)
# Click on the take Backup button
take_bckup = self.page.find_by_xpath(NavMenuLocators.backup_btn_xpath)
click = True
while click:
try:
take_bckup.click()
if self.page.wait_for_element_to_disappear(
lambda driver: driver.find_element_by_name(
NavMenuLocators.backup_filename_txt_box_name)):
click = False
except Exception as e:
pass
# Wait for the backup status alertfier
self.wait.until(
EC.visibility_of_element_located(
(By.CSS_SELECTOR,
NavMenuLocators.bcg_process_status_alertifier_css)))
status = test_utils.get_watcher_dialogue_status(self)
self.page.retry_click(
(By.CSS_SELECTOR, NavMenuLocators.status_alertifier_more_btn_css),
(By.XPATH, NavMenuLocators.process_watcher_alertfier))
self.page.wait_for_element_to_disappear(
lambda driver: driver.find_element_by_css_selector(".loading-logs"
), 10)
if status != "Successfully completed.":
self.assertEquals(status, "Successfully completed.")
backup_file = None
# Check for XSS in Backup details
if self.is_xss_check:
self._check_detailed_window_for_xss('Backup')
else:
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_canvas_css). \
text
self.assertIn(self.server['name'], str(command))
self.assertIn("from database 'pg_utility_test_db'", str(command))
# On windows a modified path may be shown so skip this test
if os.name != 'nt':
self.assertIn("test_backup", str(command))
self.assertIn("pg_dump", str(command))
if command:
backup_file = command[int(command.find('--file')) +
8:int(command.find('--host')) - 2]
test_gui_helper.close_process_watcher(self)
# Restore
tools_menu = self.driver.find_element_by_link_text(
NavMenuLocators.tools_menu_link_text)
tools_menu.click()
restore_obj = self.page.find_by_css_selector(
NavMenuLocators.restore_obj_css)
restore_obj.click()
self.wait.until(
EC.visibility_of_element_located(
(By.NAME, NavMenuLocators.restore_file_name_txt_box_name)))
self.wait.until(
EC.element_to_be_clickable(
(By.NAME,
NavMenuLocators.restore_file_name_txt_box_name))).click()
self.page.fill_input_by_field_name(
NavMenuLocators.restore_file_name_txt_box_name,
"test_backup",
loose_focus=True)
restore_btn = self.page.find_by_xpath(
NavMenuLocators.restore_button_xpath)
restore_btn.click()
self.page.wait_for_element_to_disappear(
lambda driver: driver.find_element_by_css_selector(
NavMenuLocators.restore_file_name_txt_box_name))
# Wait for the backup status alertfier
self.wait.until(
EC.visibility_of_element_located(
(By.CSS_SELECTOR,
NavMenuLocators.bcg_process_status_alertifier_css)))
status = test_utils.get_watcher_dialogue_status(self)
self.page.retry_click(
(By.CSS_SELECTOR, NavMenuLocators.status_alertifier_more_btn_css),
(By.XPATH, NavMenuLocators.process_watcher_alertfier))
self.page.wait_for_element_to_disappear(
lambda driver: driver.find_element_by_css_selector(".loading-logs"
), 10)
if status != "Successfully completed.":
self.assertEquals(status, "Successfully completed.")
# Check for XSS in Restore details
if self.is_xss_check:
self._check_detailed_window_for_xss('Restore')
else:
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_canvas_css). \
text
self.assertIn(self.server['name'], str(command))
if os.name != 'nt':
self.assertIn("test_backup", str(command))
self.assertIn("pg_restore", str(command))
test_gui_helper.close_process_watcher(self)
if backup_file is not None and os.path.isfile(backup_file):
os.remove(backup_file)
def runTest(self):
self.page.expand_database_node("Server", self.server['name'],
self.server['db_password'],
self.database_name)
# Backup
self.initiate_backup()
test_gui_helper.wait_for_process_start(self)
test_gui_helper.open_process_details(self)
backup_file = None
# Check for XSS in Backup details
if self.is_xss_check:
self._check_detailed_window_for_xss('Backup')
else:
message = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_message_css). \
text
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_css). \
text
self.assertIn(self.server['name'], str(message))
self.assertIn("from database 'pg_utility_test_db'", str(message))
# On windows a modified path may be shown so skip this test
if os.name != 'nt':
self.assertIn("test_backup", str(command))
self.assertIn("pg_dump", str(command))
if command:
backup_file = command[int(command.find('--file')) +
8:int(command.find('--host')) - 2]
test_gui_helper.close_process_watcher(self)
# Restore
self.initiate_restore()
test_gui_helper.wait_for_process_start(self)
test_gui_helper.open_process_details(self)
# Check for XSS in Restore details
if self.is_xss_check:
self._check_detailed_window_for_xss('Restore')
else:
message = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_message_css). \
text
command = self.page.find_by_css_selector(
NavMenuLocators.process_watcher_detailed_command_css). \
text
self.assertIn(self.server['name'], str(message))
if os.name != 'nt':
self.assertIn("test_backup", str(command))
self.assertIn("pg_restore", str(command))
test_gui_helper.close_process_watcher(self)
if backup_file is not None and os.path.isfile(backup_file):
os.remove(backup_file)
