def monkeypatch_teamcity_runner(): try: import unittest from teamcity import underTeamcity from teamcity.unittestpy import TeamcityTestRunner runner = TeamcityTestRunner() if underTeamcity() else unittest.TextTestRunner() org_main = unittest.main def monkey_patch(*args, **kwargs): kwargs['testRunner'] = runner org_main(*args, **kwargs) unittest.main = monkey_patch except ImportError: print "Missing teamcity-messages package for running tests in Teamcity." print "Refusing to even try enabling it!"
def monkeypatch_teamcity_runner(): try: import unittest from teamcity import underTeamcity from teamcity.unittestpy import TeamcityTestRunner runner = TeamcityTestRunner() if underTeamcity( ) else unittest.TextTestRunner() org_main = unittest.main def monkey_patch(*args, **kwargs): kwargs['testRunner'] = runner org_main(*args, **kwargs) unittest.main = monkey_patch except ImportError: print "Missing teamcity-messages package for running tests in Teamcity." print "Refusing to even try enabling it!"
def configure(self, options, conf): self.enabled = underTeamcity()
continue # Import the test file and find all TestClass clases inside it. module_name = '%s.%s' % (module, filename[:-3]) # logging.critical('adding %s' % module_name) # test_suite.addTest(unittest.TestLoader().loadTestsFromName(module_name)) test_module = __import__(module_name, {}, {}, filename[:-3]) test_suite.addTest(unittest.TestLoader().loadTestsFromModule(test_module)) return test_suite except Exception: logging.critical("Error loading tests.", exc_info=1) raise SystemExit("Error loading tests.") if __name__ == "__main__": sys.path.insert(0, os.path.join(os.environ['APPENGINEDIR'], 'lib', 'django')) os.environ['SERVER_SOFTWARE'] = 'Development/unittest' # to ensure correct reloading of config/FSM import sys if len(sys.argv) > 1: # tests specified test_suite = unittest.TestLoader().loadTestsFromNames(sys.argv[1:]) else: test_suite = suite() if teamcity.underTeamcity(): TeamcityTestRunner().run(test_suite) else: unittest.TextTestRunner(verbosity=int(os.environ.get('UNITTEST_VERBOSITY', 1))).run(test_suite)
return True def close_alert_and_get_its_text(self): try: alert = self.driver.switch_to_alert() if self.accept_next_alert: alert.accept() else: alert.dismiss() return alert.text finally: self.accept_next_alert = True def tearDown(self): self.driver.quit() self.assertEqual([], self.verificationErrors) self.display.stop() if __name__ == "__main__": unittest.main() if __name__ == '__main__': if underTeamcity(): runner = TeamcityTestRunner() else: runner = unittest.TextTestRunner() unittest.main(testRunner=runner)