def testKernelRelease(self):
mock_client = _MockClient(kernel='foobar')
with mock.patch.object(magics_impl._state, 'client', mock_client):
kernel_release = magics_impl.grr_uname_impl(kernel_release=True)
self.assertEqual(kernel_release, 'foobar')
def testMachine(self):
mock_client = _MockClient(arch='foobar')
with mock.patch.object(magics_impl._state, 'client', mock_client):
machine = magics_impl.grr_uname_impl(machine=True)
self.assertEqual(machine, 'foobar')
def grr_uname(line: Text) -> Text:
"""Returns certain system infornamtion.
Args:
line: A string representing arguments passed to the magic command.
Returns:
String representing some system information.
Raises:
NoClientSelectedError: Client is not selected to perform this operation.
"""
args = grr_uname.parser.parse_args(shlex.split(line))
return magics_impl.grr_uname_impl(args.machine, args.kernel_release)
def testNoOptionsProvided(self):
mock_client = _MockClient()
with mock.patch.object(magics_impl._state, 'client', mock_client):
with self.assertRaises(ValueError):
magics_impl.grr_uname_impl()
def testNoClientSelected(self):
with self.assertRaises(magics_impl.NoClientSelectedError):
magics_impl.grr_uname_impl()
