pandorafms/tests/console/include/policy_functions.py

# -*- 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 include.module_functions import create_module
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)