From f9364ab2bdc43c0e0fed7c53502289caea09800d Mon Sep 17 00:00:00 2001
From: axl89 <axl89@users.noreply.github.com>
Date: Tue, 9 Aug 2016 12:51:39 +0200
Subject: [PATCH] Added parallel test execution capabilities

---
 tests/run_console_tests.py | 85 ++++++++++++++++++++++++++++++++++----
 1 file changed, 76 insertions(+), 9 deletions(-)

diff --git a/tests/run_console_tests.py b/tests/run_console_tests.py
index 1571f0c97c..7e7ffef3ed 100755
--- a/tests/run_console_tests.py
+++ b/tests/run_console_tests.py
@@ -3,6 +3,7 @@ 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
 
@@ -10,6 +11,65 @@ 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]
 
+""" 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 = TestSuite()
+    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 = TestSuite()
+            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
+
+def add_test_case_to_suite(suite, tc_name):
+    # Creates a Test Suite with each Test Case added n times
+    n = 1
+    for i in range(0, n):
+        suite.addTest(tc_name)
+
+def get_suite():
+    suite = TestSuite()
+    add_test_case_to_suite(suite, My_login_test('tc_login'))
+    add_test_case_to_suite(suite, My_login_test('tc_logout'))
+    return suite
+
+class TracingStreamResult(testtools.StreamResult):
+	failures = []
+	success = []
+	skipped = []
+	errors = []
+	
+	def status(self, test_status, test_id, *args, **kwargs):
+		if test_status=='inprogress':
+			print "Running "+str(test_id)
+
+		elif test_status=='xfail' or test_status=='fail' or test_status=='exists':
+			self.failures.append(test_id)
+
+		elif test_status=='uxsuccess' or test_status=='success':
+			self.success.append(test_id)
+
+		elif test_status=='exists':
+			self.errors.append(test_id)
+
+		elif test_status=='skip':
+			self.skipped.append('test_id')
+
 #Run Enterprise tests
 is_enterprise = '1' == getenv('ENTERPRISE', False)
 
@@ -19,24 +79,31 @@ if is_enterprise:
 	tests = a.discover(start_dir='console',pattern='*.py')
 else:
 	tests = a.discover(start_dir='console',pattern='PAN*.py')
-
-c = ArticaTestResult()
-tests.run(c)
+if is_enterprise:
+	num_threads = 2
+else:
+	num_threads = 3
+suite = tests
+concurrent_suite = testtools.ConcurrentStreamTestSuite(lambda: (split_suite_into_chunks(num_threads, suite)))
+result = TracingStreamResult()
+result.startTestRun()
+concurrent_suite.run(result)
 
 #Update Saouce Labs jobs
 sauce_client = SauceClient(environ["SAUCE_USERNAME"], environ["SAUCE_ACCESS_KEY"])
-for test,error_msg in c.failures+c.skipped+c.errors:
+c = result
+for test_id in c.failures+c.skipped+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]))
+		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)
+		print "Could not annotate Sauce Labs job #%s" % str(test_id)
 		next
 
-for test,error_msg in c.success:
+for test_id in c.success:
 	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]))
+		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)
+                print "Could not annotate Sauce Labs job #%s" % str(test_id)
                 next