Fix test case for missing tasks (never make last minute changes just …

…before commiting, doh...)
locustio · Mar 11, 2020 · b330585 · b330585
1 parent 076fdf7
commit b330585
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/locust/test/test_locust_class.py b/locust/test/test_locust_class.py
@@ -36,15 +36,13 @@ class User(Locust):
             wait_time = constant(0)
             tasks = None
             _catch_exceptions = False
-
-        l = MyTasks(User(self.environment))
-
-        self.assertRaisesRegex(Exception, "No tasks defined.*", l.run)
-
+
         class MyTasks(TaskSet):
             tasks = None
 
-        User.tasks = [MyTasks]
+        l = MyTasks(User(self.environment))
+        self.assertRaisesRegex(Exception, "No tasks defined.*", l.run)
+        l.tasks = None
         self.assertRaisesRegex(Exception, "No tasks defined.*", l.run)
 
     def test_task_decorator_ratio(self):