def test_is_user_class(self):
self.assertFalse(main.is_user_class(User))
self.assertFalse(main.is_user_class(HttpUser))
self.assertFalse(main.is_user_class({}))
self.assertFalse(main.is_user_class([]))
class MyTaskSet(TaskSet):
pass
class MyHttpUser(HttpUser):
tasks = [MyTaskSet]
class MyUser(User):
tasks = [MyTaskSet]
self.assertTrue(main.is_user_class(MyHttpUser))
self.assertTrue(main.is_user_class(MyUser))
class ThriftLocust(User):
abstract = True
self.assertFalse(main.is_user_class(ThriftLocust))
def test_is_abstract(self):
self.assertTrue(FastHttpUser.abstract)
self.assertFalse(is_user_class(FastHttpUser))
