pandorafms/tests/console/include/policy_functions.py

41 lines
1.6 KiB
Python

# -*- coding: utf-8 -*-
from common_classes_60 import PandoraWebDriverTestCase
from common_functions_60 import login, click_menu_element, detect_and_pass_all_wizards, logout
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
def create_policy(driver,policy_name,group,description=None):
click_menu_element(driver,"Manage policies")
driver.find_element_by_id("submit-crt").click()
driver.find_element_by_id("text-name").send_keys(policy_name)
driver.find_element_by_xpath('//option[contains(.,"'+group+'")]').click()
if description!= None:
driver.find_element_by_id("textarea_description").send_keys(description)
driver.find_element_by_id("submit-crt").click()
def search_policy(driver,policy_name,go_to_policy=True):
click_menu_element(driver,"Manage policies")
driver.find_element_by_id("text-text_search").clear()
driver.find_element_by_id("text-text_search").send_keys(policy_name)
driver.find_element_by_id("submit-submit").click()
# If go_to_policy is True, this function enter in options of this policy
if go_to_policy == True:
driver.find_element_by_xpath('//*[@id="policies_list-0-1"]/span/strong/a/span').click()
def add_module_policy(driver,policy_name,module_type,*args,**kwargs):
search_policy(driver,policy_name,go_to_policy=True)
driver.find_element_by_xpath('//*[@id="menu_tab"]/ul/li[2]/a/img').click()
create_module(module_type,*args,**kwargs)