191 lines
7.1 KiB
Python
191 lines
7.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
from include.common_classes_60 import PandoraWebDriverTestCase
|
||
from include.common_functions_60 import login, is_element_present, click_menu_element, detect_and_pass_all_wizards, logout, gen_random_string, is_enterprise
|
||
from include.agent_functions import create_agent, search_agent, create_agent_group
|
||
from include.user_functions import create_user, create_user_profile
|
||
from include.module_functions import create_module
|
||
from include.reports_functions import create_report
|
||
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
|
||
from selenium.webdriver.remote.webelement import WebElement
|
||
import unittest2, time, re
|
||
import logging
|
||
|
||
class ACLPropagation(PandoraWebDriverTestCase):
|
||
|
||
test_name = u'ACL propagation'
|
||
tickets_associated = []
|
||
|
||
|
||
def test_ACL_propagation(self):
|
||
|
||
u"""
|
||
ACL Propagation test: Creates one group "A" with ACL propagation, then a group "B" son of "A" with no ACL propagation, and finally group "C".
|
||
The test asserts if a user with privileges to "A" can see the agent of "B" but no agents of "C".
|
||
"""
|
||
|
||
logging.basicConfig(filename="ACL.log", level=logging.INFO, filemode='w')
|
||
|
||
group_name_A = gen_random_string(6)
|
||
group_name_B = gen_random_string(6)
|
||
group_name_C = gen_random_string(6)
|
||
agent_name_A = gen_random_string(6)
|
||
agent_name_B = gen_random_string(6)
|
||
user_name = gen_random_string(6)
|
||
|
||
driver = self.driver
|
||
self.login()
|
||
detect_and_pass_all_wizards(driver)
|
||
|
||
create_agent_group(driver,group_name_A,propagate_acl=True,description="Group A, with propagate ACL, son of ALL")
|
||
create_agent_group(driver,group_name_B,parent_group=group_name_A,description="Group B, son of A")
|
||
create_agent_group(driver,group_name_C,parent_group=group_name_B,description="Group C, son of B")
|
||
|
||
create_agent(driver,agent_name_A,description="Agent in group B",group=group_name_B)
|
||
|
||
create_agent(driver,agent_name_B,description="Agent in group C",group=group_name_C)
|
||
|
||
l=[("Chief Operator",group_name_A,[])]
|
||
|
||
create_user(driver,user_name,"pandora",profile_list=l)
|
||
|
||
self.logout()
|
||
|
||
self.login(user=user_name)
|
||
|
||
detect_and_pass_all_wizards(driver)
|
||
|
||
#Is the agent listed in the agent list?
|
||
search_agent(driver,agent_name_A,go_to_agent=False)
|
||
element = driver.find_element_by_xpath('//a[contains(.,"'+agent_name_A+'")]')
|
||
self.assertIsInstance(element,WebElement)
|
||
|
||
#Is the agent accesible for the user?
|
||
search_agent(driver,agent_name_A,go_to_agent=True)
|
||
element = driver.find_element_by_xpath('//*[@id="agent_contact_main"]/thead/tr/th')
|
||
self.assertIsInstance(element,WebElement)
|
||
|
||
#Is the agent invisible to the user? (It should be invisible)
|
||
search_agent(driver,agent_name_B,go_to_agent=False)
|
||
element = driver.find_elements_by_xpath('//a[contains(.,"'+agent_name_B+'")]')
|
||
self.assertEqual(element,[])
|
||
|
||
logging.info("test_ACL_propagation is correct")
|
||
|
||
class ACLReports(PandoraWebDriverTestCase):
|
||
|
||
test_name = u'ACL reports'
|
||
tickets_associated = []
|
||
|
||
def test_ACL_reports(self):
|
||
|
||
u"""
|
||
Creates a user with Chief Operator permissions over the Applications group.
|
||
Then creates two reports: one in the Applications group and other in the Servers group. Then, it checks that the given user can only see the Application report
|
||
"""
|
||
|
||
user_name = gen_random_string(6)
|
||
report_name_A = agent_name = gen_random_string(6)
|
||
report_name_B = agent_name = gen_random_string(6)
|
||
|
||
driver = self.driver
|
||
self.login()
|
||
|
||
#Creates a user with Chief Operator - Applications profile
|
||
profile_list = []
|
||
profile_list.append(("Chief Operator","Applications",[]))
|
||
create_user(driver,user_name,user_name,email=user_name+'@pandorafms.com',profile_list=profile_list)
|
||
|
||
#Creates report
|
||
create_report(driver,report_name_A,"Applications")
|
||
create_report(driver,report_name_B,"Servers")
|
||
|
||
#Logout
|
||
self.logout()
|
||
|
||
#Login
|
||
self.login(user=user_name,passwd=user_name)
|
||
|
||
#Check that the report is visible
|
||
click_menu_element(driver,"Custom reports")
|
||
driver.find_element_by_id('text-search').clear()
|
||
driver.find_element_by_id('text-search').send_keys(report_name_A)
|
||
driver.find_element_by_id('submit-search_submit').click()
|
||
self.assertEqual(is_element_present(driver, By.ID, 'report_list-0'),True)
|
||
|
||
|
||
#Check that the report is not visible
|
||
click_menu_element(driver,"Custom reports")
|
||
driver.find_element_by_id('text-search').clear()
|
||
driver.find_element_by_id('text-search').send_keys(report_name_B)
|
||
driver.find_element_by_id('submit-search_submit').click()
|
||
|
||
time.sleep(6)
|
||
|
||
|
||
element = driver.find_element_by_xpath('//td[contains(.,"No data found.")]')
|
||
self.assertIsInstance(element,WebElement)
|
||
|
||
logging.info("test_ACL_reports is correct")
|
||
|
||
class ACLTags(PandoraWebDriverTestCase):
|
||
|
||
test_name = u'ACL tag test'
|
||
tickets_associated = []
|
||
|
||
def test_ACL_tag(self):
|
||
|
||
u"""Create agent and two modules, one without tag and with tag, create a user with tag and check this user can view module with tag and user can´t view module without tag"""
|
||
|
||
#agent_name = gen_random_string(6)
|
||
#module_name_A = gen_random_string(6)
|
||
#module_name_B = gen_random_string(6)
|
||
#user_name = gen_random_string(6)
|
||
|
||
#driver = self.driver
|
||
#self.login()
|
||
#detect_and_pass_all_wizards(driver)
|
||
|
||
#create_agent(driver,agent_name,group="Applications",ip="192.168.50.50")
|
||
|
||
#We create a module without a tag
|
||
|
||
#create_module("network_server",driver,agent_name=agent_name,module_name=module_name_A,component_group="Network Management",network_component="Host Alive",ip="192.168.50.50")
|
||
|
||
#We now create a modulo with tag "critical"
|
||
|
||
#create_module("network_server",driver,agent_name=agent_name,module_name=module_name_B,component_group="Network Management",network_component="Host Alive",ip="192.168.50.50",tag_name="critical")
|
||
|
||
|
||
#l = [("Operator (Read)","All",["critical"])]
|
||
|
||
#create_user(driver,user_name,"pandora",profile_list=l)
|
||
|
||
#self.logout()
|
||
|
||
#self.login(user=user_name)
|
||
|
||
#detect_and_pass_all_wizards(driver)
|
||
|
||
#search_agent(driver,agent_name)
|
||
|
||
#time.sleep(6)
|
||
|
||
|
||
#The user should be able to see the module with Tag
|
||
#module = driver.find_element_by_xpath('//td[contains(.,"'+module_name_B+'")]')
|
||
#self.assertIsInstance(module,WebElement)
|
||
|
||
#The user should NOT be able to see the module without tag
|
||
#modules = driver.find_elements_by_xpath('//td[contains(.,"'+module_name_A+'")]')
|
||
#self.assertEqual(modules,[])
|
||
|
||
#logging.info("test_ACL_tag is correct")
|
||
|
||
if __name__ == "__main__":
|
||
unittest2.main()
|