def test_scheduler_commands(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
self.assertEqual("LIST JOBS, KILL JOB, PAUSE, RESUME",
extension.execute(client_context, "COMMANDS"))
def test_scheduler_exception(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
self.assertEqual("Scheduler Admin Error",
extension.execute(client_context, "COMMANDS"))
def test_scheduler_other(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
self.assertEqual("Invalid Scheduler Admin command [OTHER]",
extension.execute(client_context, "OTHER"))
def test_scheduler_list_invalid(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
self.assertEqual("Invalid LIST commands, LIST JOBS",
extension.execute(client_context, "LIST"))
self.assertEqual("Unknown LIST sub command [OTHER]",
extension.execute(client_context, "LIST OTHER"))
def test_scheduler_pause_resume(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
self.assertEqual("Scheduler paused", extension.execute(client_context, "PAUSE"))
self.assertTrue(client.scheduler._paused)
self.assertEqual("Scheduler resumed", extension.execute(client_context, "RESUME"))
self.assertFalse(client.scheduler._paused)
def test_scheduler_kill_invalid(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
self.assertEqual("Invalid KILL commands, LIST JOB JOBID",
extension.execute(client_context, "KILL"))
self.assertEqual("Invalid KILL commands, LIST JOB JOBID",
extension.execute(client_context, "KILL JOB"))
self.assertEqual("Unknown KILL sub command [OTHER]",
extension.execute(client_context, "KILL OTHER XXX"))
def test_scheduler_kill_job(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
job1 = unittest.mock.Mock()
job1.next_run_time = datetime.strptime("11/04/18 19:02", "%d/%m/%y %H:%M")
job1.args = ("Arg1", "Arg2", "Arg3", "Arg4", "Arg5")
client.scheduler._jobs["1"] = job1
self.assertEqual("Job removed", extension.execute(client_context, "KILL JOB 1"))
def test_scheduler_list_jobs(self):
client = SchedulerAdminExtensionClient()
client_context = client.create_client_context("testid")
extension = SchedulerAdminExtension()
self.assertEqual("No job information available", extension.execute(client_context, "LIST JOBS"))
job1 = unittest.mock.Mock()
job1.next_run_time = datetime.strptime("11/04/18 19:02", "%d/%m/%y %H:%M")
job1.args = ("Arg1", "Arg2", "Arg3", "Arg4", "Arg5")
client.scheduler._jobs["1"] = job1
self.assertEqual("> Job ID:1, Next Run: 2018-04-11 19:02:00, Args: ('Arg1', 'Arg2', 'Arg3', 'Arg4', 'Arg5')\n", extension.execute(client_context, "LIST JOBS"))