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