Working in the testing framework

tests/console/Planned_downtime.py

@@ -1,165 +0,0 @@
-# -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
-from include.common_functions_60 import login, click_menu_element, detect_and_pass_all_wizards, logout, gen_random_string
-from include.planned_downtime_functions import *
-from include.alert_functions import *
-from include.module_functions import *
-from include.agent_functions import *
-from include.event_functions import *
-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, datetime
-
-
-class PAN13(PandoraWebDriverTestCase):
-
-	test_name = u'Planned_downtime'
-	test_description = u'Planed downtime test'
-	tickets_associated = []
-
-	def avoid_test_A_create_planned_downtime_Quiet(self):
-
-		u"""
-		Create and search planned downtime quiet
-		"""
-		driver = self.driver
-		self.login()
-		detect_and_pass_all_wizards(driver)
-
-		planned_name = gen_random_string(6)
-
-		create_planned_downtime(driver,planned_name,"Applications","Quiet","Once",description=planned_name)
-		
-		time.sleep(10)
-		
-		search_planned_downtime(driver,planned_name)	
-
-		element = driver.find_element_by_xpath('//img[@data-title="Running"]')
-                self.assertIsInstance(element,WebElement)
-	
-	def avoid_test_B_create_planned_downtime_disabled_agents(self):
-		
-                u"""
-                Create and search planned downtime disabled agents
-                """
-		driver = self.driver
-
-		planned_name = gen_random_string(6)
-		
-		create_planned_downtime(driver,planned_name,"Applications","Disabled Agents","Once",description=planned_name)
-		
-		time.sleep(10)
-		
-		search_planned_downtime(driver,planned_name)	
-	
-		element = driver.find_element_by_xpath('//img[@data-title="Running"]')
-                self.assertIsInstance(element,WebElement)
-	   
-	def avoid_test_C_create_planned_downtime_disabled_only_alerts(self):
-		
-		u"""
-		Create and search planned downtime disabled only alerts
-		"""
-		driver = self.driver
-		
-		planned_name = gen_random_string(6)
-
-		create_planned_downtime(driver,planned_name,"Applications","Disabled only Alerts","Once",description=planned_name)
-		
-		time.sleep(10)
-		
-		search_planned_downtime(driver,planned_name)	
-
- 		element = driver.find_element_by_xpath('//img[@data-title="Running"]')
-                self.assertIsInstance(element,WebElement)
-
-        def avoid_test_D_delete_planned_downtime(self):
-
-	
-		driver=self.driver
-
-		planned_name = gen_random_string(6)
-
-                create_planned_downtime(driver,planned_name,"Applications","Quiet","Once",description=planned_name)
-
-		delete_planned_downtime(driver,planned_name)
-
-	def avoid_test_E_quiet_functionality(self):
-
-		driver=self.driver
-		
-		planned_name = gen_random_string(6)
-		agent_name_A = gen_random_string(6)
-		agent_name_B = gen_random_string(6)
-
-		module_name_A_A = gen_random_string(6)
-		module_name_A_B = gen_random_string(6)
-
-		module_name_B_A = gen_random_string(6)
-		module_name_B_B = gen_random_string(6)
-
-		create_agent(driver,agent_name_A,ip="127.0.0.1",group="Applications")
-		create_agent(driver,agent_name_B,ip="127.0.0.1",group="Applications")
-
-		create_module('network_server',driver,agent_name=agent_name_A,module_name=module_name_A_A,component_group='Network Management',network_component='Host Alive')
-		create_module('network_server',driver,agent_name=agent_name_A,module_name=module_name_A_B,component_group='Network Management',network_component='Host Latency')
-
-		create_module('network_server',driver,agent_name=agent_name_B,module_name=module_name_B_A,component_group='Network Management',network_component='Host Alive')
-		create_module('network_server',driver,agent_name=agent_name_B,module_name=module_name_B_B,component_group='Network Management',network_component='Host Alive')
-
-                assign_alert_template_to_module(driver,agent_name_A,module_name_A_A,'Critical condition')
-                assign_alert_template_to_module(driver,agent_name_A,module_name_A_B,'Critical condition')
-
-                assign_alert_template_to_module(driver,agent_name_B,module_name_B_A,'Critical condition')
-                assign_alert_template_to_module(driver,agent_name_B,module_name_B_B,'Critical condition')
-
-		#Little hack to allow the planned downtime to be edited
-		fifteen_seconds_later = datetime.datetime.now() + datetime.timedelta(seconds=15)
-		fifteen_seconds_later_to_pandora = str(fifteen_seconds_later.hour)+":"+str(fifteen_seconds_later.minute)+":"+str(fifteen_seconds_later.second)
-
-		create_planned_downtime(driver,planned_name,"Applications","Quiet","Once",hour_from=fifteen_seconds_later_to_pandora,description=planned_name,agent_module_list=[(agent_name_A,[module_name_A_A]),(agent_name_B,[module_name_B_A])])
-
-		#We wait 10 seconds to the planned downtime to start. Since we do not specify a date, default dates are from now to one hour in the future.
-		time.sleep(40)
-
-		#Is the planned downtime running?
-		search_planned_downtime(driver,planned_name)
-                element = driver.find_element_by_xpath('//img[@data-title="Running"]')
-                self.assertIsInstance(element,WebElement)
-
-		force_alert_of_module(driver,agent_name_A,module_name_A_A,'Critical condition')
-		force_alert_of_module(driver,agent_name_A,module_name_A_B,'Critical condition')
-		force_alert_of_module(driver,agent_name_B,module_name_B_A,'Critical condition')
-		force_alert_of_module(driver,agent_name_B,module_name_B_B,'Critical condition')
-
-		search_events(driver,agent_name=agent_name_A,module_name=module_name_A_A)
-
-		event_who_should_not_be_present = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_A+'")]]')
-
-		self.assertEqual(event_who_should_not_be_present,[])
-
-		search_events(driver,agent_name=agent_name_A,module_name=module_name_A_B)
-
-		event_who_should_be_present = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_A+'")]]')
-		
-		self.assertNotEqual(event_who_should_be_present,[])
-
-		search_events(driver,agent_name=agent_name_B,module_name=module_name_B_A)
-
-		event_who_should_not_be_present_b = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_B+'")]]')
-
-		self.assertEqual(event_who_should_not_be_present_b,[])
-
-		search_events(driver,agent_name=agent_name_B,module_name=module_name_B_B)
-
-		event_who_should_be_present_b = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_B+'")]]')
-
-		self.assertNotEqual(event_who_should_be_present_b,[])
-
-if __name__ == "__main__":
-	unittest.main()

tests/console/collections/Collections.py

@@ -1,10 +1,12 @@
 # -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
+import sys, os
-from include.common_functions_60 import click_menu_element, detect_and_pass_all_wizards, gen_random_string, is_enterprise
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
-from include.module_functions import create_module
+from common_classes_60 import PandoraWebDriverTestCase
-from include.agent_functions import create_agent_group
+from common_functions_60 import click_menu_element, detect_and_pass_all_wizards, gen_random_string, is_enterprise
-from include.policy_functions import *
+from module_functions import create_module
-from include.collection_functions import *
+from agent_functions import create_agent_group
+from policy_functions import *
+from collection_functions import *
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
@@ -42,6 +44,7 @@ class Collections(PandoraWebDriverTestCase):
 	def test_B_edit_collection(self):
 	
 		driver = self.driver
+		self.login()
 		
 		edit_collection(driver,self.collection_name,new_name=self.new_collection_name,group="Applications",description="Edit collectionPAN11")
 		
@@ -54,6 +57,7 @@ class Collections(PandoraWebDriverTestCase):
 	def test_C_create_text_collection(self):
 	
 		driver = self.driver
+		self.login()
 	
 		create_text_in_collection(driver,self.new_collection_name,"file_collectionPAN11",text_file="test file")
 				
@@ -64,6 +68,7 @@ class Collections(PandoraWebDriverTestCase):
 	def test_D_directory_collection(self):	
 	
 		driver = self.driver
+		self.login()
 	
 		create_directory_in_collection(driver,self.new_collection_name,"directory_collectionPAN11")
 		
@@ -74,6 +79,7 @@ class Collections(PandoraWebDriverTestCase):
 	def test_E_delete_collection(self):
 	
 		driver = self.driver
+		self.login()
 		
 		delete_collection(driver,self.new_collection_name)
 	

tests/console/include/common_classes_60.py

@@ -22,13 +22,13 @@ class ArticaTestResult(TestResult):
 class PandoraWebDriverTestCase(TestCase):
 	test_name = u'' #Name of the test.
 	test_description = u'' #Description of the test
-	time_started = None
-	time_elapsed = None #Total time of the test
 	tickets_associated = []
 	sauce_username = environ["SAUCE_USERNAME"]
 	sauce_access_key = environ["SAUCE_ACCESS_KEY"]
 	sauce_client = None
 	sauce_labs_job_id = None
+	
+	currentResult = None # holds last result object passed to run method
 
 	desired_cap = {
 		'tunnel-identifier': environ["TRAVIS_JOB_NUMBER"],
@@ -36,34 +36,28 @@ class PandoraWebDriverTestCase(TestCase):
 		'browserName': "firefox",
 		'version': "46",
 	}
+	
+	def run(self,result=None,*args,**kwargs):
+		self.currentResult = result # remember result for use in tearDown
+		unittest.TestCase.run(self, result,*args,**kwargs) # call superclass run method
 
 	@classmethod
 	def setUpClass(cls):
 		cls.is_development = os.getenv('DEVELOPMENT', False)
 		cls.is_enterprise = os.getenv('ENTERPRISE', False)
-		if cls.is_development != False:
+		
-			cls.driver = webdriver.Firefox()
+	def setUp(self):
-			cls.base_url = os.getenv('DEVELOPMENT_URL')
+		if self.is_development != False:
+			self.driver = webdriver.Firefox()
+			self.base_url = os.getenv('DEVELOPMENT_URL')
 		else:
 			#Start VM in Sauce Labs
-			cls.driver = webdriver.Remote(command_executor='http://'+cls.sauce_username+':'+cls.sauce_access_key+'@ondemand.saucelabs.com:80/wd/hub',desired_capabilities=cls.desired_cap)
+			self.driver = webdriver.Remote(command_executor='http://'+self.sauce_username+':'+self.sauce_access_key+'@ondemand.saucelabs.com:80/wd/hub',desired_capabilities=self.desired_cap)
-			cls.sauce_labs_job_id = cls.driver.session_id
+			self.sauce_labs_job_id = self.driver.session_id
-			cls.base_url = "http://127.0.0.1/"
+			self.base_url = "http://127.0.0.1/"
-
-
-		
-	@classmethod
-	def tearDownClass(cls):
-		cls.driver.quit()
-
-	def setUp(self):
-		self.time_started = datetime.now()
 		self.driver.implicitly_wait(30)
 		self.verificationErrors = []
 		self.accept_next_alert = True
-		#self.is_development = self.is_development
-		#TODO Print test name
-		print "Starting test"
 		super(PandoraWebDriverTestCase, self).setUp()
 
 	def is_element_present(self, how, what):
@@ -88,16 +82,21 @@ class PandoraWebDriverTestCase(TestCase):
 		finally: self.accept_next_alert = True
 
 	def tearDown(self):
-		tack = datetime.now()
+		self.driver.quit()
-		diff = tack - self.time_started
+		ok = self.currentResult.wasSuccessful()
-		self.time_elapsed = diff.seconds
+		errors = self.currentResult.errors
+		failures = self.currentResult.failures
+		skipped = self.currentResult.skipped
 
-		self.assertEqual([], self.verificationErrors)
+		if not self.is_development:	
+			sauce_client = SauceClient(environ["SAUCE_USERNAME"], environ["SAUCE_ACCESS_KEY"])
+			sauce_client.jobs.update_job(self.sauce_labs_job_id, passed=ok==True,tags=[environ["TRAVIS_BRANCH"],self.id()],build_num=environ["TRAVIS_JOB_NUMBER"],name=str(environ["TRAVIS_COMMIT"])+"_"+str(self.id().split('.')[1])+"_"+str(self.id().split('.')[2]))
+		
+		#self.assertEqual([], self.verificationErrors) #TODO Review if this line is actually needed
 		super(PandoraWebDriverTestCase, self).tearDown()
 
 
 	def login(self,user="admin",passwd="pandora",pandora_url=None):
-		print u"Logging in"
 
 		driver = self.driver
 
@@ -109,10 +108,19 @@ class PandoraWebDriverTestCase(TestCase):
 		driver.find_element_by_id("nick").send_keys(user)
 		driver.find_element_by_id("pass").clear()
 		driver.find_element_by_id("pass").send_keys(passwd)
+		
+		#Hack
+		driver.add_cookie({'name': 'clippy', 'value': 'deleted'})
+		driver.add_cookie({'name': 'clippy_is_annoying', 'value': '1'})
 		driver.find_element_by_id("submit-login_button").click()
 
+	"""
+	def test_ZZZZZZZZZZZ(self):
+		#The hackiest way to end the driver in the LAST test (all credits to python unittest library for sorting tests alphabetically! :D)
+		self.driver.quit()
+	"""
+
 	def logout(self,pandora_url=None):
-		print u"Logging out"
 
 		driver = self.driver
 
@@ -127,3 +135,4 @@ class PandoraWebDriverTestCase(TestCase):
 		driver.get(pandora_url+"/pandora_console/index.php")
 		refresh_N_times_until_find_element(driver,2,"nick")
 
+

tests/console/include/common_functions_60.py

@@ -110,7 +110,7 @@ def detect_and_pass_newsletter_wizard(driver):
 
 def detect_and_pass_all_wizards(driver):
 	driver.implicitly_wait(2) #Optimisation workaround for skipping wizards quickly
-	detect_and_pass_pandorin(driver)
+	#detect_and_pass_pandorin(driver)
 	detect_and_pass_initial_wizard(driver)
 	detect_and_pass_newsletter_wizard(driver)
 	driver.implicitly_wait(30)

tests/console/not_execution/acl/ACL.py

@@ -1,10 +1,12 @@
 # -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
+import sys, os
-from include.common_functions_60 import login, is_element_present, click_menu_element, detect_and_pass_all_wizards, logout, gen_random_string, is_enterprise
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
-from include.agent_functions import create_agent, search_agent, create_agent_group
+from common_classes_60 import PandoraWebDriverTestCase
-from include.user_functions import create_user, create_user_profile
+from common_functions_60 import login, is_element_present, click_menu_element, detect_and_pass_all_wizards, logout, gen_random_string, is_enterprise
-from include.module_functions import create_module
+from agent_functions import create_agent, search_agent, create_agent_group
-from include.reports_functions import create_report
+from user_functions import create_user, create_user_profile
+from module_functions import create_module
+from reports_functions import create_report
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
@@ -26,7 +28,6 @@ class ACL(PandoraWebDriverTestCase):
 		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)

tests/console/not_execution/bulk_operations/Bulk_operations.py

@@ -1,7 +1,9 @@
 # -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
+import sys, os
-from include.common_functions_60 import login, click_menu_element, detect_and_pass_all_wizards
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
-from include.agent_functions import create_agent
+from common_classes_60 import PandoraWebDriverTestCase
+from common_functions_60 import login, click_menu_element, detect_and_pass_all_wizards
+from agent_functions import create_agent
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys

tests/console/not_execution/misc/Misc.py

@@ -1,12 +1,13 @@
-
 # -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
+import sys, os
-from include.common_functions_60 import login, click_menu_element, refresh_N_times_until_find_element, detect_and_pass_all_wizards, is_element_present, logout
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
-from include.reports_functions import create_report, delete_report
+from common_classes_60 import PandoraWebDriverTestCase
-from include.user_functions import create_user
+from common_functions_60 import login, click_menu_element, refresh_N_times_until_find_element, detect_and_pass_all_wizards, is_element_present, logout
-from include.agent_functions import create_agent
+from reports_functions import create_report, delete_report
-from include.module_functions import create_module
+from user_functions import create_user
-from include.event_functions import *
+from agent_functions import create_agent
+from module_functions import create_module
+from event_functions import *
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys

tests/console/not_execution/network_server/icmp_test.py

@@ -1,28 +1,33 @@
 # -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
+import sys, os
-from include.common_functions_60 import login, click_menu_element, refresh_N_times_until_find_element, detect_and_pass_all_wizards
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
-from include.agent_functions import create_agent
+from common_classes_60 import PandoraWebDriverTestCase
+from common_functions_60 import login, click_menu_element, refresh_N_times_until_find_element, detect_and_pass_all_wizards, gen_random_string
+from agent_functions import create_agent
 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 StaleElementReferenceException
-from include.module_functions import *
+from module_functions import *
 import unittest, time, re
 
 
-class PAN3(PandoraWebDriverTestCase):
+class SimpleICMPTest(PandoraWebDriverTestCase):
 
-	test_name = u'PAN_3'
+	test_name = u'Simple ICMP Test'
 	test_description = u'Creates a simple ICMP check against localhost and checks the result is 1'
 	tickets_associated = []
 
 	def test_pan3(self):
 		driver = self.driver
+		agent_name = gen_random_string(6)
+		module_name = gen_random_string(6)
+
 		self.login()
-		create_agent(driver,"PAN3_agent",ip="127.0.0.1")
+		create_agent(driver,agent_name,ip="127.0.0.1")
 		
-		create_module("network_server",driver,agent_name="PAN3_agent",module_name="PAN3_module",component_group="Network Management",network_component="Host Alive",ip="127.0.0.1")
+		create_module("network_server",driver,agent_name=agent_name,module_name=module_name,component_group="Network Management",network_component="Host Alive",ip="127.0.0.1")
 
 		driver.find_element_by_xpath('//*[@id="menu_tab"]//a[contains(@href,"ver_agente")]').click()
 

tests/console/not_execution/policies/Policies.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
+import sys, os
-from include.common_functions_60 import login, detect_and_pass_all_wizards, gen_random_string, is_enterprise
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
-from include.policy_functions import *
+from common_classes_60 import PandoraWebDriverTestCase
-from include.agent_functions import *
+from common_functions_60 import login, detect_and_pass_all_wizards, gen_random_string, is_enterprise
-from include.collection_functions import *
+from policy_functions import *
+from agent_functions import *
+from collection_functions import *
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
@@ -48,6 +50,7 @@ class Policies(PandoraWebDriverTestCase):
 		Add network server module to previous policy 
 		"""	
 		driver = self.driver
+		self.login()
 
 		add_module_policy(driver,self.policy_name,"network_server",driver,module_name=self.network_server_module_name,component_group="Network Management",network_component="Host Alive")
 
@@ -67,6 +70,7 @@ class Policies(PandoraWebDriverTestCase):
 
 
 		driver = self.driver
+		self.login()
 
 		create_policy(driver,policy_name,"Applications",description="Policy for test")
 
@@ -93,6 +97,7 @@ class Policies(PandoraWebDriverTestCase):
 		module_name_2 = gen_random_string(6)
 
 		driver = self.driver
+		self.login()
 
 		create_agent(driver,agent_name_1,description="First agent by test")
 		create_agent(driver,agent_name_2,description="Second agent 2 by test")

tests/console/not_execution/user/Users.py

@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
-from include.common_classes_60 import PandoraWebDriverTestCase
+import sys, os
-from include.common_functions_60 import login, logout, click_menu_element, detect_and_pass_all_wizards, activate_home_screen
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
+from common_classes_60 import PandoraWebDriverTestCase
+from common_functions_60 import login, logout, click_menu_element, detect_and_pass_all_wizards, activate_home_screen
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys

tests/console/planned_downtime/Planned_downtime.py

@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+import sys, os
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
+from common_classes_60 import PandoraWebDriverTestCase
+from common_functions_60 import login, click_menu_element, detect_and_pass_all_wizards, logout, gen_random_string
+from planned_downtime_functions import *
+from alert_functions import *
+from module_functions import *
+from agent_functions import *
+from event_functions import *
+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, datetime
+
+
+class Creation(PandoraWebDriverTestCase):
+
+	test_name = u'Planned downtime creation'
+	test_description = u'Planed downtime creation test'
+	tickets_associated = []
+
+	quiet_name = gen_random_string(6)
+	disabled_agents_name = gen_random_string(6)
+	disabled_only_alerts_name = gen_random_string(6)
+	
+
+	def test_A_create_planned_downtime_Quiet(self):
+
+		u"""
+		Create and search planned downtime quiet
+		"""
+		driver = self.driver
+		self.login()
+		detect_and_pass_all_wizards(driver)
+
+		planned_name = self.quiet_name
+
+		create_planned_downtime(driver,planned_name,"Applications","Quiet","Once",description=planned_name)
+		
+		time.sleep(10)
+		
+		search_planned_downtime(driver,planned_name)	
+
+		element = driver.find_element_by_xpath('//img[@data-title="Running"]')
+                self.assertIsInstance(element,WebElement)
+	
+	def ru_B_create_planned_downtime_disabled_agents(self):
+		
+                u"""
+                Create and search planned downtime disabled agents
+                """
+		driver = self.driver
+		self.login()
+
+		planned_name = self.disabled_agents_name
+		
+		create_planned_downtime(driver,planned_name,"Applications","Disabled Agents","Once",description=planned_name)
+		
+		time.sleep(10)
+		
+		search_planned_downtime(driver,planned_name)	
+	
+		element = driver.find_element_by_xpath('//img[@data-title="Running"]')
+                self.assertIsInstance(element,WebElement)
+	   
+	def ru_C_create_planned_downtime_disabled_only_alerts(self):
+		
+		u"""
+		Create and search planned downtime disabled only alerts
+		"""
+		driver = self.driver
+		self.login()
+		
+		planned_name = self.disabled_only_alerts_name
+
+		create_planned_downtime(driver,planned_name,"Applications","Disabled only Alerts","Once",description=planned_name)
+		
+		time.sleep(10)
+		
+		search_planned_downtime(driver,planned_name)	
+
+ 		element = driver.find_element_by_xpath('//img[@data-title="Running"]')
+                self.assertIsInstance(element,WebElement)
+
+        def ru_D_delete_planned_downtime(self):
+
+	
+		driver=self.driver
+
+		downtime_list = [self.disabled_only_alerts_name,self.disabled_agents_name,self.quiet_name]
+
+		for planned_name in downtime_list:
+			planned_name = gen_random_string(6)
+			delete_planned_downtime(driver,planned_name)
+			#TODO assert the downtime is deleted
+
+if __name__ == "__main__":
+	unittest.main()

tests/console/planned_downtime/Quiet_functionality.py

@@ -0,0 +1,102 @@
+# -*- coding: utf-8 -*-
+import sys, os
+sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../include")
+from common_classes_60 import PandoraWebDriverTestCase
+from common_functions_60 import login, click_menu_element, detect_and_pass_all_wizards, logout, gen_random_string
+from planned_downtime_functions import *
+from alert_functions import *
+from module_functions import *
+from agent_functions import *
+from event_functions import *
+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, datetime
+
+
+class QuietTest(PandoraWebDriverTestCase):
+
+	test_name = u'Planned downtime quiet type test'
+	test_description = u'Planed downtime quiet functionality '
+	tickets_associated = []
+
+        def test_A_quiet_functionality(self):
+
+                driver=self.driver
+		self.login()
+
+                planned_name = gen_random_string(6)
+                agent_name_A = gen_random_string(6)
+                agent_name_B = gen_random_string(6)
+
+                module_name_A_A = gen_random_string(6)
+                module_name_A_B = gen_random_string(6)
+
+                module_name_B_A = gen_random_string(6)
+                module_name_B_B = gen_random_string(6)
+
+                create_agent(driver,agent_name_A,ip="127.0.0.1",group="Applications")
+                create_agent(driver,agent_name_B,ip="127.0.0.1",group="Applications")
+
+                create_module('network_server',driver,agent_name=agent_name_A,module_name=module_name_A_A,component_group='Network Management',network_component='Host Alive')
+                create_module('network_server',driver,agent_name=agent_name_A,module_name=module_name_A_B,component_group='Network Management',network_component='Host Latency')
+
+                create_module('network_server',driver,agent_name=agent_name_B,module_name=module_name_B_A,component_group='Network Management',network_component='Host Alive')
+                create_module('network_server',driver,agent_name=agent_name_B,module_name=module_name_B_B,component_group='Network Management',network_component='Host Alive')
+
+                assign_alert_template_to_module(driver,agent_name_A,module_name_A_A,'Critical condition')
+                assign_alert_template_to_module(driver,agent_name_A,module_name_A_B,'Critical condition')
+
+                assign_alert_template_to_module(driver,agent_name_B,module_name_B_A,'Critical condition')
+                assign_alert_template_to_module(driver,agent_name_B,module_name_B_B,'Critical condition')
+
+                #Little hack to allow the planned downtime to be edited
+                fifteen_seconds_later = datetime.datetime.now() + datetime.timedelta(seconds=15)
+                fifteen_seconds_later_to_pandora = str(fifteen_seconds_later.hour)+":"+str(fifteen_seconds_later.minute)+":"+str(fifteen_seconds_later.second)
+
+                create_planned_downtime(driver,planned_name,"Applications","Quiet","Once",hour_from=fifteen_seconds_later_to_pandora,description=planned_name,agent_module_list=[(agent_name_A,[module_name_A_A]),(agent_name_B,[module_name_B_A])])
+
+                #We wait 10 seconds to the planned downtime to start. Since we do not specify a date, default dates are from now to one hour in the future.
+                time.sleep(40)
+
+                #Is the planned downtime running?
+                search_planned_downtime(driver,planned_name)
+                element = driver.find_element_by_xpath('//img[@data-title="Running"]')
+                self.assertIsInstance(element,WebElement)
+
+                force_alert_of_module(driver,agent_name_A,module_name_A_A,'Critical condition')
+                force_alert_of_module(driver,agent_name_A,module_name_A_B,'Critical condition')
+                force_alert_of_module(driver,agent_name_B,module_name_B_A,'Critical condition')
+                force_alert_of_module(driver,agent_name_B,module_name_B_B,'Critical condition')
+
+                search_events(driver,agent_name=agent_name_A,module_name=module_name_A_A)
+
+                event_who_should_not_be_present = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_A+'")]]')
+
+                self.assertEqual(event_who_should_not_be_present,[])
+
+                search_events(driver,agent_name=agent_name_A,module_name=module_name_A_B)
+				
+                event_who_should_be_present = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_A+'")]]')
+
+                self.assertNotEqual(event_who_should_be_present,[])
+
+                search_events(driver,agent_name=agent_name_B,module_name=module_name_B_A)
+
+                event_who_should_not_be_present_b = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_B+'")]]')
+
+                self.assertEqual(event_who_should_not_be_present_b,[])
+
+                search_events(driver,agent_name=agent_name_B,module_name=module_name_B_B)
+
+                event_who_should_be_present_b = driver.find_elements_by_xpath('//tbody/tr[td[3][contains(.,"Alert fired")] and td[4][contains(.,"'+agent_name_B+'")]]')
+
+                self.assertNotEqual(event_who_should_be_present_b,[])
+
+if __name__ == "__main__":
+        unittest.main()
+

tests/run_console_tests.py

@@ -3,48 +3,108 @@ from unittest import *
 from console.include.common_functions_60 import *
 from console.include.common_classes_60 import *
 from sauceclient import SauceClient
+import testtools
 from os import environ, getenv
 import subprocess, time, sys
 
-def get_test_file(test_list):
-	#return [test[0].split(' ')[0].split('.')[0].split('<')[1] for test in test_list]
-	return [test[0].test_name for test in test_list]
 
-#Run Enterprise tests
+class ArticaTestSuite(TestSuite):
-is_enterprise = '1' == getenv('ENTERPRISE', False)
+	def __init__(self,*args,**kwargs):
+		super(ArticaTestSuite,self).__init__(*args,**kwargs)
+		
+	def run(self,*args,**kwargs):
+		print "Running "+str(self.countTestCases())+" tests."
+		print "Tests are: "+str(self._tests)
+		super(ArticaTestSuite,self).run(*args,**kwargs)
 
-a = TestLoader()
+class ArticaTestLoader(TestLoader):
-
+	def __init__(self,*args,**kwargs):
-tests = a.discover(start_dir='console',pattern='*.py')
+		self.suiteClass = ArticaTestSuite
-
+		super(ArticaTestLoader, self).__init__(*args,**kwargs)
-c = ArticaTestResult()
-tests.run(c)
-
-#Update Saouce Labs jobs
-sauce_client = SauceClient(environ["SAUCE_USERNAME"], environ["SAUCE_ACCESS_KEY"])
-for test,error_msg in c.failures+c.errors:
-	try:
-		sauce_client.jobs.update_job(test.sauce_labs_job_id, passed=False,tags=[environ["TRAVIS_BRANCH"],test.id()],build_num=environ["TRAVIS_JOB_NUMBER"],name=str(environ["TRAVIS_COMMIT"])+"_"+str(test.id().split('.')[1]))
-	except:
-		print "Could not annotate Sauce Labs job #%s" % str(test)
-		next
-
-for test,error_msg in c.success+c.skipped:
-	try:
-		sauce_client.jobs.update_job(test.sauce_labs_job_id, passed=True,tags=[environ["TRAVIS_BRANCH"],test.id()],build_num=environ["TRAVIS_JOB_NUMBER"],name=str(environ["TRAVIS_COMMIT"])+"_"+str(test.id().split('.')[1]))
-	except:
-                print "Could not annotate Sauce Labs job #%s" % str(test)
-                next
 	
 
+""" Splits a Test Suite so that no more than 'n' threads will execute the tests """
+def split_suite_into_chunks(num_threads, suite):
+    # Compute num_threads such that the number of threads does not exceed the value passed to the function
+    # Keep num_threads to a reasonable number of threads
+    if num_threads < 0: num_threads = 1
+    if num_threads > 8: num_threads = 8
+    num_tests = suite.countTestCases()
+    s = []
+    s_tmp = ArticaTestSuite()
+    n = round(num_tests / num_threads)
+    for case in suite:
+        if n <= 0 and s_tmp.countTestCases() > 0:
+            s.append([s_tmp, None])
+            num_threads -= 1
+            num_tests -= s_tmp.countTestCases()
+            s_tmp = ArticaTestSuite()
+            n = round(num_tests / num_threads)
+        s_tmp.addTest(case)
+        n -= 1
+    if s_tmp.countTestCases() > 0:
+        if s_tmp.countTestCases() > 0: s.append([s_tmp, None])
+        num_tests -= s_tmp.countTestCases()
+    if num_tests != 0: print("Error: num_tests should be 0 but is %s!" % num_tests)
+    return s
 
-print "Tests failed: %s" % c.failures
+class TracingStreamResult(testtools.StreamResult):
-print "Tests succeeded: %s" % c.success
+	failures = []
-print "Tests skipped: %s" % c.skipped
+	success = []
-print "Tests with errors: %s" % c.errors
+	skipped = []
+	errors = []
+	
+	def status(self, test_status, test_id, *args, **kwargs):
+		if test_status=='inprogress':
+			print "Running test "+str(test_id)
 
-if (len(c.failures)+len(c.errors)) != 0:
+		elif test_status=='xfail' or test_status=='fail' or test_status=='exists':
+			print "Test "+str(test_id)+" has failed"
+			self.failures.append(test_id)
+
+		elif test_status=='uxsuccess' or test_status=='success':
+			print "Test "+str(test_id)+" has succeeded"
+			self.success.append(test_id)
+
+		elif test_status=='exists':
+			print "Test "+str(test_id)+" has failed (already existed)"
+			self.errors.append(test_id)
+
+		elif test_status=='skip':
+			print "Test "+str(test_id)+" was skipped"
+			self.skipped.append('test_id')
+
+is_enterprise = '1' == getenv('ENTERPRISE', False)
+if is_enterprise:
+        num_threads = 2
+else:
+        num_threads = 8
+import pdb
+a = ArticaTestLoader()
+
+#Network server tests
+tests = a.discover(start_dir='console',pattern='*.py')
+
+print str(tests.countTestCases())+" tests found"
+
+concurrent_suite = testtools.ConcurrentStreamTestSuite(lambda: (split_suite_into_chunks(num_threads, tests)))
+result = TracingStreamResult()
+
+
+try:
+	result.startTestRun()
+finally:
+	concurrent_suite.run(result)
+
+print "SUMMARY"
+print "======="
+print "Tests failed: %s" % result.failures
+print "Tests succeeded: %s" % result.success
+print "Tests skipped: %s" % result.skipped
+print "Tests with errors: %s" % result.errors
+print "======="
+
+if (len(result.failures)+len(result.errors)) != 0:
 	sys.exit(1)
-
 else:
 	sys.exit(0)