From 0e0ddb7d52809988338f1a70efa25b4fb5649124 Mon Sep 17 00:00:00 2001 From: debugtalk Date: Mon, 26 Nov 2018 17:39:25 +0800 Subject: [PATCH] fix compatibility with locust --- httprunner/runner.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/httprunner/runner.py b/httprunner/runner.py index 7dd05360..6ea969c3 100644 --- a/httprunner/runner.py +++ b/httprunner/runner.py @@ -71,12 +71,18 @@ class Runner(object): def clear_test_data(self): """ clear request and response data """ + if not isinstance(self.http_client_session, HttpSession): + return + self.evaluated_validators = [] self.http_client_session.init_meta_data() def get_test_data(self): """ 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["validators"] = self.evaluated_validators return meta_data