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 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()