pandorafms/tests/console/ACL.py

# -*- 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 unittest, time, re

class ACL(PandoraWebDriverTestCase):

	test_name = u'ACL'
	tickets_associated = []


	def test_A_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".
		"""
		print "funciono"
        	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,[])

		#We need to logout because we've loged in with a non-admin user
		self.logout()

 	def test_B_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 reporting")
                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 reporting")
                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)

                
		#Delete reports
                self.logout()

	def test_C_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,[])
 
if __name__ == "__main__":
	unittest.main()