import fastunit as unittest import requests, time class MyTest(unittest.TestCase): # 继承unittest.TestCase def test_b_run(self): r = requests.get("http://www.baidu.com") print(2, r) assert (False) time.sleep(2) if __name__ == '__main__': unittest.main()
cmd, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, ) timeout = timeout_in_minutes * 60 my_timer = Timer(timeout, lambda process: process.kill(), [p]) try: my_timer.start() stdout, stderr = p.communicate() stdout, stderr = stdout.decode('utf-8'), stderr.decode('utf-8') finally: my_timer.cancel() logging.debug("Ran %s: retcode: %d, stdout: %s, stderr: %s", cmd, p.returncode, stdout, stderr) return p.returncode, stdout, stderr @staticmethod def generate_verbose_test_name(testcase_func, param_num, param): return "{} [mode={}, version={}, random_prefix={}]".format( testcase_func.__name__, param.args[0], param.args[1].replace(".", "_"), DataprocTestCase.random_str()) if __name__ == '__main__': if PARALLEL_RUN: fastunit.main() else: unittest.main()