def test_restart_worker_with_all(self): app = AppController(None, Settings()) app._restartWorker = Mock() app._restartWorkers = Mock() app.api = Mock() app.api.read_worker.return_value = {'wrk_id': 'wrk1', 'command': 'command', 'params': 'params', 'size': 'size'} args = Mock() args.name = 'app/dep' args.wrk_id = False args.all = True app.restartWorker(args) self.assertTrue(app._restartWorkers.called)
def test_restart_worker_gone_error(self): app = AppController(None, Settings()) app._restartWorker = Mock() app._restartWorkers = Mock() app.api = Mock() app.api.read_worker.side_effect = GoneError args = Mock() args.name = 'app/dep' args.wrk_id = 'wrkgone' args.all = False self.assertRaises(InputErrorException, app.restartWorker, args) self.assertFalse(app._restartWorker.called) self.assertFalse(app._restartWorkers.called)
def test__restart_worker(self): app = AppController(None, Settings()) app.api = Mock() app._restartWorker('app_name', 'deployment_name', 'wrk_id', 'command', 'params', 'size') self.assertTrue(app.api.delete_worker.called) self.assertTrue(app.api.create_worker.called)