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))