def checks(tasksets, locusts): # Test that tasksets are all tasksets and none are locusts. taskset_tuples = [(name, obj) for name, obj in tasksets.items()] for taskset in taskset_tuples: self.assertEqual(True, is_taskset(taskset)) self.assertEqual(False, is_locust(taskset)) # Test that all locusts are locusts locust_tuples = [(name, obj) for name, obj in locusts.items()] for locust in locust_tuples: self.assertEqual(True, is_locust(locust)) self.assertEqual(False, is_taskset(locust)) # Check that ExampleModel is found as a taskset # Check that ExampleModelLocust is found as a locust self.assertTrue( 'ExampleModel' in [name for name, _ in taskset_tuples]) locust_names = [name for name, _ in locust_tuples] self.assertTrue('ExampleModelLocust' in locust_names) self.assertFalse('_PrivateExample' in locust_names)
def test_is_locust(self): self.assertFalse(main.is_locust(("Locust", Locust))) self.assertFalse(main.is_locust(("HttpLocust", HttpLocust))) self.assertFalse(main.is_locust(("random_dict", {}))) self.assertFalse(main.is_locust(("random_list", []))) class MyTaskSet(TaskSet): pass class MyHttpLocust(HttpLocust): task_set = MyTaskSet class MyLocust(Locust): task_set = MyTaskSet self.assertTrue(main.is_locust(("MyHttpLocust", MyHttpLocust))) self.assertTrue(main.is_locust(("MyLocust", MyLocust))) class ThriftLocust(Locust): pass self.assertFalse(main.is_locust(("ThriftLocust", ThriftLocust)))
def test_is_locust(self): self.assertFalse(main.is_locust(("Locust", Locust))) self.assertFalse(main.is_locust(("HttpLocust", HttpLocust))) self.assertFalse(main.is_locust(("random_dict", {}))) self.assertFalse(main.is_locust(("random_list", []))) class MyTaskSet(TaskSet): pass class MyHttpLocust(HttpLocust): task_set = MyTaskSet class MyLocust(Locust): task_set = MyTaskSet self.assertTrue(main.is_locust(("MyHttpLocust", MyHttpLocust))) self.assertTrue(main.is_locust(("MyLocust", MyLocust))) class ThriftLocust(Locust): pass self.assertFalse(main.is_locust(("ThriftLocust", ThriftLocust)))
def test_is_locust(self): self.assertFalse(main.is_locust(Locust)) self.assertFalse(main.is_locust(HttpLocust)) self.assertFalse(main.is_locust({})) self.assertFalse(main.is_locust([])) class MyTaskSet(TaskSet): pass class MyHttpLocust(HttpLocust): tasks = [MyTaskSet] class MyLocust(Locust): tasks = [MyTaskSet] self.assertTrue(main.is_locust(MyHttpLocust)) self.assertTrue(main.is_locust(MyLocust)) class ThriftLocust(Locust): abstract = True self.assertFalse(main.is_locust(ThriftLocust))
def test_is_locust(self): assert not main.is_locust(("Locust", Locust)) assert not main.is_locust(("HttpLocust", HttpLocust)) assert not main.is_locust(("random_dict", {})) assert not main.is_locust(("random_list", [])) class MyTaskSet(TaskSet): pass class MyHttpLocust(HttpLocust): task_set = MyTaskSet class MyLocust(Locust): task_set = MyTaskSet assert main.is_locust(("MyHttpLocust", MyHttpLocust)) assert main.is_locust(("MyLocust", MyLocust)) class ThriftLocust(Locust): pass assert not main.is_locust(("ThriftLocust", ThriftLocust))
def test_is_abstract(self): self.assertTrue(FastHttpLocust.abstract) self.assertFalse(is_locust(FastHttpLocust))