pandorafms/tests/console/include/alert_functions.py

100 lines
4.0 KiB
Python

# -*- coding: utf-8 -*-
from selenium import selenium
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait, Select
from selenium.webdriver.support import expected_conditions as EC
from module_functions import search_module
from common_functions_60 import *
import random, time
import string
def assign_alert_template_to_module (driver,agent_name,module_name,template_name):
search_module(driver,agent_name,module_name)
driver.find_element_by_xpath('//ul[@class="mn"]/li/a/img[@data-title="Alerts"]').click()
Select(driver.find_element_by_id("id_agent_module")).select_by_visible_text(module_name)
Select(driver.find_element_by_id("template")).select_by_visible_text(template_name)
driver.find_element_by_id("submit-add").click()
def force_alert_of_module(driver,agent_name,module_name,template_name):
click_menu_element(driver,"Agent detail")
driver.find_element_by_id("text-search").clear()
driver.find_element_by_id("text-search").send_keys(agent_name)
driver.find_element_by_id("submit-srcbutton").click()
driver.find_element_by_css_selector("b").click()
driver.find_element_by_xpath('//ul[@class="mn"]/li/a/img[@data-title="Alerts"]').click()
driver.find_element_by_xpath('//tr[td[3][contains(.,"'+module_name+'")] and td[4][contains(.,"'+template_name+'")]]/td[2]/a').click()
time.sleep(10)
def create_new_action_to_alert(driver,action_name,action_group,command,threshold=None,field1=None,field2=None,field3=None):
click_menu_element(driver,"Actions")
driver.find_element_by_id("submit-create").click()
driver.find_element_by_id("text-name").clear()
driver.find_element_by_id("text-name").send_keys(action_name)
driver.find_element_by_xpath('//option[contains(.,"'+action_group+'")]').click()
driver.find_element_by_xpath('//option[contains(.,"'+command+'")]').click()
if threshold != None:
driver.find_element_by_id("text-action_threshold").clear()
driver.find_element_by_id("text-action_threshold").send_keys(threshold)
if command == "eMail" and field1 != None and field2 != None and field3 != None:
driver.find_element_by_id("textarea_field1_value").clear()
driver.find_element_by_id("textarea_field1_value").send_keys(field1)
driver.find_element_by_id("textarea_field2_value").clear()
driver.find_element_by_id("textarea_field2_value").send_keys(field2)
driver.find_element_by_id("textarea_field3_value").clear()
driver.find_element_by_id("textarea_field3_value").send_keys(field3)
if command == "Syslog" and field1 != None and field2 != None:
driver.find_element_by_id("textarea_field1_value").clear()
driver.find_element_by_id("textarea_field1_value").send_keys(field1)
driver.find_element_by_id("textarea_field2_value").clear()
driver.find_element_by_id("textarea_field2_value").send_keys(field2)
driver.find_element_by_id("submit-create").click()
def create_new_command_to_alert(driver,command_name,command,list_field_description,list_field_values,description=None):
click_menu_element(driver,"Commands")
driver.find_element_by_id("submit-create").click()
driver.find_element_by_id("text-name").clear()
Select(driver.find_element_by_id("text-name")).send_keys(command_name)
driver.find_element_by_id("textarea_command").clear()
Select(driver.find_element_by_id("textarea_command")).send_keys(command)
if description != None:
driver.find_element_by_id("textarea_description").clear()
Select(driver.find_element_by_id("textarea_description")).send_keys(description)
i=1
for field_description in list_field_description:
driver.find_element_by_id("text-field"+i+"_description").clear()
Select(driver.find_element_by_id("text-field"+i+"_description")).send_keys(field_description)
i=1
for field_value in list_field_values:
driver.find_element_by_id("text-field"+i+"_description").clear()
Select(driver.find_element_by_id("text-field"+i+"_description")).send_keys(field_value)
driver.find_element_by_id("submit-create").click()