How to use test_exception_in_task method in locust

Best Python code snippet using locust

test_runners.py

Source:test_runners.py

...160                num_clients += Message.unserialize(msg).data["num_clients"]161            162            self.assertEqual(2, num_clients, "Total number of locusts that would have been spawned is not 2")163    164    def test_exception_in_task(self):165        class HeyAnException(Exception):166            pass167        168        class MyLocust(Locust):169            class task_set(TaskSet):170                @task171                def will_error(self):172                    raise HeyAnException(":(")173        174        runner = LocalLocustRunner([MyLocust], self.options)175        176        l = MyLocust()177        l._catch_exceptions = False178        ...

test_kamma.py

Source:test_kamma.py

...134        self.count[0] = self.count[0] + 1135        if self.count[0] < self.num_failures:136            raise Exception('I don\'t want to work, try {}'.format(self.count[0]))137138    def test_exception_in_task(self):139        worker = kamma.Worker(queue_path=TEST_PATH)140        worker.add_task_callback(callback=self.task0, retry_wait=kamma.wait_fixed(0), retry_stop=kamma.stop_after_attempt(self.num_failures+1))141        worker.push_task(callback=self.task0)142        worker.run_async()143        worker.wait_empty_event()144        worker.stop()145        self.assertEqual(self.num_failures, self.count[0])146147148class KammaTestsOnAbortion(unittest.TestCase):149    def setUp(self):150        _clear_queue()151        self.abortion_called = False152        self.failure_called = False
...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.