# -*- 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)