import sys from test_login_local import TestLoginLocal from test_registration import TestRegistration from unittest import TestSuite, TextTestRunner, TestLoader test_cases = [ TestLoginLocal, TestRegistration ] def suite(gn2_url, es_url): the_suite = TestSuite() for case in test_cases: the_suite.addTests(initTest(case, gn2_url, es_url)) return the_suite def initTest(klass, gn2_url, es_url): loader = TestLoader() methodNames = loader.getTestCaseNames(klass) return [klass(mname, gn2_url, es_url) for mname in methodNames] def main(gn2_url, es_url): runner = TextTestRunner() runner.run(suite(gn2_url, es_url)) if __name__ == "__main__": if len(sys.argv) < 3: raise Exception("Required arguments missing:\n\tTry running `run-integration-test.py `") else: main(sys.argv[1], sys.argv[2])