pandorafms/tests/console/include/user_functions.py

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

import random, time
import string

def add_user_profile(driver,user_name,profile,group):
	click_menu_element(driver,"Users management")
	driver.find_element_by_css_selector("b").click()
	driver.find_element_by_id("text-filter_search").clear()
	driver.find_element_by_id("text-filter_search").send_keys(user_name)
	driver.find_element_by_id("submit-search").click()
	driver.find_element_by_xpath('//*[@id="table3-0-6"]/a[2]').click()
	Select(driver.find_element_by_id("assign_profile")).select_by_visible_text(profile)

	if group == "All":
		Select(driver.find_element_by_id("assign_group")).select_by_visible_text(group)
	else:
		#TODO This will not work when choosing a group within a group within another group
		Select(driver.find_element_by_id("assign_group")).select_by_visible_text("    "+group)

	#driver.find_element_by_id("image-add2").click()
	driver.find_element_by_xpath('//*[@name="add"]').click()


def create_user(driver,user_name,userpwd,email=None,profile_list=None):
	u"""
	Profile list es una LISTA de TUPLAS:
			l = [("Chief Operator","All"),("Read Operator","Servers")]
	"""
	click_menu_element(driver,"Users management")
	driver.find_element_by_id("submit-crt").click()
	driver.find_element_by_name("id_user").clear()
	driver.find_element_by_name("id_user").send_keys(user_name)
	driver.find_element_by_name("password_new").clear()
	driver.find_element_by_name("password_new").send_keys(userpwd)
	driver.find_element_by_name("password_confirm").clear()
	driver.find_element_by_name("password_confirm").send_keys(userpwd)
	driver.find_element_by_name("email").clear()
	if email != None:
		driver.find_element_by_name("email").clear()
		driver.find_element_by_name("email").send_keys(email)
	driver.find_element_by_id("submit-crtbutton").click()

	if profile_list != None:
		for profile_name,group_name in profile_list:
			add_user_profile(driver,user_name,profile_name,group_name)

def search_user(driver,user_name):
	click_menu_element(driver,"Users management")
	driver.find_element_by_css_selector("b").click()
	driver.find_element_by_id('text-filter_search').clear()
	driver.find_element_by_id("text-filter_search").send_keys(user_name)
	driver.find_element_by_id("submit-search").click()

		
def activate_home_screen(driver,mode):

	click_menu_element(driver,"Edit my user")
	Select(driver.find_element_by_id("section")).select_by_visible_text(mode)
	driver.find_element_by_id("submit-uptbutton").click()