def testWait(self):
mock_client = mock.MagicMock()
with mock.patch.object(magics_impl._state, 'client', mock_client):
magics_impl.grr_request_approval_impl('foo', ['bar'], wait=True)
mock_client.request_approval_and_wait.assert_called_once_with(
reason='foo', approvers=['bar'])
self.assertFalse(mock_client.request_approval.called)
def grr_request_approval(line: Text) -> None:
"""Sends approval request to the selected client for the current user.
Args:
line: A string representing arguments passed to the magic command.
Returns:
Nothing.
Raises:
NoClientSelectedError: Client is not selected to perform this operation.
"""
args = grr_request_approval.parser.parse_args(shlex.split(line))
magics_impl.grr_request_approval_impl(args.reason, args.approvers,
args.wait)
def testNoClientSelected(self):
with self.assertRaises(magics_impl.NoClientSelectedError):
magics_impl.grr_request_approval_impl('foo', ['bar'])