From ed964c83729657f711b4931681c962909022b783 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Wed, 7 Nov 2018 21:15:26 +0800
Subject: [PATCH] bugfix: catch AssertionError in load test

---
 httprunner/templates/locustfile_template | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/httprunner/templates/locustfile_template b/httprunner/templates/locustfile_template
index f5f481c5..1b61b4be 100644
--- a/httprunner/templates/locustfile_template
+++ b/httprunner/templates/locustfile_template
@@ -23,7 +23,7 @@ class WebPageTasks(TaskSet):
         for teststep in teststeps:
             try:
                 self.test_runner.run_test(teststep)
-            except (MyBaseError, MyBaseFailure) as ex:
+            except (AssertionError, MyBaseError, MyBaseFailure) as ex:
                 request_failure.fire(
                     request_type=teststep.get("request", {}).get("method"),
                     name=teststep.get("name"),