CI Integration. TeamCity

Author: pitbulk

setup.py

@@ -38,7 +38,8 @@
     ],
     extras_require={
         'test': (
-            'coverage==3.7.1',
+            'teamcity-messages==1.17',
+            'coverage==4.0.3',
             'freezegun==0.3.5',
             'pylint==1.3.1',
             'pep8==1.5.7',

tests/src/OneLogin/saml2_tests/auth_test.py

@@ -7,6 +7,8 @@
 import json
 from os.path import dirname, join, exists
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 from urlparse import urlparse, parse_qs
 
 from onelogin.saml2.auth import OneLogin_Saml2_Auth
@@ -858,3 +860,11 @@ def testBuildResponseSignature(self):
             auth2.build_response_signature(message, relay_state)
         except Exception as e:
             self.assertIn("Trying to sign the SAMLResponse but can't load the SP private key", e.message)
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/authn_request_test.py

@@ -7,6 +7,8 @@
 import json
 from os.path import dirname, join, exists
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 from urlparse import urlparse, parse_qs
 from zlib import decompress
 
@@ -257,3 +259,11 @@ def testCreateEncSAMLRequest(self):
         self.assertRegexpMatches(inflated, '<saml:Issuer>http://stuff.com/endpoints/metadata.php</saml:Issuer>')
         self.assertRegexpMatches(inflated, 'Format="urn:oasis:names:tc:SAML:2.0:nameid-format:encrypted"')
         self.assertRegexpMatches(inflated, 'ProviderName="SP prueba"')
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/error_test.py

@@ -4,14 +4,23 @@
 # All rights reserved.
 
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 from onelogin.saml2.errors import OneLogin_Saml2_Error
 
 
 class OneLogin_Saml2_Error_Test(unittest.TestCase):
     """
     Tests the OneLogin_Saml2_Error Constructor.
     """
-
     def runTest(self):
         exception = OneLogin_Saml2_Error('test')
         self.assertEqual(exception.message, 'test')
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/logout_request_test.py

@@ -7,6 +7,8 @@
 import json
 from os.path import dirname, join, exists
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 from urlparse import urlparse, parse_qs
 from xml.dom.minidom import parseString
 
@@ -438,3 +440,11 @@ def testIsValidSign(self):
             self.assertFalse(valid)
         except Exception as e:
             self.assertIn('In order to validate the sign on the Logout Request, the x509cert of the IdP is required', e.message)
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/logout_response_test.py

@@ -6,6 +6,8 @@
 import json
 from os.path import dirname, join, exists
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 from urlparse import urlparse, parse_qs
 from xml.dom.minidom import parseString
 
@@ -381,3 +383,11 @@ def testIsValid(self):
 
         response_3 = OneLogin_Saml2_Logout_Response(settings, message_3)
         self.assertTrue(response_3.is_valid(request_data))
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/metadata_test.py

@@ -9,6 +9,8 @@
 from time import strftime
 from datetime import datetime
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 
 from onelogin.saml2.metadata import OneLogin_Saml2_Metadata
 from onelogin.saml2.settings import OneLogin_Saml2_Settings
@@ -222,3 +224,11 @@ def testAddX509KeyDescriptors(self):
             self.assertFalse(True)
         except Exception as e:
             self.assertIn('Error parsing metadata', e.message)
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/response_test.py

@@ -10,6 +10,8 @@
 import json
 from os.path import dirname, join, exists
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 from xml.dom.minidom import parseString
 
 from onelogin.saml2.response import OneLogin_Saml2_Response
@@ -1175,3 +1177,11 @@ def testIsValidSignWithEmptyReferenceURI(self):
         xml = self.file_contents(join(self.data_path, 'responses', 'response_without_reference_uri.xml.base64'))
         response = OneLogin_Saml2_Response(settings, xml)
         self.assertTrue(response.is_valid(self.get_request_data()))
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/settings_test.py

@@ -6,6 +6,8 @@
 import json
 from os.path import dirname, join, exists, sep
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 
 from onelogin.saml2.errors import OneLogin_Saml2_Error
 from onelogin.saml2.settings import OneLogin_Saml2_Settings
@@ -669,3 +671,11 @@ def testIsDebugActive(self):
         settings_info['debug'] = True
         settings_3 = OneLogin_Saml2_Settings(settings_info)
         self.assertTrue(settings_3.is_debug_active())
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)

tests/src/OneLogin/saml2_tests/signed_response_test.py

@@ -7,6 +7,8 @@
 import json
 from os.path import dirname, join, exists
 import unittest
+from teamcity import is_running_under_teamcity
+from teamcity.unittestpy import TeamcityTestRunner
 
 from onelogin.saml2.response import OneLogin_Saml2_Response
 from onelogin.saml2.settings import OneLogin_Saml2_Settings
@@ -52,3 +54,11 @@ def testResponseAndAssertionSigned(self):
         response = OneLogin_Saml2_Response(settings, b64encode(message))
 
         self.assertEquals('someone@example.com', response.get_nameid())
+
+
+if __name__ == '__main__':
+    if is_running_under_teamcity():
+        runner = TeamcityTestRunner()
+    else:
+        runner = unittest.TextTestRunner()
+        unittest.main(testRunner=runner)