def test_monitor_error(self):
self.client.is_batch_ended.side_effect = livy.RequestError(0, "test")
self.assertEqual(
1, module.main(["--api-url", "http://example.com", "1234"]))
self.client.is_batch_ended.side_effect = KeyboardInterrupt()
self.assertEqual(
1, module.main(["--api-url", "http://example.com", "1234"]))
def test_get_batch_information_errors(self):
self.client.get_batch_information.side_effect = livy.RequestError(
0, "test")
self.assertEqual(
1, module.main(["--api-url", "http://example.com", "1234"]))
self.client.get_batch_information.side_effect = KeyboardInterrupt()
self.assertEqual(
1, module.main(["--api-url", "http://example.com", "1234"]))
def test_task_finished(self):
self.client.get_batch_information.return_value = {
"id": 1234,
"state": "failed",
}
self.assertEqual(
1, module.main(["--api-url", "http://example.com", "1234"]))
def test_user_cancal(self):
self.check_user_confirm.return_value = False
self.assertEqual(
1, module.main(["--api-url", "http://example.com", "1234"]))
def test_success(self):
self.client.is_batch_ended.side_effect = [False, True]
self.assertEqual(
0, module.main(["--api-url", "http://example.com", "1234"]))