def testWithClientSet(self):
mock_client = _MockClient()
mock_client.set_hostname('foobar')
with mock.patch.object(magics_impl._state, 'client', mock_client):
hostname = magics_impl.grr_hostname_impl()
self.assertEqual(hostname, 'foobar')
def grr_hostname(line: Text) -> Text:
"""Returns hostname of the selected client.
Args:
line: A string representing arguments passed to the magic command.
Returns:
String representing hostname of a client.
Raises:
NoClientSelectedError: Client is not selected to perform this operation.
"""
del line # Unused.
return magics_impl.grr_hostname_impl()
def testNoClientSelected(self):
with self.assertRaises(magics_impl.NoClientSelectedError):
magics_impl.grr_hostname_impl()