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)
