fix compatibility with locust

This commit is contained in:
httprunner
2018-11-26 17:39:25 +08:00
parent b8d7969611
commit abf9b03535

View File

@@ -71,12 +71,18 @@ class Runner(object):
def clear_test_data(self): def clear_test_data(self):
""" clear request and response data """ clear request and response data
""" """
if not isinstance(self.http_client_session, HttpSession):
return
self.evaluated_validators = [] self.evaluated_validators = []
self.http_client_session.init_meta_data() self.http_client_session.init_meta_data()
def get_test_data(self): def get_test_data(self):
""" get request/response data and validate results """ get request/response data and validate results
""" """
if not isinstance(self.http_client_session, HttpSession):
raise exceptions.FunctionNotFound("get_test_data is only valid in HttpSession!")
meta_data = self.http_client_session.meta_data meta_data = self.http_client_session.meta_data
meta_data["validators"] = self.evaluated_validators meta_data["validators"] = self.evaluated_validators
return meta_data return meta_data